视频源脚本

2026-03-21 11:47:22 +08:00
parent c10e28f5ae
commit b66d607675
3 changed files with 1351 additions and 1 deletions
--- a/src/app/api/admin/source-script/route.ts
+++ b/src/app/api/admin/source-script/route.ts
@@ -0,0 +1,130 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
+
+import { NextRequest, NextResponse } from 'next/server';
+
+import { getAuthInfoFromCookie } from '@/lib/auth';
+import { db } from '@/lib/db';
+import {
+  deleteSourceScript,
+  getDefaultSourceScriptTemplate,
+  importSourceScripts,
+  listSourceScripts,
+  restoreSourceScriptHistory,
+  saveSourceScript,
+  testSourceScript,
+  toggleSourceScriptEnabled,
+} from '@/lib/source-script';
+
+export const runtime = 'nodejs';
+
+async function assertAdmin(request: NextRequest) {
+  const storageType = process.env.NEXT_PUBLIC_STORAGE_TYPE || 'localstorage';
+  if (storageType === 'localstorage') {
+    throw new Error('不支持本地存储进行管理员配置');
+  }
+
+  const authInfo = getAuthInfoFromCookie(request);
+  if (!authInfo?.username) {
+    return null;
+  }
+
+  if (authInfo.username === process.env.USERNAME) {
+    return authInfo.username;
+  }
+
+  const userInfoV2 = await db.getUserInfoV2(authInfo.username);
+  if (!userInfoV2 || (userInfoV2.role !== 'admin' && userInfoV2.role !== 'owner') || userInfoV2.banned) {
+    return null;
+  }
+
+  return authInfo.username;
+}
+
+export async function GET(request: NextRequest) {
+  try {
+    const username = await assertAdmin(request);
+    if (!username) {
+      return NextResponse.json({ error: 'Unauthorized' }, { status: 401 });
+    }
+
+    const items = await listSourceScripts();
+    return NextResponse.json({
+      items,
+      template: getDefaultSourceScriptTemplate(),
+    });
+  } catch (error) {
+    return NextResponse.json(
+      { error: (error as Error).message || '获取脚本列表失败' },
+      { status: 500 }
+    );
+  }
+}
+
+export async function POST(request: NextRequest) {
+  try {
+    const username = await assertAdmin(request);
+    if (!username) {
+      return NextResponse.json({ error: 'Unauthorized' }, { status: 401 });
+    }
+
+    const body = (await request.json()) as Record<string, any>;
+    const action = body.action as string;
+
+    switch (action) {
+      case 'save': {
+        const saved = await saveSourceScript({
+          id: body.id,
+          key: body.key,
+          name: body.name,
+          description: body.description,
+          code: body.code,
+          enabled: body.enabled,
+        });
+        return NextResponse.json({ ok: true, item: saved });
+      }
+      case 'delete': {
+        await deleteSourceScript(body.id);
+        return NextResponse.json({ ok: true });
+      }
+      case 'toggle_enabled': {
+        const item = await toggleSourceScriptEnabled(body.id);
+        return NextResponse.json({ ok: true, item });
+      }
+      case 'restore': {
+        const item = await restoreSourceScriptHistory(body.id, body.version);
+        return NextResponse.json({ ok: true, item });
+      }
+      case 'test': {
+        const result = await testSourceScript({
+          code: body.code,
+          hook: body.hook,
+          payload: body.payload || {},
+          name: body.name,
+          key: body.key,
+          configValues: body.configValues,
+        });
+
+        if (!result.ok) {
+          return NextResponse.json(result, { status: 400 });
+        }
+
+        return NextResponse.json(result);
+      }
+      case 'import': {
+        const imported = await importSourceScripts(
+          Array.isArray(body.items) ? body.items : []
+        );
+        return NextResponse.json({ ok: true, items: imported });
+      }
+      default:
+        return NextResponse.json({ error: '未知操作' }, { status: 400 });
+    }
+  } catch (error) {
+    return NextResponse.json(
+      {
+        error: (error as Error).message || '脚本操作失败',
+      },
+      { status: 500 }
+    );
+  }
+}