f120b93e1c
This is useful if you want to use a custom dialer to proxy requests via an external server since the HTTPProxy can only be used with tcp trackers and not udp.
58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package tracker
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"net/url"
|
|
|
|
"github.com/anacrolix/log"
|
|
trHttp "github.com/anacrolix/torrent/tracker/http"
|
|
"github.com/anacrolix/torrent/tracker/udp"
|
|
)
|
|
|
|
type Client interface {
|
|
Announce(context.Context, AnnounceRequest, AnnounceOpt) (AnnounceResponse, error)
|
|
Close() error
|
|
}
|
|
|
|
type AnnounceOpt = trHttp.AnnounceOpt
|
|
|
|
type NewClientOpts struct {
|
|
Http trHttp.NewClientOpts
|
|
// Overrides the network in the scheme. Probably a legacy thing.
|
|
UdpNetwork string
|
|
Logger log.Logger
|
|
ListenPacket func(network, addr string) (net.PacketConn, error)
|
|
}
|
|
|
|
func NewClient(urlStr string, opts NewClientOpts) (Client, error) {
|
|
_url, err := url.Parse(urlStr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
switch _url.Scheme {
|
|
case "http", "https":
|
|
return trHttp.NewClient(_url, opts.Http), nil
|
|
case "udp", "udp4", "udp6":
|
|
network := _url.Scheme
|
|
if opts.UdpNetwork != "" {
|
|
network = opts.UdpNetwork
|
|
}
|
|
cc, err := udp.NewConnClient(udp.NewConnClientOpts{
|
|
Network: network,
|
|
Host: _url.Host,
|
|
Logger: opts.Logger,
|
|
ListenPacket: opts.ListenPacket,
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &udpClient{
|
|
cl: cc,
|
|
requestUri: _url.RequestURI(),
|
|
}, nil
|
|
default:
|
|
return nil, ErrBadScheme
|
|
}
|
|
}
|