feat: use proxy for update if configured

Author: lich0821

internal/service/update.go

@@ -29,10 +29,20 @@ func NewUpdateService(cfg *config.Config, s *storage.SQLiteStorage, version stri
     }
 }
 
+// syncProxyConfig syncs proxy configuration to updater
+func (u *UpdateService) syncProxyConfig() {
+    if proxyCfg := u.config.GetProxy(); proxyCfg != nil {
+        u.updater.SetProxyURL(proxyCfg.URL)
+    } else {
+        u.updater.SetProxyURL("")
+    }
+}
+
 // CheckForUpdates checks if a new version is available
 func (u *UpdateService) CheckForUpdates() string {
     logger.Info("CheckForUpdates called, current version: %s", u.version)
 
+    u.syncProxyConfig()
     info, err := u.updater.CheckForUpdates()
     if err != nil {
         logger.Error("CheckForUpdates failed: %v", err)
@@ -109,6 +119,7 @@ func (u *UpdateService) SkipVersion(version string) error {
 
 // DownloadUpdate downloads the update file
 func (u *UpdateService) DownloadUpdate(url, filename string) error {
+    u.syncProxyConfig()
     return u.updater.DownloadUpdate(url, filename)
 }
 

internal/updater/downloader.go

@@ -27,6 +27,7 @@ type Downloader struct {
 	progress   DownloadProgress
 	mu         sync.RWMutex
 	cancelChan chan struct{}
+	proxyURL   string
 }
 
 // NewDownloader creates a new downloader
@@ -36,6 +37,13 @@ func NewDownloader() *Downloader {
 	}
 }
 
+// SetProxyURL sets the proxy URL for downloads
+func (d *Downloader) SetProxyURL(proxyURL string) {
+	d.mu.Lock()
+	d.proxyURL = proxyURL
+	d.mu.Unlock()
+}
+
 // Download downloads a file from URL to destination
 func (d *Downloader) Download(url, destPath string) error {
 	d.mu.Lock()
@@ -44,6 +52,7 @@ func (d *Downloader) Download(url, destPath string) error {
 		FilePath: destPath,
 	}
 	d.cancelChan = make(chan struct{})
+	proxyURL := d.proxyURL
 	d.mu.Unlock()
 
 	// Create destination directory
@@ -73,6 +82,13 @@ func (d *Downloader) Download(url, destPath string) error {
 		},
 	}
 
+	// Apply proxy if configured
+	if proxyURL != "" {
+		if transport, err := createProxyTransport(proxyURL); err == nil {
+			client.Transport = transport
+		}
+	}
+
 	req, err := http.NewRequest("GET", url, nil)
 	if err != nil {
 		d.setError(fmt.Sprintf("failed to create request: %v", err))

internal/updater/github.go

@@ -5,9 +5,12 @@ import (
 	"fmt"
 	"io"
 	"net/http"
+	"net/url"
 	"runtime"
 	"strings"
 	"time"
+
+	"golang.org/x/net/proxy"
 )
 
 const (
@@ -32,9 +35,15 @@ type Asset struct {
 }
 
 // GetLatestRelease fetches the latest release from GitHub
-func GetLatestRelease() (*ReleaseInfo, error) {
+func GetLatestRelease(proxyURL string) (*ReleaseInfo, error) {
 	client := &http.Client{Timeout: httpTimeout}
 
+	if proxyURL != "" {
+		if transport, err := createProxyTransport(proxyURL); err == nil {
+			client.Transport = transport
+		}
+	}
+
 	req, err := http.NewRequest("GET", githubAPIURL, nil)
 	if err != nil {
 		return nil, fmt.Errorf("failed to create request: %w", err)
@@ -102,3 +111,35 @@ func getPlatformPattern() string {
 		return fmt.Sprintf("%s-%s", goos, goarch)
 	}
 }
+
+// createProxyTransport creates an http.Transport with proxy support
+func createProxyTransport(proxyURL string) (*http.Transport, error) {
+	parsed, err := url.Parse(proxyURL)
+	if err != nil {
+		return nil, err
+	}
+
+	transport := &http.Transport{}
+
+	switch parsed.Scheme {
+	case "socks5", "socks5h":
+		auth := &proxy.Auth{}
+		if parsed.User != nil {
+			auth.User = parsed.User.Username()
+			auth.Password, _ = parsed.User.Password()
+		} else {
+			auth = nil
+		}
+		dialer, err := proxy.SOCKS5("tcp", parsed.Host, auth, proxy.Direct)
+		if err != nil {
+			return nil, err
+		}
+		transport.Dial = dialer.Dial
+	case "http", "https":
+		transport.Proxy = http.ProxyURL(parsed)
+	default:
+		return nil, fmt.Errorf("unsupported proxy scheme: %s", parsed.Scheme)
+	}
+
+	return transport, nil
+}

internal/updater/updater.go

@@ -40,6 +40,7 @@ type Updater struct {
 	currentVersion string
 	downloader     *Downloader
 	downloadPath   string
+	proxyURL       string
 }
 
 // New creates a new updater
@@ -51,9 +52,15 @@ func New(currentVersion string) *Updater {
 	}
 }
 
+// SetProxyURL sets the proxy URL for HTTP requests
+func (u *Updater) SetProxyURL(proxyURL string) {
+	u.proxyURL = proxyURL
+	u.downloader.SetProxyURL(proxyURL)
+}
+
 // CheckForUpdates checks if a new version is available
 func (u *Updater) CheckForUpdates() (*UpdateInfo, error) {
-	release, err := GetLatestRelease()
+	release, err := GetLatestRelease(u.proxyURL)
 	if err != nil {
 		return nil, fmt.Errorf("failed to check updates: %w", err)
 	}