AST

cmdlets/delete_url.py

@@ -0,0 +1,82 @@
+from __future__ import annotations
+
+from typing import Any, Dict, Sequence
+import json
+
+from . import register
+from helper import hydrus as hydrus_wrapper
+from ._shared import Cmdlet, CmdletArg, normalize_hash
+from helper.logger import log
+
+CMDLET = Cmdlet(
+    name="delete-url",
+    summary="Remove a URL association from a Hydrus file.",
+    usage="delete-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 remove from the file."),
+    ],
+    details=[
+        "- Removes the URL from the Hydrus file's known URL list.",
+    ],
+)
+
+
+def _parse_hash_and_rest(args: Sequence[str]) -> tuple[str | None, list[str]]:
+    override_hash: str | None = None
+    rest: list[str] = []
+    i = 0
+    while i < len(args):
+        a = args[i]
+        low = str(a).lower()
+        if low in {"-hash", "--hash", "hash"} and i + 1 < len(args):
+            override_hash = str(args[i + 1]).strip()
+            i += 2
+            continue
+        rest.append(a)
+        i += 1
+    return override_hash, rest
+
+
+@register(["del-url", "delete-url", "delete_url"])  # aliases
+def delete(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
+    override_hash, rest = _parse_hash_and_rest(args)
+    if not rest:
+        log("Requires a URL argument")
+        return 1
+    url = str(rest[0] or '').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.delete_url(hash_hex, url)
+    except Exception as exc:
+        log(f"Hydrus del-url failed: {exc}")
+        return 1
+    log(f"Deleted URL: {url}")
+    return 0