2023年11月9日12:19:00 修复无法登录

王明元
1 parent c1c8fc18
Showing 2 changed files with 50 additions and 1 deletions
src/main/java/cn/fw/freya/controller/CrawlController.java
src/main/java/cn/fw/freya/controller/LoginController.java
@@ -35,7 +35,7 @@ public class CrawlController {
      * @param type      账户类型(1:快手, 2:抖音, 3:懂车帝, 4:Bilibili)
      * @return 登录二维码
      */
-    @GetMapping("/preLogin")
+    @GetMapping("/getLoginQRCode")
     public ResponseMessage<String> getLoginQRCode(@NotBlank(message = "登陆账号不能为空") String accountNo,
                                             @NotNull(message = "平台类型不能为空") Integer type) {
         return success(crawlBizService.getLoginQRCode(accountNo, type));
+package cn.fw.freya.controller;
+
+import cn.fw.freya.common.ResponseMessage;
+import cn.fw.freya.service.CrawlBizService;
+import lombok.RequiredArgsConstructor;
+import org.springframework.validation.annotation.Validated;
+import org.springframework.web.bind.annotation.*;
+
+import javax.validation.constraints.NotBlank;
+import javax.validation.constraints.NotNull;
+
+/**
+ * @author kurisu
+ */
+@RestController
+@RequiredArgsConstructor
+@Validated
+@CrossOrigin
+@RequestMapping("/account")
+public class LoginController {
+
+    private final CrawlBizService crawlBizService;
+
+    /**
+     * 获取登录二维码
+     *
+     * @param account 账户号
+     * @param type    账户类型(1:快手, 2:抖音, 3:懂车帝, 4:Bilibili)
+     * @return 登录二维码
+     */
+    @GetMapping("/getCode")
+    public ResponseMessage<String> getCode(@NotBlank(message = "登陆账号不能为空") String account,
+                                           @NotNull(message = "平台类型不能为空") Integer type) {
+        return ResponseMessage.success(crawlBizService.getLoginQRCode(account, type));
+    }
+
+    /**
+     * 登录, 保存用户cookies
+     *
+     * @param account 账户号
+     * @param type    账户类型(1:快手, 2:抖音, 3:懂车帝, 4:Bilibili)
+     * @return 是否成功
+     */
+    @PostMapping("/login")
+    public ResponseMessage<Boolean> login(@NotBlank(message = "登陆账号不能为空") String account,
+                                          @NotNull(message = "平台类型不能为空") Integer type) {
+        return ResponseMessage.success(crawlBizService.doLogin(account, type));
+    }
+}