Add v2 network package

This commit is contained in:
9seconds
2026-02-27 15:50:21 +01:00
parent 282896be09
commit 42927c8bdc
15 changed files with 714 additions and 0 deletions
+36
View File
@@ -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
}