feat: update

backend/app/http/wechats.go

@@ -1,10 +1,14 @@
 package http
 
 import (
+	"time"
+
+	"quyun/app/models"
 	"quyun/database/schemas/public/model"
 	"quyun/providers/wechat"
 
 	"github.com/gofiber/fiber/v3"
+	"github.com/gofiber/fiber/v3/log"
 )
 
 // @provider
@@ -17,5 +21,31 @@ type wechats struct {
 // @Bind url query
 // @Bind user local
 func (ctl *wechats) GetJsSDK(ctx fiber.Ctx, url string, user *model.Users) (*wechat.JsSDK, error) {
+	if user.AuthToken.Data.StableExpiresAt.After(time.Now()) {
+		token, err := ctl.wechat.RefreshAccessToken(user.AuthToken.Data.RefreshToken)
+		if err != nil {
+			return nil, err
+		}
+		log.Infof("refresh token: %+v", token)
+
+		stableToken, err := ctl.wechat.GetStableAccessToken()
+		if err != nil {
+			return nil, err
+		}
+		user.AuthToken.Data.StableAccessToken = stableToken.AccessToken
+
+		oldToken := user.AuthToken.Data
+		oldToken.AccessToken = token.AccessToken
+		oldToken.ExpiresAt = time.Now().Add(time.Second * time.Duration(token.ExpiresIn))
+
+		oldToken.StableAccessToken = stableToken.AccessToken
+		oldToken.StableExpiresAt = time.Now().Add(time.Second * time.Duration(stableToken.ExpiresIn))
+
+		if err := models.Users.UpdateUserToken(ctx.Context(), user.ID, oldToken); err != nil {
+			return nil, err
+		}
+		user.AuthToken.Data = oldToken
+	}
+
 	return ctl.wechat.GetJsSDK(user.AuthToken.Data.StableAccessToken, url)
 }

backend/app/middlewares/mid_auth.go

@@ -3,6 +3,7 @@ package middlewares
 import (
 	"net/url"
 	"strings"
+	"time"
 
 	"quyun/app/models"
 	"quyun/pkg/utils"
@@ -70,6 +71,17 @@ func (f *Middlewares) Auth(ctx fiber.Ctx) error {
 		}
 		return ctx.Redirect().To(fullUrl)
 	}
+
+	// TOKEN 过期
+	if user.AuthToken.Data.ExpiresAt.Before(time.Now()) {
+		// remove cookie
+		ctx.ClearCookie("token")
+		if ctx.XHR() {
+			return ctx.SendStatus(fiber.StatusUnauthorized)
+		}
+		return ctx.Redirect().To(fullUrl)
+	}
+
 	ctx.Locals("user", user)
 
 	return ctx.Next()

backend/app/models/users.go

@@ -5,6 +5,7 @@ import (
 	"time"
 
 	"quyun/app/requests"
+	"quyun/database/fields"
 	"quyun/database/schemas/public/model"
 	"quyun/database/schemas/public/table"
 
@@ -385,3 +386,21 @@ func (m *usersModel) UpdateUsername(ctx context.Context, id int64, username stri
 	}
 	return nil
 }
+
+// UpdateUserToken
+func (m *usersModel) UpdateUserToken(ctx context.Context, id int64, token fields.UserAuthToken) error {
+	tbl := table.Users
+	stmt := tbl.
+		UPDATE(tbl.AuthToken).
+		SET(fields.ToJson(token)).
+		WHERE(
+			tbl.ID.EQ(Int64(id)),
+		)
+	m.log.Infof("sql: %s", stmt.DebugSql())
+
+	if _, err := stmt.ExecContext(ctx, db); err != nil {
+		m.log.Errorf("error updating user token: %v", err)
+		return err
+	}
+	return nil
+}