Add custom auto-update urls

internal/cli/run_proxy.go

@@ -258,6 +258,8 @@ func runProxy(conf *config.Config, version string) error { //nolint: funlen
 		DomainFrontingProxyProtocol: conf.GetDomainFrontingProxyProtocol(false),
 		PreferIP:                    conf.PreferIP.Get(mtglib.DefaultPreferIP),
 		AutoUpdate:                  conf.AutoUpdate.Get(false),
+		AutoUpdateURLv4:             conf.AutoUpdateURLv4.String(),
+		AutoUpdateURLv6:             conf.AutoUpdateURLv6.String(),
 
 		AllowFallbackOnUnknownDC: conf.AllowFallbackOnUnknownDC.Get(false),
 		TolerateTimeSkewness:     conf.TolerateTimeSkewness.Value,

internal/config/config.go

@@ -30,6 +30,8 @@ type Config struct {
 	ProxyProtocolListener       TypeBool        `json:"proxyProtocolListener"`
 	PreferIP                    TypePreferIP    `json:"preferIp"`
 	AutoUpdate                  TypeBool        `json:"autoUpdate"`
+	AutoUpdateURLv4             TypeURL         `json:"autoUpdateURLv4"`
+	AutoUpdateURLv6             TypeURL         `json:"autoUpdateURLv6"`
 	DomainFrontingPort          TypePort        `json:"domainFrontingPort"`
 	DomainFrontingIP            TypeIP          `json:"domainFrontingIp"`
 	DomainFrontingProxyProtocol TypeBool        `json:"domainFrontingProxyProtocol"`

internal/config/parse.go

@@ -16,6 +16,8 @@ type tomlConfig struct {
 	ProxyProtocolListener       bool   `toml:"proxy-protocol-listener" json:"proxyProtocolListener"`
 	PreferIP                    string `toml:"prefer-ip" json:"preferIp,omitempty"`
 	AutoUpdate                  bool   `toml:"auto-update" json:"autoUpdate,omitempty"`
+	AutoUpdateURLv4             string `toml:"auto-update-url-v4" json:"autoUpdateURLv4,omitempty"`
+	AutoUpdateURLv6             string `toml:"auto-update-url-v6" json:"autoUpdateURLv6,omitempty"`
 	DomainFrontingPort          uint   `toml:"domain-fronting-port" json:"domainFrontingPort,omitempty"`
 	DomainFrontingIP            string `toml:"domain-fronting-ip" json:"domainFrontingIp,omitempty"`
 	DomainFrontingProxyProtocol bool   `toml:"domain-fronting-proxy-protocol" json:"domainFrontingProxyProtocol,omitempty"`

internal/config/type_url.go

@@ -0,0 +1,53 @@
+package config
+
+import (
+	"fmt"
+	"net/url"
+)
+
+type TypeURL struct {
+	Value *url.URL
+}
+
+func (t *TypeURL) Set(value string) error {
+	parsedURL, err := url.Parse(value)
+	if err != nil {
+		return fmt.Errorf("value is not correct URL (%s): %w", value, err)
+	}
+
+	if parsedURL.Host == "" {
+		return fmt.Errorf("url has to have a schema: %s", value)
+	}
+
+	if parsedURL.Scheme != "http" && parsedURL.Scheme != "https" {
+		return fmt.Errorf("unsupported schema: %s", parsedURL.Scheme)
+	}
+
+	t.Value = parsedURL
+
+	return nil
+}
+
+func (t *TypeURL) Get(defaultValue *url.URL) *url.URL {
+	if t.Value == nil {
+		return defaultValue
+	}
+
+	return t.Value
+}
+
+func (t *TypeURL) UnmarshalText(data []byte) error {
+	return t.Set(string(data))
+}
+
+func (t TypeURL) MarshalText() ([]byte, error) {
+	return []byte(t.String()), nil
+}
+
+func (t TypeURL) String() string {
+	if t.Value == nil {
+		return ""
+	}
+
+	return t.Value.String()
+}

mtglib/proxy.go

@@ -361,8 +361,17 @@ func NewProxy(opts ProxyOpts) (*Proxy, error) {
 	proxy.doppelGanger.Run()
 
 	if opts.AutoUpdate {
-		proxy.configUpdater.Run(ctx, dc.PublicConfigUpdateURLv4, "tcp4")
-		proxy.configUpdater.Run(ctx, dc.PublicConfigUpdateURLv6, "tcp6")
+		ipv4UpdateURL := opts.AutoUpdateURLv4
+		if ipv4UpdateURL == "" {
+			ipv4UpdateURL = dc.PublicConfigUpdateURLv4
+		}
+		proxy.configUpdater.Run(ctx, ipv4UpdateURL, "tcp4")
+
+		ipv6UpdateURL := opts.AutoUpdateURLv6
+		if ipv6UpdateURL == "" {
+			ipv6UpdateURL = dc.PublicConfigUpdateURLv6
+		}
+		proxy.configUpdater.Run(ctx, ipv6UpdateURL, "tcp6")
 	}
 
 	pool, err := ants.NewPoolWithFunc(opts.getConcurrency(),

mtglib/proxy_opts.go

@@ -91,6 +91,16 @@ type ProxyOpts struct {
 	// This is an optional setting.
 	AutoUpdate bool
 
+	// AutoUpdate URL to update Telegram IPv4 proxy list
+	//
+	// This is an optional setting.
+	AutoUpdateURLv4 string
+
+	// AutoUpdate URL to update Telegram IPv6 proxy list
+	//
+	// This is an optional setting.
+	AutoUpdateURLv6 string
+
 	// DomainFrontingPort is a port we use to connect to a fronting domain.
 	//
 	// This is required because secret does not specify a port. It specifies a