sssssss

cmdlets/get_url.py

@@ -2,6 +2,8 @@ from __future__ import annotations
 
 from typing import Any, Dict, Sequence
 import json
+import sys
+from pathlib import Path
 
 from . import register
 import models
@@ -9,16 +11,18 @@ import pipeline as ctx
 from helper import hydrus as hydrus_wrapper
 from ._shared import Cmdlet, CmdletArg, normalize_hash
 from helper.logger import log
+from config import get_local_storage_path
+from helper.local_library import LocalLibraryDB
 
 CMDLET = Cmdlet(
     name="get-url",
-    summary="List URLs associated with a Hydrus file.",
+    summary="List URLs associated with a file (Hydrus or Local).",
     usage="get-url [-hash <sha256>]",
     args=[
         CmdletArg("-hash", description="Override the Hydrus file hash (SHA256) to target instead of the selected result."),
     ],
     details=[
-        "- Prints the known URLs for the selected Hydrus file.",
+        "- Prints the known URLs for the selected file.",
     ],
 )
 
@@ -55,36 +59,81 @@ def get_urls(result: Any, args: Sequence[str], config: Dict[str, Any]) -> int:
             return 0
     except Exception:
         pass
-    override_hash, _ = _parse_hash_and_rest(args)
-    hash_hex = normalize_hash(override_hash) if override_hash else normalize_hash(get_field(result, "hash_hex", None))
-    if not hash_hex:
-        log("Selected result does not include a Hydrus hash")
-        return 1
-    try:
-        client = hydrus_wrapper.get_client(config)
-    except Exception as exc:
-        log(f"Hydrus client unavailable: {exc}")
-        return 1
     
-    if client is None:
-        log("Hydrus client unavailable")
-        return 1
-    try:
-        payload = client.fetch_file_metadata(hashes=[hash_hex], include_file_urls=True)
-    except Exception as exc:
-        log(f"Hydrus metadata fetch failed: {exc}")
-        return 1
-    items = payload.get("metadata") if isinstance(payload, dict) else None
-    meta = items[0] if (isinstance(items, list) and items and isinstance(items[0], dict)) else None
-    urls = (meta.get("known_urls") if isinstance(meta, dict) else None) or []
-    if urls:
-        ctx.emit("URLs:")
-        for u in urls:
+    override_hash, _ = _parse_hash_and_rest(args)
+    
+    # Handle @N selection which creates a list - extract the first item
+    if isinstance(result, list) and len(result) > 0:
+        result = result[0]
+
+    found_urls = []
+    
+    # 1. Try Local Library
+    file_path = get_field(result, "file_path") or get_field(result, "path")
+    if file_path and not override_hash:
+        try:
+            path_obj = Path(file_path)
+            if path_obj.exists():
+                storage_path = get_local_storage_path(config)
+                if storage_path:
+                    with LocalLibraryDB(storage_path) as db:
+                        metadata = db.get_metadata(path_obj)
+                        if metadata and metadata.get("known_urls"):
+                            found_urls.extend(metadata["known_urls"])
+        except Exception as e:
+            log(f"Error checking local library: {e}", file=sys.stderr)
+
+    # 2. Try Hydrus
+    hash_hex = normalize_hash(override_hash) if override_hash else normalize_hash(get_field(result, "hash_hex", None))
+    
+    # If we haven't found URLs yet, or if we want to merge them (maybe?), let's check Hydrus if we have a hash
+    # But usually if it's local, we might not want to check Hydrus unless requested.
+    # However, the user said "they can just work together".
+    
+    if hash_hex:
+        try:
+            client = hydrus_wrapper.get_client(config)
+            if client:
+                payload = client.fetch_file_metadata(hashes=[hash_hex], include_file_urls=True)
+                items = payload.get("metadata") if isinstance(payload, dict) else None
+                meta = items[0] if (isinstance(items, list) and items and isinstance(items[0], dict)) else None
+                hydrus_urls = (meta.get("known_urls") if isinstance(meta, dict) else None) or []
+                for u in hydrus_urls:
+                    if u not in found_urls:
+                        found_urls.append(u)
+        except Exception as exc:
+            # Only log error if we didn't find local URLs either, or if it's a specific error
+            if not found_urls:
+                log(f"Hydrus lookup failed: {exc}", file=sys.stderr)
+
+    if found_urls:
+        for u in found_urls:
             text = str(u).strip()
             if text:
-                ctx.emit(f"- {text}")
-    else:
-        ctx.emit("No URLs found.")
+                # Emit a rich object that looks like a string but carries context
+                # We use a dict with 'title' which ResultTable uses for display
+                # and 'url' which is the actual data
+                # We also include the source file info so downstream cmdlets can use it
+                
+                # Create a result object that mimics the structure expected by delete-url
+                # delete-url expects a file object usually, but here we are emitting URLs.
+                # If we emit a dict with 'url' and 'source_file', delete-url can use it.
+                
+                rich_result = {
+                    "title": text, # Display as just the URL
+                    "url": text,
+                    "source_file": result, # Pass the original file context
+                    "file_path": get_field(result, "file_path") or get_field(result, "path"),
+                    "hash_hex": hash_hex
+                }
+                ctx.emit(rich_result)
+        return 0
+    
+    if not hash_hex and not file_path:
+        log("Selected result does not include a file path or Hydrus hash", file=sys.stderr)
+        return 1
+        
+    ctx.emit("No URLs found.")
     return 0