v1.0.0

src/store/modules/auth/helper.ts

@@ -0,0 +1,24 @@
+import { ss } from '@/utils/storage'
+
+const LOCAL_NAME = 'SECRET_TOKEN'
+
+export function getToken() {
+  return ss.get(LOCAL_NAME)
+}
+
+export function setToken(token: string) {
+  return ss.set(LOCAL_NAME, token)
+}
+
+export function setUserInfo(userInfo: User.Info) {
+  return ss.set(LOCAL_NAME, userInfo)
+}
+
+export function getUserInfo() {
+  return ss.get(LOCAL_NAME)
+}
+
+export function removeToken() {
+  // ss.clear()
+  return ss.remove(LOCAL_NAME)
+}

src/store/modules/auth/index.ts

@@ -0,0 +1,63 @@
+import { defineStore } from 'pinia'
+import { getToken, getUserInfo, removeToken, setToken } from './helper'
+import { store } from '@/store'
+import { fetchSession } from '@/api'
+
+interface SessionResponse {
+  auth: boolean
+  model: 'ChatGPTAPI' | 'ChatGPTUnofficialProxyAPI'
+}
+
+export interface AuthState {
+  token: string | undefined
+  userInfo: User.Info | undefined
+  session: SessionResponse | null
+
+}
+
+export const useAuthStore = defineStore('auth-store', {
+  state: (): AuthState => ({
+    userInfo: getUserInfo(),
+    token: getToken(),
+    session: null,
+  }),
+
+  getters: {
+    isChatGPTAPI(state): boolean {
+      return state.session?.model === 'ChatGPTAPI'
+    },
+  },
+
+  actions: {
+    async getSession() {
+      try {
+        const { data } = await fetchSession<SessionResponse>()
+        this.session = { ...data }
+        return Promise.resolve(data)
+      }
+      catch (error) {
+        return Promise.reject(error)
+      }
+    },
+
+    setToken(token: string) {
+      this.token = token
+      setToken(token)
+    },
+
+    setUserInfo(userInfo: User.Info) {
+      this.userInfo = userInfo
+      this.setUserInfo(userInfo)
+    },
+
+    // 清除所有的本地储存
+    removeToken() {
+      this.token = undefined
+      removeToken()
+    },
+  },
+})
+
+export function useAuthStoreWithout() {
+  return useAuthStore(store)
+}