bit1001/tdl
1
1
Fork 0
mirror of https://github.com/bit1001/tdl.git synced 2024-09-20 06:35:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(utils): proxy

Browse source
This commit is contained in:
iyear 2022-09-07 00:04:58 +08:00
parent 7ab0e1b3e2
commit a3e64b4eea
3 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
app/dl/dlurl/url.go
View file

@ -25,7 +25,7 @@ func Run(ctx context.Context, ns, proxy string, partSize, threads, limit int, ur
c := telegram.NewClient(consts.AppID, consts.AppHash, telegram.Options{
Resolver: dcs.Plain(dcs.PlainOptions{
Dial: utils.GetDial(proxy).DialContext,
Dial: utils.Proxy.GetDial(proxy).DialContext,
}),
Device: consts.Device,
SessionStorage: storage.NewSession(kvd, false),

2
app/login/login.go
View file

@ -25,7 +25,7 @@ func Run(ctx context.Context, ns, proxy string) error {
c := telegram.NewClient(consts.AppID, consts.AppHash, telegram.Options{
Resolver: dcs.Plain(dcs.PlainOptions{
Dial: utils.GetDial(proxy).DialContext,
Dial: utils.Proxy.GetDial(proxy).DialContext,
}),
Device: consts.Device,
SessionStorage: storage.NewSession(kvd, true),

8
pkg/utils/proxy.go
View file

@ -5,8 +5,12 @@ import (
"net/url"
)
func GetDial(p string) proxy.ContextDialer {
u, err := url.Parse(p)
type _proxy struct{}
var Proxy = _proxy{}
func (p _proxy) GetDial(_url string) proxy.ContextDialer {
u, err := url.Parse(_url)
if err != nil {
return proxy.Direct
}