dfdsf

cmdlets/search_provider.py

@@ -1,15 +1,23 @@
 """search-provider cmdlet: Search external providers (bandcamp, libgen, soulseek, youtube)."""
 from __future__ import annotations
 
-from typing import Any, Dict, List, Sequence
+from typing import Any, Dict, List, Sequence, Optional
 import sys
+import json
+import uuid
+import importlib
 
-from helper.logger import log, debug
-from helper.provider import get_search_provider, list_search_providers
+from SYS.logger import log, debug
+from Provider.registry import get_search_provider, list_search_providers
 
 from ._shared import Cmdlet, CmdletArg, should_show_help
 import pipeline as ctx
 
+# Optional dependencies
+try:
+    from config import get_local_storage_path
+except Exception:  # pragma: no cover
+    get_local_storage_path = None  # type: ignore
 
 class Search_Provider(Cmdlet):
     """Search external content providers."""
@@ -88,30 +96,74 @@ class Search_Provider(Cmdlet):
                 if available:
                     log(f"  - {name}", file=sys.stderr)
             return 1
-        
-        # Execute search
-        try:
-            debug(f"[search-provider] Calling {provider_name}.search()")
-            results = provider.search(query, limit=limit)
-            debug(f"[search-provider] Got {len(results)} results")
-            
-            if not results:
-                log(f"No results found for query: {query}", file=sys.stderr)
-                return 0
-            
-            # Emit results for pipeline
-            for search_result in results:
-                ctx.emit(search_result.to_dict())
-            
-            log(f"Found {len(results)} result(s) from {provider_name}", file=sys.stderr)
-            return 0
-            
-        except Exception as e:
-            log(f"Error searching {provider_name}: {e}", file=sys.stderr)
-            import traceback
-            debug(traceback.format_exc())
+
+        from API.folder import API_folder_store
+        worker_id = str(uuid.uuid4())
+        library_root = get_local_storage_path(config or {})
+        if not library_root:
+            log("No library root configured", file=sys.stderr)
             return 1
 
+        # Use context manager to ensure database is always closed
+        with API_folder_store(library_root) as db:
+            try:
+                db.insert_worker(
+                    worker_id,
+                    "search-provider",
+                    title=f"Search: {query}",
+                    description=f"Provider: {provider_name}, Query: {query}",
+                    pipe=ctx.get_current_command_text()
+                )
+
+                results_list = []
+                import result_table
+                importlib.reload(result_table)
+                from result_table import ResultTable
+
+                table_title = f"Search: {query} [{provider_name}]"
+                preserve_order = provider_name.lower() in ('youtube', 'openlibrary')
+                table = ResultTable(table_title).set_preserve_order(preserve_order)
+                table.set_table(provider_name)
+
+                debug(f"[search-provider] Calling {provider_name}.search()")
+                results = provider.search(query, limit=limit)
+                debug(f"[search-provider] Got {len(results)} results")
+                
+                if not results:
+                    log(f"No results found for query: {query}", file=sys.stderr)
+                    db.append_worker_stdout(worker_id, json.dumps([], indent=2))
+                    db.update_worker_status(worker_id, 'completed')
+                    return 0
+                
+                # Emit results for pipeline
+                for search_result in results:
+                    item_dict = search_result.to_dict() if hasattr(search_result, 'to_dict') else dict(search_result)
+                    
+                    # Ensure table field is set (should be by provider, but just in case)
+                    if 'table' not in item_dict:
+                        item_dict['table'] = provider_name
+                    
+                    table.add_result(search_result) # ResultTable handles SearchResult objects
+                    results_list.append(item_dict)
+                    ctx.emit(item_dict)
+                
+                ctx.set_last_result_table(table, results_list)
+                db.append_worker_stdout(worker_id, json.dumps(results_list, indent=2))
+                db.update_worker_status(worker_id, 'completed')
+                
+                log(f"Found {len(results)} result(s) from {provider_name}", file=sys.stderr)
+                return 0
+                
+            except Exception as e:
+                log(f"Error searching {provider_name}: {e}", file=sys.stderr)
+                import traceback
+                debug(traceback.format_exc())
+                try:
+                    db.update_worker_status(worker_id, 'error')
+                except Exception:
+                    pass
+                return 1
+
 
 # Register cmdlet instance
 Search_Provider_Instance = Search_Provider()