From c41955952e0f2bade0ea9654f78c949a3a6acb0f Mon Sep 17 00:00:00 2001
From: mtvpls <mtvpls@redaaa.ip-ddns.com>
Date: Sat, 21 Mar 2026 12:08:47 +0800
Subject: [PATCH] =?UTF-8?q?=E8=84=9A=E6=9C=AC=E5=A4=9A=E6=BA=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/app/admin/page.tsx   | 24 ++++++++++++++-------
 src/lib/source-script.ts | 45 ++++++++++++++++++++++++++++++++--------
 2 files changed, 53 insertions(+), 16 deletions(-)

diff --git a/src/app/admin/page.tsx b/src/app/admin/page.tsx
index 3346c7f..48c50be 100644
--- a/src/app/admin/page.tsx
+++ b/src/app/admin/page.tsx
@@ -6001,9 +6001,9 @@ const VideoSourceScriptLab = () => {
     enabled: true,
     history: [],
   });
-  const [testHook, setTestHook] = useState<'search' | 'detail' | 'resolvePlayUrl'>('search');
+  const [testHook, setTestHook] = useState<'getSources' | 'search' | 'recommend' | 'detail' | 'resolvePlayUrl'>('getSources');
   const [testPayload, setTestPayload] = useState(
-    JSON.stringify({ keyword: '凡人修仙传', page: 1 }, null, 2)
+    JSON.stringify({}, null, 2)
   );
   const [testOutput, setTestOutput] = useState('');
   const importInputRef = useRef<HTMLInputElement | null>(null);
@@ -6302,13 +6302,18 @@ const VideoSourceScriptLab = () => {
 
   useEffect(() => {
     setTestPayload(
-      testHook === 'search'
-        ? JSON.stringify({ keyword: '凡人修仙传', page: 1 }, null, 2)
+      testHook === 'getSources'
+        ? JSON.stringify({}, null, 2)
+        : testHook === 'search'
+        ? JSON.stringify({ keyword: '凡人修仙传', page: 1, sourceId: 'main' }, null, 2)
+        : testHook === 'recommend'
+        ? JSON.stringify({ page: 1 }, null, 2)
         : testHook === 'detail'
-          ? JSON.stringify({ id: 'demo-id' }, null, 2)
+          ? JSON.stringify({ id: 'demo-id', sourceId: 'main' }, null, 2)
           : JSON.stringify(
               {
-                sourceId: 'demo-id',
+                sourceId: 'main',
+                lineId: 'default',
                 playUrl: 'https://example.com/video.m3u8',
                 episodeIndex: 0,
               },
@@ -6488,14 +6493,19 @@ const VideoSourceScriptLab = () => {
                 </label>
                 <select
                   value={testHook}
-                  onChange={(e) => setTestHook(e.target.value as 'search' | 'detail' | 'resolvePlayUrl')}
+                  onChange={(e) => setTestHook(e.target.value as 'getSources' | 'search' | 'recommend' | 'detail' | 'resolvePlayUrl')}
                   className='px-3 py-2 border border-gray-300 dark:border-gray-600 rounded-lg bg-white dark:bg-gray-800 text-gray-900 dark:text-gray-100'
                 >
+                  <option value='getSources'>getSources</option>
                   <option value='search'>search</option>
+                  <option value='recommend'>recommend</option>
                   <option value='detail'>detail</option>
                   <option value='resolvePlayUrl'>resolvePlayUrl</option>
                 </select>
               </div>
+              <p className='text-xs text-gray-500 dark:text-gray-400'>
+                现在脚本可以自己管理多个源和线路，测试入参可传 `sourceId`、`lineId`。
+              </p>
               <textarea
                 value={testPayload}
                 onChange={(e) => setTestPayload(e.target.value)}
diff --git a/src/lib/source-script.ts b/src/lib/source-script.ts
index 19ebba2..128fd24 100644
--- a/src/lib/source-script.ts
+++ b/src/lib/source-script.ts
@@ -55,9 +55,17 @@ const DEFAULT_SCRIPT_TEMPLATE = `return {
     author: 'admin'
   },
 
-  async search(ctx, { keyword, page }) {
-    ctx.log.info('search', keyword, page);
+  async getSources(ctx) {
+    return [
+      { id: 'main', name: '主站' },
+      { id: 'backup', name: '备用站' }
+    ];
+  },
+
+  async search(ctx, { keyword, page, sourceId }) {
+    ctx.log.info('search', keyword, page, sourceId);
     return {
+      sourceId,
       list: [],
       page,
       pageCount: 1,
@@ -65,21 +73,40 @@ const DEFAULT_SCRIPT_TEMPLATE = `return {
     };
   },
 
-  async detail(ctx, { id }) {
-    ctx.log.info('detail', id);
+  async recommend(ctx, { page }) {
+    ctx.log.info('recommend', page);
+    return {
+      list: [],
+      page: page || 1,
+      pageCount: 1,
+      total: 0
+    };
+  },
+
+  async detail(ctx, { id, sourceId }) {
+    ctx.log.info('detail', id, sourceId);
     return {
       id,
+      sourceId,
       title: '',
       poster: '',
       year: '',
       desc: '',
-      episodes: [],
-      episodes_titles: []
+      playbacks: [
+        {
+          sourceId: sourceId || 'main',
+          sourceName: '主站',
+          lineId: 'default',
+          lineName: '默认线路',
+          episodes: [],
+          episodes_titles: []
+        }
+      ]
     };
   },
 
-  async resolvePlayUrl(ctx, { playUrl, sourceId, episodeIndex }) {
-    ctx.log.info('resolvePlayUrl', sourceId, episodeIndex, playUrl);
+  async resolvePlayUrl(ctx, { playUrl, sourceId, lineId, episodeIndex }) {
+    ctx.log.info('resolvePlayUrl', sourceId, lineId, episodeIndex, playUrl);
     return {
       url: playUrl,
       type: 'auto',
@@ -550,7 +577,7 @@ export async function restoreSourceScriptHistory(id: string, version: string) {
 
 export async function testSourceScript(input: {
   code: string;
-  hook: 'search' | 'detail' | 'resolvePlayUrl';
+  hook: 'getSources' | 'search' | 'recommend' | 'detail' | 'resolvePlayUrl';
   payload: Record<string, any>;
   name?: string;
   key?: string;