feat: attach token from query

Signed-off-by: Innei <[email protected]>

src/main.ts

@@ -18,6 +18,9 @@ import App from './App'
 import './index.css'
 
 import { router } from './router'
+import { attachTokenFromQuery } from './utils'
+
+attachTokenFromQuery()
 
 const app = createApp(App)
 

src/utils/auth.ts

@@ -1,5 +1,7 @@
 import Cookies from 'js-cookie'
 
+import { router } from '~/router'
+
 export const TokenKey = 'mx-token'
 
 /**
@@ -22,3 +24,23 @@ export function setToken(token: string) {
 export function removeToken() {
   return Cookies.remove(TokenKey)
 }
+
+export const attachTokenFromQuery = () => {
+  const token = new URLSearchParams(window.location.search).get('token')
+  if (token) {
+    setToken(token)
+    router.isReady().then(() => {
+      const parsedUrl = new URL(window.location.href)
+      parsedUrl.searchParams.delete('token')
+      const query = {} as any
+      for (const [key, value] of parsedUrl.searchParams.entries()) {
+        query[key] = value
+      }
+
+      router.replace({
+        path: parsedUrl.pathname,
+        query,
+      })
+    })
+  }
+}