feat: prevent the user from entering the knowledge base if he is not logged in (#45)

web/src/pages/setting/model.ts

@@ -1,6 +1,7 @@
-import { Effect, Reducer, Subscription } from 'umi';
-import { message } from 'antd';
 import userService from '@/services/userService';
+import authorizationUtil from '@/utils/authorizationUtil';
+import { message } from 'antd';
+import { Effect, Reducer, Subscription } from 'umi';
 
 export interface settingModelState {
   isShowPSwModal: boolean;
@@ -9,10 +10,10 @@ export interface settingModelState {
   isShowSSModal: boolean;
   llm_factory: string;
   loading: boolean;
-  tenantIfo: any,
-  llmInfo: any,
-  myLlm: any[],
-  factoriesList: any[]
+  tenantIfo: any;
+  llmInfo: any;
+  myLlm: any[];
+  factoriesList: any[];
 }
 
 export interface settingModelType {
@@ -45,32 +46,31 @@ const Model: settingModelType = {
     tenantIfo: {},
     llmInfo: {},
     myLlm: [],
-    factoriesList: []
+    factoriesList: [],
   },
   subscriptions: {
     setup({ dispatch, history }) {
-      history.listen(location => {
-      });
-    }
+      history.listen((location) => {});
+    },
   },
   effects: {
     *setting({ payload = {}, callback }, { call, put }) {
       const { data, response } = yield call(userService.setting, payload);
-      const { retcode, data: res, retmsg } = data
+      const { retcode, data: res, retmsg } = data;
       if (retcode === 0) {
         message.success('密码修改成功！');
-        callback && callback()
+        callback && callback();
       }
     },
     *getUserInfo({ payload = {} }, { call, put }) {
       const { data, response } = yield call(userService.user_info, payload);
-      const { retcode, data: res, retmsg } = data
+      const { retcode, data: res, retmsg } = data;
       const userInfo = {
         avatar: res.avatar,
         name: res.nickname,
-        email: res.email
+        email: res.email,
       };
-      localStorage.setItem('userInfo', JSON.stringify(userInfo))
+      authorizationUtil.setUserInfo(userInfo);
       if (retcode === 0) {
         // localStorage.setItem('userInfo',res.)
       }
@@ -79,91 +79,100 @@ const Model: settingModelType = {
       yield put({
         type: 'updateState',
         payload: {
-          loading: true
-        }
+          loading: true,
+        },
       });
-      const { data, response } = yield call(userService.get_tenant_info, payload);
-      const { retcode, data: res, retmsg } = data
+      const { data, response } = yield call(
+        userService.get_tenant_info,
+        payload,
+      );
+      const { retcode, data: res, retmsg } = data;
       // llm_id 对应chat_id
       // asr_id 对应speech2txt
 
       yield put({
         type: 'updateState',
         payload: {
-          loading: false
-        }
+          loading: false,
+        },
       });
       if (retcode === 0) {
-        res.chat_id = res.llm_id
-        res.speech2text_id = res.asr_id
+        res.chat_id = res.llm_id;
+        res.speech2text_id = res.asr_id;
         yield put({
           type: 'updateState',
           payload: {
-            tenantIfo: res
-          }
+            tenantIfo: res,
+          },
         });
       }
     },
     *set_tenant_info({ payload = {} }, { call, put }) {
-      const { data, response } = yield call(userService.set_tenant_info, payload);
-      const { retcode, data: res, retmsg } = data
+      const { data, response } = yield call(
+        userService.set_tenant_info,
+        payload,
+      );
+      const { retcode, data: res, retmsg } = data;
       // llm_id 对应chat_id
       // asr_id 对应speech2txt
       if (retcode === 0) {
         yield put({
           type: 'updateState',
           payload: {
-            isShowSSModal: false
-          }
+            isShowSSModal: false,
+          },
         });
         yield put({
-          type: 'getTenantInfo'
-        })
+          type: 'getTenantInfo',
+        });
       }
     },
 
     *factories_list({ payload = {} }, { call, put }) {
-      const { data, response } = yield call(userService.factories_list, payload);
-      const { retcode, data: res, retmsg } = data
+      const { data, response } = yield call(
+        userService.factories_list,
+        payload,
+      );
+      const { retcode, data: res, retmsg } = data;
       if (retcode === 0) {
         yield put({
           type: 'updateState',
           payload: {
-            factoriesList: res
-          }
+            factoriesList: res,
+          },
         });
       }
     },
     *llm_list({ payload = {} }, { call, put }) {
       const { data, response } = yield call(userService.llm_list, payload);
-      const { retcode, data: res, retmsg } = data
+      const { retcode, data: res, retmsg } = data;
       if (retcode === 0) {
         yield put({
           type: 'updateState',
           payload: {
-            llmInfo: res
-          }
+            llmInfo: res,
+          },
         });
       }
     },
     *my_llm({ payload = {} }, { call, put }) {
       const { data, response } = yield call(userService.my_llm, payload);
-      const { retcode, data: res, retmsg } = data
+      const { retcode, data: res, retmsg } = data;
       if (retcode === 0) {
         yield put({
           type: 'updateState',
           payload: {
-            myLlm: res
-          }
+            myLlm: res,
+          },
         });
       }
     },
     *set_api_key({ payload = {}, callback }, { call, put }) {
       const { data, response } = yield call(userService.set_api_key, payload);
-      const { retcode, data: res, retmsg } = data
+      const { retcode, data: res, retmsg } = data;
       if (retcode === 0) {
         message.success('设置API KEY成功！');
-        callback && callback()
+        callback && callback();
       }
     },
   },
@@ -171,9 +180,9 @@ const Model: settingModelType = {
     updateState(state, { payload }) {
       return {
         ...state,
-        ...payload
+        ...payload,
       };
-    }
-  }
+    },
+  },
 };
 export default Model;