fix: only one default role exists (#216)

Author: ayangweb

.eslintrc-auto-import.json

@@ -32,7 +32,7 @@
     "executeSQL": true,
     "generalKeys": true,
     "getActivePinia": true,
-    "getAiIamge": true,
+    "getAiImage": true,
     "getAiMessage": true,
     "getContributorsApi": true,
     "getCurrentInstance": true,

src/components/Function/components/ToggleAnswer.vue

@@ -13,7 +13,7 @@ const handleClick = () => {
 
     updateSessionData(getLastItem(sessionDataList.value))
   } else {
-    currentSession.value?.type === 'text' ? getAiMessage() : getAiIamge()
+    currentSession.value?.type === 'text' ? getAiMessage() : getAiImage()
   }
 }
 </script>

src/components/Input/index.vue

@@ -30,7 +30,7 @@ const onKeydown = (event: KeyboardEvent) => {
       popoverVisible.value = false
 
       if (currentSession.value?.type === 'text') getAiMessage(value)
-      else getAiIamge(value)
+      else getAiImage(value)
 
       textAreaElement.value?.blur()
       textAreaValue.value = ''

src/locales/es.json

@@ -158,6 +158,6 @@
     "edit": "Editar",
     "delete": "Eliminar",
     "cancel": "Cancelar",
-    "confirm": "Confirmar"    
+    "confirm": "Confirmar"
   }
 }

src/sqls/index.ts

@@ -1,6 +1,11 @@
 import { appConfigDir } from '@tauri-apps/api/path'
 import Database from 'tauri-plugin-sql-api'
-import type { TableName, TablePayload, WherePayload } from '@/types'
+import type {
+  TableName,
+  TablePayload,
+  WherePayload,
+  RolePayload
+} from '@/types'
 
 const dbFile = import.meta.env.DEV ? 'sql.dev.db' : 'sql.db'
 const db = await Database.load(`sqlite:${await appConfigDir()}${dbFile}`)
@@ -134,16 +139,14 @@ export const insertSQL = async (
   tableName: TableName,
   payload: TablePayload
 ) => {
-  if (tableName === 'role') {
-    const findPayload = Object.keys(payload).reduce((result, key) => {
-      const newKey = key as keyof TablePayload
+  const { changeDefaultRole } = useRoleStore()
 
-      return result.concat({ key: newKey, value: payload[newKey] })
-    }, [] as WherePayload[])
+  if (tableName === 'role' && (payload as RolePayload).is_default) {
+    const findDefaultRole = await selectSQL('role', [
+      { key: 'is_default', value: true }
+    ])
 
-    const findRole = await selectSQL('role', findPayload)
-
-    if (findRole.length) return
+    if (findDefaultRole.length) return changeDefaultRole()
   }
 
   const insertKeys = [],

src/stores/role.ts

@@ -113,6 +113,25 @@ export const useRoleStore = defineStore(
       )[0]
     }
 
+    // 改变默认角色的信息（适配语言变更）
+    const changeDefaultRole = async () => {
+      const findDefaultRole = await selectSQL('role', [
+        { key: 'is_default', value: true }
+      ])
+
+      if (!findDefaultRole.length) return
+
+      const { t } = i18n.global
+
+      await updateSQL('role', {
+        ...findDefaultRole[0],
+        name: t('session.role.default.name'),
+        description: t('session.role.default.description')
+      })
+
+      if (currentRole.value?.is_default) changeCurrentRole()
+    }
+
     return {
       currentRole,
       roleList,
@@ -125,7 +144,8 @@ export const useRoleStore = defineStore(
       addRole,
       updateRole,
       deleteRole,
-      changeCurrentRole
+      changeCurrentRole,
+      changeDefaultRole
     }
   },
   {

src/stores/settings.ts

@@ -86,6 +86,8 @@ export const useSettingsStore = defineStore(
 
     // 语言切换
     const setLanguage = async (lang?: Locales) => {
+      const { changeDefaultRole } = useRoleStore()
+
       let systemLang = ((await invoke('get_user_language')) as string).split(
         '-'
       )[0] as Locales
@@ -95,6 +97,8 @@ export const useSettingsStore = defineStore(
       }
 
       locale.value = lang || currentLang.value || systemLang
+
+      changeDefaultRole()
     }
 
     onMounted(() => {

src/utils/getMessage.ts

@@ -134,7 +134,7 @@ export const getAiMessage = async (value?: string) => {
  * 获取 ai 作图
  * @param value 消息内容
  */
-export const getAiIamge = async (value?: string) => {
+export const getAiImage = async (value?: string) => {
   const key = getOpenAIKey()
   if (!key) return