Merge branch 'master' of https://gitee.com/chuangxxt/share-copilot

Signed-off-by: luyoyu <8829470+chuangxxt@user.noreply.gitee.com>

README.md

@@ -1,12 +1,15 @@
-[中文](README.md) | [English](README-EN.md) 
+ **[中文](README.md) | [English](README-EN.md)** 
 
 
+更新代码提示接口代理：[代码提示代理说明](readme/codeTipsProxy.md)
+
 # share-copilot
 
 - 作为代理服务器，中转copilot插件的API的请求（也可以代理无良商贩的，自行研究）
 - 支持vscode插件和jetbrains插件
 - 支持多用户共享一个token
 - 优化请求逻辑，降低suspended概率（别万人骑基本不会）
+![输入图片说明](readme/jiaggou.png)
 
 ```sh
 简单说一下
@@ -82,7 +85,7 @@ go build
 ### 1.安装
 
 ```sh
-wget https://gitee.com/chuangxxt/share-copilot/releases/download/1.0.2/share-copilot-linux-amd64.zip
+wget https://gitee.com/chuangxxt/share-copilot/releases/download/v2.0.4/share-copilot-linux-amd64.zip
 ```
 ```sh
 unzip share-copilot-linux-amd64.zip 
@@ -139,7 +142,9 @@ verification//自定义验证
             "yours_token_2",
             "yours_token_3"
 		]},
-	"verification":""
+	"verification":"",
+         "diyMsg":"",
+	"isModMsg":false
 }
 
 ```
@@ -174,7 +179,9 @@ scop v  #查看是否运行
         "ghu_GZgKFwraHorAxXXUvsUclOhxiYERPsSJeNuF",
         "ghu_SPUTCLvkMKoeMstPJmhSlYsYvCojhkFjGubl"
 		]},
-	"verification":"i_am_free"
+	"verification":"i_am_free",
+        "diyMsg":"",
+	"isModMsg":false
 }
 ```
 

readme/codeTipsProxy.md

@@ -0,0 +1,68 @@
+# 代码提示 代理说明
+
+ **-以windows为例** 
+
+## **vscode：**
+
+#### 1.修改文件：**
+
+`%userprofile%\.vscode\extensions\github.copilot-xxxx\dist\extension.js`
+
+
+**1.1.extension.js全文搜索替换下面两个接口，替换为自己的url** 
+
+
+```js
+"https://copilot-proxy.githubusercontent.com" 
+替换为"https://127.0.0.1" (你的url)
+这个是代码提示接口
+---------------------------------------------------------------
+"https://copilot-telemetry.githubusercontent.com/telemetry"
+替换为"https://127.0.0.1/telemetry" (你的url/telemetry)
+这个应该是copilot把你代码上传拿去深度学习的,直接改成null好像也不影响使用
+```
+
+**1.2.也可以使用[脚本](https://gitee.com/chuangxxt/share-copilot/releases/download/v1.0.4/vscode%E4%B8%80%E9%94%AE%E9%85%8D%E7%BD%AE%E8%84%9A%E6%9C%AC.bat)一键修改，需要将extension.js原来增加的内容删掉，再配置脚本一键修改，脚本会自动备份原来的文件**
+
+
+```bash
+@echo off
+setlocal
+
+:: 将127.0.0.1改为自己的Api url  GITHUB_TOKEN改为自己代理服务器中间件验证内容
+set "GITHUB_TOKEN=123456"
+set "GITHUB_API_URL=http://127.0.0.1"
+set "completionsUrl=http://127.0.0.1"
+set "telemetryUrl=http://127.0.0.1/telemetry"
+
+set "CODESPACES=true"
+set "GITHUB_SERVER_URL=https://github.com"
+
+...略...
+```
+
+#### 2.测试
+
+全部接口代理后是走一个ip了，虽然不知道有没有意义
+
+不过可以搞点有意思的东西，修改一下配置文件
+
+```json
+"diyMsg":"//授权已经过期，重新获取-> https://gitee.com/chuangxxt/share-copilot",
+"isModMsg":true
+```
+![输入图片说明](ceshi.gif)
+
+
+
+## **jetbrains：**
+
+代理服务器是支持的，插件我不会改，等个有缘人
+
+我修改了插件里面js文件里面的接口，改为自己的，然后重新构建成exe，
+打包后有问题，可以运行，代理联网等都正常
+
+但是代码提示无法在ide里面显示
+目前找不到原因（不懂node.js）
+
+![](https://foruda.gitee.com/images/1695118855841319485/b2c3c89e_8829470.png)

src/global.go

@@ -19,6 +19,8 @@ type Config struct {
 		Token        []string `json:"token"`
 	} `json:"copilot_config"`
 	Verification string `json:"verification"`
+	DiyMsg       string `json:"diyMsg"`
+	IsModMsg     bool   `json:"isModMsg"`
 }
 
 // 全局变量
@@ -38,12 +40,7 @@ var (
 	requestCount = 0
 	//请求成功计数
 	successCount = 0
-)
 
-// 测试的玩意儿
-var (
-	diyMsg        = ";//授权已经过期，重新获取-> https://www.sharecopilot.top"
 	completionUrl = "https://copilot-proxy.githubusercontent.com/v1/engines/copilot-codex/completions"
 	telemetryUrl  = "https://copilot-telemetry.githubusercontent.com/telemetry"
-	isModMsg      = true
 )

src/middleware.go

@@ -28,7 +28,7 @@ func VerifyRequestMiddleware() gin.HandlerFunc {
 		if configFile.Verification != "" {
 			authHeader := c.GetHeader("Authorization")
 			if authHeader != "token "+configFile.Verification {
-				c.JSON(401, gin.H{"error": "Unauthorized"})
+				c.AbortWithStatusJSON(401, gin.H{"error": "Unauthorized"})
 				return
 			}
 		}

src/server.go

@@ -1,7 +1,6 @@
 package main
 
 import (
-	"bytes"
 	"crypto/tls"
 	"encoding/json"
 	"fmt"
@@ -13,7 +12,6 @@ import (
 	"net/url"
 	"os"
 	"path/filepath"
-	"regexp"
 	"strconv"
 )
 
@@ -84,65 +82,33 @@ func diyBadRequest(c *gin.Context, code int, errorMessage string) {
 
 // registerProxyRoute 反向代理路由
 func registerProxyRoute(r *gin.Engine, routePath, targetURL string) {
-	target, err := url.Parse(targetURL)
-	if err != nil {
-		panic(err)
+	target, _ := url.Parse(targetURL)
+	isModMsgEnabled := configFile.IsModMsg && targetURL == completionUrl
+	director := func(req *http.Request) {
+		req.URL.Host = target.Host
+		req.Host = target.Host
+		req.URL.Scheme = "https"
 	}
 	proxy := &httputil.ReverseProxy{
-		Director: func(req *http.Request) {
-			req.URL.Host = target.Host
-			req.URL.Scheme = "https"
-			req.Host = target.Host
-		},
-		ErrorHandler: func(w http.ResponseWriter, r *http.Request, err error) {
-		},
+		Director:     director,
+		ErrorHandler: func(w http.ResponseWriter, r *http.Request, err error) {},
 	}
 	r.Any(routePath, func(c *gin.Context) {
-		proxy.Director = func(req *http.Request) {
-			req.URL.Host = target.Host
-			req.URL.Scheme = "https"
-			req.Host = target.Host
+		if isModMsgEnabled {
+			modProxyResp(c)
+		} else {
+			proxy.ServeHTTP(c.Writer, c.Request)
 		}
-		if isModMsg {
-			modProxyResp(c, proxy)
-		}
-		proxy.ServeHTTP(c.Writer, c.Request)
 	})
 }
 
-// 修改代码提示内容
-func modProxyResp(c *gin.Context, proxy *httputil.ReverseProxy) {
-	// 在代理响应之前修改响应
-	proxy.ModifyResponse = func(response *http.Response) error {
-		responseBuffer := new(bytes.Buffer)
-		_, readErr := responseBuffer.ReadFrom(response.Body)
-		if readErr != nil {
-			return readErr
-		}
-		responseData := responseBuffer.Bytes()
-		// 定义正则表达式模式
-		pattern := `(.*?)data:`
-		// 编译正则表达式
-		regex := regexp.MustCompile(pattern)
-		// 查找匹配项
-		match := regex.FindStringSubmatch(string(responseData))
-		var replacedData = ""
-		if len(match) >= 2 {
-			replacedData = match[1]
-		} else {
-			fmt.Println("No match found.")
-		}
-		// 将字符串转换为[]rune类型的字符数组
-		runes := []rune(diyMsg)
-		newStr := ""
-		// 遍历字符数组，对每个字符进行处理
-		for _, r := range runes {
-			// 在字符前后添加内容，生成新的字符串
-			newStr += fmt.Sprintf("data: {\"id\":\"cmpl-7xy1GLgssjHEubVrPyt534VRYVF0t\",\"model\":\"cushman-ml\",\"created\":1694526422,\"choices\":[{\"text\":\"%c\",\"index\":0,\"finish_reason\":null,\"logprobs\":null}]}\n", r)
-		}
-		newStr = replacedData + newStr + "data: [DONE]\n"
-		fmt.Println(newStr)
-		c.Data(response.StatusCode, response.Header.Get("Content-Type"), []byte(newStr))
-		return nil
+func modProxyResp(c *gin.Context) {
+	c.Header("Content-Type", "text/plain;charset=utf-8")
+	c.Header("X-Forwarded-Proto", "http")
+	runes := []rune(configFile.DiyMsg)
+	for _, r := range runes {
+		modStr := fmt.Sprintf("data: {\"id\":\"cmpl-7xy1GLgssjHEubVrPyt534VRYVF0t\",\"model\":\"cushman-ml\",\"created\":1694526422,\"choices\":[{\"text\":\"%c\",\"index\":0,\"finish_reason\":null,\"logprobs\":null}]}\n", r)
+		c.String(200, modStr)
 	}
+	c.String(200, "data: [DONE]\n")
 }