mirror of
https://github.com/ScuroNeko/mtg.git
synced 2026-08-31 12:44:02 +03:00
Add v2 network package
This commit is contained in:
@@ -0,0 +1,36 @@
|
||||
package network
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"net/url"
|
||||
|
||||
"github.com/9seconds/mtg/v2/essentials"
|
||||
"golang.org/x/net/proxy"
|
||||
)
|
||||
|
||||
type proxyNetwork struct {
|
||||
Network
|
||||
client proxy.ContextDialer
|
||||
}
|
||||
|
||||
func (p proxyNetwork) DialContext(ctx context.Context, network, address string) (essentials.Conn, error) {
|
||||
conn, err := p.client.DialContext(ctx, network, address)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return essentials.WrapNetConn(conn), nil
|
||||
}
|
||||
|
||||
func NewProxyNetwork(base Network, proxyURL *url.URL) (*proxyNetwork, error) {
|
||||
socks, err := proxy.FromURL(proxyURL, base.NativeDialer())
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("cannot build proxy dialer: %w", err)
|
||||
}
|
||||
|
||||
return &proxyNetwork{
|
||||
Network: base,
|
||||
client: socks.(proxy.ContextDialer),
|
||||
}, nil
|
||||
}
|
||||
Reference in New Issue
Block a user