From e1a95fbc6250ca91fd148ac491150bf8826c4a68 Mon Sep 17 00:00:00 2001 From: ben Gutier Date: Tue, 12 Sep 2023 16:07:49 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=96=87=E4=BB=B6=E5=A4=B9?= =?UTF-8?q?=E5=90=8D=E7=A7=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/define.go | 32 ++++++++++ src/getCopilotToke.go | 142 ++++++++++++++++++++++++++++++++++++++++++ src/go.mod | 39 ++++++++++++ src/go.sum | 104 +++++++++++++++++++++++++++++++ src/main.go | 21 +++++++ src/server.go | 121 +++++++++++++++++++++++++++++++++++ src/showMsg.go | 48 ++++++++++++++ 7 files changed, 507 insertions(+) create mode 100644 src/define.go create mode 100644 src/getCopilotToke.go create mode 100644 src/go.mod create mode 100644 src/go.sum create mode 100644 src/main.go create mode 100644 src/server.go create mode 100644 src/showMsg.go diff --git a/src/define.go b/src/define.go new file mode 100644 index 0000000..e1fd34a --- /dev/null +++ b/src/define.go @@ -0,0 +1,32 @@ +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/src/getCopilotToke.go b/src/getCopilotToke.go new file mode 100644 index 0000000..f6827cd --- /dev/null +++ b/src/getCopilotToke.go @@ -0,0 +1,142 @@ +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/src/go.mod b/src/go.mod new file mode 100644 index 0000000..87aa372 --- /dev/null +++ b/src/go.mod @@ -0,0 +1,39 @@ +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/src/go.sum b/src/go.sum new file mode 100644 index 0000000..144c731 --- /dev/null +++ b/src/go.sum @@ -0,0 +1,104 @@ +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/src/main.go b/src/main.go new file mode 100644 index 0000000..411290e --- /dev/null +++ b/src/main.go @@ -0,0 +1,21 @@ +package main + +func main() { + // 初始化配置文件 + configFile = initConfig() + + // 创建Gin引擎 + engine := setupGinEngine() + + // 初始化有效的token列表 + initValidTokenList() + + // 定义路由 + setupRoutes(engine) + + // 初始化并启动服务器 + initAndStartServer(engine) + + // 显示信息 + showMsg() +} diff --git a/src/server.go b/src/server.go new file mode 100644 index 0000000..6395ab4 --- /dev/null +++ b/src/server.go @@ -0,0 +1,121 @@ +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/src/showMsg.go b/src/showMsg.go new file mode 100644 index 0000000..b56a42f --- /dev/null +++ b/src/showMsg.go @@ -0,0 +1,48 @@ +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) // + } +}