mirror of
				https://gitee.com/chuangxxt/share-copilot
				synced 2025-10-31 17:29:49 +00:00 
			
		
		
		
	2023.9.9
This commit is contained in:
		
						commit
						0929683600
					
				
							
								
								
									
										115
									
								
								README.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										115
									
								
								README.md
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,115 @@ | |||||||
|  | # share-copilot  | ||||||
|  | 
 | ||||||
|  | ``` | ||||||
|  | 测试环境:Linux4.18.0-305.3.1.el8.x86_64 GNU/Linux | ||||||
|  | ``` | ||||||
|  | ### 1.安装 | ||||||
|  | 
 | ||||||
|  | ``` | ||||||
|  | git clone https://gitee.com/chuangxxt/share-copilot.git | ||||||
|  | ``` | ||||||
|  | ``` | ||||||
|  | cd share-copilot | ||||||
|  | ``` | ||||||
|  | ``` | ||||||
|  | bash install.sh | ||||||
|  | ``` | ||||||
|  | 
 | ||||||
|  | ### 2.配置 | ||||||
|  | 
 | ||||||
|  | ##### 2.1 config.json文件说明 | ||||||
|  | 
 | ||||||
|  | ```js | ||||||
|  | domain//绑定域名 | ||||||
|  | 
 | ||||||
|  | host //ip | ||||||
|  | 
 | ||||||
|  | port //端口 80或443 | ||||||
|  | 
 | ||||||
|  | certPath//公钥路径-443必须 | ||||||
|  | 
 | ||||||
|  | keyPath//私钥路径-443必须 | ||||||
|  | 
 | ||||||
|  | github_api_url//默认不改 | ||||||
|  | 
 | ||||||
|  | token//自己的副驾驶token,可放多个,请求时随机使用 | ||||||
|  | //登录插件后在%userprofile%\AppData\Local\github-copilot\host.json     ghu_开头 | ||||||
|  | 
 | ||||||
|  | verification//自定义验证 | ||||||
|  | ``` | ||||||
|  | 
 | ||||||
|  | ##### 2.2 示例  | ||||||
|  | 
 | ||||||
|  | ```json | ||||||
|  | { | ||||||
|  | 	"server": { | ||||||
|  | 		"domain": "example.com", | ||||||
|  | 		"host": "0.0.0.0", | ||||||
|  | 		"port": 443, | ||||||
|  | 		"certPath":"./example.pem", | ||||||
|  | 		"keyPath":"./example.pem" | ||||||
|  | 	}, | ||||||
|  | 	"copilot_config":{ | ||||||
|  | 		"github_api_url": "https://api.github.com/copilot_internal/v2/token", | ||||||
|  | 		"token":[ | ||||||
|  | 			"yours_token_1", | ||||||
|  |             "yours_token_2", | ||||||
|  |             "yours_token_3" | ||||||
|  | 		]}, | ||||||
|  | 	"verification":"" | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | ``` | ||||||
|  | 格式不能错,不确定的去这里验证 ->https://www.json.cn/ | ||||||
|  | 
 | ||||||
|  | ### 3.启动  | ||||||
|  | 
 | ||||||
|  | ``` | ||||||
|  | scop r  --运行 | ||||||
|  | scop rb --后台运行 | ||||||
|  | scop st --停止 | ||||||
|  | scop v  --查看状态 | ||||||
|  | ``` | ||||||
|  | 
 | ||||||
|  | ### 4.完整示例一: | ||||||
|  | 
 | ||||||
|  | config.json | ||||||
|  | 
 | ||||||
|  | ```json | ||||||
|  | { | ||||||
|  | 	"server": { | ||||||
|  | 		"domain": "api.example.com", | ||||||
|  | 		"host": "0.0.0.0", | ||||||
|  | 		"port": 443, | ||||||
|  | 		"certPath":"./example.pem", | ||||||
|  | 		"keyPath":"./example.pem" | ||||||
|  | 	}, | ||||||
|  | 	"copilot_config":{ | ||||||
|  | 		"github_api_url": "https://api.github.com/copilot_internal/v2/token", | ||||||
|  | 		"token":[ | ||||||
|  | 		"ghu_xMNAYLcJAPqAfiGoobrWffkJoNcGMVJtETKA", | ||||||
|  |         "ghu_GZgKFwraHorAxXXUvsUclOhxiYERPsSJeNuF", | ||||||
|  |         "ghu_SPUTCLvkMKoeMstPJmhSlYsYvCojhkFjGubl" | ||||||
|  | 		]}, | ||||||
|  | 	"verification":"i_am_free" | ||||||
|  | } | ||||||
|  | ``` | ||||||
|  | 
 | ||||||
|  | 运行截图 | ||||||
|  | 
 | ||||||
|  |  | ||||||
|  | 
 | ||||||
|  | 时间-合计请求次数: 0 | 成功次数 0 | 失败次数 0   统计次数在00:00:00归零 | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | 修改%userprofile%\AppData\Local\github-copilot\host.json | ||||||
|  | 
 | ||||||
|  | ```json | ||||||
|  | {"github.com":{"user":"suibian","oauth_token":"i_am_free","dev_override":{"copilot_token_url":"https://api.example.com/copilot_internal/v2/token"}}}  | ||||||
|  | ``` | ||||||
|  | 测试:  | ||||||
|  | 
 | ||||||
|  |  | ||||||
							
								
								
									
										15
									
								
								config.json
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										15
									
								
								config.json
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,15 @@ | |||||||
|  | { | ||||||
|  | 	"server": { | ||||||
|  | 		"domain": "example.con", | ||||||
|  | 		"host": "0.0.0.0", | ||||||
|  | 		"port": 443, | ||||||
|  | 		"certPath":"./example.pem", | ||||||
|  | 		"keyPath":"./example.pem" | ||||||
|  | 	}, | ||||||
|  | 	"copilot_config":{ | ||||||
|  | 		"github_api_url": "https://api.github.com/copilot_internal/v2/token", | ||||||
|  | 		"token":[ | ||||||
|  | 			"yours_token" | ||||||
|  | 		]}, | ||||||
|  | 	"verification":"" | ||||||
|  | } | ||||||
							
								
								
									
										46
									
								
								install.sh
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										46
									
								
								install.sh
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,46 @@ | |||||||
|  | #!/bin/bash | ||||||
|  | SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )" | ||||||
|  | START_SCRIPT_PATH="/usr/local/bin/scop" | ||||||
|  | cat > "$START_SCRIPT_PATH" <<EOF | ||||||
|  | #!/bin/bash | ||||||
|  | EXECUTABLE_DIR="$SCRIPT_DIR" | ||||||
|  | EXECUTABLE="\$EXECUTABLE_DIR/share-copilot" | ||||||
|  | case "\$1" in | ||||||
|  |   r) | ||||||
|  |     echo "Running share-copilot..." | ||||||
|  |     "\$EXECUTABLE" | ||||||
|  |     ;; | ||||||
|  |   rb) | ||||||
|  |     echo "Running share-copilot in the background..." | ||||||
|  |     "\$EXECUTABLE" >/dev/null 2>&1 & | ||||||
|  |     ;; | ||||||
|  |   st) | ||||||
|  |     echo "Stopping share-copilot..." | ||||||
|  |     pkill -f "\$EXECUTABLE" | ||||||
|  |     ;; | ||||||
|  |   v) | ||||||
|  |     if pgrep -f "\$EXECUTABLE" >/dev/null; then | ||||||
|  |       echo "share-copilot is running" | ||||||
|  |     else | ||||||
|  |       echo "share-copilot is not running" | ||||||
|  |     fi | ||||||
|  |     ;; | ||||||
|  |   *) | ||||||
|  |     echo "Usage: \$0 {r|rb|st|v}" | ||||||
|  |     exit 1 | ||||||
|  |     ;; | ||||||
|  | esac | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | exit 0 | ||||||
|  | EOF | ||||||
|  | chmod +x "$START_SCRIPT_PATH" | ||||||
|  | echo "----------------- | ||||||
|  | install success! | ||||||
|  | ----------------- | ||||||
|  | scop r  --运行 | ||||||
|  | scop rb --后台运行 | ||||||
|  | scop st --停止 | ||||||
|  | scop v  --查看状态 | ||||||
|  | -----------------" | ||||||
							
								
								
									
										
											BIN
										
									
								
								share-copilot
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										
											BIN
										
									
								
								share-copilot
									
									
									
									
									
										Normal file
									
								
							
										
											Binary file not shown.
										
									
								
							
							
								
								
									
										10
									
								
								source/build.bat
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										10
									
								
								source/build.bat
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,10 @@ | |||||||
|  | @echo off | ||||||
|  | SET CGO_ENABLED=0 | ||||||
|  | SET GOOS=linux | ||||||
|  | SET GOARCH=amd64 | ||||||
|  | go build | ||||||
|  | @echo off | ||||||
|  | SET CGO_ENABLED=0 | ||||||
|  | SET GOOS=windows | ||||||
|  | SET GOARCH=amd64 | ||||||
|  | go build | ||||||
							
								
								
									
										35
									
								
								source/go.mod
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										35
									
								
								source/go.mod
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,35 @@ | |||||||
|  | module share-copilot | ||||||
|  | 
 | ||||||
|  | go 1.21 | ||||||
|  | 
 | ||||||
|  | require ( | ||||||
|  | 	github.com/bytedance/sonic v1.10.0 // indirect | ||||||
|  | 	github.com/chenzhuoyu/base64x v0.0.0-20230717121745-296ad89f973d // indirect | ||||||
|  | 	github.com/chenzhuoyu/iasm v0.9.0 // indirect | ||||||
|  | 	github.com/fatih/color v1.15.0 // indirect | ||||||
|  | 	github.com/gabriel-vasile/mimetype v1.4.2 // indirect | ||||||
|  | 	github.com/gin-contrib/sse v0.1.0 // indirect | ||||||
|  | 	github.com/gin-gonic/gin v1.9.1 // 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.15.3 // indirect | ||||||
|  | 	github.com/go-resty/resty/v2 v2.7.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.5 // 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/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.1.0 // indirect | ||||||
|  | 	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect | ||||||
|  | 	github.com/ugorji/go/codec v1.2.11 // indirect | ||||||
|  | 	golang.org/x/arch v0.5.0 // indirect | ||||||
|  | 	golang.org/x/crypto v0.13.0 // indirect | ||||||
|  | 	golang.org/x/net v0.15.0 // indirect | ||||||
|  | 	golang.org/x/sys v0.12.0 // indirect | ||||||
|  | 	golang.org/x/text v0.13.0 // indirect | ||||||
|  | 	google.golang.org/protobuf v1.31.0 // indirect | ||||||
|  | 	gopkg.in/yaml.v3 v3.0.1 // indirect | ||||||
|  | ) | ||||||
							
								
								
									
										133
									
								
								source/go.sum
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										133
									
								
								source/go.sum
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,133 @@ | |||||||
|  | github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM= | ||||||
|  | github.com/bytedance/sonic v1.10.0-rc/go.mod h1:ElCzW+ufi8qKqNW0FY314xriJhyJhuoJ3gFZdAHF7NM= | ||||||
|  | github.com/bytedance/sonic v1.10.0 h1:qtNZduETEIWJVIyDl01BeNxur2rW9OwTQ/yBqFRkKEk= | ||||||
|  | github.com/bytedance/sonic v1.10.0/go.mod h1:iZcSUejdk5aukTND/Eu/ivjQuEL0Cu9/rf50Hi0u/g4= | ||||||
|  | github.com/chenzhuoyu/base64x v0.0.0-20211019084208-fb5309c8db06/go.mod h1:DH46F32mSOjUmXrMHnKwZdA8wcEefY7UVqBKYGjpdQY= | ||||||
|  | github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311/go.mod h1:b583jCggY9gE99b6G5LEC39OIiVsWj+R97kbl5odCEk= | ||||||
|  | github.com/chenzhuoyu/base64x v0.0.0-20230717121745-296ad89f973d h1:77cEq6EriyTZ0g/qfRdp61a3Uu/AWrgIq2s0ClJV1g0= | ||||||
|  | github.com/chenzhuoyu/base64x v0.0.0-20230717121745-296ad89f973d/go.mod h1:8EPpVsBuRksnlj1mLy4AWzRNQYxauNi62uWcE3to6eA= | ||||||
|  | github.com/chenzhuoyu/iasm v0.9.0 h1:9fhXjVzq5hUy2gkhhgHl95zG2cEAhw9OSGs8toWWAwo= | ||||||
|  | github.com/chenzhuoyu/iasm v0.9.0/go.mod h1:Xjy2NpN3h7aUqeqM+woSuuvxmIe6+DDsiNLIrkAmYog= | ||||||
|  | github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||||||
|  | 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/fogleman/gg v1.3.0 h1:/7zJX8F6AaYQc57WQCyN9cAIz+4bCJGO9B+dyW29am8= | ||||||
|  | github.com/fogleman/gg v1.3.0/go.mod h1:R/bRT+9gY/C5z7JzPU0zXsXHKM4/ayA+zqcVNZzPa1k= | ||||||
|  | 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/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.15.3 h1:S+sSpunYjNPDuXkWbK+x+bA7iXiW296KG4dL3X7xUZo= | ||||||
|  | github.com/go-playground/validator/v10 v10.15.3/go.mod h1:9iXMNT7sEkjXb0I+enO7QXmzG6QCsPWY4zveKFVRSyU= | ||||||
|  | github.com/go-resty/resty/v2 v2.3.0/go.mod h1:UpN9CgLZNsv4e9XG50UU8xdI0F43UQ4HmxLBDwaroHU= | ||||||
|  | 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/freetype v0.0.0-20170609003504-e2365dfdc4a0 h1:DACJavvAHhabrF08vX0COfcOBJRhZ8lUbR+ZWIs0Y5g= | ||||||
|  | github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0/go.mod h1:E/TSTwGwJL78qG/PmXZO1EjYhfJinVAhrmmHX6Z8B9k= | ||||||
|  | github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk= | ||||||
|  | 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.5 h1:0E5MSMDEoAulmXNFquVs//DdoomxaoTY1kUhbc/qbZg= | ||||||
|  | github.com/klauspost/cpuid/v2 v2.2.5/go.mod h1:Lcz8mBdAVJIBVzewtcLocK12l3Y+JytZYpaMropDUws= | ||||||
|  | github.com/knz/go-libedit v1.10.1/go.mod h1:MZTVkCWyz0oBc7JOWP3wNAzd002ZbM/5hgShxwh4x8M= | ||||||
|  | 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/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/pelletier/go-toml/v2 v2.1.0 h1:FnwAJ4oYMvbT/34k9zzHuZNrhlz48GB3/s6at6/MHO4= | ||||||
|  | github.com/pelletier/go-toml/v2 v2.1.0/go.mod h1:tJU2Z3ZkXwnxa4DPO899bsyIoywizdUvyaeZurnPPDc= | ||||||
|  | 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.4/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= | ||||||
|  | github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY= | ||||||
|  | golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8= | ||||||
|  | golang.org/x/arch v0.5.0 h1:jpGode6huXQxcskEIpOCvrU+tzo81b6+oFLUYXWtH/Y= | ||||||
|  | golang.org/x/arch v0.5.0/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8= | ||||||
|  | golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= | ||||||
|  | golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc= | ||||||
|  | golang.org/x/crypto v0.13.0 h1:mvySKfSWJ+UKUii46M40LOvyWfN0s2U+46/jDd0e6Ck= | ||||||
|  | golang.org/x/crypto v0.13.0/go.mod h1:y6Z2r+Rw4iayiXXAIxJIDAJ1zMW4yaTpebo8fPOliYc= | ||||||
|  | golang.org/x/image v0.12.0 h1:w13vZbU4o5rKOFFR8y7M+c4A5jXDC0uXTdHYRP8X2DQ= | ||||||
|  | golang.org/x/image v0.12.0/go.mod h1:Lu90jvHG7GfemOIcldsh9A2hS01ocl6oNO7ype5mEnk= | ||||||
|  | golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4= | ||||||
|  | golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs= | ||||||
|  | golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= | ||||||
|  | golang.org/x/net v0.0.0-20200513185701-a91f0712d120/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A= | ||||||
|  | golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= | ||||||
|  | golang.org/x/net v0.0.0-20211029224645-99673261e6eb/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y= | ||||||
|  | golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c= | ||||||
|  | golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs= | ||||||
|  | golang.org/x/net v0.15.0 h1:ugBLEUaxABaB5AJqW9enI0ACdci2RUd4eP51NTBvuJ8= | ||||||
|  | golang.org/x/net v0.15.0/go.mod h1:idbUs1IY1+zTqbi8yxTbhexhEEk5ur9LInksu6HrEpk= | ||||||
|  | golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= | ||||||
|  | golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= | ||||||
|  | golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= | ||||||
|  | golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= | ||||||
|  | golang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= | ||||||
|  | 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-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||||
|  | golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||||
|  | golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||||
|  | golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||||
|  | golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||||
|  | golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||||
|  | golang.org/x/sys v0.12.0 h1:CM0HF96J0hcLAwsHPJZjfdNzs0gftsLfgKt57wWHJ0o= | ||||||
|  | golang.org/x/sys v0.12.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||||
|  | golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= | ||||||
|  | golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8= | ||||||
|  | golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k= | ||||||
|  | golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= | ||||||
|  | golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= | ||||||
|  | golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= | ||||||
|  | golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ= | ||||||
|  | golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8= | ||||||
|  | golang.org/x/text v0.13.0 h1:ablQoSUd0tRdKxZewP80B+BaqeKJuVhuRxj/dkrun3k= | ||||||
|  | golang.org/x/text v0.13.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE= | ||||||
|  | golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= | ||||||
|  | golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= | ||||||
|  | golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc= | ||||||
|  | golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU= | ||||||
|  | golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= | ||||||
|  | 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.31.0 h1:g0LDEJHgrBl9N9r17Ru3sqWhkIx2NB67okBHPwC7hs8= | ||||||
|  | google.golang.org/protobuf v1.31.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I= | ||||||
|  | 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= | ||||||
|  | nullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50= | ||||||
|  | rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4= | ||||||
							
								
								
									
										244
									
								
								source/main.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										244
									
								
								source/main.go
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,244 @@ | |||||||
|  | package main | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"crypto/tls" | ||||||
|  | 	"encoding/json" | ||||||
|  | 	"fmt" | ||||||
|  | 	"github.com/fatih/color" | ||||||
|  | 	"github.com/gin-gonic/gin" | ||||||
|  | 	"github.com/go-resty/resty/v2" | ||||||
|  | 	"io" | ||||||
|  | 	"log" | ||||||
|  | 	"math/rand" | ||||||
|  | 	"net/http" | ||||||
|  | 	"os" | ||||||
|  | 	"strconv" | ||||||
|  | 	"strings" | ||||||
|  | 	"sync" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  | 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 ( | ||||||
|  | 	requestCountMutex sync.Mutex | ||||||
|  | 	requestCount      int | ||||||
|  | 	successCount      int | ||||||
|  | 	configFile        = Config{} | ||||||
|  | ) | ||||||
|  | func main() { | ||||||
|  | 	gin.SetMode(gin.ReleaseMode) | ||||||
|  | 	gin.DefaultWriter = io.Discard | ||||||
|  | 	//初始化配置文件
 | ||||||
|  | 	configFile = initConfig() | ||||||
|  | 	// 创建Gin引擎
 | ||||||
|  | 	engine := gin.New() | ||||||
|  | 	// 自定义错误处理程序
 | ||||||
|  | 	engine.Use(func(c *gin.Context) { | ||||||
|  | 		c.Next() | ||||||
|  | 		err := c.Errors.Last() | ||||||
|  | 		if err != nil { | ||||||
|  | 			log.Printf("Error: %v", err.Err) | ||||||
|  | 			c.JSON(http.StatusInternalServerError, gin.H{"error": "Internal Server Error"}) | ||||||
|  | 		} | ||||||
|  | 	}) | ||||||
|  | 	// 设置信任的代理
 | ||||||
|  | 	err := engine.SetTrustedProxies([]string{"127.0.0.1"}) | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	// 配置支持的应用程序协议
 | ||||||
|  | 	server := &http.Server{ | ||||||
|  | 		Addr: ":443", | ||||||
|  | 		TLSConfig: &tls.Config{ | ||||||
|  | 			NextProtos: []string{"http/1.1", "http/1.2", "http/2"}, // 支持的应用程序协议列表
 | ||||||
|  | 		}, | ||||||
|  | 		Handler: engine, | ||||||
|  | 	} | ||||||
|  | 	// 定义路由
 | ||||||
|  | 	domainDefault := engine.Group("/", DomainMiddleware(configFile.Server.Domain)) | ||||||
|  | 	domainDefault.GET("/copilot_internal/v2/token", getToken()) | ||||||
|  | 	// 启动Gin服务器并监听端口
 | ||||||
|  | 	listenAddress := configFile.Server.Host + ":" + strconv.Itoa(configFile.Server.Port) | ||||||
|  | 	go func() { | ||||||
|  | 		if configFile.Server.Port != 443 { | ||||||
|  | 			err := engine.Run(listenAddress) | ||||||
|  | 			if err != nil { | ||||||
|  | 				log.Fatal(err) | ||||||
|  | 			} | ||||||
|  | 		} else { | ||||||
|  | 			err := server.ListenAndServeTLS(configFile.Server.CertPath, configFile.Server.KeyPath) | ||||||
|  | 			if err != nil { | ||||||
|  | 				log.Fatal(err) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
|  | 	// 显示url
 | ||||||
|  | 	displayMsg() | ||||||
|  | 	ticker := time.NewTicker(1 * time.Second) | ||||||
|  | 	defer ticker.Stop() | ||||||
|  | 	for range ticker.C { | ||||||
|  | 		showRequestCount() | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | func getToken() gin.HandlerFunc { | ||||||
|  | 	return func(c *gin.Context) { | ||||||
|  | 		// 请求计数
 | ||||||
|  | 		incrementRequestCount() | ||||||
|  | 		// 获取请求头中的Authorization令牌
 | ||||||
|  | 		if configFile.Verification != "" { | ||||||
|  | 			token := c.GetHeader("Authorization") | ||||||
|  | 			tokenStr := strings.ReplaceAll(token, " ", "") | ||||||
|  | 			configCert := strings.ReplaceAll(configFile.Verification, " ", "") | ||||||
|  | 			if tokenStr != "token"+configCert { | ||||||
|  | 				// 拒绝不符合Verification的请求
 | ||||||
|  | 				c.JSON(http.StatusBadRequest, gin.H{ | ||||||
|  | 					"message":           "Bad credentials", | ||||||
|  | 					"documentation_url": "https://docs.github.com/rest"}) | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		// 设置请求头
 | ||||||
|  | 		headers := map[string]string{ | ||||||
|  | 			"Authorization":         "token " + getRandomToken(configFile.CopilotConfig.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 { | ||||||
|  | 			// 处理请求错误
 | ||||||
|  | 			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 		// 检查响应状态码
 | ||||||
|  | 		if response.StatusCode() != http.StatusOK { | ||||||
|  | 			// 处理非200响应
 | ||||||
|  | 			c.JSON(response.StatusCode(), gin.H{"error": "Request failed"}) | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 		// 请求成功统计
 | ||||||
|  | 		incrementSuccessCount() | ||||||
|  | 		// 从响应中获取内容并发送回客户端
 | ||||||
|  | 		responseBody := response.String() | ||||||
|  | 		// 设置响应头 重要
 | ||||||
|  | 		c.Header("Content-Type", "application/json") | ||||||
|  | 		c.String(http.StatusOK, responseBody) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | func initConfig() Config { | ||||||
|  | 	// 打开或创建 error.log 文件
 | ||||||
|  | 	logFile, err := os.OpenFile("error.log", os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0644) | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	defer logFile.Close() | ||||||
|  | 	// 设置日志输出到文件
 | ||||||
|  | 	log.SetOutput(logFile) | ||||||
|  | 	// 读取配置文件
 | ||||||
|  | 	configFile, err := os.Open("./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 | ||||||
|  | } | ||||||
|  | func displayMsg() { | ||||||
|  | 	c := color.New(color.FgGreen) | ||||||
|  | 	fmt.Printf("\n------------------------------------------------------------\n") | ||||||
|  | 	if configFile.Server.Port == 80 { | ||||||
|  | 		_, err := c.Println("[ApiUri] http://" + configFile.Server.Domain + "/copilot_internal/v2/token\r") | ||||||
|  | 		if err != nil { | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 	} else if configFile.Server.Port == 443 { | ||||||
|  | 		_, err := c.Println("[ApiUrl] https://" + configFile.Server.Domain + "/copilot_internal/v2/token\r") | ||||||
|  | 		if err != nil { | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 	} else { | ||||||
|  | 		_, err := c.Println("[ApiUrl] http://" + configFile.Server.Domain + ":" + strconv.Itoa(configFile.Server.Port) + "/copilot_internal/v2/token\r") | ||||||
|  | 		if err != nil { | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	fmt.Printf("------------------------------------------------------------\n") | ||||||
|  | } | ||||||
|  | func incrementRequestCount() { | ||||||
|  | 	requestCount++ | ||||||
|  | } | ||||||
|  | func incrementSuccessCount() { | ||||||
|  | 	successCount++ | ||||||
|  | } | ||||||
|  | func showRequestCount() { | ||||||
|  | 	requestCountMutex.Lock() | ||||||
|  | 	count := requestCount | ||||||
|  | 	sCount := successCount | ||||||
|  | 	requestCountMutex.Unlock() | ||||||
|  | 	currentTime := time.Now().Format("2006-01-02 15:04:05") | ||||||
|  | 	if time.Now().Format("15:04:05") == "00:00:00" { | ||||||
|  | 		resetRequestCount() | ||||||
|  | 	} | ||||||
|  | 	// 黄色文本
 | ||||||
|  | 	timeStr := fmt.Sprintf("\x1b[33m%s\x1b[0m", currentTime) | ||||||
|  | 	// 蓝色文本
 | ||||||
|  | 	countStr := fmt.Sprintf("\x1b[34m%d\x1b[0m", count) | ||||||
|  | 	// 绿色文本
 | ||||||
|  | 	successCountStr := fmt.Sprintf("\x1b[32m%d\x1b[0m", sCount) | ||||||
|  | 	// 红色文本
 | ||||||
|  | 	failureCountStr := fmt.Sprintf("\x1b[31m%d\x1b[0m", count-sCount) | ||||||
|  | 	fmt.Printf("\033[1G%s - Total Count: %s | Success Count %s | Fail Count %s   ", timeStr, countStr, successCountStr, failureCountStr) | ||||||
|  | } | ||||||
|  | func resetRequestCount() { | ||||||
|  | 	requestCountMutex.Lock() | ||||||
|  | 	defer requestCountMutex.Unlock() | ||||||
|  | 	requestCount = 0 | ||||||
|  | 	successCount = 0 | ||||||
|  | } | ||||||
|  | func getRandomToken(tokens []string) string { | ||||||
|  | 	if len(tokens) == 0 { | ||||||
|  | 		return "" // 返回空字符串或处理其他错误情况
 | ||||||
|  | 	} | ||||||
|  | 	r := rand.New(rand.NewSource(time.Now().UnixNano())) | ||||||
|  | 	randomIndex := r.Intn(len(tokens)) | ||||||
|  | 	return tokens[randomIndex] | ||||||
|  | } | ||||||
|  | func DomainMiddleware(domain string) gin.HandlerFunc { | ||||||
|  | 	return func(c *gin.Context) { | ||||||
|  | 		// 检查域名是否匹配
 | ||||||
|  | 		requestDomain := strings.Split(c.Request.Host, ":")[0] | ||||||
|  | 		if requestDomain == domain { | ||||||
|  | 			c.Next() | ||||||
|  | 		} else { | ||||||
|  | 			c.String(403, "Forbidden") | ||||||
|  | 			c.Abort() | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										3
									
								
								uninstall.sh
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										3
									
								
								uninstall.sh
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,3 @@ | |||||||
|  | START_SCRIPT_PATH="/usr/local/bin/scop" | ||||||
|  | rm -f "$START_SCRIPT_PATH" | ||||||
|  | echo "卸载完成!" | ||||||
		Loading…
	
		Reference in New Issue
	
	Block a user