✨ copilot proxy

Author: v_frgfeng

api/common/common.go

@@ -0,0 +1,4 @@
+package common
+
+type CommonApi struct {
+}

api/copilot/copilot.go

@@ -478,3 +478,31 @@ func ClearAuthCount() {
 		}
 	}
 }
+
+var pacContent string
+var pacTpl = `function FindProxyForURL(url, host) {
+	if (host == 'githubusercontent.com' || host == 'githubcopilot.com' || host == 'cocopilot.net' || host == 'google.com') {
+		return '%s'
+	}
+	return 'DIRECT'
+}
+`
+var directPac = `function FindProxyForURL(url, host) {
+	return 'DIRECT'
+}
+`
+
+func (co *CopilotApi) ProxyPacHandler(ctx *gin.Context) {
+	proxy := global.Config.Copilot.Proxy.HTTP
+	if pacContent == "" && proxy != "" {
+		pacContent = fmt.Sprintf(pacTpl, proxy)
+	}
+	if pacContent == "" {
+		pacContent = directPac
+	}
+	if proxy != "" && pacContent == directPac {
+		pacContent = fmt.Sprintf(pacTpl, proxy)
+	}
+	ctx.Header("Content-Type", "application/x-ns-proxy-autoconfig")
+	ctx.String(http.StatusOK, pacContent)
+}

api/enter.go

@@ -1,6 +1,7 @@
 package api
 
 import (
+	"github.com/dalefengs/chat-api-proxy/api/common"
 	"github.com/dalefengs/chat-api-proxy/api/copilot"
 	"github.com/dalefengs/chat-api-proxy/api/genai"
 	"github.com/dalefengs/chat-api-proxy/api/openai"
@@ -10,6 +11,7 @@ type ApiGroup struct {
 	copilot.CopilotApi
 	openai.OpenApi
 	genai.GenApi
+	common.CommonApi
 }
 
 var ApiGroupApp = new(ApiGroup)

config.docker.yaml

@@ -21,7 +21,11 @@ copilot:
 
   # cocopilot
   coTokenUrl: https://api.cocopilot.net/copilot_internal/v2/token
+  proxy:
+    HTTP: ""
+    HTTPS: ""
 
+    
 gemini:
   # https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key=${API_KEY}
   # https://api-gm.xfjy.in

config.yaml

@@ -22,6 +22,10 @@ copilot:
   # cocopilot
   coTokenUrl: https://api.cocopilot.net/copilot_internal/v2/token
 
+  proxy:
+    HTTP: ""
+    HTTPS: ""
+
 gemini:
   # https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key=${API_KEY}
   # https://api-gm.xfjy.in

core/environment.go

@@ -22,4 +22,10 @@ func InitEnvironmentVar() {
 	if env := os.Getenv("CO_TOKEN_URL"); env != "" {
 		global.Config.Copilot.CoTokenURL = env
 	}
+	if env := os.Getenv("HTTP_PROXY"); env != "" {
+		global.Config.Copilot.Proxy.HTTP = env
+	}
+	if env := os.Getenv("HTTPS_PROXY"); env != "" {
+		global.Config.Copilot.Proxy.HTTPS = env
+	}
 }

global/config/copilot.go

@@ -4,4 +4,5 @@ type Copilot struct {
 	TokenURL       string `yaml:"tokenUrl"`
 	CoTokenURL     string `yaml:"coTokenUrl"`
 	CompletionsURL string `yaml:"completionsUrl"`
+	Proxy          Proxy  `yaml:"proxy"`
 }

global/config/system.go

@@ -5,3 +5,8 @@ type System struct {
 	Port         int    `mapstructure:"port" json:"port" yaml:"port"` // 端口值
 	RouterPrefix string `mapstructure:"router-prefix" json:"router-prefix" yaml:"router-prefix"`
 }
+
+type Proxy struct {
+	HTTP  string `mapstructure:"http" json:"http" yaml:"http"`
+	HTTPS string `mapstructure:"https" json:"https" yaml:"https"`
+}

router/copilot.go

@@ -5,23 +5,13 @@ import (
 	"github.com/gin-gonic/gin"
 )
 
-func (r *Router) InitRouter(Router *gin.RouterGroup) (R gin.IRoutes) {
-	baseRouter := Router.Group("")
-	copilotApi := api.ApiGroupApp.CopilotApi
-	{
-		baseRouter.GET("/copilot_internal/v2/token", copilotApi.CoTokenHandler)
-		baseRouter.POST("/v1/chat/completions", copilotApi.CompletionsHandler)
-		baseRouter.POST("/chat/completions", copilotApi.CompletionsOfficialHandler)
-	}
-	return baseRouter
-}
-
 func (r *Router) InitCopilotRouter(Router *gin.RouterGroup) (R gin.IRoutes) {
 	baseRouter := Router.Group("copilot")
 	copilotApi := api.ApiGroupApp.CopilotApi
 	{
 		baseRouter.GET("/copilot_internal/v2/token", copilotApi.TokenHandler) // 官方获取 token
 		baseRouter.POST("/v1/chat/completions", copilotApi.CompletionsHandler)
+		baseRouter.GET("/proxy.pac", copilotApi.ProxyPacHandler) // 自动配置代理
 	}
 	return baseRouter
 }

router/router.go

@@ -0,0 +1,18 @@
+package router
+
+import (
+	"github.com/dalefengs/chat-api-proxy/api"
+	"github.com/gin-gonic/gin"
+)
+
+func (r *Router) InitRouter(Router *gin.RouterGroup) (R gin.IRoutes) {
+	baseRouter := Router.Group("")
+	copilotApi := api.ApiGroupApp.CopilotApi
+	_ = api.ApiGroupApp.CommonApi
+	{
+		baseRouter.GET("/copilot_internal/v2/token", copilotApi.CoTokenHandler)
+		baseRouter.POST("/v1/chat/completions", copilotApi.CompletionsHandler)
+		baseRouter.POST("/chat/completions", copilotApi.CompletionsOfficialHandler)
+	}
+	return baseRouter
+}