Add proxy support and instance-specific configurations

Enhanced the application with proxy capabilities, allowing instance-specific and global proxy configurations. Updated browser manager and API to incorporate proxy settings, introduced a new proxy server package, and refined configuration management to support these changes.

Author: luispater

.gitignore

@@ -10,4 +10,5 @@ docs/
 chromium/
 cert/ca.crt
 cert/ca.key
-*.log
+*.log
+.junie/

README.md

@@ -88,16 +88,18 @@ browser:
     - "--lang=en-US"
     - "--accept-lang=en-US"
   user-data-dir: "/anyAIProxyAPI/user-data-dir"
+  proxy-url: "http://user:[email protected]:8080/" # proxy url for browser, if instance-alone is false, this proxy setting will be ignored
 api-port: "2048"
 headless: false
+instance-alone: true # if true, each instance will have its own browser instance
 logfile: "any-ai-proxy.log"
 tokens: # Global tokens for API validation (optional)
   - "global-token-1"
   - "global-token-2"
 instance:
   - name: "gemini-aistudio"
     adapter: "gemini-aistudio"
-    proxy-url: ""
+    proxy-url: "socks5://user:[email protected]:1080/" # proxy url for each instance browser, if instance-alone is true, this proxy setting will be used
     url: "https://aistudio.google.com/prompts/new_chat"
     sniff-url:
       - "https://alkalimakersuite-pa.clients6.google.com/$rpc/google.internal.alkali.applications.makersuite.v1.MakerSuiteService/GenerateContent"
@@ -113,7 +115,7 @@ instance:
       - "gemini-token-4"
   - name: "chatgpt"
     adapter: "chatgpt"
-    proxy-url: ""
+    proxy-url: "" # proxy url for each instance browser, if this setting is empty, the browser will be directly connected to the internet
     url: "https://chatgpt.com/"
     sniff-url:
       - "https://chatgpt.com/backend-api/conversation"
@@ -146,12 +148,15 @@ instance:
   - `fingerprint-chromium-path`: Path to Fingerprint Chromium browser
   - `args`: Browser launch arguments
   - `user-data-dir`: User data directory
+  - `proxy-url`: Proxy URL for browser, if `instance-alone` is false, this proxy setting will be ignored
 - `api-port`: Port for the API server
 - `headless`: Run browser in headless mode
+- `instance-alone`: Run each instance will have its own browser instance
 - `tokens`: Global tokens for API validation (optional)
 - `instance`: Array of AI service instances to manage. Each instance has its own configuration
   - `name`: Instance name
   - `adapter`: Adapter name (corresponds to different AI services)
+  - `proxy-url`: Proxy URL for each instance browser, if `instance-alone` is false, this proxy setting will be ignored
   - `url`: AI service URL
   - `sniff-url`: URL patterns for intercepting responses
   - `auth`: Authentication configuration
@@ -304,7 +309,9 @@ For detailed information about the runner system, see [runner.md](runner.md).
 │   ├── browser/               # Browser management
 │   │   └── chrome/            # ChromeDP manager
 │   ├── config/                # Configuration handling
+│   ├── html/                  # HTML content
 │   ├── method/                # Automation methods
+│   ├── proxy/                 # Proxy server
 │   ├── runner/                # Workflow execution engine
 │   └── utils/                 # Utility functions
 ├── runner/                    # Workflow configurations

internal/api/handlers.go

@@ -13,6 +13,7 @@ import (
 	"net/http"
 	"os"
 	"path/filepath"
+	"strconv"
 	"strings"
 	"sync"
 	"time"
@@ -108,6 +109,26 @@ func (h *APIHandlers) TakeScreenshot(c *gin.Context) {
 	}
 }
 
+func (h *APIHandlers) ProxyPac(c *gin.Context) {
+	c.Status(http.StatusOK)
+	c.Header("Content-Type", "application/x-ns-proxy-autoconfig")
+	if h.appConfig.InstanceAlone {
+		index, ok := c.GetQuery("index")
+		if ok {
+			i, err := strconv.ParseUint(index, 10, 32)
+			if err != nil {
+				_, _ = c.Writer.Write([]byte(`function FindProxyForURL(url, host) {return "PROXY 127.0.0.1:3120";}`))
+				return
+			}
+
+			port := 3120 + i
+			_, _ = c.Writer.Write([]byte(fmt.Sprintf(`function FindProxyForURL(url, host) {return "PROXY 127.0.0.1:%d";}`, port)))
+		}
+	} else {
+		_, _ = c.Writer.Write([]byte(`function FindProxyForURL(url, host) {return "PROXY 127.0.0.1:3120";}`))
+	}
+}
+
 // ChatCompletions handles the /v1/chat/completions endpoint
 func (h *APIHandlers) ChatCompletions(c *gin.Context) {
 	rawJson, err := c.GetRawData()

internal/api/server.go

@@ -106,6 +106,7 @@ func (s *Server) setupRoutes() {
 	})
 
 	s.engine.GET("/screenshot", s.handlers.TakeScreenshot)
+	s.engine.GET("/proxy.pac", s.handlers.ProxyPac)
 }
 
 // Start starts the API server

internal/browser/chrome/manager.go

@@ -24,7 +24,7 @@ type Manager struct {
 
 // NewManager creates a new Chromedp Manager instance.
 // It initializes the allocator context but does not launch the browser yet.
-func NewManager(appConfig *config.AppConfig) (*Manager, error) {
+func NewManager(appConfig *config.AppConfig, indies ...int) (*Manager, error) {
 	if appConfig == nil {
 		return nil, fmt.Errorf("appConfig cannot be nil")
 	}
@@ -41,9 +41,8 @@ func NewManager(appConfig *config.AppConfig) (*Manager, error) {
 		chromedp.NoFirstRun,
 		chromedp.NoDefaultBrowserCheck,
 		chromedp.Flag("fingerprint", "1000"),
-		// chromedp.Flag("ignore-certificate-errors", true),
-		// chromedp.Flag("proxy-pac-url", "http://127.0.0.1:2048/proxy.pac"),
 
+		// chromedp.Flag("ignore-certificate-errors", true),
 		// chromedp.Flag("disable-background-timer-throttling", true),
 		// chromedp.Flag("disable-backgrounding-occluded-windows", true),
 		// chromedp.Flag("disable-renderer-backgrounding", true),
@@ -56,6 +55,16 @@ func NewManager(appConfig *config.AppConfig) (*Manager, error) {
 		// chromedp.Flag("remote-debugging-port", "9222"), // 如果需要远程调试
 	}
 
+	if len(indies) > 0 {
+		if appConfig.Instance[indies[0]].ProxyURL != "" {
+			opts = append(opts, chromedp.Flag("proxy-pac-url", fmt.Sprintf("http://127.0.0.1:2048/proxy.pac?index=%d", indies[0])))
+		}
+	} else {
+		if appConfig.Browser.ProxyURL != "" {
+			opts = append(opts, chromedp.Flag("proxy-pac-url", "http://127.0.0.1:2048/proxy.pac"))
+		}
+	}
+
 	if execPath != "" {
 		opts = append(opts, chromedp.ExecPath(execPath))
 	}

internal/config/config.go

@@ -22,6 +22,7 @@ type AppConfigBrowser struct {
 	FingerprintChromiumPath string   `yaml:"fingerprint-chromium-path"`
 	Args                    []string `yaml:"args"`
 	UserDataDir             string   `yaml:"user-data-dir,omitempty"`
+	ProxyURL                string   `yaml:"proxy-url"`
 }
 type AppConfigRunner struct {
 	Init            string `yaml:"init"`