dfd

cmdlets/add_url.py

@@ -13,6 +13,7 @@ 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
+from helper.logger import debug
 
 CMDLET = Cmdlet(
     name="add-url",
@@ -124,6 +125,39 @@ def add(result: Any, args: Sequence[str], config: Dict[str, Any]) -> int:
                 return 1
 
     if success:
+        # If we just mutated the currently displayed item, refresh URLs via get-url
+        try:
+            from cmdlets import get_url as get_url_cmd  # type: ignore
+        except Exception:
+            get_url_cmd = None
+        if get_url_cmd:
+            try:
+                subject = ctx.get_last_result_subject()
+                if subject is not None:
+                    def norm(val: Any) -> str:
+                        return str(val).lower()
+                    target_hash = norm(hash_hex) if hash_hex else None
+                    target_path = norm(file_path) if 'file_path' in locals() else None
+                    subj_hashes = []
+                    subj_paths = []
+                    if isinstance(subject, dict):
+                        subj_hashes = [norm(v) for v in [subject.get("hydrus_hash"), subject.get("hash"), subject.get("hash_hex"), subject.get("file_hash")] if v]
+                        subj_paths = [norm(v) for v in [subject.get("file_path"), subject.get("path"), subject.get("target")] if v]
+                    else:
+                        subj_hashes = [norm(getattr(subject, f, None)) for f in ("hydrus_hash", "hash", "hash_hex", "file_hash") if getattr(subject, f, None)]
+                        subj_paths = [norm(getattr(subject, f, None)) for f in ("file_path", "path", "target") if getattr(subject, f, None)]
+                    is_match = False
+                    if target_hash and target_hash in subj_hashes:
+                        is_match = True
+                    if target_path and target_path in subj_paths:
+                        is_match = True
+                    if is_match:
+                        refresh_args: list[str] = []
+                        if hash_hex:
+                            refresh_args.extend(["-hash", hash_hex])
+                        get_url_cmd._run(subject, refresh_args, config)
+            except Exception:
+                debug("URL refresh skipped (error)")
         return 0
         
     if not hash_hex and not file_path: