两步验证失败增加ip记录

back/services/auth.ts

@@ -155,7 +155,7 @@ export default class AuthService {
     return isValid;
   }
 
-  public twoFactorLogin({ username, password, code }, req) {
+  public async twoFactorLogin({ username, password, code }, req) {
     const authInfo = this.getAuthInfo();
     const isValid = authenticator.verify({
       token: code,
@@ -165,6 +165,11 @@ export default class AuthService {
       this.updateAuthInfo(authInfo, { twoFactorChecked: true });
       return this.login({ username, password }, req);
     } else {
+      const { ip, address } = await getNetIp(req);
+      this.updateAuthInfo(authInfo, {
+        lastip: ip,
+        lastaddr: address,
+      });
       return { code: 430, message: '验证失败' };
     }
   }

src/layouts/index.tsx

@@ -34,6 +34,9 @@ export default function (props: any) {
       .then((data) => {
         if (data.data.username) {
           setUser(data.data);
+          if (props.location.pathname === '/') {
+            history.push('/crontab');
+          }
         }
       })
       .catch((e) => {
@@ -56,12 +59,6 @@ export default function (props: any) {
     document.title = '控制面板';
   }, []);
 
-  useEffect(() => {
-    if (props.location.pathname === '/') {
-      history.push('/crontab');
-    }
-  }, [props.location.pathname]);
-
   useEffect(() => {
     const _theme = localStorage.getItem('qinglong_dark_theme') || 'auto';
     setFetchMethod(window.fetch);