diff --git a/README.md b/README.md index 0056389..c018b1d 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,8 @@ +测试版已经支持全部接口代理， +所有也就是代码提示也走代理。:stuck_out_tongue: + +![输入图片说明](readme/screenshots.gif) +*** # share-copilot - 作为代理服务器，中转copilot插件的API的请求 @@ -5,6 +10,8 @@ - 支持多用户共享一个token - 优化请求逻辑，降低token被ban概率（别万人骑基本不可能封） ![软件系统网络架构.png](https://img1.imgtp.com/2023/09/10/qTL8A2u9.png) + +*** # 一、自行编译： ``` 测试环境:Linux4.18.0-305.3.1.el8.x86_64 GNU/Linux @@ -28,7 +35,7 @@ go version cd /share-copilot/source go build ``` - +*** # 二、使用现成： ### 1.安装 diff --git a/readme/screenshots.gif b/readme/screenshots.gif new file mode 100644 index 0000000..8f90102 Binary files /dev/null and b/readme/screenshots.gif differ diff --git a/source/.gitignore b/source/.gitignore deleted file mode 100644 index 1117f4d..0000000 --- a/source/.gitignore +++ /dev/null @@ -1,8 +0,0 @@ -/*.pem -/*.idea -/*.gitignore -/*.json -/config.json -/*.exe -/share-copilot -/*.bat diff --git a/source/define.go b/source/define.go deleted file mode 100644 index e1fd34a..0000000 --- a/source/define.go +++ /dev/null @@ -1,32 +0,0 @@ -package main - -import ( - "sync" -) - -type Config struct { - Server struct { - Domain string `json:"domain"` - Host string `json:"host"` - Port int `json:"port"` - CertPath string `json:"certPath"` - KeyPath string `json:"keyPath"` - } `json:"server"` - CopilotConfig struct { - GithubApiUrl string `json:"github_api_url"` - Token []string `json:"token"` - } `json:"copilot_config"` - Verification string `json:"verification"` -} - -var ( - //初始化需要返回给客户端的响应体 - tokenMap = make(map[string]map[string]interface{}) - //有效的token列表 - validTokenList = make(map[string]bool) - requestCountMutex sync.Mutex - githubApiCount = 0 - requestCount = 0 - successCount = 0 - configFile Config -) diff --git a/source/getCopilotToke.go b/source/getCopilotToke.go deleted file mode 100644 index f6827cd..0000000 --- a/source/getCopilotToke.go +++ /dev/null @@ -1,142 +0,0 @@ -package main - -import ( - "encoding/json" - "errors" - "github.com/gin-gonic/gin" - "github.com/go-resty/resty/v2" - "math/rand" - "net/http" - "strings" - "sync" - "time" -) - -// 初始化有效的github token列表 -func initValidTokenList() { - //为了安全起见，应该等待请求完成并处理其响应。 - var wg sync.WaitGroup - for _, token := range configFile.CopilotConfig.Token { - wg.Add(1) - go func(token string) { - defer wg.Done() - if getGithubApi(token) { - validTokenList[token] = true - } - }(token) - } - wg.Wait() -} - -// 请求github api -func getGithubApi(token string) bool { - githubApiCount++ - // 设置请求头 - headers := map[string]string{ - "Authorization": "token " + token, - /*"editor-version": c.GetHeader("editor-version"), - "editor-plugin-version": c.GetHeader("editor-plugin-version"), - "user-agent": c.GetHeader("user-agent"), - "accept": c.GetHeader("accept"), - "accept-encoding": c.GetHeader("accept-encoding"),*/ - } - // 发起GET请求 - response, err := resty.New().R(). - SetHeaders(headers). - Get(configFile.CopilotConfig.GithubApiUrl) - if err != nil { - // 处理请求错误 - return false - } - // 判断响应状态码 - if response.StatusCode() == http.StatusOK { - // 响应状态码为200 OK - respDataMap := map[string]interface{}{} - err = json.Unmarshal(response.Body(), &respDataMap) - if err != nil { - // 处理JSON解析错误 - return false - } - //token map - tokenMap[token] = respDataMap - return true - } else { - // 处理其他状态码 - delete(validTokenList, token) - return false - } -} - -// 获取copilot token -func getGithubToken() gin.HandlerFunc { - return func(c *gin.Context) { - requestCount++ - if err := verifyRequest(c); err != nil { - badRequest(c) - return - } - token := getRandomToken(validTokenList) - if respDataMap, exists := getTokenData(token); exists { - if !isTokenExpired(respDataMap) { - proxyResp(c, respDataMap) - return - } - } - if getGithubApi(token) { - proxyResp(c, tokenMap[token]) - } else { - badRequest(c) - } - } -} - -// 验证请求代理请求token -func verifyRequest(c *gin.Context) error { - if configFile.Verification != "" { - token := c.GetHeader("Authorization") - tokenStr := strings.ReplaceAll(token, " ", "") - configCert := strings.ReplaceAll(configFile.Verification, " ", "") - if tokenStr != "token"+configCert { - return errors.New("verification failed") - } - } - return nil -} - -// 从map中获取github token对应的copilot token -func getTokenData(token string) (map[string]interface{}, bool) { - respDataMap, exists := tokenMap[token] - return respDataMap, exists -} - -// 检测copilot token是否过期 -func isTokenExpired(respDataMap map[string]interface{}) bool { - if expiresAt, ok := respDataMap["expires_at"].(float64); ok { - currentTime := time.Now().Unix() - expiresAtInt64 := int64(expiresAt) - return expiresAtInt64 <= currentTime+60 - } - return true -} - -// 重置请求计数 -func resetRequestCount() { - requestCountMutex.Lock() - defer requestCountMutex.Unlock() - requestCount = 0 - successCount = 0 -} - -// 从map中随机获取一个github token -func getRandomToken(m map[string]bool) string { - keys := make([]string, 0, len(m)) - for k := range m { - keys = append(keys, k) - } - if len(keys) == 0 { - return "" // 返回空字符串或处理其他错误情况 - } - r := rand.New(rand.NewSource(time.Now().UnixNano())) - randomIndex := r.Intn(len(keys)) - return keys[randomIndex] -} diff --git a/source/go.mod b/source/go.mod deleted file mode 100644 index 87aa372..0000000 --- a/source/go.mod +++ /dev/null @@ -1,39 +0,0 @@ -module share-copilot - -go 1.21 - -require ( - github.com/fatih/color v1.15.0 - github.com/gin-gonic/gin v1.9.1 - github.com/go-resty/resty/v2 v2.7.0 - github.com/nsf/termbox-go v1.1.1 -) - -require ( - github.com/bytedance/sonic v1.9.1 // indirect - github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 // indirect - github.com/gabriel-vasile/mimetype v1.4.2 // indirect - github.com/gin-contrib/sse v0.1.0 // indirect - github.com/go-playground/locales v0.14.1 // indirect - github.com/go-playground/universal-translator v0.18.1 // indirect - github.com/go-playground/validator/v10 v10.14.0 // indirect - github.com/goccy/go-json v0.10.2 // indirect - github.com/json-iterator/go v1.1.12 // indirect - github.com/klauspost/cpuid/v2 v2.2.4 // indirect - github.com/leodido/go-urn v1.2.4 // indirect - github.com/mattn/go-colorable v0.1.13 // indirect - github.com/mattn/go-isatty v0.0.19 // indirect - github.com/mattn/go-runewidth v0.0.9 // indirect - github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect - github.com/modern-go/reflect2 v1.0.2 // indirect - github.com/pelletier/go-toml/v2 v2.0.8 // indirect - github.com/twitchyliquid64/golang-asm v0.15.1 // indirect - github.com/ugorji/go/codec v1.2.11 // indirect - golang.org/x/arch v0.3.0 // indirect - golang.org/x/crypto v0.9.0 // indirect - golang.org/x/net v0.10.0 // indirect - golang.org/x/sys v0.8.0 // indirect - golang.org/x/text v0.9.0 // indirect - google.golang.org/protobuf v1.30.0 // indirect - gopkg.in/yaml.v3 v3.0.1 // indirect -) diff --git a/source/go.sum b/source/go.sum deleted file mode 100644 index 144c731..0000000 --- a/source/go.sum +++ /dev/null @@ -1,104 +0,0 @@ -github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM= -github.com/bytedance/sonic v1.9.1 h1:6iJ6NqdoxCDr6mbY8h18oSO+cShGSMRGCEo7F2h0x8s= -github.com/bytedance/sonic v1.9.1/go.mod h1:i736AoUSYt75HyZLoJW9ERYxcy6eaN6h4BZXU064P/U= -github.com/chenzhuoyu/base64x v0.0.0-20211019084208-fb5309c8db06/go.mod h1:DH46F32mSOjUmXrMHnKwZdA8wcEefY7UVqBKYGjpdQY= -github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 h1:qSGYFH7+jGhDF8vLC+iwCD4WpbV1EBDSzWkJODFLams= -github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311/go.mod h1:b583jCggY9gE99b6G5LEC39OIiVsWj+R97kbl5odCEk= -github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= -github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= -github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= -github.com/fatih/color v1.15.0 h1:kOqh6YHBtK8aywxGerMG2Eq3H6Qgoqeo13Bk2Mv/nBs= -github.com/fatih/color v1.15.0/go.mod h1:0h5ZqXfHYED7Bhv2ZJamyIOUej9KtShiJESRwBDUSsw= -github.com/gabriel-vasile/mimetype v1.4.2 h1:w5qFW6JKBz9Y393Y4q372O9A7cUSequkh1Q7OhCmWKU= -github.com/gabriel-vasile/mimetype v1.4.2/go.mod h1:zApsH/mKG4w07erKIaJPFiX0Tsq9BFQgN3qGY5GnNgA= -github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE= -github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI= -github.com/gin-gonic/gin v1.9.1 h1:4idEAncQnU5cB7BeOkPtxjfCSye0AAm1R0RVIqJ+Jmg= -github.com/gin-gonic/gin v1.9.1/go.mod h1:hPrL7YrpYKXt5YId3A/Tnip5kqbEAP+KLuI3SUcPTeU= -github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s= -github.com/go-playground/assert/v2 v2.2.0/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4= -github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA= -github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY= -github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY= -github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY= -github.com/go-playground/validator/v10 v10.14.0 h1:vgvQWe3XCz3gIeFDm/HnTIbj6UGmg/+t63MyGU2n5js= -github.com/go-playground/validator/v10 v10.14.0/go.mod h1:9iXMNT7sEkjXb0I+enO7QXmzG6QCsPWY4zveKFVRSyU= -github.com/go-resty/resty/v2 v2.7.0 h1:me+K9p3uhSmXtrBZ4k9jcEAfJmuC8IivWHwaLZwPrFY= -github.com/go-resty/resty/v2 v2.7.0/go.mod h1:9PWDzw47qPphMRFfhsyk0NnSgvluHcljSMVIq3w7q0I= -github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU= -github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I= -github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk= -github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU= -github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= -github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= -github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM= -github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo= -github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg= -github.com/klauspost/cpuid/v2 v2.2.4 h1:acbojRNwl3o09bUq+yDCtZFc1aiwaAAxtcn8YkZXnvk= -github.com/klauspost/cpuid/v2 v2.2.4/go.mod h1:RVVoqg1df56z8g3pUjL/3lE5UfnlrJX8tyFgg4nqhuY= -github.com/leodido/go-urn v1.2.4 h1:XlAE/cm/ms7TE/VMVoduSpNBoyc2dOxHs5MZSwAN63Q= -github.com/leodido/go-urn v1.2.4/go.mod h1:7ZrI8mTSeBSHl/UaRyKQW1qZeMgak41ANeCNaVckg+4= -github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA= -github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg= -github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM= -github.com/mattn/go-isatty v0.0.19 h1:JITubQf0MOLdlGRuRq+jtsDlekdYPia9ZFsB8h/APPA= -github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= -github.com/mattn/go-runewidth v0.0.9 h1:Lm995f3rfxdpd6TSmuVCHVb/QhupuXlYr8sCI/QdE+0= -github.com/mattn/go-runewidth v0.0.9/go.mod h1:H031xJmbD/WCDINGzjvQ9THkh0rPKHF+m2gUSrubnMI= -github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= -github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg= -github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= -github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M= -github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk= -github.com/nsf/termbox-go v1.1.1 h1:nksUPLCb73Q++DwbYUBEglYBRPZyoXJdrj5L+TkjyZY= -github.com/nsf/termbox-go v1.1.1/go.mod h1:T0cTdVuOwf7pHQNtfhnEbzHbcNyCEcVU4YPpouCbVxo= -github.com/pelletier/go-toml/v2 v2.0.8 h1:0ctb6s9mE31h0/lhu+J6OPmVeDxJn+kYnJc2jZR9tGQ= -github.com/pelletier/go-toml/v2 v2.0.8/go.mod h1:vuYfssBdrU2XDZ9bYydBu6t+6a6PYNcZljzZR9VXg+4= -github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= -github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= -github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= -github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= -github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= -github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= -github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= -github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= -github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= -github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= -github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= -github.com/stretchr/testify v1.8.3 h1:RP3t2pwF7cMEbC1dqtB6poj3niw/9gnV4Cjg5oW5gtY= -github.com/stretchr/testify v1.8.3/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo= -github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI= -github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08= -github.com/ugorji/go/codec v1.2.11 h1:BMaWp1Bb6fHwEtbplGBGJ498wD+LKlNSl25MjdZY4dU= -github.com/ugorji/go/codec v1.2.11/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg= -golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8= -golang.org/x/arch v0.3.0 h1:02VY4/ZcO/gBOH6PUaoiptASxtXU10jazRCP865E97k= -golang.org/x/arch v0.3.0/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8= -golang.org/x/crypto v0.9.0 h1:LF6fAI+IutBocDJ2OT0Q1g8plpYljMZ4+lty+dsqw3g= -golang.org/x/crypto v0.9.0/go.mod h1:yrmDGqONDYtNj3tH8X9dzUun2m2lzPa9ngI6/RUPGR0= -golang.org/x/net v0.0.0-20211029224645-99673261e6eb/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y= -golang.org/x/net v0.10.0 h1:X2//UzNDwYmtCLn7To6G58Wr6f5ahEAQgKNzv9Y951M= -golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg= -golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= -golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= -golang.org/x/sys v0.0.0-20220704084225-05e143d24a9e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= -golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= -golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= -golang.org/x/sys v0.8.0 h1:EBmGv8NaZBZTWvrbjNoL6HVt+IVy3QDQpJs7VRIw3tU= -golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= -golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= -golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= -golang.org/x/text v0.9.0 h1:2sjJmO8cDvYveuX97RDLsxlyUxLl+GHoLxBiRdHllBE= -golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8= -golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= -golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4= -golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= -google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw= -google.golang.org/protobuf v1.30.0 h1:kPPoIgf3TsEvrm0PFe15JQ+570QVxYzEvvHqChK+cng= -google.golang.org/protobuf v1.30.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I= -gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM= -gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= -gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= -gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= -gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= -rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4= diff --git a/source/main.go b/source/main.go deleted file mode 100644 index 411290e..0000000 --- a/source/main.go +++ /dev/null @@ -1,21 +0,0 @@ -package main - -func main() { - // 初始化配置文件 - configFile = initConfig() - - // 创建Gin引擎 - engine := setupGinEngine() - - // 初始化有效的token列表 - initValidTokenList() - - // 定义路由 - setupRoutes(engine) - - // 初始化并启动服务器 - initAndStartServer(engine) - - // 显示信息 - showMsg() -} diff --git a/source/server.go b/source/server.go deleted file mode 100644 index 6395ab4..0000000 --- a/source/server.go +++ /dev/null @@ -1,121 +0,0 @@ -package main - -import ( - "crypto/tls" - "encoding/json" - "github.com/gin-gonic/gin" - "io" - "log" - "net/http" - "os" - "path/filepath" - "strconv" - "strings" -) - -// 初始化配置文件 -func initConfig() Config { - // 读取配置文件 - exePath, err := os.Executable() - if err != nil { - panic(err) - } - // 获取执行文件所在目录 - exeDir := filepath.Dir(exePath) - configFile, err := os.Open(exeDir + "/config.json") - if err != nil { - panic("file \"./config.json\" not found") - } - defer func(configFile *os.File) { - err := configFile.Close() - if err != nil { - panic("close file \"./config.json\" err") - } - }(configFile) - decoder := json.NewDecoder(configFile) - config := Config{} - err = decoder.Decode(&config) - if err != nil { - panic("config format err") - } - return config -} - -// 创建和配置Gin引擎 -func setupGinEngine() *gin.Engine { - gin.SetMode(gin.ReleaseMode) - gin.DefaultWriter = io.Discard - engine := gin.New() - // 设置信任的代理 - if err := engine.SetTrustedProxies([]string{"127.0.0.1"}); err != nil { - log.Fatal(err) - } - return engine -} - -// 定义路由和中间件 -func setupRoutes(engine *gin.Engine) { - domainDefault := engine.Group("/", DomainMiddleware(configFile.Server.Domain)) - domainDefault.GET("/copilot_internal/v2/token", getGithubToken()) -} - -// DomainMiddleware 域名中间件 -func DomainMiddleware(domain string) gin.HandlerFunc { - return func(c *gin.Context) { - // 检查域名是否匹配 - requestDomain := strings.Split(c.Request.Host, ":")[0] - if requestDomain == domain || requestDomain == "127.0.0.1" { - c.Next() - } else { - c.String(403, "Forbidden") - c.Abort() - } - } -} - -// 初始化和启动服务器 -func initAndStartServer(engine *gin.Engine) { - listenAddress := configFile.Server.Host + ":" + strconv.Itoa(configFile.Server.Port) - server := createTLSServer(engine, listenAddress) - go func() { - if configFile.Server.Port != 443 { - err := engine.Run(listenAddress) - log.Fatal(err) - } else { - err := server.ListenAndServeTLS(configFile.Server.CertPath, configFile.Server.KeyPath) - log.Fatal(err) - } - }() -} - -// 创建TLS服务器配置 -func createTLSServer(engine *gin.Engine, address string) *http.Server { - return &http.Server{ - Addr: address, - TLSConfig: &tls.Config{ - NextProtos: []string{"http/1.1", "http/1.2", "http/2"}, - }, - Handler: engine, - } -} - -// 本服务器响应 -func proxyResp(c *gin.Context, respDataMap map[string]interface{}) { - // 将map转换为JSON字符串 - responseJSON, err := json.Marshal(respDataMap) - if err != nil { - c.JSON(http.StatusInternalServerError, gin.H{"error": "JSON marshaling error"}) - } - // 请求成功统计 - successCount++ - // 将JSON字符串作为响应体返回 - c.Header("Content-Type", "application/json") - c.String(http.StatusOK, string(responseJSON)) -} - -// 请求错误 -func badRequest(c *gin.Context) { - c.JSON(http.StatusBadRequest, gin.H{ - "message": "Bad credentials", - "documentation_url": "https://docs.github.com/rest"}) -} diff --git a/source/showMsg.go b/source/showMsg.go deleted file mode 100644 index b56a42f..0000000 --- a/source/showMsg.go +++ /dev/null @@ -1,48 +0,0 @@ -package main - -import ( - "fmt" - "github.com/fatih/color" - "strconv" - "time" -) - -// 控制台显示信息 -func showMsg() { - var url = "" - if configFile.Server.Port == 80 { - url = "http://" + configFile.Server.Domain - } else if configFile.Server.Port == 443 { - url = "https://" + configFile.Server.Domain - } else { - url = "http://" + configFile.Server.Domain + ":" + strconv.Itoa(configFile.Server.Port) - } - var jetStr = color.WhiteString("[Jetbrains]") - var vsStr = color.WhiteString("[Vscode]") - var valid = color.WhiteString("[Valid tokens]") - fmt.Println(jetStr + ": " + color.HiBlueString(url+"/copilot_internal/v2/token")) - fmt.Println(vsStr + ": " + color.HiBlueString(url)) - fmt.Println(valid + ": " + color.HiBlueString(strconv.Itoa(len(validTokenList)))) - fmt.Println(color.WhiteString("-----------------------------------------------------------------------")) - for { - requestCountMutex.Lock() - sCount := successCount - tCount := requestCount - gCount := githubApiCount - requestCountMutex.Unlock() - currentTime := time.Now().Format("2006-01-02 15:04:05") - if "00:00:00" == currentTime { - resetRequestCount() - } - var s2 = color.WhiteString("[Succeed]") - var s3 = color.WhiteString("[Failed]") - var s4 = color.WhiteString("[GithubApi]") - // 打印文本 - fmt.Printf("\033[G%s - %s: %s %s: %s %s: %s ", - color.HiYellowString(currentTime), - s2, color.GreenString(strconv.Itoa(sCount)), - s3, color.RedString(strconv.Itoa(tCount-sCount)), - s4, color.CyanString(strconv.Itoa(gCount))) - time.Sleep(1 * time.Second) // - } -} diff --git a/source/test.go b/source/test.go deleted file mode 100644 index 1913fb8..0000000 --- a/source/test.go +++ /dev/null @@ -1,86 +0,0 @@ -package main - -import ( - "encoding/json" - "github.com/gin-gonic/gin" - "net/http" -) - -func getGithubTest(c *gin.Context, token string) { - githubApiCount++ - data1 := ` - { - "chat_enabled": false, - "code_quote_enabled": false, - "code_quote_v2_enabled": false, - "copilotignore_enabled": false, - "expires_at": 3194360727, - "prompt_8k": true, - "public_suggestions": "disabled", - "refresh_in": 1500, - "sku": "free_educational", - "telemetry": "disabled", - "token": "tid=;exp=1694360727;sku=free_educational;st=dotcom;8kp=1:", - "tracking_id": "" - } - ` - data2 := ` - { - "chat_enabled": false, - "code_quote_enabled": false, - "code_quote_v2_enabled": false, - "copilotignore_enabled": false, - "expires_at": 2294360727, - "prompt_8k": true, - "public_suggestions": "disabled", - "refresh_in": 1500, - "sku": "free_educational", - "telemetry": "disabled", - "token": "tid=;exp=1694360727;sku=free_educational;st=dotcom;8kp=1:", - "tracking_id": "" - } - ` - data3 := ` - { - "chat_enabled": false, - "code_quote_enabled": false, - "code_quote_v2_enabled": false, - "copilotignore_enabled": false, - "expires_at": 3394360727, - "prompt_8k": true, - "public_suggestions": "disabled", - "refresh_in": 1500, - "sku": "free_educational", - "telemetry": "disabled", - "token": "tid=;exp=1694360727;sku=free_educational;st=dotcom;8kp=1:", - "tracking_id": "" - } - ` - //响应体map - var respDataMap = make(map[string]interface{}) - if token == "1" { - err := json.Unmarshal([]byte(data1), &respDataMap) - if err != nil { - // 处理JSON解析错误 - c.JSON(http.StatusInternalServerError, gin.H{"error": "JSON parsing error"}) - return - } - } else if token == "2" { - err := json.Unmarshal([]byte(data2), &respDataMap) - if err != nil { - // 处理JSON解析错误 - c.JSON(http.StatusInternalServerError, gin.H{"error": "JSON parsing error"}) - return - } - } else { - err := json.Unmarshal([]byte(data3), &respDataMap) - if err != nil { - // 处理JSON解析错误 - c.JSON(http.StatusInternalServerError, gin.H{"error": "JSON parsing error"}) - return - } - } - - //token map - tokenMap[token] = respDataMap -}