=test_update_homework_tests_and_work_log

src/modules/settings/actions.ts

@@ -0,0 +1,204 @@
+"use server"
+
+import { z } from "zod"
+import { revalidatePath } from "next/cache"
+import { createId } from "@paralleldrive/cuid2"
+import { count, desc, eq } from "drizzle-orm"
+
+import { auth } from "@/auth"
+import { db } from "@/shared/db"
+import { aiProviders } from "@/shared/db/schema"
+import type { ActionState } from "@/shared/types/action-state"
+import { encryptAiApiKey, getAiErrorMessage, testAiProviderById, testAiProviderConfig } from "@/shared/lib/ai"
+
+const ProviderSchema = z.enum(["zhipu", "openai", "gemini", "custom"])
+
+const AiProviderFormSchema = z.object({
+  id: z.string().optional(),
+  provider: ProviderSchema,
+  baseUrl: z.string().url().optional().or(z.literal("")),
+  model: z.string().min(1),
+  apiKey: z.string().min(1).optional(),
+  isDefault: z.boolean().optional(),
+})
+
+const AiProviderTestSchema = AiProviderFormSchema.extend({
+  apiKey: z.string().optional(),
+}).superRefine((data, ctx) => {
+  if (!data.apiKey?.trim() && !data.id?.trim()) {
+    ctx.addIssue({
+      code: z.ZodIssueCode.custom,
+      path: ["apiKey"],
+      message: "API key is required",
+    })
+  }
+})
+
+export type AiProviderSummary = {
+  id: string
+  provider: z.infer<typeof ProviderSchema>
+  baseUrl: string | null
+  model: string
+  apiKeyLast4: string | null
+  isDefault: boolean
+  updatedAt: Date
+}
+
+const ensureUser = async () => {
+  const session = await auth()
+  const userId = String(session?.user?.id ?? "").trim()
+  if (!userId) throw new Error("Unauthorized")
+  return { id: userId }
+}
+
+const normalizeBaseUrl = (value: string | undefined) => {
+  const raw = String(value ?? "").trim()
+  if (!raw.length) return null
+  const trimmed = raw.replace(/\/+$/, "")
+  return trimmed
+    .replace(/\/v1\/chat\/completions$/i, "")
+    .replace(/\/chat\/completions$/i, "")
+}
+
+export async function getAiProviderSummaries(): Promise<AiProviderSummary[]> {
+  await ensureUser()
+  const rows = await db
+    .select({
+      id: aiProviders.id,
+      provider: aiProviders.provider,
+      baseUrl: aiProviders.baseUrl,
+      model: aiProviders.model,
+      apiKeyLast4: aiProviders.apiKeyLast4,
+      isDefault: aiProviders.isDefault,
+      updatedAt: aiProviders.updatedAt,
+    })
+    .from(aiProviders)
+    .orderBy(desc(aiProviders.updatedAt))
+  return rows
+}
+
+export async function upsertAiProviderAction(
+  data: z.infer<typeof AiProviderFormSchema>
+): Promise<ActionState<string>> {
+  try {
+    const user = await ensureUser()
+    const parsed = AiProviderFormSchema.safeParse(data)
+    if (!parsed.success) {
+      return { success: false, message: "Invalid form data" }
+    }
+
+    const payload = parsed.data
+    const baseUrl = normalizeBaseUrl(payload.baseUrl)
+    if (payload.provider !== "openai" && !baseUrl) {
+      return { success: false, message: "Base URL is required for this provider" }
+    }
+
+    const [defaultRow] = await db
+      .select({ value: count() })
+      .from(aiProviders)
+      .where(eq(aiProviders.isDefault, true))
+    const defaultCount = Number(defaultRow?.value ?? 0)
+    const hasDefault = defaultCount > 0
+
+    if (payload.id) {
+      const id = payload.id
+      const [existing] = await db
+        .select({
+          id: aiProviders.id,
+          apiKeyEncrypted: aiProviders.apiKeyEncrypted,
+          apiKeyLast4: aiProviders.apiKeyLast4,
+          isDefault: aiProviders.isDefault,
+        })
+        .from(aiProviders)
+        .where(eq(aiProviders.id, id))
+        .limit(1)
+      if (!existing) return { success: false, message: "AI provider not found" }
+
+      const nextKey = payload.apiKey?.trim()
+      const encrypted = nextKey ? encryptAiApiKey(nextKey) : existing.apiKeyEncrypted
+      const last4 = nextKey ? nextKey.slice(-4) : existing.apiKeyLast4
+
+      const nextIsDefault =
+        payload.isDefault === false && existing.isDefault && defaultCount <= 1 ? true : payload.isDefault ?? existing.isDefault
+
+      await db.transaction(async (tx) => {
+        if (payload.isDefault) {
+          await tx.update(aiProviders).set({ isDefault: false }).where(eq(aiProviders.isDefault, true))
+        }
+        await tx
+          .update(aiProviders)
+          .set({
+            provider: payload.provider,
+            baseUrl,
+            model: payload.model,
+            apiKeyEncrypted: encrypted,
+            apiKeyLast4: last4,
+            isDefault: nextIsDefault,
+            updatedBy: user.id,
+          })
+          .where(eq(aiProviders.id, id))
+      })
+
+      revalidatePath("/settings")
+      return { success: true, message: "AI provider updated", data: id }
+    }
+
+    if (!payload.apiKey) {
+      return { success: false, message: "API key is required" }
+    }
+
+    const id = createId()
+    const encrypted = encryptAiApiKey(payload.apiKey.trim())
+    const last4 = payload.apiKey.trim().slice(-4)
+    const makeDefault = payload.isDefault ?? !hasDefault
+
+    await db.transaction(async (tx) => {
+      if (makeDefault) {
+        await tx.update(aiProviders).set({ isDefault: false }).where(eq(aiProviders.isDefault, true))
+      }
+      await tx.insert(aiProviders).values({
+        id,
+        provider: payload.provider,
+        baseUrl,
+        model: payload.model,
+        apiKeyEncrypted: encrypted,
+        apiKeyLast4: last4,
+        isDefault: makeDefault,
+        createdBy: user.id,
+        updatedBy: user.id,
+      })
+    })
+
+    revalidatePath("/settings")
+    return { success: true, message: "AI provider created", data: id }
+  } catch {
+    return { success: false, message: "Failed to save AI provider" }
+  }
+}
+
+export async function testAiProviderAction(
+  data: z.infer<typeof AiProviderTestSchema>
+): Promise<ActionState<null>> {
+  try {
+    await ensureUser()
+    const parsed = AiProviderTestSchema.safeParse(data)
+    if (!parsed.success) {
+      return { success: false, message: "Invalid form data" }
+    }
+    const payload = parsed.data
+    const baseUrl = normalizeBaseUrl(payload.baseUrl)
+    if (payload.provider !== "openai" && !baseUrl) {
+      return { success: false, message: "Base URL is required for this provider" }
+    }
+    const model = payload.model.trim()
+    const apiKey = payload.apiKey?.trim()
+    if (apiKey) {
+      await testAiProviderConfig({ apiKey, baseUrl: baseUrl ?? undefined, model })
+    } else if (payload.id) {
+      await testAiProviderById(payload.id, { baseUrl: baseUrl ?? undefined, model })
+    }
+    return { success: true, message: "AI connection ok", data: null }
+  } catch (error) {
+    return { success: false, message: getAiErrorMessage(error) }
+  }
+}

src/modules/settings/components/ai-provider-settings-card.tsx

@@ -0,0 +1,405 @@
+"use client"
+
+import { useCallback, useEffect, useMemo, useRef, useState, useTransition } from "react"
+import { z } from "zod"
+import { zodResolver } from "@hookform/resolvers/zod"
+import { useForm } from "react-hook-form"
+import { toast } from "sonner"
+import { Loader2, Save, Sparkles } from "lucide-react"
+
+import { Card, CardContent, CardDescription, CardFooter, CardHeader, CardTitle } from "@/shared/components/ui/card"
+import { Button } from "@/shared/components/ui/button"
+import { Input } from "@/shared/components/ui/input"
+import { Checkbox } from "@/shared/components/ui/checkbox"
+import {
+  Form,
+  FormControl,
+  FormDescription,
+  FormField,
+  FormItem,
+  FormLabel,
+  FormMessage,
+} from "@/shared/components/ui/form"
+import {
+  Select,
+  SelectContent,
+  SelectItem,
+  SelectTrigger,
+  SelectValue,
+} from "@/shared/components/ui/select"
+import { getAiProviderSummaries, testAiProviderAction, upsertAiProviderAction, type AiProviderSummary } from "@/modules/settings/actions"
+
+const ProviderSchema = z.enum(["zhipu", "openai", "gemini", "custom"])
+
+const AiProviderFormSchema = z.object({
+  id: z.string().optional(),
+  provider: ProviderSchema,
+  baseUrl: z.string().optional(),
+  model: z.string().min(1, "Model is required"),
+  apiKey: z.string().optional(),
+  isDefault: z.boolean().optional(),
+})
+
+type AiProviderFormValues = z.infer<typeof AiProviderFormSchema>
+
+const providerLabels: Record<z.infer<typeof ProviderSchema>, string> = {
+  zhipu: "智谱",
+  openai: "OpenAI",
+  gemini: "Gemini",
+  custom: "Custom",
+}
+
+const NEW_PROVIDER_VALUE = "__new__"
+
+export function AiProviderSettingsCard({
+  onProvidersChanged,
+  initialMode = "first",
+}: {
+  onProvidersChanged?: (rows: AiProviderSummary[]) => void
+  initialMode?: "new" | "first"
+}) {
+  const [isPending, startTransition] = useTransition()
+  const [providers, setProviders] = useState<AiProviderSummary[]>([])
+  const [selectedId, setSelectedId] = useState<string>("")
+  const [testStatus, setTestStatus] = useState<"idle" | "testing" | "passed" | "failed">("idle")
+  const [lastTestedSignature, setLastTestedSignature] = useState<string>("")
+  const loadedRef = useRef(false)
+
+  const form = useForm<AiProviderFormValues>({
+    resolver: zodResolver(AiProviderFormSchema),
+    defaultValues: {
+      id: "",
+      provider: "openai",
+      baseUrl: "",
+      model: "",
+      apiKey: "",
+      isDefault: false,
+    },
+  })
+
+  const selectedProvider = useMemo(
+    () => providers.find((item) => item.id === selectedId) ?? null,
+    [providers, selectedId]
+  )
+
+  const buildSignature = useCallback((values: AiProviderFormValues) => {
+    return JSON.stringify({
+      provider: values.provider,
+      baseUrl: values.baseUrl?.trim() || "",
+      model: values.model.trim(),
+      apiKey: values.apiKey?.trim() || "",
+    })
+  }, [])
+
+  const resetToNew = useCallback(() => {
+    setSelectedId("")
+    setTestStatus("idle")
+    setLastTestedSignature("")
+    form.reset({
+      id: "",
+      provider: "openai",
+      baseUrl: "",
+      model: "",
+      apiKey: "",
+      isDefault: false,
+    })
+  }, [form])
+
+  useEffect(() => {
+    if (loadedRef.current) return
+    loadedRef.current = true
+    startTransition(async () => {
+      try {
+        const rows = await getAiProviderSummaries()
+        setProviders(rows)
+        onProvidersChanged?.(rows)
+        if (initialMode === "new") {
+          resetToNew()
+          return
+        }
+        if (rows.length > 0 && !selectedId) {
+          const next = rows[0]
+          setSelectedId(next.id)
+          form.reset({
+            id: next.id,
+            provider: next.provider,
+            baseUrl: next.baseUrl ?? "",
+            model: next.model,
+            apiKey: "",
+            isDefault: next.isDefault,
+          })
+        }
+      } catch {
+        toast.error("Failed to load AI providers")
+      }
+    })
+  }, [form, selectedId, onProvidersChanged, initialMode, resetToNew])
+
+  const handleSelectChange = (value: string) => {
+    if (value === NEW_PROVIDER_VALUE) {
+      resetToNew()
+      return
+    }
+    setSelectedId(value)
+    setTestStatus("idle")
+    setLastTestedSignature("")
+    const next = providers.find((item) => item.id === value)
+    if (!next) return
+    form.reset({
+      id: next.id,
+      provider: next.provider,
+      baseUrl: next.baseUrl ?? "",
+      model: next.model,
+      apiKey: "",
+      isDefault: next.isDefault,
+    })
+  }
+
+  useEffect(() => {
+    const subscription = form.watch(() => {
+      if (!lastTestedSignature) return
+      const currentSignature = buildSignature(form.getValues())
+      if (currentSignature !== lastTestedSignature) {
+        setTestStatus("idle")
+      }
+    })
+    return () => subscription.unsubscribe()
+  }, [form, buildSignature, lastTestedSignature])
+
+  const handleTest = () => {
+    const values = form.getValues()
+    const apiKey = values.apiKey?.trim()
+    if (!apiKey && !values.id?.trim()) {
+      toast.error("Please enter API key to test")
+      return
+    }
+    setTestStatus("testing")
+    startTransition(async () => {
+      const payload = {
+        id: values.id?.trim() || undefined,
+        provider: values.provider,
+        baseUrl: values.baseUrl?.trim() || undefined,
+        model: values.model.trim(),
+        apiKey: apiKey || undefined,
+        isDefault: values.isDefault ?? false,
+      }
+      const result = await testAiProviderAction(payload)
+      if (result.success) {
+        setTestStatus("passed")
+        setLastTestedSignature(buildSignature(values))
+        toast.success(result.message ?? "Test passed")
+      } else {
+        setTestStatus("failed")
+        toast.error(result.message ?? "Test failed")
+      }
+    })
+  }
+
+  const onSubmit = (values: AiProviderFormValues) => {
+    const signature = buildSignature(values)
+    if (testStatus !== "passed" || signature !== lastTestedSignature) {
+      toast.error("Please test the configuration before saving")
+      return
+    }
+    startTransition(async () => {
+      const payload = {
+        id: values.id?.trim() || undefined,
+        provider: values.provider,
+        baseUrl: values.baseUrl?.trim() || undefined,
+        model: values.model.trim(),
+        apiKey: values.apiKey?.trim() || undefined,
+        isDefault: values.isDefault ?? false,
+      }
+      const result = await upsertAiProviderAction(payload)
+      if (result.success) {
+        toast.success(result.message ?? "Saved")
+        setTestStatus("idle")
+        setLastTestedSignature("")
+        const rows = await getAiProviderSummaries()
+        setProviders(rows)
+        onProvidersChanged?.(rows)
+        const nextId = result.data ?? payload.id ?? ""
+        setSelectedId(nextId)
+        const next = rows.find((item) => item.id === nextId)
+        if (next) {
+          form.reset({
+            id: next.id,
+            provider: next.provider,
+            baseUrl: next.baseUrl ?? "",
+            model: next.model,
+            apiKey: "",
+            isDefault: next.isDefault,
+          })
+        }
+      } else {
+        toast.error(result.message ?? "Failed to save")
+      }
+    })
+  }
+
+  return (
+    <Card>
+      <CardHeader>
+        <CardTitle className="flex items-center gap-2">
+          <Sparkles className="h-4 w-4 text-purple-500" />
+          AI Providers
+        </CardTitle>
+        <CardDescription>Manage AI vendors and default model configuration.</CardDescription>
+      </CardHeader>
+      <CardContent className="space-y-6">
+        <div className="grid gap-4 sm:grid-cols-2">
+          <div className="space-y-2">
+            <FormLabel>Existing Providers</FormLabel>
+            <Select value={selectedId || NEW_PROVIDER_VALUE} onValueChange={handleSelectChange}>
+              <SelectTrigger>
+                <SelectValue placeholder="Create new or select existing" />
+              </SelectTrigger>
+              <SelectContent>
+                <SelectItem value={NEW_PROVIDER_VALUE}>Create new</SelectItem>
+                {providers.map((item) => (
+                  <SelectItem key={item.id} value={item.id}>
+                    {providerLabels[item.provider]} · {item.model}
+                  </SelectItem>
+                ))}
+              </SelectContent>
+            </Select>
+          </div>
+          <div className="space-y-2">
+            <FormLabel>Key Status</FormLabel>
+            <div className="rounded-md border px-3 py-2 text-sm text-muted-foreground">
+              {selectedProvider?.apiKeyLast4
+                ? `Stored • ****${selectedProvider.apiKeyLast4}`
+                : "No key stored"}
+            </div>
+          </div>
+        </div>
+
+        <Form {...form}>
+          <div className="grid gap-6">
+            <div className="grid gap-4 sm:grid-cols-2">
+              <FormField
+                control={form.control}
+                name="id"
+                render={({ field }) => (
+                  <FormItem>
+                    <FormLabel>ID</FormLabel>
+                    <FormControl>
+                      <Input {...field} value={field.value ?? ""} disabled />
+                    </FormControl>
+                    <FormDescription>Auto-generated for each provider.</FormDescription>
+                  </FormItem>
+                )}
+              />
+              <FormField
+                control={form.control}
+                name="provider"
+                render={({ field }) => (
+                  <FormItem>
+                    <FormLabel>品牌方</FormLabel>
+                    <Select value={field.value} onValueChange={field.onChange}>
+                      <FormControl>
+                        <SelectTrigger>
+                          <SelectValue placeholder="Select provider" />
+                        </SelectTrigger>
+                      </FormControl>
+                      <SelectContent>
+                        <SelectItem value="zhipu">智谱</SelectItem>
+                        <SelectItem value="openai">OpenAI</SelectItem>
+                        <SelectItem value="gemini">Gemini</SelectItem>
+                        <SelectItem value="custom">Custom</SelectItem>
+                      </SelectContent>
+                    </Select>
+                    <FormMessage />
+                  </FormItem>
+                )}
+              />
+              <FormField
+                control={form.control}
+                name="baseUrl"
+                render={({ field }) => (
+                  <FormItem>
+                    <FormLabel>API URL</FormLabel>
+                    <FormControl>
+                      <Input {...field} placeholder="https://open.bigmodel.cn/api/paas/v4" />
+                    </FormControl>
+                    <FormDescription>填写基础地址，不要包含 /chat/completions。</FormDescription>
+                    <FormMessage />
+                  </FormItem>
+                )}
+              />
+              <FormField
+                control={form.control}
+                name="model"
+                render={({ field }) => (
+                  <FormItem>
+                    <FormLabel>Model</FormLabel>
+                    <FormControl>
+                      <Input {...field} placeholder="gpt-4o-mini" />
+                    </FormControl>
+                    <FormMessage />
+                  </FormItem>
+                )}
+              />
+              <FormField
+                control={form.control}
+                name="apiKey"
+                render={({ field }) => (
+                  <FormItem className="sm:col-span-2">
+                    <FormLabel>API Key</FormLabel>
+                    <FormControl>
+                      <Input type="password" {...field} placeholder="Paste new key to replace" />
+                    </FormControl>
+                    <FormDescription>不会回显历史 Key，留空表示不更新。</FormDescription>
+                    <FormMessage />
+                  </FormItem>
+                )}
+              />
+            </div>
+
+            <FormField
+              control={form.control}
+              name="isDefault"
+              render={({ field }) => (
+                <FormItem className="flex items-center gap-2">
+                  <FormControl>
+                    <Checkbox checked={!!field.value} onCheckedChange={(value) => field.onChange(value === true)} />
+                  </FormControl>
+                  <FormLabel>设为默认</FormLabel>
+                </FormItem>
+              )}
+            />
+
+            <CardFooter className="flex justify-between border-t px-0 pt-4">
+              <Button type="button" variant="outline" onClick={handleTest} disabled={isPending || testStatus === "testing"}>
+                {testStatus === "testing" ? (
+                  <>
+                    <Loader2 className="mr-2 h-4 w-4 animate-spin" />
+                    Testing...
+                  </>
+                ) : (
+                  <>
+                    <Sparkles className="mr-2 h-4 w-4" />
+                    Test
+                  </>
+                )}
+              </Button>
+              <Button type="button" onClick={form.handleSubmit(onSubmit)} disabled={isPending}>
+                {isPending ? (
+                  <>
+                    <Loader2 className="mr-2 h-4 w-4 animate-spin" />
+                    Saving...
+                  </>
+                ) : (
+                  <>
+                    <Save className="mr-2 h-4 w-4" />
+                    Save Changes
+                  </>
+                )}
+              </Button>
+            </CardFooter>
+          </div>
+        </Form>
+      </CardContent>
+    </Card>
+  )
+}