From b6f45dd7a7c1f74aef07fbaec7598e3ed5fffc75 Mon Sep 17 00:00:00 2001
From: Lazy <2211717435@qq.com>
Date: Wed, 30 Aug 2023 21:16:27 +0800
Subject: [PATCH] Feat: add update user apikey

---
 frontend/src/components/MyselfInfoDialog.vue | 63 ++++++++++++--------
 frontend/src/services/apis/index.ts          | 29 ++++++---
 2 files changed, 59 insertions(+), 33 deletions(-)

diff --git a/frontend/src/components/MyselfInfoDialog.vue b/frontend/src/components/MyselfInfoDialog.vue
index 0221444e..25d3ea8d 100644
--- a/frontend/src/components/MyselfInfoDialog.vue
+++ b/frontend/src/components/MyselfInfoDialog.vue
@@ -3,37 +3,31 @@ import { t } from "@/lang/i18n";
 import { useAppStateStore } from "@/stores/useAppStateStore";
 import { useAppToolsStore } from "@/stores/useAppToolsStore";
 import { reactive, toRaw } from "vue";
-const { state } = useAppStateStore();
+import { setUserApiKey } from "@/services/apis";
+import { message } from "ant-design-vue";
+const { state, updateUserInfo } = useAppStateStore();
 const { state: tools } = useAppToolsStore();
 
-// const menus = [
-//   {
-//     title: t("基本信息"),
-//     key: "baseInfo",
-//     icon: ProjectOutlined
-//   },
-//   {
-//     title: t("第三方接口"),
-//     key: "api",
-//     icon: RobotOutlined
-//   },
-//   {
-//     title: t("账号安全"),
-//     key: "password",
-//     icon: RobotOutlined
-//   }
-// ];
+const { execute, isLoading } = setUserApiKey();
 
 const formState = reactive({
   resetPassword: false,
-  apikey: "",
   oldPassword: "",
   password1: "",
   password2: ""
 });
 
-const handleGenerateApiKey = () => {
-  formState.apikey = "XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX";
+const handleGenerateApiKey = async (enable: boolean) => {
+  const res = await execute({
+    data: {
+      enable
+    }
+  });
+
+  if (res.value) {
+    updateUserInfo();
+    return message.success(t("更新成功"));
+  }
 };
 
 const handleChangePassword = () => {};
@@ -45,8 +39,8 @@ const onSubmit = () => {
 <template>
   <a-modal
     v-model:open="tools.showUserInfoDialog"
-    :title="t('用户信息')"
     centered
+    :title="t('用户信息')"
     :footer="null"
     @ok="tools.showUserInfoDialog = false"
   >
@@ -124,10 +118,29 @@ const onSubmit = () => {
               )
             }}
           </a-typography-paragraph>
-          <a-typography-paragraph v-if="formState.apikey">
-            <pre>{{ formState.apikey }}</pre>
+          <a-typography-paragraph v-if="state.userInfo?.apiKey">
+            <pre>{{ state.userInfo.apiKey }}</pre>
           </a-typography-paragraph>
-          <a-button size="default" danger @click="handleGenerateApiKey">生成</a-button>
+          <a-typography-paragraph v-else>
+            <pre>{{ t("未启用") }}</pre>
+          </a-typography-paragraph>
+          <a-button
+            class="mr-10"
+            size="default"
+            :loading="isLoading"
+            @click="handleGenerateApiKey(true)"
+          >
+            {{ t("生成") }}
+          </a-button>
+          <a-popconfirm
+            v-if="state.userInfo?.apiKey"
+            :title="t('你确定要关闭APIKey吗')"
+            ok-text="Yes"
+            cancel-text="No"
+            @confirm="handleGenerateApiKey(false)"
+          >
+            <a-button size="default" danger> {{ t("关闭") }} </a-button>
+          </a-popconfirm>
         </a-form-item>
       </a-form>
     </div>
diff --git a/frontend/src/services/apis/index.ts b/frontend/src/services/apis/index.ts
index 6c09db9b..18b6c866 100644
--- a/frontend/src/services/apis/index.ts
+++ b/frontend/src/services/apis/index.ts
@@ -7,12 +7,12 @@ import type { BaseUserInfo } from "@/types/user";
 // 用户登录
 export const loginUser = useDefineApi<
   | {
-      // Post
-      data: {
-        username: string;
-        password: string;
-      };
-    }
+    // Post
+    data: {
+      username: string;
+      password: string;
+    };
+  }
   | undefined,
   // Response
   {
@@ -61,11 +61,24 @@ export const settingInfo = useDefineApi<any, Settings>({
 // 提交设置信息
 export const setSettingInfo = useDefineApi<
   | {
-      data: Settings;
-    }
+    data: Settings;
+  }
   | undefined,
   string
 >({
   url: "/api/overview/setting",
   method: "PUT"
 });
+
+// 设置用户APIKey
+export const setUserApiKey = useDefineApi<
+  {
+    data: {
+      enable: boolean;
+    };
+  },
+  string
+>({
+  url: "/api/auth/api",
+  method: "PUT"
+});
\ No newline at end of file