From 300609472b10e4151eba5b23bb8c5ce1ca8fed8c Mon Sep 17 00:00:00 2001
From: carry <2641257231@qq.com>
Date: Fri, 7 Feb 2025 01:39:43 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E6=88=90=E4=BA=86pinia=E7=8A=B6?=
 =?UTF-8?q?=E6=80=81=E5=92=8C=E9=89=B4=E6=9D=83api?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/api/authService.ts | 15 ++++++++
 src/store/userStore.ts | 77 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 92 insertions(+)
 create mode 100644 src/api/authService.ts
 create mode 100644 src/store/userStore.ts

diff --git a/src/api/authService.ts b/src/api/authService.ts
new file mode 100644
index 0000000..395c5f0
--- /dev/null
+++ b/src/api/authService.ts
@@ -0,0 +1,15 @@
+import axios from 'axios';
+
+const API_BASE_URL = '/api/auth';
+
+export const authService = {
+  async login(username: string, password: string) {
+    const response = await axios.post(`${API_BASE_URL}/login`, { username, password });
+    return response.data;
+  },
+
+  async refreshToken(refreshToken: string) {
+    const response = await axios.post(`${API_BASE_URL}/refresh`, { refresh_token: refreshToken });
+    return response.data;
+  }
+};
\ No newline at end of file
diff --git a/src/store/userStore.ts b/src/store/userStore.ts
new file mode 100644
index 0000000..9fc9b20
--- /dev/null
+++ b/src/store/userStore.ts
@@ -0,0 +1,77 @@
+import { defineStore } from 'pinia';
+import { ref } from 'vue';
+import { authService } from '../api/authService';
+import piniaPluginPersistedstate from 'pinia-plugin-persistedstate';
+import { createPinia } from 'pinia';
+
+const pinia = createPinia();
+pinia.use(piniaPluginPersistedstate);
+
+export const useAuthStore = defineStore('auth', () => {
+  const accessToken = ref('');
+  const refreshToken = ref('');
+  const role = ref('');
+  const username = ref('');
+  const userId = ref<number | null>(null);
+
+  function setTokens(tokens: { access_token: string; refresh_token: string }) {
+    accessToken.value = tokens.access_token;
+    refreshToken.value = tokens.refresh_token;
+  }
+
+  function setRole(userRole: string) {
+    role.value = userRole;
+  }
+
+  async function login(usernameParam: string, password: string) {
+    try {
+      const { access_token, refresh_token, role: userRole, username: userUsername, id: userIdValue } = await authService.login(usernameParam, password);
+      setTokens({ access_token, refresh_token });
+      setRole(userRole);
+      username.value = userUsername;
+      userId.value = userIdValue;
+    } catch (error) {
+      console.error('Login failed:', error);
+    }
+  }
+
+  async function logout() {
+    accessToken.value = '';
+    refreshToken.value = '';
+    role.value = '';
+    username.value = '';
+    userId.value = null;
+  }
+
+  async function refreshTokenMethod() {
+    try {
+      const { access_token, refresh_token } = await authService.refreshToken(refreshToken.value);
+      setTokens({ access_token, refresh_token });
+    } catch (error) {
+      console.error('Token refresh failed:', error);
+    }
+  }
+
+  return { accessToken, refreshTokenToken: refreshToken, role, username, userId, login, logout, refreshTokenMethod };
+}, {
+  persist: true,
+});
+
+// 使用插件进行持久化
+useAuthStore().$subscribe((mutation, state) => {
+  localStorage.setItem('authStore', JSON.stringify(state));
+});
+
+useAuthStore().$onAction(({ name, store, args, after, onError }) => {
+  if (name === 'login' || name === 'refreshTokenMethod') {
+    after(() => {
+      localStorage.setItem('authStore', JSON.stringify(store.$state));
+    });
+  }
+});
+
+// 初始化时从本地存储恢复状态
+const persistedState = JSON.parse(localStorage.getItem('authStore') || '{}');
+if (persistedState.accessToken && persistedState.refreshTokenToken && persistedState.role && persistedState.username && persistedState.userId !== null) {
+  useAuthStore().$patch(persistedState);
+}