feat: auto refresh cookie

Author: AnotiaWang

build.sh

@@ -0,0 +1,8 @@
+echo "Building for Linux.."
+GOOS=linux GOARCH=amd64 go build -o oa-helper_linux_amd64 .
+GOOD=linux GOARCH=arm64 go build -o oa-helper_linux_arm64 .
+echo "Building for Windows.."
+GOOS=windows GOARCH=amd64 go build -o oa-helper_windows_amd64.exe .
+echo "Building for Mac.."
+GOOS=darwin GOARCH=amd64 go build -o oa-helper_darwin_amd64 .
+GOOS=darwin GOARCH=arm64 go build -o oa-helper_darwin_arm64 .

main.go

@@ -16,32 +16,19 @@ func main() {
 		panic(err)
 	}
 	InitChatGPTClient(model.Conf.ApiKey)
+	i := 0
 	for true {
 		err = StartTask()
 		if err != nil {
 			logx.Error("Error while doing task: ", err)
 		}
+		if i++; i == 10 {
+			i = 0
+			err = RefreshCookie()
+			if err != nil {
+				logx.Error("Error while refreshing cookie: ", err)
+			}
+		}
 		time.Sleep(2 * time.Second)
 	}
-	// 	s, err := GetLabelsFromChatGPT(`{
-	//     "spam": 0,
-	//     "fails_task": 0,
-	//     "lang_mismatch": 0,
-	//     "not_appropriate": 0,
-	//     "pii": 0,
-	//     "hate_speech": 0,
-	//     "sexual_content": 0,
-	//     "quality": 0.75,
-	//     "helpfulness": 1,
-	//     "creativity": 0.5,
-	//     "humor": 0,
-	//     "toxicity": 0,
-	//     "violence": 0
-	// }`)
-	// 	if err != nil {
-	// 		panic(err)
-	// 	}
-	// 	for l, v := range s {
-	// 		println(l, " ", v)
-	// 	}
 }

model/config.go

@@ -95,3 +95,16 @@ func LoadConfig() error {
 
 	return nil
 }
+
+func UpdateCookie(cookie string) error {
+	Conf.OaCookie = cookie
+	file, err := json.MarshalIndent(Conf, "", "    ")
+	if err != nil {
+		return err
+	}
+	err = os.WriteFile("config.json", file, 0644)
+	if err != nil {
+		return err
+	}
+	return nil
+}

open-assistant.go

@@ -7,6 +7,7 @@ import (
 	"github.com/zeromicro/go-zero/core/jsonx"
 	"github.com/zeromicro/go-zero/core/logx"
 	"open-assistant-helper-go/model"
+	"strings"
 )
 
 var rty = resty.New()
@@ -28,6 +29,24 @@ func CancelTask(id string) error {
 	return nil
 }
 
+func RefreshCookie() error {
+	logx.Infof("RefreshCookie")
+	resp, err := rty.R().
+		SetHeaders(map[string]string{
+			"cookie": model.Conf.OaCookie,
+		}).
+		Get("https://open-assistant.io/api/auth/session")
+	if err != nil {
+		return err
+	}
+	cs := resp.Header()["Set-Cookie"]
+	if len(cs) == 0 {
+		return fmt.Errorf("no cookie")
+	}
+	c := strings.Join(cs, "")
+	return model.UpdateCookie(c)
+}
+
 func GetLabelsFromChatGPT(resp string) (map[model.OALabel]float32, error) {
 	var j map[string]interface{}
 	err := jsonx.Unmarshal([]byte(resp), &j)