session接口触发时主动刷新accessToken

api/session.go

@@ -1,8 +1,13 @@
 package api
 
 import (
+	backendapi "chatgpt-mirror-server/backend-api"
+	"chatgpt-mirror-server/config"
+	"chatgpt-mirror-server/modules/chatgpt/model"
 	"net/http"
+	"time"
 
+	"github.com/cool-team-official/cool-admin-go/cool"
 	"github.com/gogf/gf/v2/encoding/gjson"
 	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/net/ghttp"
@@ -23,10 +28,27 @@ func Session(r *ghttp.Request) {
 		return
 	}
 	officialSession := gjson.New(record["officialSession"].String())
-
-	officialSession.Set("accessToken", userToken.String())
-	officialSession.Set("user.email", "[email protected]")
-	officialSession.Set("user.name", expireTime)
-	officialSession.Remove("refreshCookie")
-	r.Response.WriteJsonExit(officialSession)
+	getSessionUrl := config.CHATPROXY(ctx) + "/getsession"
+	refreshCookie := officialSession.Get("refreshCookie").String()
+	sessionVar := g.Client().SetHeader("authkey", config.AUTHKEY(ctx)).PostVar(ctx, getSessionUrl, g.Map{
+		"username":      record["email"].String(),
+		"password":      record["password"].String(),
+		"authkey":       config.AUTHKEY(ctx),
+		"refreshCookie": refreshCookie,
+	})
+	sessionJson := gjson.New(sessionVar)
+	if sessionJson.Get("accessToken").String() == "" {
+		g.Log().Error(ctx, "get session error", sessionJson)
+		r.Response.WriteStatus(http.StatusUnauthorized)
+		return
+	}
+	cool.DBM(model.NewChatgptSession()).Where("email=?", record["email"].String()).Update(g.Map{
+		"officialSession": sessionJson.String(),
+	})
+	backendapi.AccessTokenCache.Set(ctx, userToken.String(), sessionJson.Get("accessToken").String(), 10*24*time.Hour)
+	sessionJson.Set("accessToken", userToken.String())
+	sessionJson.Set("user.email", "[email protected]")
+	sessionJson.Set("user.name", expireTime)
+	sessionJson.Remove("refreshCookie")
+	r.Response.WriteJsonExit(sessionJson)
 }