增加只搜私人影库

src/app/api/search/route.ts

@@ -26,6 +26,7 @@ export async function GET(request: NextRequest) {
   const { searchParams } = new URL(request.url);
   const query = searchParams.get('q');
   const includeSpecialSources = searchParams.get('special') === '1';
+  const privateOnly = searchParams.get('privateOnly') === '1';
 
   if (!query) {
     const cacheTime = await getCacheTime();
@@ -43,7 +44,9 @@ export async function GET(request: NextRequest) {
   }
 
   const config = await getConfig();
-  const apiSites = await getAvailableApiSites(authInfo.username, includeSpecialSources);
+  const apiSites = privateOnly
+    ? []
+    : await getAvailableApiSites(authInfo.username, includeSpecialSources);
   const [canAccessOpenList, canAccessEmby] = await Promise.all([
     hasFeaturePermission(authInfo.username, 'private_library'),
     hasFeaturePermission(authInfo.username, 'emby'),
@@ -191,7 +194,7 @@ export async function GET(request: NextRequest) {
     })
   );
 
-  const scriptSummaries = await listEnabledSourceScripts();
+  const scriptSummaries = privateOnly ? [] : await listEnabledSourceScripts();
   const scriptPromises = scriptSummaries.map((script) =>
     Promise.race([
       (async () => {

src/app/api/search/ws/route.ts

@@ -26,6 +26,7 @@ export async function GET(request: NextRequest) {
   const { searchParams } = new URL(request.url);
   const query = searchParams.get('q');
   const includeSpecialSources = searchParams.get('special') === '1';
+  const privateOnly = searchParams.get('privateOnly') === '1';
 
   if (!query) {
     return new Response(
@@ -40,7 +41,9 @@ export async function GET(request: NextRequest) {
   }
 
   const config = await getConfig();
-  const apiSites = await getAvailableApiSites(authInfo.username, includeSpecialSources);
+  const apiSites = privateOnly
+    ? []
+    : await getAvailableApiSites(authInfo.username, includeSpecialSources);
   const [canAccessOpenList, canAccessEmby] = await Promise.all([
     hasFeaturePermission(authInfo.username, 'private_library'),
     hasFeaturePermission(authInfo.username, 'emby'),
@@ -75,7 +78,7 @@ export async function GET(request: NextRequest) {
     config.EmbyConfig.Sources.length > 0 &&
     config.EmbyConfig.Sources.some(s => s.enabled && s.ServerURL)
   );
-  const enabledScripts = await listEnabledSourceScripts();
+  const enabledScripts = privateOnly ? [] : await listEnabledSourceScripts();
 
   // 共享状态
   let streamClosed = false;
@@ -114,11 +117,13 @@ export async function GET(request: NextRequest) {
         }
       }
 
+      const totalSourceCount = sortedApiSites.length + (hasOpenList ? 1 : 0) + embySourcesCount + enabledScripts.length;
+
       // 发送开始事件
       const startEvent = `data: ${JSON.stringify({
         type: 'start',
         query,
-        totalSources: sortedApiSites.length + (hasOpenList ? 1 : 0) + embySourcesCount + enabledScripts.length,
+        totalSources: totalSourceCount,
         timestamp: Date.now()
       })}\n\n`;
 
@@ -130,6 +135,30 @@ export async function GET(request: NextRequest) {
       let completedSources = 0;
       const allResults: any[] = [];
 
+      const maybeComplete = () => {
+        if (completedSources !== totalSourceCount || streamClosed) return;
+        const completeEvent = `data: ${JSON.stringify({
+          type: 'complete',
+          totalResults: allResults.length,
+          completedSources,
+          timestamp: Date.now()
+        })}\n\n`;
+
+        if (safeEnqueue(encoder.encode(completeEvent))) {
+          streamClosed = true;
+          try {
+            controller.close();
+          } catch (error) {
+            console.warn('Failed to close controller:', error);
+          }
+        }
+      };
+
+      if (totalSourceCount === 0) {
+        maybeComplete();
+        return;
+      }
+
       // 搜索 Emby（如果配置了）- 异步带超时，支持多源
       if (hasEmby) {
         (async () => {
@@ -192,6 +221,7 @@ export async function GET(request: NextRequest) {
                     streamClosed = true;
                   }
                 }
+                maybeComplete();
 
                 return results;
               } catch (error) {
@@ -211,6 +241,7 @@ export async function GET(request: NextRequest) {
                   })}\n\n`;
                   safeEnqueue(encoder.encode(sourceEvent));
                 }
+                maybeComplete();
                 return [];
               }
             });
@@ -232,6 +263,7 @@ export async function GET(request: NextRequest) {
                 })}\n\n`;
                 safeEnqueue(encoder.encode(sourceEvent));
               }
+              maybeComplete();
             }
           }
         })();
@@ -310,6 +342,7 @@ export async function GET(request: NextRequest) {
                 allResults.push(...safeResults);
               }
             }
+            maybeComplete();
           })
           .catch((error) => {
             console.error('[Search WS] 搜索 OpenList 超时:', error);
@@ -324,6 +357,7 @@ export async function GET(request: NextRequest) {
               })}\n\n`;
               safeEnqueue(encoder.encode(sourceEvent));
             }
+            maybeComplete();
           });
       }
 
@@ -402,7 +436,7 @@ export async function GET(request: NextRequest) {
         }
 
         // 检查是否所有源都已完成
-        if (completedSources === sortedApiSites.length + (hasOpenList ? 1 : 0) + embySourcesCount + enabledScripts.length) {
+        if (completedSources === totalSourceCount) {
           if (!streamClosed) {
             // 发送最终完成事件
             const completeEvent = `data: ${JSON.stringify({
@@ -414,6 +448,7 @@ export async function GET(request: NextRequest) {
 
             if (safeEnqueue(encoder.encode(completeEvent))) {
               // 只有在成功发送完成事件后才关闭流
+              streamClosed = true;
               try {
                 controller.close();
               } catch (error) {
@@ -514,7 +549,7 @@ export async function GET(request: NextRequest) {
           }
         }
 
-        if (completedSources === sortedApiSites.length + (hasOpenList ? 1 : 0) + embySourcesCount + enabledScripts.length) {
+        if (completedSources === totalSourceCount) {
           if (!streamClosed) {
             const completeEvent = `data: ${JSON.stringify({
               type: 'complete',
@@ -524,6 +559,7 @@ export async function GET(request: NextRequest) {
             })}\n\n`;
 
             if (safeEnqueue(encoder.encode(completeEvent))) {
+              streamClosed = true;
               try {
                 controller.close();
               } catch (error) {

src/app/search/page.tsx

@@ -127,10 +127,20 @@ function SearchPageClient() {
   const [isFromCache, setIsFromCache] = useState(false);
   // 精确搜索开关
   const [exactSearch, setExactSearch] = useState(true);
+  const [advancedOpen, setAdvancedOpen] = useState(false);
+  const [privateLibraryOnly, setPrivateLibraryOnly] = useState(false);
+  const [privateLibraryOnlyReady, setPrivateLibraryOnlyReady] = useState(false);
+  const privateLibraryOnlyLoadedRef = useRef(false);
+  const advancedButtonRefs = useRef<HTMLButtonElement[]>([]);
+  const advancedDropdownRefs = useRef<HTMLDivElement[]>([]);
 
   // 生成缓存键
   const getCacheKey = (query: string) => {
-    const suffix = isSpecialSourcesEnabledOnDevice() ? '_special' : '';
+    const suffixParts = [
+      isSpecialSourcesEnabledOnDevice() ? 'special' : '',
+      privateLibraryOnly ? 'private' : '',
+    ].filter(Boolean);
+    const suffix = suffixParts.length > 0 ? `_${suffixParts.join('_')}` : '';
     return `search_cache_${query.trim()}${suffix}`;
   };
 
@@ -812,6 +822,35 @@ function SearchPageClient() {
     }
   }, [resultDisplayMode]);
 
+  useEffect(() => {
+    if (typeof window !== 'undefined' && privateLibraryOnlyLoadedRef.current) {
+      localStorage.setItem('searchPrivateLibraryOnly', String(privateLibraryOnly));
+    }
+  }, [privateLibraryOnly]);
+
+  useEffect(() => {
+    if (!advancedOpen) return;
+
+    const handleClickOutside = (event: MouseEvent) => {
+      const target = event.target as Node;
+      const clickedButton = advancedButtonRefs.current.some((ref) =>
+        ref?.contains(target)
+      );
+      const clickedDropdown = advancedDropdownRefs.current.some((ref) =>
+        ref?.contains(target)
+      );
+
+      if (!clickedButton && !clickedDropdown) {
+        setAdvancedOpen(false);
+      }
+    };
+
+    document.addEventListener('mousedown', handleClickOutside);
+    return () => {
+      document.removeEventListener('mousedown', handleClickOutside);
+    };
+  }, [advancedOpen]);
+
   const getSearchResultUrl = (params: {
     title: string;
     year?: string;
@@ -1064,6 +1103,15 @@ function SearchPageClient() {
       if (savedExactSearch !== null) {
         setExactSearch(savedExactSearch === 'true');
       }
+
+      const savedPrivateLibraryOnly = localStorage.getItem(
+        'searchPrivateLibraryOnly'
+      );
+      if (savedPrivateLibraryOnly !== null) {
+        setPrivateLibraryOnly(savedPrivateLibraryOnly === 'true');
+      }
+      privateLibraryOnlyLoadedRef.current = true;
+      setPrivateLibraryOnlyReady(true);
     }
 
     // 监听搜索历史更新事件
@@ -1145,8 +1193,8 @@ function SearchPageClient() {
   ]);
 
   useEffect(() => {
-    // 等待转换器初始化完成
+    // 等待转换器和私人影库搜索设置初始化完成
-    if (!converterReady) {
+    if (!converterReady || !privateLibraryOnlyReady) {
       return;
     }
 
@@ -1292,9 +1340,10 @@ function SearchPageClient() {
 
       if (currentFluidSearch) {
         // 流式搜索：打开新的流式连接
-        const es = new EventSource(
+        const searchUrl = `/api/search/ws?q=${encodeURIComponent(trimmed)}${
-          appendSpecialSourceParam(`/api/search/ws?q=${encodeURIComponent(trimmed)}`)
+          privateLibraryOnly ? '&privateOnly=1' : ''
-        );
+        }`;
+        const es = new EventSource(appendSpecialSourceParam(searchUrl));
         eventSourceRef.current = es;
 
         es.onmessage = (event) => {
@@ -1399,9 +1448,10 @@ function SearchPageClient() {
         };
       } else {
         // 传统搜索：使用普通接口
-        fetch(
+        const searchUrl = `/api/search?q=${encodeURIComponent(trimmed)}${
-          appendSpecialSourceParam(`/api/search?q=${encodeURIComponent(trimmed)}`)
+          privateLibraryOnly ? '&privateOnly=1' : ''
-        )
+        }`;
+        fetch(appendSpecialSourceParam(searchUrl))
           .then((response) => response.json())
           .then((data) => {
             if (currentQueryRef.current !== trimmed) return;
@@ -1434,7 +1484,13 @@ function SearchPageClient() {
       setShowResults(false);
       setShowSuggestions(false);
     }
-  }, [searchParams, forceRefresh, converterReady]);
+  }, [
+    searchParams,
+    forceRefresh,
+    converterReady,
+    privateLibraryOnlyReady,
+    privateLibraryOnly,
+  ]);
 
   useEffect(() => {
     if (!featureFlagsReady) return;
@@ -1832,6 +1888,77 @@ function SearchPageClient() {
               <AcgSearch keyword={searchQuery} controlsOnly />
             </div>
           )}
+
+          {activeTab === 'video' && !showResults && (
+            <div className='mx-auto mt-4 flex max-w-2xl justify-end'>
+              <div className='relative'>
+                <button
+                  ref={(el) => {
+                    if (el) advancedButtonRefs.current[0] = el;
+                  }}
+                  type='button'
+                  onClick={() => setAdvancedOpen((prev) => !prev)}
+                  className={`flex items-center gap-1.5 rounded-lg px-3 py-1.5 text-sm transition-colors ${
+                    advancedOpen
+                      ? 'bg-green-50 text-green-600 dark:bg-gray-700/70 dark:text-green-400'
+                      : 'text-gray-600 hover:bg-green-50 hover:text-green-600 dark:text-gray-400 dark:hover:bg-gray-700/50 dark:hover:text-green-400'
+                  }`}
+                  aria-expanded={advancedOpen}
+                >
+                  <span>高级</span>
+                  <ChevronUp
+                    className={`h-4 w-4 transition-transform ${
+                      advancedOpen ? '' : 'rotate-180'
+                    }`}
+                  />
+                </button>
+                {advancedOpen && (
+                  <div
+                    ref={(el) => {
+                      if (el) advancedDropdownRefs.current[0] = el;
+                    }}
+                    className='absolute right-0 z-30 mt-2 w-56 rounded-xl border border-gray-200 bg-white p-3 shadow-lg dark:border-gray-700 dark:bg-gray-900'
+                  >
+                    <label className='flex cursor-pointer select-none items-center justify-between gap-3 rounded-lg px-1 py-2'>
+                      <span className='text-sm text-gray-700 dark:text-gray-300'>
+                        聚合
+                      </span>
+                      <div className='relative'>
+                        <input
+                          type='checkbox'
+                          className='peer sr-only'
+                          checked={viewMode === 'agg'}
+                          onChange={() =>
+                            setViewMode(viewMode === 'agg' ? 'all' : 'agg')
+                          }
+                        />
+                        <div className='h-5 w-9 rounded-full bg-gray-300 transition-colors peer-checked:bg-green-500 dark:bg-gray-600'></div>
+                        <div className='absolute left-0.5 top-0.5 h-4 w-4 rounded-full bg-white transition-transform peer-checked:translate-x-4'></div>
+                      </div>
+                    </label>
+                    <label className='flex cursor-pointer select-none items-center justify-between gap-3 rounded-lg px-1 py-2'>
+                      <span className='text-sm text-gray-700 dark:text-gray-300'>
+                        只搜私人影库
+                      </span>
+                      <div className='relative'>
+                        <input
+                          type='checkbox'
+                          className='peer sr-only'
+                          checked={privateLibraryOnly}
+                          onChange={(e) =>
+                            setPrivateLibraryOnly(e.target.checked)
+                          }
+                        />
+                        <div className='h-5 w-9 rounded-full bg-gray-300 transition-colors peer-checked:bg-green-500 dark:bg-gray-600'></div>
+                        <div className='absolute left-0.5 top-0.5 h-4 w-4 rounded-full bg-white transition-transform peer-checked:translate-x-4'></div>
+                      </div>
+                    </label>
+                  </div>
+                )}
+              </div>
+            </div>
+          )}
+
         </div>
 
         {pansouCloudFilterOpen &&
@@ -1884,7 +2011,11 @@ function SearchPageClient() {
         {/* 搜索结果或搜索历史 */}
         <div
           className={`max-w-[95%] mx-auto overflow-visible ${
-            activeTab === 'pansou' ? 'mt-4' : 'mt-12'
+            activeTab === 'video' && !showResults
+              ? 'mt-2'
+              : activeTab === 'pansou'
+              ? 'mt-4'
+              : 'mt-12'
           }`}
         >
           {showResults ? (
@@ -1895,58 +2026,58 @@ function SearchPageClient() {
                   {/* 标题 */}
                   <div className='mb-4 flex items-start justify-between gap-4'>
                     <div className='min-w-0'>
-                      <h2 className='text-xl font-bold text-gray-800 dark:text-gray-200'>
+                      <h2 className='flex flex-wrap items-start gap-x-3 gap-y-1 text-xl font-bold text-gray-800 dark:text-gray-200'>
-                        搜索结果
+                        <span className='inline-flex items-center gap-2'>
-                        {isFromCache ? (
+                          搜索结果
-                          <span className='ml-2 rounded-md bg-green-50 px-2 py-0.5 text-xs font-medium text-green-600 dark:bg-green-900/30 dark:text-green-400'>
+                          {isFromCache && (
-                            缓存
+                            <span className='rounded-md bg-green-50 px-2 py-0.5 text-xs font-medium text-green-600 dark:bg-green-900/30 dark:text-green-400'>
-                          </span>
+                              缓存
-                        ) : (
+                            </span>
-                          <>
+                          )}
-                            {totalSources > 0 && useFluidSearch && (
-                              <span className='ml-2 text-sm font-normal text-gray-500 dark:text-gray-400'>
-                                源 {completedSources}/{totalSources}
-                              </span>
-                            )}
-                            {isLoading && useFluidSearch && (
-                              <span className='ml-2 inline-block align-middle'>
-                                <span className='inline-block h-3 w-3 animate-spin rounded-full border-2 border-gray-300 border-t-green-500'></span>
-                              </span>
-                            )}
-                          </>
-                        )}
-                      </h2>
-                      <div className='mt-2 flex flex-wrap items-center gap-2 text-xs'>
-                        <span className='inline-flex items-center rounded-full bg-gray-100 px-2.5 py-1 font-medium text-gray-700 dark:bg-gray-800 dark:text-gray-200'>
-                          {resultCountMeta.modeLabel}{' '}
-                          {resultCountMeta.visibleCount.toLocaleString()}{' '}
-                          {resultCountMeta.unit}
                         </span>
-                        {resultCountMeta.isFiltered && (
+                        <span className='flex flex-col text-xs font-medium leading-5 text-gray-500 dark:text-gray-400'>
-                          <span className='inline-flex items-center rounded-full bg-white/80 px-2.5 py-1 font-medium text-gray-500 ring-1 ring-gray-200 dark:bg-gray-900/70 dark:text-gray-400 dark:ring-gray-700'>
+                          <span>
-                            筛选前 {resultCountMeta.totalCount.toLocaleString()}{' '}
+                            {resultCountMeta.modeLabel}{' '}
+                            {resultCountMeta.visibleCount.toLocaleString()}{' '}
                             {resultCountMeta.unit}
+                            {resultCountMeta.isFiltered && (
+                              <span className='ml-1 text-gray-400 dark:text-gray-500'>
+                                / 筛选前{' '}
+                                {resultCountMeta.totalCount.toLocaleString()}{' '}
+                                {resultCountMeta.unit}
+                              </span>
+                            )}
                           </span>
-                        )}
+                          {!isFromCache && totalSources > 0 && useFluidSearch && (
-                      </div>
+                            <span className='inline-flex items-center gap-1'>
+                              源 {completedSources}/{totalSources}
+                              {isLoading && (
+                                <span className='inline-block h-3 w-3 animate-spin rounded-full border-2 border-gray-300 border-t-green-500'></span>
+                              )}
+                            </span>
+                          )}
+                        </span>
+                      </h2>
+                    </div>
+                    <div className='flex shrink-0 items-center gap-2'>
+                      {searchQuery && (
+                        <button
+                          onClick={() => {
+                            setForceRefresh(true);
+                          }}
+                          disabled={isLoading}
+                          className='flex items-center gap-1.5 rounded-lg px-3 py-1.5 text-sm text-gray-600 transition-colors hover:bg-green-50 hover:text-green-600 disabled:cursor-not-allowed disabled:opacity-50 dark:text-gray-400 dark:hover:bg-gray-700/50 dark:hover:text-green-400'
+                          aria-label='强制刷新搜索结果'
+                        >
+                          <RefreshCw
+                            className={`h-4 w-4 ${
+                              isLoading ? 'animate-spin' : ''
+                            }`}
+                          />
+                          <span>刷新</span>
+                        </button>
+                      )}
                     </div>
-                    {searchQuery && (
-                      <button
-                        onClick={() => {
-                          setForceRefresh(true);
-                        }}
-                        disabled={isLoading}
-                        className='flex items-center gap-1.5 rounded-lg px-3 py-1.5 text-sm text-gray-600 transition-colors hover:bg-green-50 hover:text-green-600 disabled:cursor-not-allowed disabled:opacity-50 dark:text-gray-400 dark:hover:bg-gray-700/50 dark:hover:text-green-400'
-                        aria-label='强制刷新搜索结果'
-                      >
-                        <RefreshCw
-                          className={`h-4 w-4 ${
-                            isLoading ? 'animate-spin' : ''
-                          }`}
-                        />
-                        <span>刷新</span>
-                      </button>
-                    )}
                   </div>
                   <div className='mb-4 flex items-center gap-3'>
                     <div className='min-w-0 flex-1'>
@@ -1964,24 +2095,70 @@ function SearchPageClient() {
                         />
                       )}
                     </div>
-                    <div className='flex shrink-0 items-center justify-end self-center'>
+                    <div className='relative flex shrink-0 items-center justify-end self-center'>
-                      <label className='flex shrink-0 cursor-pointer select-none items-center gap-2'>
+                      <button
-                        <span className='text-xs text-gray-700 dark:text-gray-300 sm:text-sm'>
+                        ref={(el) => {
-                          聚合
+                          if (el) advancedButtonRefs.current[1] = el;
-                        </span>
+                        }}
-                        <div className='relative'>
+                        type='button'
-                          <input
+                        onClick={() => setAdvancedOpen((prev) => !prev)}
-                            type='checkbox'
+                        className={`flex items-center gap-1.5 rounded-lg px-3 py-1.5 text-sm transition-colors ${
-                            className='peer sr-only'
+                          advancedOpen
-                            checked={viewMode === 'agg'}
+                            ? 'bg-green-50 text-green-600 dark:bg-gray-700/70 dark:text-green-400'
-                            onChange={() =>
+                            : 'text-gray-600 hover:bg-green-50 hover:text-green-600 dark:text-gray-400 dark:hover:bg-gray-700/50 dark:hover:text-green-400'
-                              setViewMode(viewMode === 'agg' ? 'all' : 'agg')
+                        }`}
-                            }
+                        aria-expanded={advancedOpen}
-                          />
+                      >
-                          <div className='h-5 w-9 rounded-full bg-gray-300 transition-colors peer-checked:bg-green-500 dark:bg-gray-600'></div>
+                        <span>高级</span>
-                          <div className='absolute left-0.5 top-0.5 h-4 w-4 rounded-full bg-white transition-transform peer-checked:translate-x-4'></div>
+                        <ChevronUp
+                          className={`h-4 w-4 transition-transform ${
+                            advancedOpen ? '' : 'rotate-180'
+                          }`}
+                        />
+                      </button>
+                      {advancedOpen && (
+                        <div
+                          ref={(el) => {
+                            if (el) advancedDropdownRefs.current[1] = el;
+                          }}
+                          className='absolute right-0 top-full z-30 mt-2 w-56 rounded-xl border border-gray-200 bg-white p-3 shadow-lg dark:border-gray-700 dark:bg-gray-900'
+                        >
+                          <label className='flex cursor-pointer select-none items-center justify-between gap-3 rounded-lg px-1 py-2'>
+                            <span className='text-sm text-gray-700 dark:text-gray-300'>
+                              聚合
+                            </span>
+                            <div className='relative'>
+                              <input
+                                type='checkbox'
+                                className='peer sr-only'
+                                checked={viewMode === 'agg'}
+                                onChange={() =>
+                                  setViewMode(viewMode === 'agg' ? 'all' : 'agg')
+                                }
+                              />
+                              <div className='h-5 w-9 rounded-full bg-gray-300 transition-colors peer-checked:bg-green-500 dark:bg-gray-600'></div>
+                              <div className='absolute left-0.5 top-0.5 h-4 w-4 rounded-full bg-white transition-transform peer-checked:translate-x-4'></div>
+                            </div>
+                          </label>
+                          <label className='flex cursor-pointer select-none items-center justify-between gap-3 rounded-lg px-1 py-2'>
+                            <span className='text-sm text-gray-700 dark:text-gray-300'>
+                              只搜私人影库
+                            </span>
+                            <div className='relative'>
+                              <input
+                                type='checkbox'
+                                className='peer sr-only'
+                                checked={privateLibraryOnly}
+                                onChange={(e) =>
+                                  setPrivateLibraryOnly(e.target.checked)
+                                }
+                              />
+                              <div className='h-5 w-9 rounded-full bg-gray-300 transition-colors peer-checked:bg-green-500 dark:bg-gray-600'></div>
+                              <div className='absolute left-0.5 top-0.5 h-4 w-4 rounded-full bg-white transition-transform peer-checked:translate-x-4'></div>
+                            </div>
+                          </label>
                         </div>
-                      </label>
+                      )}
                     </div>
                   </div>
                   <div className='mb-8 flex justify-center'>