From 8e1de8d29f6efe535cbbafdbfe6dc0332ef961b8 Mon Sep 17 00:00:00 2001
From: tbphp <admin@tbphp.net>
Date: Wed, 2 Jul 2025 19:14:23 +0800
Subject: [PATCH] feat: login

---
 web/src/App.vue                              |  7 ++-
 web/src/api/logs.ts                          | 14 +++++
 web/src/components/Layout.vue                |  4 +-
 web/src/components/Logout.vue                | 14 ++++-
 web/src/main.ts                              |  4 +-
 web/src/{utils/router.ts => router/index.ts} | 19 ++++++
 web/src/services/auth.ts                     | 33 ++++++++++
 web/src/types/models.ts                      | 63 ++++++++++++++++++++
 web/src/utils/http.ts                        | 12 +++-
 web/src/views/Login.vue                      | 55 ++++++++++++++++-
 10 files changed, 216 insertions(+), 9 deletions(-)
 create mode 100644 web/src/api/logs.ts
 rename web/src/{utils/router.ts => router/index.ts} (61%)
 create mode 100644 web/src/services/auth.ts
 create mode 100644 web/src/types/models.ts

diff --git a/web/src/App.vue b/web/src/App.vue
index d275699..e5b936c 100644
--- a/web/src/App.vue
+++ b/web/src/App.vue
@@ -1,7 +1,12 @@
 <script setup lang="ts">
 import Layout from "@/components/Layout.vue";
+import { NDialogProvider, NMessageProvider } from "naive-ui";
 </script>
 
 <template>
-  <layout />
+  <n-message-provider>
+    <n-dialog-provider>
+      <layout />
+    </n-dialog-provider>
+  </n-message-provider>
 </template>
diff --git a/web/src/api/logs.ts b/web/src/api/logs.ts
new file mode 100644
index 0000000..11c12ec
--- /dev/null
+++ b/web/src/api/logs.ts
@@ -0,0 +1,14 @@
+import type { Group, LogFilter, LogsResponse } from "@/types/models";
+import http from "@/utils/http";
+
+export const logApi = {
+  // 获取日志列表
+  getLogs: (params: LogFilter): Promise<LogsResponse> => {
+    return http.get("/logs", { params });
+  },
+
+  // 获取分组列表（用于筛选）
+  getGroups: (): Promise<Group[]> => {
+    return http.get("/groups");
+  },
+};
diff --git a/web/src/components/Layout.vue b/web/src/components/Layout.vue
index 6b08f53..f92d791 100644
--- a/web/src/components/Layout.vue
+++ b/web/src/components/Layout.vue
@@ -1,5 +1,5 @@
 <template>
-  <n-layout>
+  <n-layout v-if="authService.isLoggedIn()">
     <n-layout-header class="flex items-center">
       <h1 class="layout-header-title">T.COM</h1>
       <nav-bar />
@@ -9,11 +9,13 @@
       <router-view />
     </n-layout-content>
   </n-layout>
+  <router-view v-else />
 </template>
 
 <script setup lang="ts">
 import NavBar from "@/components/NavBar.vue";
 import Logout from "@/components/Logout.vue";
+import { authService } from "@/services/auth";
 </script>
 
 <style scoped>
diff --git a/web/src/components/Logout.vue b/web/src/components/Logout.vue
index da4e421..ab1d992 100644
--- a/web/src/components/Logout.vue
+++ b/web/src/components/Logout.vue
@@ -1,3 +1,15 @@
 <template>
-  <n-button quaternary round>退出</n-button>
+  <n-button quaternary round @click="handleLogout">退出</n-button>
 </template>
+
+<script setup lang="ts">
+import { authService } from "@/services/auth";
+import { useRouter } from "vue-router";
+
+const router = useRouter();
+
+const handleLogout = () => {
+  authService.logout();
+  router.push("/login");
+};
+</script>
diff --git a/web/src/main.ts b/web/src/main.ts
index 68c4333..8aacc86 100644
--- a/web/src/main.ts
+++ b/web/src/main.ts
@@ -1,7 +1,7 @@
+import App from "@/App.vue";
+import router from "@/router";
 import naive from "naive-ui";
 import { createApp } from "vue";
-import App from "./App.vue";
 import "./style.css";
-import router from "./utils/router";
 
 createApp(App).use(router).use(naive).mount("#app");
diff --git a/web/src/utils/router.ts b/web/src/router/index.ts
similarity index 61%
rename from web/src/utils/router.ts
rename to web/src/router/index.ts
index 1749f39..9988010 100644
--- a/web/src/utils/router.ts
+++ b/web/src/router/index.ts
@@ -1,3 +1,4 @@
+import { authService } from "@/services/auth";
 import { createRouter, createWebHistory, type RouteRecordRaw } from "vue-router";
 
 const routes: Array<RouteRecordRaw> = [
@@ -21,6 +22,11 @@ const routes: Array<RouteRecordRaw> = [
     name: "settings",
     component: () => import("@/views/Settings.vue"),
   },
+  {
+    path: "/login",
+    name: "login",
+    component: () => import("@/views/Login.vue"),
+  },
 ];
 
 const router = createRouter({
@@ -28,4 +34,17 @@ const router = createRouter({
   routes,
 });
 
+router.beforeEach((to, _from, next) => {
+  const loggedIn = authService.isLoggedIn();
+  if (to.path !== "/login" && !loggedIn) {
+    return next({ path: "/login" });
+  }
+
+  if (to.path === "/login" && loggedIn) {
+    return next({ path: "/" });
+  }
+
+  next();
+});
+
 export default router;
diff --git a/web/src/services/auth.ts b/web/src/services/auth.ts
new file mode 100644
index 0000000..08724b2
--- /dev/null
+++ b/web/src/services/auth.ts
@@ -0,0 +1,33 @@
+import http from "@/utils/http";
+
+const AUTH_KEY = "authKey";
+
+const login = async (authKey: string): Promise<boolean> => {
+  try {
+    await http.post("/auth/login", { auth_key: authKey });
+    localStorage.setItem(AUTH_KEY, authKey);
+    return true;
+  } catch (error) {
+    console.error("Login failed:", error);
+    return false;
+  }
+};
+
+const logout = (): void => {
+  localStorage.removeItem(AUTH_KEY);
+};
+
+const getAuthKey = (): string | null => {
+  return localStorage.getItem(AUTH_KEY);
+};
+
+const isLoggedIn = (): boolean => {
+  return !!localStorage.getItem(AUTH_KEY);
+};
+
+export const authService = {
+  login,
+  logout,
+  getAuthKey,
+  isLoggedIn,
+};
diff --git a/web/src/types/models.ts b/web/src/types/models.ts
new file mode 100644
index 0000000..80af94a
--- /dev/null
+++ b/web/src/types/models.ts
@@ -0,0 +1,63 @@
+// 数据模型定义
+export interface APIKey {
+  id: number;
+  group_id: number;
+  key_value: string;
+  status: "active" | "inactive" | "error";
+  request_count: number;
+  failure_count: number;
+  last_used_at?: string;
+  created_at: string;
+  updated_at: string;
+}
+
+export interface Group {
+  id: number;
+  name: string;
+  description: string;
+  channel_type: "openai" | "gemini";
+  config: string;
+  api_keys?: APIKey[];
+  created_at: string;
+  updated_at: string;
+}
+
+export interface RequestLog {
+  id: string;
+  timestamp: string;
+  group_id: number;
+  key_id: number;
+  source_ip: string;
+  status_code: number;
+  request_path: string;
+  request_body_snippet: string;
+}
+
+export interface LogsResponse {
+  total: number;
+  page: number;
+  size: number;
+  data: RequestLog[];
+}
+
+export interface LogFilter {
+  page: number;
+  size: number;
+  group_id?: number;
+  start_time?: string;
+  end_time?: string;
+  status_code?: number;
+  source_ip?: string;
+}
+
+export interface DashboardStats {
+  total_requests: number;
+  success_requests: number;
+  success_rate: number;
+  group_stats: GroupRequestStat[];
+}
+
+export interface GroupRequestStat {
+  group_name: string;
+  request_count: number;
+}
diff --git a/web/src/utils/http.ts b/web/src/utils/http.ts
index 8bfe8f2..9e73f9a 100644
--- a/web/src/utils/http.ts
+++ b/web/src/utils/http.ts
@@ -1,4 +1,6 @@
+import { authService } from "@/services/auth";
 import axios from "axios";
+import { useRouter } from "vue-router";
 
 const http = axios.create({
   baseURL: "/api",
@@ -8,9 +10,9 @@ const http = axios.create({
 
 // 请求拦截器
 http.interceptors.request.use(config => {
-  const token = localStorage.getItem("token");
-  if (token) {
-    config.headers.Authorization = `Bearer ${token}`;
+  const authKey = authService.getAuthKey();
+  if (authKey) {
+    config.headers.Authorization = `Bearer ${authKey}`;
   }
   return config;
 });
@@ -19,6 +21,10 @@ http.interceptors.request.use(config => {
 http.interceptors.response.use(
   response => response.data,
   error => {
+    if (error.response && error.response.status === 401) {
+      authService.logout();
+      useRouter().push("/login");
+    }
     console.error("API Error:", error);
     return Promise.reject(error);
   }
diff --git a/web/src/views/Login.vue b/web/src/views/Login.vue
index 7537bb0..9606bbc 100644
--- a/web/src/views/Login.vue
+++ b/web/src/views/Login.vue
@@ -1,3 +1,56 @@
 <template>
-  <h1>Login</h1>
+  <div class="login-container">
+    <n-card class="login-card" title="Login">
+      <n-space vertical>
+        <n-input
+          v-model:value="authKey"
+          type="password"
+          placeholder="Auth Key"
+          @keyup.enter="handleLogin"
+        />
+        <n-button type="primary" block @click="handleLogin" :loading="loading">Login</n-button>
+      </n-space>
+    </n-card>
+  </div>
 </template>
+
+<script setup lang="ts">
+import { authService } from "@/services/auth";
+import { NButton, NCard, NInput, NSpace, useMessage } from "naive-ui";
+import { ref } from "vue";
+import { useRouter } from "vue-router";
+
+const authKey = ref("");
+const loading = ref(false);
+const router = useRouter();
+const message = useMessage();
+
+const handleLogin = async () => {
+  if (!authKey.value) {
+    message.error("Please enter Auth Key");
+    return;
+  }
+  loading.value = true;
+  const success = await authService.login(authKey.value);
+  loading.value = false;
+  if (success) {
+    router.push("/");
+  } else {
+    message.error("Login failed, please check your Auth Key");
+  }
+};
+</script>
+
+<style scoped>
+.login-container {
+  display: flex;
+  justify-content: center;
+  align-items: center;
+  height: 100vh;
+  background-color: #f0f2f5;
+}
+
+.login-card {
+  width: 400px;
+}
+</style>