feat: 登录优化

canyon-project · Nov 1, 2023 · 99de329 · 99de329
1 parent 59e6062
commit 99de329
Showing 1 changed file with 29 additions and 3 deletions.
diff --git a/packages/app-ui/src/layouts/MainBox.tsx b/packages/app-ui/src/layouts/MainBox.tsx
@@ -68,7 +68,33 @@ const MainBox = () => {
     },
   );
 
-  const { data: meData } = useQuery(MeDocument);
+  const { data: meData } = useRequest(
+    () =>
+      axios
+        .post<any>(
+          '/graphql',
+          {
+            operationName: 'Me',
+            variables: {},
+            query:
+              'query Me {\n  me {\n    id\n    username\n    password\n    nickname\n    avatar\n    th_refresh_token\n    th_access_token\n    email\n    th_id\n    role\n    created_at\n    __typename\n  }\n}',
+          },
+          {
+            headers: {
+              Authorization: `Bearer ` + (localStorage.getItem('token') || ''),
+            },
+          },
+        )
+        .then((res) => res.data),
+    {
+      onSuccess(res) {
+        if (!['/welcome', '/login'].includes(loc.pathname) && !res.data) {
+          localStorage.clear();
+          nav(`/welcome`);
+        }
+      },
+    },
+  );
 
   const nav = useNavigate();
   const loc = useLocation();
@@ -166,13 +192,13 @@ const MainBox = () => {
                   <Dropdown menu={{ items, onClick }}>
                     <div>
                       <HoverEffect>
-                        <Avatar size={'small'} src={meData?.me.avatar} />
+                        <Avatar size={'small'} src={meData?.data?.me.avatar} />
                         <span
                           css={css`
                             margin-left: 6px;
                           `}
                         >
-                          {meData?.me.nickname}
+                          {meData?.data?.me.nickname}
                         </span>
                       </HoverEffect>
                     </div>