|
|
@@ -3,6 +3,7 @@ package network
|
|
3
|
3
|
import (
|
|
4
|
4
|
"context"
|
|
5
|
5
|
"fmt"
|
|
|
6
|
+ "net/http"
|
|
6
|
7
|
"net/url"
|
|
7
|
8
|
|
|
8
|
9
|
"github.com/9seconds/mtg/v2/essentials"
|
|
|
@@ -15,6 +16,10 @@ type proxyNetwork struct {
|
|
15
|
16
|
client proxy.ContextDialer
|
|
16
|
17
|
}
|
|
17
|
18
|
|
|
|
19
|
+func (p proxyNetwork) Dial(network, address string) (essentials.Conn, error) {
|
|
|
20
|
+ return p.DialContext(context.Background(), network, address)
|
|
|
21
|
+}
|
|
|
22
|
+
|
|
18
|
23
|
func (p proxyNetwork) DialContext(ctx context.Context, network, address string) (essentials.Conn, error) {
|
|
19
|
24
|
conn, err := p.client.DialContext(ctx, network, address)
|
|
20
|
25
|
if err != nil {
|
|
|
@@ -24,6 +29,16 @@ func (p proxyNetwork) DialContext(ctx context.Context, network, address string)
|
|
24
|
29
|
return essentials.WrapNetConn(conn), nil
|
|
25
|
30
|
}
|
|
26
|
31
|
|
|
|
32
|
+func (p proxyNetwork) MakeHTTPClient(
|
|
|
33
|
+ dialFunc func(context.Context, string, string) (essentials.Conn, error),
|
|
|
34
|
+) *http.Client {
|
|
|
35
|
+ if dialFunc == nil {
|
|
|
36
|
+ dialFunc = p.DialContext
|
|
|
37
|
+ }
|
|
|
38
|
+
|
|
|
39
|
+ return p.Network.MakeHTTPClient(dialFunc)
|
|
|
40
|
+}
|
|
|
41
|
+
|
|
27
|
42
|
func NewProxyNetwork(base mtglib.Network, proxyURL *url.URL) (*proxyNetwork, error) {
|
|
28
|
43
|
socks, err := proxy.FromURL(proxyURL, base.NativeDialer())
|
|
29
|
44
|
if err != nil {
|