64 lines
1.7 KiB
Go
64 lines
1.7 KiB
Go
package store
|
|
|
|
import (
|
|
"log/slog"
|
|
|
|
"github.com/anacrolix/dht/v2/bep44"
|
|
tlog "github.com/anacrolix/log"
|
|
"github.com/anacrolix/torrent"
|
|
"github.com/anacrolix/torrent/storage"
|
|
|
|
"git.kmsign.ru/royalcat/tstor/src/config"
|
|
dlog "git.kmsign.ru/royalcat/tstor/src/log"
|
|
)
|
|
|
|
// MOVE
|
|
func NewClient(st storage.ClientImpl, fis bep44.Store, cfg *config.TorrentClient, id [20]byte) (*torrent.Client, error) {
|
|
l := slog.With("component", "torrent-client")
|
|
|
|
// TODO download and upload limits
|
|
torrentCfg := torrent.NewDefaultClientConfig()
|
|
torrentCfg.PeerID = string(id[:])
|
|
torrentCfg.DefaultStorage = st
|
|
torrentCfg.AlwaysWantConns = true
|
|
torrentCfg.AcceptPeerConnections = true
|
|
torrentCfg.DisableAggressiveUpload = false
|
|
|
|
torrentCfg.Seed = true
|
|
// torrentCfg.DownloadRateLimiter = rate.NewLimiter(rate.Inf, 0)
|
|
// torrentCfg
|
|
|
|
tl := tlog.NewLogger()
|
|
tl.SetHandlers(&dlog.Torrent{L: l})
|
|
torrentCfg.Logger = tl
|
|
torrentCfg.Callbacks.NewPeer = append(torrentCfg.Callbacks.NewPeer, func(p *torrent.Peer) {
|
|
l := l.With("ip", p.RemoteAddr.String())
|
|
if p.Torrent() != nil {
|
|
l = l.With("torrent", p.Torrent().Name())
|
|
}
|
|
|
|
l.Debug("new peer")
|
|
|
|
})
|
|
|
|
torrentCfg.Callbacks.PeerClosed = append(torrentCfg.Callbacks.PeerClosed, func(p *torrent.Peer) {
|
|
l := l.With("ip", p.RemoteAddr.String())
|
|
if p.Torrent() != nil {
|
|
l = l.With("torrent", p.Torrent().Name())
|
|
}
|
|
|
|
l.Debug("peer closed")
|
|
})
|
|
|
|
// torrentCfg.Callbacks.PeerConnClosed = append(torrentCfg.Callbacks.PeerConnClosed, func(c *torrent.PeerConn) {
|
|
// l.Debug("peer closed", "ip", c.RemoteAddr.String())
|
|
// })
|
|
|
|
// torrentCfg.ConfigureAnacrolixDhtServer = func(cfg *dht.ServerConfig) {
|
|
// cfg.Store = fis
|
|
// cfg.Exp = 2 * time.Hour
|
|
// cfg.NoSecurity = false
|
|
// }
|
|
|
|
return torrent.NewClient(torrentCfg)
|
|
}
|