AST

cmdlets/add_url.py

@@ -0,0 +1,78 @@
+from __future__ import annotations
+
+from typing import Any, Dict, Sequence
+import json
+
+from . import register
+import models
+import pipeline as ctx
+from helper import hydrus as hydrus_wrapper
+from ._shared import Cmdlet, CmdletArg, normalize_hash
+from helper.logger import log
+
+CMDLET = Cmdlet(
+    name="add-url",
+    summary="Associate a URL with a Hydrus file.",
+    usage="add-url [-hash <sha256>] <url>",
+    args=[
+        CmdletArg("-hash", description="Override the Hydrus file hash (SHA256) to target instead of the selected result."),
+        CmdletArg("url", required=True, description="The URL to associate with the file."),
+    ],
+    details=[
+        "- Adds the URL to the Hydrus file's known URL list.",
+    ],
+)
+
+
+@register(["add-url", "ass-url", "associate-url", "add_url"])  # aliases
+def add(result: Any, args: Sequence[str], config: Dict[str, Any]) -> int:
+    # Help
+    try:
+        if any(str(a).lower() in {"-?", "/?", "--help", "-h", "help", "--cmdlet"} for a in args):
+            log(json.dumps(CMDLET, ensure_ascii=False, indent=2))
+            return 0
+    except Exception:
+        pass
+        
+    from ._shared import parse_cmdlet_args
+    parsed = parse_cmdlet_args(args, CMDLET)
+    override_hash = parsed.get("hash")
+    url = parsed.get("url")
+    
+    if not url:
+        log("Requires a URL argument")
+        return 1
+    
+    url = str(url).strip()
+    if not url:
+        log("Requires a non-empty URL")
+        return 1
+    
+    # Handle @N selection which creates a list - extract the first item
+    if isinstance(result, list) and len(result) > 0:
+        result = result[0]
+    
+    hash_hex = normalize_hash(override_hash) if override_hash else normalize_hash(getattr(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:
+        client.associate_url(hash_hex, url)
+    except Exception as exc:
+        log(f"Hydrus add-url failed: {exc}")
+        return 1
+    preview = hash_hex[:12] + ('…' if len(hash_hex) > 12 else '')
+    ctx.emit(f"Associated URL with {preview}: {url}")
+    return 0
+
+
+