Implement SOCKS5 and HTTP proxy support in main.go, enhancing outgoing request handling. Update README.md to document new proxy environment variables. Update go.mod to include golang.org/x/net dependency.
Publish telemt-bot Docker image / build-and-push (push) Failing after 45s
Publish telemt-bot Docker image / build-and-push (push) Failing after 45s
This commit is contained in:
@@ -22,6 +22,7 @@
|
|||||||
- `TELEMT_LINK_HOST` - хост для генерации `tg://proxy` ссылок, если API их не вернуло.
|
- `TELEMT_LINK_HOST` - хост для генерации `tg://proxy` ссылок, если API их не вернуло.
|
||||||
- `TELEMT_LINK_PORT` - порт для ссылок (по умолчанию `9443`).
|
- `TELEMT_LINK_PORT` - порт для ссылок (по умолчанию `9443`).
|
||||||
- `TELEMT_TLS_DOMAIN` - TLS-домен для генерации `EE-TLS` ссылки (опционально).
|
- `TELEMT_TLS_DOMAIN` - TLS-домен для генерации `EE-TLS` ссылки (опционально).
|
||||||
|
- `ALL_PROXY`, `HTTPS_PROXY`, `HTTP_PROXY` - прокси для исходящих запросов (Telegram API, Telemt API). Поддерживаются `socks5://`, `socks5h://`, `http://`, `https://`. Приоритет: ALL_PROXY → HTTPS_PROXY → HTTP_PROXY.
|
||||||
|
|
||||||
## Быстрый старт (локально, Linux/bash)
|
## Быстрый старт (локально, Linux/bash)
|
||||||
|
|
||||||
|
|||||||
@@ -1,3 +1,5 @@
|
|||||||
module telemt-bot
|
module telemt-bot
|
||||||
|
|
||||||
go 1.22
|
go 1.22
|
||||||
|
|
||||||
|
require golang.org/x/net v0.28.0
|
||||||
|
|||||||
@@ -8,6 +8,7 @@ import (
|
|||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"log"
|
"log"
|
||||||
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/url"
|
"net/url"
|
||||||
"os"
|
"os"
|
||||||
@@ -17,6 +18,8 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
"syscall"
|
"syscall"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"golang.org/x/net/proxy"
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
@@ -111,22 +114,73 @@ type createUserResponse struct {
|
|||||||
Secret string `json:"secret"`
|
Secret string `json:"secret"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func makeHTTPTransport() (*http.Transport, error) {
|
||||||
|
tr := &http.Transport{
|
||||||
|
MaxIdleConns: 64,
|
||||||
|
MaxIdleConnsPerHost: 32,
|
||||||
|
IdleConnTimeout: 90 * time.Second,
|
||||||
|
TLSHandshakeTimeout: 5 * time.Second,
|
||||||
|
ForceAttemptHTTP2: true,
|
||||||
|
DisableCompression: false,
|
||||||
|
}
|
||||||
|
|
||||||
|
proxyURL := getProxyURL()
|
||||||
|
if proxyURL == "" {
|
||||||
|
return tr, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
u, err := url.Parse(proxyURL)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("invalid proxy URL %q: %w", proxyURL, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
switch u.Scheme {
|
||||||
|
case "socks5", "socks5h":
|
||||||
|
dialer, err := proxy.FromURL(u, proxy.Direct)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("socks5 proxy %q: %w", proxyURL, err)
|
||||||
|
}
|
||||||
|
if ctxDialer, ok := dialer.(proxy.ContextDialer); ok {
|
||||||
|
tr.DialContext = ctxDialer.DialContext
|
||||||
|
} else {
|
||||||
|
tr.DialContext = func(ctx context.Context, network, addr string) (net.Conn, error) {
|
||||||
|
return dialer.Dial(network, addr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
log.Printf("using SOCKS5 proxy: %s", u.Host)
|
||||||
|
case "http", "https":
|
||||||
|
tr.Proxy = http.ProxyURL(u)
|
||||||
|
log.Printf("using HTTP proxy: %s", u.Host)
|
||||||
|
default:
|
||||||
|
return nil, fmt.Errorf("unsupported proxy scheme %q (use socks5, socks5h, http, or https)", u.Scheme)
|
||||||
|
}
|
||||||
|
|
||||||
|
return tr, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func getProxyURL() string {
|
||||||
|
for _, key := range []string{"ALL_PROXY", "all_proxy", "HTTPS_PROXY", "https_proxy", "HTTP_PROXY", "http_proxy"} {
|
||||||
|
if v := strings.TrimSpace(os.Getenv(key)); v != "" {
|
||||||
|
return v
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
cfg, err := loadConfig()
|
cfg, err := loadConfig()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatalf("config error: %v", err)
|
log.Fatalf("config error: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
transport, err := makeHTTPTransport()
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("transport error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
httpClient := &http.Client{
|
httpClient := &http.Client{
|
||||||
Transport: &http.Transport{
|
Transport: transport,
|
||||||
MaxIdleConns: 64,
|
Timeout: cfg.TelegramTimeout,
|
||||||
MaxIdleConnsPerHost: 32,
|
|
||||||
IdleConnTimeout: 90 * time.Second,
|
|
||||||
TLSHandshakeTimeout: 5 * time.Second,
|
|
||||||
ForceAttemptHTTP2: true,
|
|
||||||
DisableCompression: false,
|
|
||||||
},
|
|
||||||
Timeout: cfg.TelegramTimeout,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
b := &bot{
|
b := &bot{
|
||||||
|
|||||||
Reference in New Issue
Block a user