Pull1 (aurora-develop#53)

* Update README.md * up * 修正聊天接口自动continue功能，添加authorization中间件，增加Authorization系统变量。 * 修正聊天接口自动continue功能，添加authorization中间件，增加Authorization系统变量。 * 优化authorization中间件，增加自定义free_tokens.txt文件，由于有些uuid和正常账号重复了，会导致401 * 优化authorization中间件，新增自定义free_tokens.txt文件。 * 优化authorization中间件，新增自定义free_tokens.txt文件。 * docs: readme.md update docker-compose.yml * 增加session_token刷新access_token,和refresh刷新。 * 修复解析代理地址时未判断错误。 * Delete README.md * Delete .env.template * Update response.go (aurora-develop#49) (aurora-develop#51) * docs: readme.md update docker-compose.yml (aurora-develop#43) * Create LICENSE (aurora-develop#44) * fix: default free accounts * doc: remove useless readme * doc: readme * doc: readme * Update response.go (aurora-develop#49) --------- Co-authored-by: madoka <[email protected]> Co-authored-by: LanQian <[email protected]> --------- Co-authored-by: lishihua <[email protected]> Co-authored-by: madoka <[email protected]> Co-authored-by: LanQian <[email protected]>
lixplor · Apr 4, 2024 · 41612eb · 41612eb
1 parent c936eb8
commit 41612eb
Show file tree

Hide file tree

Showing 7 changed files with 173 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -66,8 +66,10 @@ PROXY_URL=your_proxy_url
 
 感谢各位大佬的pr支持，感谢。
 
+
 ## 参考项目
 
+
 https://github.com/xqdoo00o/ChatGPT-to-API
 
 ## License

diff --git a/auth.go b/auth.go
@@ -4,9 +4,10 @@ import (
 	"aurora/internal/tokens"
 	"bufio"
 	"fmt"
-	"github.com/google/uuid"
 	"os"
 	"strconv"
+
+	"github.com/google/uuid"
 )
 
 func readAccessToken() {
@@ -28,7 +29,7 @@ func readAccessToken() {
 		}
 	}
 
-	// 增加自定义free_tokens.txt，支持文件游客账号的uuid
+	// 增加自定义free_tokens.txt，支持文件设置每个账号的uuid
 	if _, err := os.Stat("free_tokens.txt"); err == nil {
 		// Each line is a proxy, put in proxies array
 		file, _ := os.Open("free_tokens.txt")

diff --git a/handlers.go b/handlers.go
@@ -21,6 +21,58 @@ func optionsHandler(c *gin.Context) {
 	})
 }
 
+func refresh_handler(c *gin.Context) {
+	var refresh_token official_types.OpenAIRefreshToken
+	err := c.BindJSON(&refresh_token)
+	if err != nil {
+		c.JSON(400, gin.H{"error": gin.H{
+			"message": "Request must be proper JSON",
+			"type":    "invalid_request_error",
+			"param":   nil,
+			"code":    err.Error(),
+		}})
+		return
+	}
+	proxy_url := ProxyIP.GetProxyIP()
+	openaiRefreshToken, status, err := chatgpt.GETTokenForRefreshToken(refresh_token.RefreshToken, proxy_url)
+	if err != nil {
+		c.JSON(400, gin.H{
+			"message": "Request must be proper JSON",
+			"type":    "invalid_request_error",
+			"param":   nil,
+			"code":    err.Error(),
+		})
+		return
+	}
+	c.JSON(status, openaiRefreshToken)
+}
+
+func session_handler(c *gin.Context) {
+	var session_token official_types.OpenAISessionToken
+	err := c.BindJSON(&session_token)
+	if err != nil {
+		c.JSON(400, gin.H{
+			"message": "Request must be proper JSON",
+			"type":    "invalid_request_error",
+			"param":   nil,
+			"code":    err.Error(),
+		})
+		return
+	}
+	proxy_url := ProxyIP.GetProxyIP()
+	openaiSessionToken, status, err := chatgpt.GETTokenForSessionToken(session_token.SessionToken, proxy_url)
+	if err != nil {
+		c.JSON(400, gin.H{"error": gin.H{
+			"message": "Request must be proper JSON",
+			"type":    "invalid_request_error",
+			"param":   nil,
+			"code":    err.Error(),
+		}})
+		return
+	}
+	c.JSON(status, openaiSessionToken)
+}
+
 func nightmare(c *gin.Context) {
 	var original_request official_types.APIRequest
 	err := c.BindJSON(&original_request)

diff --git a/internal/chatgpt/request.go b/internal/chatgpt/request.go
@@ -722,3 +722,96 @@ func Handler(c *gin.Context, response *http.Response, secret *tokens.Secret, uui
 		ParentID:       original_response.Message.ID,
 	}
 }
+
+type AuthSession struct {
+	User struct {
+		Id           string        `json:"id"`
+		Name         string        `json:"name"`
+		Email        string        `json:"email"`
+		Image        string        `json:"image"`
+		Picture      string        `json:"picture"`
+		Idp          string        `json:"idp"`
+		Iat          int           `json:"iat"`
+		Mfa          bool          `json:"mfa"`
+		Groups       []interface{} `json:"groups"`
+		IntercomHash string        `json:"intercom_hash"`
+	} `json:"user"`
+	Expires      time.Time `json:"expires"`
+	AccessToken  string    `json:"accessToken"`
+	AuthProvider string    `json:"authProvider"`
+}
+
+func GETTokenForRefreshToken(refresh_token string, proxy string) (interface{}, int, error) {
+	if proxy != "" {
+		client.SetProxy(proxy)
+	}
+	url := "https://auth0.openai.com/oauth/token"
+
+	data := map[string]interface{}{
+		"redirect_uri":  "com.openai.chat://auth0.openai.com/ios/com.openai.chat/callback",
+		"grant_type":    "refresh_token",
+		"client_id":     "pdlLIX2Y72MIl2rhLhTE9VV9bN905kBh",
+		"refresh_token": refresh_token,
+	}
+
+	reqBody, err := json.Marshal(data)
+	if err != nil {
+		return nil, 0, err
+	}
+
+	req, _ := http.NewRequest("POST", url, bytes.NewBuffer(reqBody))
+	req.Header.Set("authority", "auth0.openai.com")
+	req.Header.Add("Accept-Language", "en-US,en;q=0.9")
+	req.Header.Set("Content-Type", "application/json")
+	req.Header.Set("User-Agent", "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36")
+	req.Header.Set("Accept", "*/*")
+	resp, err := client.Do(req)
+	if err != nil {
+		return nil, 0, err
+	}
+	defer resp.Body.Close()
+	var result interface{}
+	err = json.NewDecoder(resp.Body).Decode(&result)
+	if err != nil {
+		return nil, 0, err
+	}
+	return result, resp.StatusCode, nil
+}
+
+func GETTokenForSessionToken(session_token string, proxy string) (interface{}, int, error) {
+	if proxy != "" {
+		client.SetProxy(proxy)
+	}
+	url := "https://chat.openai.com/api/auth/session"
+	req, _ := http.NewRequest("GET", url, nil)
+	req.Header.Set("authority", "chat.openai.com")
+	req.Header.Set("accept-language", "zh-CN,zh;q=0.9")
+	req.Header.Set("User-Agent", userAgent)
+	req.Header.Set("Accept", "*/*")
+	req.Header.Set("oai-language", "en-US")
+	req.Header.Set("origin", "https://chat.openai.com")
+	req.Header.Set("referer", "https://chat.openai.com/")
+	req.Header.Set("cookie", "__Secure-next-auth.session-token="+session_token)
+	resp, err := client.Do(req)
+	if err != nil {
+		return nil, 0, err
+	}
+	defer resp.Body.Close()
+	var result AuthSession
+	json.NewDecoder(resp.Body).Decode(&result)
+
+	cookies := parseCookies(resp.Cookies())
+	if value, ok := cookies["__Secure-next-auth.session-token"]; ok {
+		session_token = value
+	}
+	openai_sessionToken := official_types.NewOpenAISessionToken(session_token, result.AccessToken)
+	return openai_sessionToken, resp.StatusCode, nil
+}
+
+func parseCookies(cookies []*http.Cookie) map[string]string {
+	cookieDict := make(map[string]string)
+	for _, cookie := range cookies {
+		cookieDict[cookie.Name] = cookie.Value
+	}
+	return cookieDict
+}
diff --git a/main.go b/main.go
@@ -53,7 +53,7 @@ func checkProxy() {
 		scanner := bufio.NewScanner(file)
 		for scanner.Scan() {
 			proxy := scanner.Text()
-			parsedURL, _ := url.Parse(proxy)
+			parsedURL, err := url.Parse(proxy)
 			if err != nil {
 				fmt.Println("无法解析URL:", err)
 				return
@@ -95,6 +95,8 @@ func main() {
 		})
 	})
 
+	router.POST("/auth/session", session_handler)
+	router.POST("/auth/refresh", refresh_handler)
 	router.OPTIONS("/v1/chat/completions", optionsHandler)
 	router.POST("/v1/chat/completions", Authorization, nightmare)
 	router.GET("/v1/models", Authorization, engines_handler)

diff --git a/typings/official/request.go b/typings/official/request.go
@@ -11,3 +11,11 @@ type api_message struct {
 	Role    string `json:"role"`
 	Content string `json:"content"`
 }
+
+type OpenAISessionToken struct {
+	SessionToken string `json:"session_token"`
+}
+
+type OpenAIRefreshToken struct {
+	RefreshToken string `json:"refresh_token"`
+}
diff --git a/typings/official/response.go b/typings/official/response.go
@@ -104,3 +104,15 @@ func NewChatCompletion(full_test string) ChatCompletion {
 		},
 	}
 }
+
+type OpenAIAccessTokenWithSession struct {
+	SessionToken string `json:"session_token"`
+	AccessToken  string `json:"access_token"`
+}
+
+func NewOpenAISessionToken(session_token string, access_token string) *OpenAIAccessTokenWithSession {
+	return &OpenAIAccessTokenWithSession{
+		SessionToken: session_token,
+		AccessToken:  access_token,
+	}
+}