first commit

Co-Authored-By: Codex <noreply@openai.com>
2026-05-19 19:50:36 +08:00
commit 472bf8b732
133 changed files with 16869 additions and 0 deletions
@@ -0,0 +1,109 @@
+"use client";
+
+import { useEffect, useState } from "react";
+import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
+import { App } from "antd";
+
+import { deleteAdminAsset, fetchAdminAssets, saveAdminAsset, type AdminAsset } from "@/services/api/admin";
+import { useUserStore } from "@/stores/use-user-store";
+
+const defaultPageSize = 10;
+
+export function useAdminAssets() {
+  const { message } = App.useApp();
+  const queryClient = useQueryClient();
+  const token = useUserStore((state) => state.token);
+  const clearSession = useUserStore((state) => state.clearSession);
+  const [keyword, setKeyword] = useState("");
+  const [type, setType] = useState("");
+  const [tag, setTag] = useState<string[]>([]);
+  const [page, setPage] = useState(1);
+  const [pageSize, setPageSize] = useState(defaultPageSize);
+
+  const query = useQuery({
+    queryKey: ["admin", "assets", token, keyword, type, tag, page, pageSize],
+    queryFn: () => fetchAdminAssets(token, { keyword, type, tag, page, pageSize }),
+    enabled: Boolean(token),
+    retry: false,
+  });
+
+  const saveMutation = useMutation({
+    mutationFn: (asset: Partial<AdminAsset>) => saveAdminAsset(token, asset),
+    onSuccess: async (_, asset) => {
+      await queryClient.invalidateQueries({ queryKey: ["admin", "assets"] });
+      message.success(asset.id ? "素材已保存" : "素材已新增");
+    },
+    onError: (error) => {
+      message.error(error instanceof Error ? error.message : "保存失败");
+    },
+  });
+
+  const deleteMutation = useMutation({
+    mutationFn: (id: string) => deleteAdminAsset(token, id),
+    onSuccess: async () => {
+      await queryClient.invalidateQueries({ queryKey: ["admin", "assets"] });
+      message.success("素材已删除");
+    },
+    onError: (error) => {
+      message.error(error instanceof Error ? error.message : "删除失败");
+    },
+  });
+
+  useEffect(() => {
+    if (query.isError) {
+      const errorMessage = query.error instanceof Error ? query.error.message : "读取素材失败";
+      message.error(errorMessage);
+      if (errorMessage.includes("未登录") || errorMessage.includes("权限不足") || errorMessage.includes("登录状态无效")) {
+        clearSession();
+      }
+    }
+  }, [clearSession, message, query.error, query.isError]);
+
+  const updateFilters = (next: Partial<{ keyword: string; type: string; tag: string[]; page: number; pageSize: number }>) => {
+    const queryState = { keyword, type, tag, page, pageSize, ...next };
+    if (next.keyword !== undefined || next.type !== undefined || next.tag !== undefined || next.pageSize !== undefined) {
+      queryState.page = 1;
+    }
+    setKeyword(queryState.keyword);
+    setType(queryState.type);
+    setTag(queryState.tag);
+    setPage(queryState.page);
+    setPageSize(queryState.pageSize);
+  };
+
+  const refreshAssets = async () => {
+    await query.refetch();
+  };
+
+  const saveAsset = async (asset: Partial<AdminAsset>) => {
+    await saveMutation.mutateAsync(asset);
+  };
+
+  const deleteAsset = async (id: string) => {
+    await deleteMutation.mutateAsync(id);
+  };
+
+  const data = query.data;
+  const isLoading = query.isFetching || saveMutation.isPending || deleteMutation.isPending;
+
+  return {
+    assets: data?.items || [],
+    tags: data?.tags || [],
+    keyword,
+    kind: type,
+    tag,
+    page,
+    pageSize,
+    total: data?.total || 0,
+    isLoading,
+    searchAssets: (value = keyword) => updateFilters({ keyword: value }),
+    changeKind: (value: string) => updateFilters({ type: value, tag: [] }),
+    changeTag: (value: string[]) => updateFilters({ tag: value }),
+    changePage: (value: number) => updateFilters({ page: value }),
+    changePageSize: (value: number) => updateFilters({ pageSize: value }),
+    resetFilters: () => updateFilters({ keyword: "", type: "", tag: [], page: 1, pageSize: defaultPageSize }),
+    refreshAssets,
+    saveAsset,
+    deleteAsset,
+  };
+}
@@ -0,0 +1,137 @@
+"use client";
+
+import { useEffect, useState } from "react";
+import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
+import { App } from "antd";
+
+import { deleteAdminPrompt, fetchAdminPrompts, fetchAdminPromptCategories, saveAdminPrompt, syncAdminPromptCategory, type AdminPromptCategory } from "@/services/api/admin";
+import type { Prompt } from "@/services/api/prompts";
+import { useUserStore } from "@/stores/use-user-store";
+
+const defaultPageSize = 10;
+
+export function useAdminPrompts() {
+  const { message } = App.useApp();
+  const queryClient = useQueryClient();
+  const token = useUserStore((state) => state.token);
+  const clearSession = useUserStore((state) => state.clearSession);
+  const [keyword, setKeyword] = useState("");
+  const [category, setCategory] = useState("");
+  const [tag, setTag] = useState<string[]>([]);
+  const [page, setPage] = useState(1);
+  const [pageSize, setPageSize] = useState(defaultPageSize);
+
+  const categoriesQuery = useQuery({
+    queryKey: ["admin", "prompt-categories", token],
+    queryFn: () => fetchAdminPromptCategories(token),
+    enabled: Boolean(token),
+    retry: false,
+  });
+
+  const promptsQuery = useQuery({
+    queryKey: ["admin", "prompts", token, keyword, category, tag, page, pageSize],
+    queryFn: () => fetchAdminPrompts(token, { keyword, category, tag, page, pageSize }),
+    enabled: Boolean(token),
+    retry: false,
+  });
+
+  const syncMutation = useMutation({
+    mutationFn: (category: string) => syncAdminPromptCategory(token, category),
+    onSuccess: async (categories) => {
+      queryClient.setQueryData<AdminPromptCategory[]>(["admin", "prompt-categories", token], categories);
+      await queryClient.invalidateQueries({ queryKey: ["admin", "prompts"] });
+      message.success("远程提示词源已同步");
+    },
+    onError: (error) => {
+      message.error(error instanceof Error ? error.message : "同步失败");
+    },
+  });
+
+  const saveMutation = useMutation({
+    mutationFn: (prompt: Partial<Prompt>) => saveAdminPrompt(token, prompt),
+    onSuccess: async (_, prompt) => {
+      await queryClient.invalidateQueries({ queryKey: ["admin", "prompt-categories"] });
+      await queryClient.invalidateQueries({ queryKey: ["admin", "prompts"] });
+      message.success(prompt.id ? "提示词已保存" : "提示词已新增");
+    },
+    onError: (error) => {
+      message.error(error instanceof Error ? error.message : "保存失败");
+    },
+  });
+
+  const deleteMutation = useMutation({
+    mutationFn: (id: string) => deleteAdminPrompt(token, id),
+    onSuccess: async () => {
+      await queryClient.invalidateQueries({ queryKey: ["admin", "prompt-categories"] });
+      await queryClient.invalidateQueries({ queryKey: ["admin", "prompts"] });
+      message.success("提示词已删除");
+    },
+    onError: (error) => {
+      message.error(error instanceof Error ? error.message : "删除失败");
+    },
+  });
+
+  useEffect(() => {
+    if (categoriesQuery.isError) {
+      const errorMessage = categoriesQuery.error instanceof Error ? categoriesQuery.error.message : "读取提示词分类失败";
+      message.error(errorMessage);
+      if (errorMessage.includes("未登录") || errorMessage.includes("权限不足") || errorMessage.includes("登录状态无效")) {
+        clearSession();
+      }
+    }
+  }, [categoriesQuery.error, categoriesQuery.isError, clearSession, message]);
+
+  useEffect(() => {
+    if (promptsQuery.isError) {
+      const errorMessage = promptsQuery.error instanceof Error ? promptsQuery.error.message : "读取提示词失败";
+      message.error(errorMessage);
+      if (errorMessage.includes("未登录") || errorMessage.includes("权限不足") || errorMessage.includes("登录状态无效")) {
+        clearSession();
+      }
+    }
+  }, [clearSession, message, promptsQuery.error, promptsQuery.isError]);
+
+  const updateFilters = (next: Partial<{ keyword: string; category: string; tag: string[]; page: number; pageSize: number }>) => {
+    const queryState = { keyword, category, tag, page, pageSize, ...next };
+    if (next.keyword !== undefined || next.category !== undefined || next.tag !== undefined || next.pageSize !== undefined) {
+      queryState.page = 1;
+    }
+    setKeyword(queryState.keyword);
+    setCategory(queryState.category);
+    setTag(queryState.tag);
+    setPage(queryState.page);
+    setPageSize(queryState.pageSize);
+  };
+
+  const refreshPrompts = async () => {
+    await categoriesQuery.refetch();
+    await promptsQuery.refetch();
+  };
+
+  const data = promptsQuery.data;
+  const isLoading = categoriesQuery.isFetching || promptsQuery.isFetching || saveMutation.isPending || deleteMutation.isPending;
+
+  return {
+    categories: categoriesQuery.data || [],
+    prompts: data?.items || [],
+    tags: data?.tags || [],
+    keyword,
+    category,
+    tag,
+    page,
+    pageSize,
+    total: data?.total || 0,
+    isLoading,
+    isSyncing: syncMutation.isPending,
+    syncCategory: (category: string) => syncMutation.mutateAsync(category),
+    searchPrompts: (value = keyword) => updateFilters({ keyword: value }),
+    changeCategory: (value: string) => updateFilters({ category: value, tag: [] }),
+    changeTag: (value: string[]) => updateFilters({ tag: value }),
+    changePage: (value: number) => updateFilters({ page: value }),
+    changePageSize: (value: number) => updateFilters({ pageSize: value }),
+    resetFilters: () => updateFilters({ keyword: "", category: "", tag: [], page: 1, pageSize: defaultPageSize }),
+    refreshPrompts,
+    savePrompt: (prompt: Partial<Prompt>) => saveMutation.mutateAsync(prompt),
+    deletePrompt: (id: string) => deleteMutation.mutateAsync(id),
+  };
+}