release: v0.1.3

internal/handler/ckwk.go

@@ -4,6 +4,7 @@ import (
 	"ckwk/internal/ckwk"
 	"ckwk/internal/dto"
 	"ckwk/pkg/log"
+	"errors"
 	"fmt"
 	"net/http"
 
@@ -44,25 +45,22 @@ func (h *WKHandler) Login(ctx *gin.Context) {
 		ctx.JSON(200, dto.Error(-1, "登录失败：请提供账号密码或有效的 Token，并确保 Host 正确"))
 		return
 	}
-
-	userinfo, err := wk.UserInfoGet()
-	if err != nil {
-		ctx.JSON(200, dto.Error(-1, err.Error()))
-		return
-	}
-
-	courses, err := wk.CourseGet(req.Status)
-	if err != nil {
-		ctx.JSON(200, dto.Error(-1, err.Error()))
-		return
+	if req.Token == "" {
+		ok, err := wk.Login()
+		if err != nil {
+			ctx.JSON(200, dto.Error(-1, err.Error()))
+			return
+		}
+		if !ok {
+			ctx.JSON(200, dto.Error(-1, "登录失败"))
+			return
+		}
 	}
 
 	sessionID := h.Session.Store(wk)
 
 	ctx.JSON(200, dto.Success(map[string]any{
 		"session_id": sessionID,
-		"user":       userinfo,
-		"courses":    courses,
 	}))
 }
 
@@ -76,6 +74,10 @@ func (h *WKHandler) Online(ctx *gin.Context) {
 
 	flag, err := wk.Online()
 	if err != nil {
+		if errors.Is(err, ckwk.ErrSessionRemoved) {
+			ctx.JSON(http.StatusUnauthorized, dto.Error(401, err.Error()))
+			return
+		}
 		ctx.JSON(200, dto.Error(-1, err.Error()))
 		return
 	}
@@ -153,6 +155,10 @@ func (h *WKHandler) Study(ctx *gin.Context) {
 
 	result, err := wk.Study(req.NodeID, req.StudyID, req.StudyTime, req.Status)
 	if err != nil {
+		if errors.Is(err, ckwk.ErrSessionRemoved) {
+			ctx.JSON(http.StatusUnauthorized, dto.Error(401, err.Error()))
+			return
+		}
 		ctx.JSON(200, dto.Error(-1, err.Error()))
 		return
 	}