From 78b6416e15dfd671f7281bc4815d05cdc6e6e1ee Mon Sep 17 00:00:00 2001
From: keqingmoe <me@keqing.moe>
Date: Tue, 31 Dec 2024 00:28:51 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E7=94=A8=E6=88=B7=E6=9D=83?=
 =?UTF-8?q?=E9=99=90=E6=9F=A5=E8=AF=A2=E5=8A=9F=E8=83=BD=EF=BC=8C=E8=A7=A3?=
 =?UTF-8?q?=E7=A0=81JWT=E4=BB=A5=E8=8E=B7=E5=8F=96=E7=94=A8=E6=88=B7ID?=
 =?UTF-8?q?=EF=BC=8C=E5=B9=B6=E6=A0=B9=E6=8D=AE=E6=9D=83=E9=99=90=E6=9B=B4?=
 =?UTF-8?q?=E6=96=B0=E7=94=A8=E6=88=B7=E6=9D=83=E9=99=90=E7=8A=B6=E6=80=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ui/src/components/study/ProblemList.vue | 42 ++++++++++++++++++++++++-
 1 file changed, 41 insertions(+), 1 deletion(-)

diff --git a/ui/src/components/study/ProblemList.vue b/ui/src/components/study/ProblemList.vue
index 75719e9..68979bb 100644
--- a/ui/src/components/study/ProblemList.vue
+++ b/ui/src/components/study/ProblemList.vue
@@ -11,7 +11,7 @@
           刷新
         </v-btn>
         <v-dialog v-model="dialog0" max-width="500px">
-          <template v-slot:activator="{ props }">
+          <template v-if="userPermission == '1'" v-slot:activator="{ props }">
             <v-btn class="mb-2" color="primary" dark v-bind="props">
               新建题目
             </v-btn>
@@ -124,6 +124,7 @@ import { useAuthStore } from '@/store/auth';
 import axios, { Axios, AxiosError } from 'axios';
 import { computed, nextTick, onMounted, ref, watch } from 'vue';
 import * as mathjs from 'mathjs';
+import { jwtDecode, type JwtPayload } from 'jwt-decode';
 
 const loading = ref(false);
 const refresh = async () => {
@@ -278,6 +279,45 @@ watch(dialogDelete, (val) => {
 const authStore = useAuthStore();
 const storedToken = ref(authStore.token);
 
+interface KqmJwt extends JwtPayload {
+  user_id: string;
+};
+
+const userId = computed(() => {
+  if (storedToken.value != '') {
+    let data: KqmJwt = jwtDecode(storedToken.value);
+    return data.user_id;
+  }
+  return '';
+});
+
+type UserPermissionResponse = { success?: string, permission?: string, error?: string };
+
+const queryPermission = async () => {
+  try {
+    const formData = new FormData;
+    formData.append("user_id", userId.value);
+    let res = await axios.post('/api/auth/permission', formData);
+    return res.data as UserPermissionResponse;
+  } catch (e) {
+    let ex = e as AxiosError;
+    return ex.response?.data as UserPermissionResponse;
+  }
+}
+
+const userPermission = ref('');
+
+const updateUserPermission = async () => {
+  let res = await queryPermission();
+  if (res?.success) {
+    userPermission.value = res.permission as string;
+  } else {
+    userPermission.value = '';
+  }
+}
+
+watch(userId, updateUserPermission, { immediate: true });
+
 const decorate = <T>(ex: AxiosError) => {
   if (ex.response?.data) {
     return ex.response?.data as T;