将next-auth替换为better-auth

src/lib/server/ai.ts

@@ -0,0 +1,62 @@
+"use server";
+
+import { format } from "util";
+
+async function callZhipuAPI(
+  messages: { role: string; content: string }[],
+  model = process.env.ZHIPU_MODEL_NAME,
+) {
+  const url = "https://open.bigmodel.cn/api/paas/v4/chat/completions";
+
+  const response = await fetch(url, {
+    method: "POST",
+    headers: {
+      Authorization: "Bearer " + process.env.ZHIPU_API_KEY,
+      "Content-Type": "application/json",
+    },
+    body: JSON.stringify({
+      model: model,
+      messages: messages,
+      temperature: 0.2,
+      thinking: {
+        type: "disabled",
+      },
+    }),
+  });
+
+  if (!response.ok) {
+    throw new Error(`API 调用失败: ${response.status} ${response.statusText}`);
+  }
+
+  return await response.json();
+}
+
+export async function getLLMAnswer(prompt: string) {
+  return (
+    await callZhipuAPI([
+      {
+        role: "user",
+        content: prompt,
+      },
+    ])
+  ).choices[0].message.content.trim() as string;
+}
+
+export async function simpleGetLLMAnswer(
+  prompt: string,
+  searchParams: URLSearchParams,
+  args: string[],
+) {
+  if (args.some((arg) => typeof searchParams.get(arg) !== "string")) {
+    return Response.json({
+      status: "error",
+      message: "Missing required parameters",
+    });
+  }
+  return Response.json({
+    status: "success",
+    message: await getLLMAnswer(
+      format(prompt, ...args.map((v) => searchParams.get(v))),
+    ),
+  });
+}

src/lib/server/services/folderService.ts

@@ -0,0 +1,72 @@
+"use server";
+
+import { FolderCreateInput, FolderUpdateInput } from "../../../../generated/prisma/models";
+import prisma from "../../db";
+
+export async function getFoldersByUserId(userId: string) {
+  const folders = await prisma.folder.findMany({
+    where: {
+      userId: userId,
+    },
+  });
+  return folders;
+}
+
+export async function renameFolderById(id: number, newName: string) {
+  await prisma.folder.update({
+    where: {
+      id: id,
+    },
+    data: {
+      name: newName,
+    },
+  });
+}
+
+export async function getFoldersWithTotalPairsByUserId(userId: string) {
+  const folders = await prisma.folder.findMany({
+    where: { userId },
+    include: {
+      _count: {
+        select: { pairs: true },
+      },
+    },
+  });
+
+  return folders.map(folder => ({
+    ...folder,
+    total: folder._count?.pairs ?? 0,
+  }));
+}
+
+export async function createFolder(folder: FolderCreateInput) {
+  await prisma.folder.create({
+    data: folder,
+  });
+}
+
+export async function deleteFolderById(id: number) {
+  await prisma.folder.delete({
+    where: {
+      id: id,
+    },
+  });
+}
+
+export async function updateFolderById(id: number, data: FolderUpdateInput) {
+  await prisma.folder.update({
+    where: {
+      id: id,
+    },
+    data: data,
+  });
+}
+
+export async function getUserIdByFolderId(id: number) {
+  const folder = await prisma.folder.findUnique({
+    where: {
+      id: id,
+    },
+  });
+  return folder?.userId;
+}

src/lib/server/services/pairService.ts

@@ -0,0 +1,48 @@
+"use server";
+
+import { PairCreateInput, PairUpdateInput } from "../../../../generated/prisma/models";
+import prisma from "../../db";
+
+export async function createPair(data: PairCreateInput) {
+  await prisma.pair.create({
+    data: data,
+  });
+}
+
+export async function deletePairById(id: number) {
+  await prisma.pair.delete({
+    where: {
+      id: id,
+    },
+  });
+}
+
+export async function updatePairById(
+  id: number,
+  data: PairUpdateInput,
+) {
+  await prisma.pair.update({
+    where: {
+      id: id,
+    },
+    data: data,
+  });
+}
+
+export async function getPairCountByFolderId(folderId: number) {
+  const count = await prisma.pair.count({
+    where: {
+      folderId: folderId,
+    },
+  });
+  return count;
+}
+
+export async function getPairsByFolderId(folderId: number) {
+  const textPairs = await prisma.pair.findMany({
+    where: {
+      folderId: folderId,
+    },
+  });
+  return textPairs;
+}

src/lib/server/services/userService.ts

@@ -0,0 +1,28 @@
+import prisma from "@/lib/db";
+import { UserCreateInput } from "../../../../generated/prisma/models";
+
+export async function createUserIfNotExists(email: string, name?: string | null) {
+    const user = await prisma.user.upsert({
+        where: {
+            email: email,
+        },
+        update: {},
+        create: {
+            email: email,
+            name: name || "New User",
+        } as UserCreateInput,
+    });
+    return user;
+}
+
+export async function getUserIdByEmail(email: string) {
+    const user = await prisma.user.findUnique({
+        where: {
+            email: email,
+        },
+        select: {
+            id: true,
+        },
+    });
+    return user ? user.id : null;
+}

src/lib/server/translatorActions.ts

@@ -0,0 +1,29 @@
+"use server";
+
+import { getLLMAnswer } from "./ai";
+
+export const genIPA = async (text: string) => {
+  return (
+    "[" +
+    (
+      await getLLMAnswer(
+        `${text}\n请生成以上文本的严式国际音标，然后直接发给我，不要附带任何说明，不要擅自增减符号。`,
+      )
+    )
+      .replaceAll("[", "")
+      .replaceAll("]", "") +
+    "]"
+  );
+};
+
+export const genLocale = async (text: string) => {
+  return await getLLMAnswer(
+    `${text}\n推断以上文本的地区（locale），然后直接发给我，形如如zh-CN，不要附带任何说明，不要擅自增减符号。`,
+  );
+};
+
+export const genTranslation = async (text: string, targetLanguage: string) => {
+  return await getLLMAnswer(
+    `${text}\n请将以上文本翻译到${targetLanguage}，然后直接发给我，不要附带任何说明，不要擅自增减符号。`,
+  );
+};