Add a dropdown menu that allows users to select a library.

app/main.py

@@ -1,5 +1,4 @@
 import os
-from typing import Tuple
 from app.utils.config import server_config
 from app.utils.playlist_merge import SyncMode, sync_all_playlists, TEST_PLAYLIST_DIR
 from fastapi import FastAPI, Request, Form
@@ -48,7 +47,7 @@ SYNC_MODE_OPTIONS = [
 ]
 
 
-def _get_cloud_playlists() -> Tuple[list[dict], str, dict]:
+def _get_cloud_playlists() -> tuple[list[dict], str, dict, str, list[str]]:
     """Fetch playlists and connection state from the remote Plex server."""
 
     server_config.load()
@@ -58,10 +57,12 @@ def _get_cloud_playlists() -> Tuple[list[dict], str, dict]:
         "name": "未设置",
         "domain": server_config.url or "未设置",
     }
+    selected_library = server_config.library_name
+    music_libraries: list[str] = []
 
     # no server url configured
     if not server_config.url:
-        return playlists, status, server_info
+        return playlists, status, server_info, selected_library, music_libraries
 
     status = "failed"
     try:
@@ -79,7 +80,16 @@ def _get_cloud_playlists() -> Tuple[list[dict], str, dict]:
             }
         )
 
-        for playlist in plex_client.server.playlists():
+        music_libraries = plex_client.get_libs_name_list()
+        if not music_libraries:
+            server_config.set_and_save_config(library_name="")
+            return playlists, status, server_info, "", music_libraries
+
+        if not selected_library or selected_library not in music_libraries:
+            selected_library = music_libraries[0]
+            server_config.set_and_save_config(library_name=selected_library)
+
+        for playlist in plex_client.get_lib_playlists(selected_library) or []:
             track_count = getattr(playlist, "itemCount", None)
             if track_count is None:
                 try:
@@ -97,7 +107,8 @@ def _get_cloud_playlists() -> Tuple[list[dict], str, dict]:
         status = "failed"
 
     playlists.sort(key=lambda item: item["name"].lower())
-    return playlists, status, server_info
+    return playlists, status, server_info, selected_library, music_libraries
+
 
 def _build_home_context(
     request: Request,
@@ -109,7 +120,13 @@ def _build_home_context(
 ):
     server_config.load()
     local_playlists = scan_local_playlists(local_path)
-    cloud_playlists, connection_status, server_info = _get_cloud_playlists()
+    (
+        cloud_playlists,
+        connection_status,
+        server_info,
+        selected_library,
+        music_libraries,
+    ) = _get_cloud_playlists()
 
     return {
         "request": request,
@@ -120,6 +137,8 @@ def _build_home_context(
         "cloud_playlists": cloud_playlists,
         "connection_status": connection_status,
         "server_info": server_info,
+        "selected_library": selected_library,
+        "music_libraries": music_libraries,
         "sync_modes": SYNC_MODE_OPTIONS,
         "selected_mode": selected_mode,
         "message": message,
@@ -225,6 +244,25 @@ async def save_path_rules(
 @app.get("/login", response_class=HTMLResponse)
 async def login_page(request: Request):
     server_config.load()
+    music_libraries = []
+    selected_library = server_config.library_name
+    if server_config.url and server_config.token:
+        try:
+            plex_client.connect(
+                token=server_config.token,
+                scheme=server_config.scheme,
+                url=server_config.url,
+                port=server_config.port,
+            )
+            music_libraries = plex_client.get_libs_name_list()
+            if music_libraries:
+                if selected_library not in music_libraries:
+                    selected_library = music_libraries[0]
+            else:
+                selected_library = ""
+        except Exception:
+            selected_library = ""
+            music_libraries = []
     return templates.TemplateResponse(
         "login.html",
         {
@@ -235,6 +273,8 @@ async def login_page(request: Request):
             "scheme": server_config.scheme,
             "server_url": server_config.url,
             "port": server_config.port,
+            "music_libraries": music_libraries,
+            "selected_library": selected_library,
         },
     )
 
@@ -248,6 +288,7 @@ async def login(
     scheme: str = Form("https"),
     url: str = Form(...),
     port: str = Form("32400"),
+    library_name: str = Form(""),
 ):
     # 尝试连接到 Plex 服务器
     try:
@@ -261,9 +302,30 @@ async def login(
             port=port,
         )
         # 成功连接后保存配置到配置文件
+        music_libraries: list[str] = []
+        selected_library = ""
         if plex_client.connected:
+            try:
+                music_libraries = plex_client.get_libs_name_list()
+            except Exception as exc:
+                logger.warning(f"Unable to fetch music libraries: {exc}")
+                music_libraries = []
+            if music_libraries:
+                if library_name and library_name in music_libraries:
+                    selected_library = library_name
+                else:
+                    selected_library = music_libraries[0]
             server_config.set_and_save_config(
-                token=token_success, scheme=scheme, url=url, port=port
+                token=token_success,
+                scheme=scheme,
+                url=url,
+                port=port,
+                library_name=selected_library,
+            )
+        else:
+            music_libraries = []
+            server_config.set_and_save_config(
+                token=token_success, scheme=scheme, url=url, port=port, library_name=""
             )
         return templates.TemplateResponse(
             "login.html",
@@ -277,9 +339,12 @@ async def login(
                 "scheme": server_config.scheme,
                 "server_url": server_config.url,
                 "port": server_config.port,
+                "music_libraries": music_libraries,
+                "selected_library": selected_library,
             },
         )
     except Exception as e:
+        music_libraries = []
         return templates.TemplateResponse(
             "login.html",
             {
@@ -291,6 +356,8 @@ async def login(
                 "scheme": scheme,
                 "server_url": url,
                 "port": port,
+                "music_libraries": music_libraries,
+                "selected_library": "",
             },
         )