frontend 兼容

ai-chat-web/src/store/index.ts

@@ -0,0 +1,10 @@
+import type { App } from 'vue'
+import { createPinia } from 'pinia'
+
+export const store = createPinia()
+
+export function setupStore(app: App) {
+  app.use(store)
+}
+
+export * from './modules'

ai-chat-web/src/store/modules/app/helper.ts

@@ -0,0 +1,26 @@
+import { ss } from '@/utils/storage'
+
+const LOCAL_NAME = 'appSetting'
+
+export type Theme = 'light' | 'dark' | 'auto'
+
+export type Language = 'zh-CN' | 'zh-TW' | 'en-US'
+
+export interface AppState {
+  siderCollapsed: boolean
+  theme: Theme
+  language: Language
+}
+
+export function defaultSetting(): AppState {
+  return { siderCollapsed: false, theme: 'light', language: 'zh-CN' }
+}
+
+export function getLocalSetting(): AppState {
+  const localSetting: AppState | undefined = ss.get(LOCAL_NAME)
+  return { ...defaultSetting(), ...localSetting }
+}
+
+export function setLocalSetting(setting: AppState): void {
+  ss.set(LOCAL_NAME, setting)
+}

ai-chat-web/src/store/modules/app/index.ts

@@ -0,0 +1,34 @@
+import { defineStore } from 'pinia'
+import type { AppState, Language, Theme } from './helper'
+import { getLocalSetting, setLocalSetting } from './helper'
+import { store } from '@/store'
+
+export const useAppStore = defineStore('app-store', {
+  state: (): AppState => getLocalSetting(),
+  actions: {
+    setSiderCollapsed(collapsed: boolean) {
+      this.siderCollapsed = collapsed
+      this.recordState()
+    },
+
+    setTheme(theme: Theme) {
+      this.theme = theme
+      this.recordState()
+    },
+
+    setLanguage(language: Language) {
+      if (this.language !== language) {
+        this.language = language
+        this.recordState()
+      }
+    },
+
+    recordState() {
+      setLocalSetting(this.$state)
+    },
+  },
+})
+
+export function useAppStoreWithOut() {
+  return useAppStore(store)
+}

ai-chat-web/src/store/modules/auth/helper.ts

@@ -0,0 +1,15 @@
+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 removeToken() {
+  return ss.remove(LOCAL_NAME)
+}

ai-chat-web/src/store/modules/auth/index.ts

@@ -0,0 +1,54 @@
+import { defineStore } from 'pinia'
+import { getToken, 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
+  session: SessionResponse | null
+}
+
+export const useAuthStore = defineStore('auth-store', {
+  state: (): AuthState => ({
+    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)
+    },
+
+    removeToken() {
+      this.token = undefined
+      removeToken()
+    },
+  },
+})
+
+export function useAuthStoreWithout() {
+  return useAuthStore(store)
+}

ai-chat-web/src/store/modules/chat/helper.ts

@@ -0,0 +1,22 @@
+import { ss } from '@/utils/storage'
+
+const LOCAL_NAME = 'chatStorage'
+
+export function defaultState(): Chat.ChatState {
+  const uuid = 1002
+  return {
+    active: uuid,
+    usingContext: true,
+    history: [{ uuid, title: 'New Chat', isEdit: false }],
+    chat: [{ uuid, data: [] }],
+  }
+}
+
+export function getLocalState(): Chat.ChatState {
+  const localState = ss.get(LOCAL_NAME)
+  return { ...defaultState(), ...localState }
+}
+
+export function setLocalState(state: Chat.ChatState) {
+  ss.set(LOCAL_NAME, state)
+}

ai-chat-web/src/store/modules/chat/index.ts

@@ -0,0 +1,194 @@
+import { defineStore } from 'pinia'
+import { getLocalState, setLocalState } from './helper'
+import { router } from '@/router'
+
+export const useChatStore = defineStore('chat-store', {
+  state: (): Chat.ChatState => getLocalState(),
+
+  getters: {
+    getChatHistoryByCurrentActive(state: Chat.ChatState) {
+      const index = state.history.findIndex(item => item.uuid === state.active)
+      if (index !== -1)
+        return state.history[index]
+      return null
+    },
+
+    getChatByUuid(state: Chat.ChatState) {
+      return (uuid?: number) => {
+        if (uuid)
+          return state.chat.find(item => item.uuid === uuid)?.data ?? []
+        return state.chat.find(item => item.uuid === state.active)?.data ?? []
+      }
+    },
+  },
+
+  actions: {
+    setUsingContext(context: boolean) {
+      this.usingContext = context
+      this.recordState()
+    },
+
+    addHistory(history: Chat.History, chatData: Chat.Chat[] = []) {
+      this.history.unshift(history)
+      this.chat.unshift({ uuid: history.uuid, data: chatData })
+      this.active = history.uuid
+      this.reloadRoute(history.uuid)
+    },
+
+    updateHistory(uuid: number, edit: Partial<Chat.History>) {
+      const index = this.history.findIndex(item => item.uuid === uuid)
+      if (index !== -1) {
+        this.history[index] = { ...this.history[index], ...edit }
+        this.recordState()
+      }
+    },
+
+    async deleteHistory(index: number) {
+      this.history.splice(index, 1)
+      this.chat.splice(index, 1)
+
+      if (this.history.length === 0) {
+        this.active = null
+        this.reloadRoute()
+        return
+      }
+
+      if (index > 0 && index <= this.history.length) {
+        const uuid = this.history[index - 1].uuid
+        this.active = uuid
+        this.reloadRoute(uuid)
+        return
+      }
+
+      if (index === 0) {
+        if (this.history.length > 0) {
+          const uuid = this.history[0].uuid
+          this.active = uuid
+          this.reloadRoute(uuid)
+        }
+      }
+
+      if (index > this.history.length) {
+        const uuid = this.history[this.history.length - 1].uuid
+        this.active = uuid
+        this.reloadRoute(uuid)
+      }
+    },
+
+    async setActive(uuid: number) {
+      this.active = uuid
+      return await this.reloadRoute(uuid)
+    },
+
+    getChatByUuidAndIndex(uuid: number, index: number) {
+      if (!uuid || uuid === 0) {
+        if (this.chat.length)
+          return this.chat[0].data[index]
+        return null
+      }
+      const chatIndex = this.chat.findIndex(item => item.uuid === uuid)
+      if (chatIndex !== -1)
+        return this.chat[chatIndex].data[index]
+      return null
+    },
+
+    addChatByUuid(uuid: number, chat: Chat.Chat) {
+      if (!uuid || uuid === 0) {
+        if (this.history.length === 0) {
+          const uuid = Date.now()
+          this.history.push({ uuid, title: chat.text, isEdit: false })
+          this.chat.push({ uuid, data: [chat] })
+          this.active = uuid
+          this.recordState()
+        }
+        else {
+          this.chat[0].data.push(chat)
+          if (this.history[0].title === 'New Chat')
+            this.history[0].title = chat.text
+          this.recordState()
+        }
+      }
+
+      const index = this.chat.findIndex(item => item.uuid === uuid)
+      if (index !== -1) {
+        this.chat[index].data.push(chat)
+        if (this.history[index].title === 'New Chat')
+          this.history[index].title = chat.text
+        this.recordState()
+      }
+    },
+
+    updateChatByUuid(uuid: number, index: number, chat: Chat.Chat) {
+      if (!uuid || uuid === 0) {
+        if (this.chat.length) {
+          this.chat[0].data[index] = chat
+          this.recordState()
+        }
+        return
+      }
+
+      const chatIndex = this.chat.findIndex(item => item.uuid === uuid)
+      if (chatIndex !== -1) {
+        this.chat[chatIndex].data[index] = chat
+        this.recordState()
+      }
+    },
+
+    updateChatSomeByUuid(uuid: number, index: number, chat: Partial<Chat.Chat>) {
+      if (!uuid || uuid === 0) {
+        if (this.chat.length) {
+          this.chat[0].data[index] = { ...this.chat[0].data[index], ...chat }
+          this.recordState()
+        }
+        return
+      }
+
+      const chatIndex = this.chat.findIndex(item => item.uuid === uuid)
+      if (chatIndex !== -1) {
+        this.chat[chatIndex].data[index] = { ...this.chat[chatIndex].data[index], ...chat }
+        this.recordState()
+      }
+    },
+
+    deleteChatByUuid(uuid: number, index: number) {
+      if (!uuid || uuid === 0) {
+        if (this.chat.length) {
+          this.chat[0].data.splice(index, 1)
+          this.recordState()
+        }
+        return
+      }
+
+      const chatIndex = this.chat.findIndex(item => item.uuid === uuid)
+      if (chatIndex !== -1) {
+        this.chat[chatIndex].data.splice(index, 1)
+        this.recordState()
+      }
+    },
+
+    clearChatByUuid(uuid: number) {
+      if (!uuid || uuid === 0) {
+        if (this.chat.length) {
+          this.chat[0].data = []
+          this.recordState()
+        }
+        return
+      }
+
+      const index = this.chat.findIndex(item => item.uuid === uuid)
+      if (index !== -1) {
+        this.chat[index].data = []
+        this.recordState()
+      }
+    },
+
+    async reloadRoute(uuid?: number) {
+      this.recordState()
+      await router.push({ name: 'Chat', params: { uuid } })
+    },
+
+    recordState() {
+      setLocalState(this.$state)
+    },
+  },
+})

ai-chat-web/src/store/modules/index.ts

@@ -0,0 +1,6 @@
+export * from './app'
+export * from './chat'
+export * from './user'
+export * from './prompt'
+export * from './settings'
+export * from './auth'

ai-chat-web/src/store/modules/prompt/helper.ts

@@ -0,0 +1,18 @@
+import { ss } from '@/utils/storage'
+
+const LOCAL_NAME = 'promptStore'
+
+export type PromptList = []
+
+export interface PromptStore {
+  promptList: PromptList
+}
+
+export function getLocalPromptList(): PromptStore {
+  const promptStore: PromptStore | undefined = ss.get(LOCAL_NAME)
+  return promptStore ?? { promptList: [] }
+}
+
+export function setLocalPromptList(promptStore: PromptStore): void {
+  ss.set(LOCAL_NAME, promptStore)
+}

ai-chat-web/src/store/modules/prompt/index.ts

@@ -0,0 +1,17 @@
+import { defineStore } from 'pinia'
+import type { PromptStore } from './helper'
+import { getLocalPromptList, setLocalPromptList } from './helper'
+
+export const usePromptStore = defineStore('prompt-store', {
+  state: (): PromptStore => getLocalPromptList(),
+
+  actions: {
+    updatePromptList(promptList: []) {
+      this.$patch({ promptList })
+      setLocalPromptList({ promptList })
+    },
+    getPromptList() {
+      return this.$state
+    },
+  },
+})

ai-chat-web/src/store/modules/settings/helper.ts

@@ -0,0 +1,27 @@
+import { ss } from '@/utils/storage'
+
+const LOCAL_NAME = 'settingsStorage'
+
+export interface SettingsState {
+  systemMessage: string
+}
+
+export function defaultSetting(): SettingsState {
+  const currentDate = new Date().toISOString().split('T')[0]
+  return {
+    systemMessage: `You are ChatGPT, a large language model trained by OpenAI. Answer as concisely as possible.\nKnowledge cutoff: 2021-09-01\nCurrent date: ${currentDate}`,
+  }
+}
+
+export function getLocalState(): SettingsState {
+  const localSetting: SettingsState | undefined = ss.get(LOCAL_NAME)
+  return { ...defaultSetting(), ...localSetting }
+}
+
+export function setLocalState(setting: SettingsState): void {
+  ss.set(LOCAL_NAME, setting)
+}
+
+export function removeLocalState() {
+  ss.remove(LOCAL_NAME)
+}

ai-chat-web/src/store/modules/settings/index.ts

@@ -0,0 +1,22 @@
+import { defineStore } from 'pinia'
+import type { SettingsState } from './helper'
+import { defaultSetting, getLocalState, removeLocalState, setLocalState } from './helper'
+
+export const useSettingStore = defineStore('setting-store', {
+  state: (): SettingsState => getLocalState(),
+  actions: {
+    updateSetting(settings: Partial<SettingsState>) {
+      this.$state = { ...this.$state, ...settings }
+      this.recordState()
+    },
+
+    resetSetting() {
+      this.$state = defaultSetting()
+      removeLocalState()
+    },
+
+    recordState() {
+      setLocalState(this.$state)
+    },
+  },
+})

ai-chat-web/src/store/modules/user/helper.ts

@@ -0,0 +1,32 @@
+import { ss } from '@/utils/storage'
+
+const LOCAL_NAME = 'userStorage'
+
+export interface UserInfo {
+  avatar: string
+  name: string
+  description: string
+}
+
+export interface UserState {
+  userInfo: UserInfo
+}
+
+export function defaultSetting(): UserState {
+  return {
+    userInfo: {
+      avatar: 'https://raw.githubusercontent.com/Chanzhaoyu/chatgpt-web/main/src/assets/avatar.jpg',
+      name: 'ChenZhaoYu',
+      description: 'Star on <a href="https://github.com/Chanzhaoyu/chatgpt-bot" class="text-blue-500" target="_blank" >Github</a>',
+    },
+  }
+}
+
+export function getLocalState(): UserState {
+  const localSetting: UserState | undefined = ss.get(LOCAL_NAME)
+  return { ...defaultSetting(), ...localSetting }
+}
+
+export function setLocalState(setting: UserState): void {
+  ss.set(LOCAL_NAME, setting)
+}

ai-chat-web/src/store/modules/user/index.ts

@@ -0,0 +1,22 @@
+import { defineStore } from 'pinia'
+import type { UserInfo, UserState } from './helper'
+import { defaultSetting, getLocalState, setLocalState } from './helper'
+
+export const useUserStore = defineStore('user-store', {
+  state: (): UserState => getLocalState(),
+  actions: {
+    updateUserInfo(userInfo: Partial<UserInfo>) {
+      this.userInfo = { ...this.userInfo, ...userInfo }
+      this.recordState()
+    },
+
+    resetUserInfo() {
+      this.userInfo = { ...defaultSetting().userInfo }
+      this.recordState()
+    },
+
+    recordState() {
+      setLocalState(this.$state)
+    },
+  },
+})