From bd45ea82b730e4327448fd93666a05fd82b041cd Mon Sep 17 00:00:00 2001
From: Antonio <acheong@student.dalat.org>
Date: Mon, 26 Jun 2023 17:05:43 +0800
Subject: [PATCH] arkose

---
 conversion/requests/chatgpt/convert.go |  2 ++
 internal/chatgpt/request.go            | 43 ++++++++++++++++++++++++++
 2 files changed, 45 insertions(+)

diff --git a/conversion/requests/chatgpt/convert.go b/conversion/requests/chatgpt/convert.go
index 05bdde8..904c926 100644
--- a/conversion/requests/chatgpt/convert.go
+++ b/conversion/requests/chatgpt/convert.go
@@ -1,6 +1,7 @@
 package chatgpt
 
 import (
+	chatgpt_req "freechatgpt/internal/chatgpt"
 	chatgpt_types "freechatgpt/typings/chatgpt"
 	official_types "freechatgpt/typings/official"
 	"strings"
@@ -12,6 +13,7 @@ func ConvertAPIRequest(api_request official_types.APIRequest) chatgpt_types.Chat
 		chatgpt_request.Model = "text-davinci-002-render-sha"
 	}
 	if strings.HasPrefix(api_request.Model, "gpt-4") {
+		chatgpt_req.Get_arkose_token()
 		chatgpt_request.Model = api_request.Model
 	}
 	if api_request.Model == "gpt-4" {
diff --git a/internal/chatgpt/request.go b/internal/chatgpt/request.go
index fc5e6fb..7a4b397 100644
--- a/internal/chatgpt/request.go
+++ b/internal/chatgpt/request.go
@@ -66,6 +66,49 @@ func random_int(min int, max int) int {
 	return min + rand.Intn(max-min)
 }
 
+type arkose_response struct {
+	Token string `json:"token"`
+}
+
+func Get_arkose_token() (string, error) {
+	resp, err := client.Get("http://bypass.churchless.tech/api/arkose")
+	if err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+	payload, _ := io.ReadAll(resp.Body)
+	// println(string(payload))
+	url := "https://tcr9i.chat.openai.com/fc/gt2/public_key/35536E1E-65B4-4D96-9D97-6ADB7EFF8147"
+	req, _ := http.NewRequest(http.MethodPost, url, bytes.NewReader(payload))
+	req.Header.Set("Host", "tcr9i.chat.openai.com")
+	req.Header.Set("User-Agent", "Mozilla/5.0 (Windows NT 10.0; rv:114.0) Gecko/20100101 Firefox/114.0")
+	req.Header.Set("Accept", "*/*")
+	req.Header.Set("Accept-Language", "en-US,en;q=0.5")
+	req.Header.Set("Accept-Encoding", "gzip, deflate, br")
+	// form
+	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
+	req.Header.Set("Origin", "https://tcr9i.chat.openai.com")
+	req.Header.Set("DNT", "1")
+	req.Header.Set("Connection", "keep-alive")
+	req.Header.Set("Referer", "https://tcr9i.chat.openai.com/v2/1.5.2/enforcement.64b3a4e29686f93d52816249ecbf9857.html")
+	req.Header.Set("Sec-Fetch-Dest", "empty")
+	req.Header.Set("Sec-Fetch-Mode", "cors")
+	req.Header.Set("Sec-Fetch-Site", "same-origin")
+	req.Header.Set("TE", "trailers")
+	resp, err = client.Do(req)
+	if err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+	var arkose arkose_response
+	err = json.NewDecoder(resp.Body).Decode(&arkose)
+	if err != nil {
+		return "", err
+	}
+	println(arkose.Token)
+	return arkose.Token, nil
+}
+
 func POSTconversation(message chatgpt_types.ChatGPTRequest, access_token string) (*http.Response, error) {
 	if http_proxy != "" && len(proxies) == 0 {
 		client.SetProxy(http_proxy)