from __future__ import annotations from typing import Any, Dict, Sequence import sys from . import register import pipeline as ctx from ._shared import Cmdlet, CmdletArg, SharedArgs, parse_cmdlet_args, get_field, normalize_hash from helper.logger import log from helper.store import FileStorage class Delete_Url(Cmdlet): """Delete URL associations from files via hash+store.""" NAME = "delete-url" SUMMARY = "Remove a URL association from a file" USAGE = "@1 | delete-url " ARGS = [ SharedArgs.HASH, SharedArgs.STORE, CmdletArg("url", required=True, description="URL to remove"), ] DETAIL = [ "- Removes URL association from file identified by hash+store", "- Multiple url can be comma-separated", ] def run(self, result: Any, args: Sequence[str], config: Dict[str, Any]) -> int: """Delete URL from file via hash+store backend.""" parsed = parse_cmdlet_args(args, self) # Extract hash and store from result or args file_hash = parsed.get("hash") or get_field(result, "hash") store_name = parsed.get("store") or get_field(result, "store") url_arg = parsed.get("url") if not file_hash: log("Error: No file hash provided") return 1 if not store_name: log("Error: No store name provided") return 1 if not url_arg: log("Error: No URL provided") return 1 # Normalize hash file_hash = normalize_hash(file_hash) if not file_hash: log("Error: Invalid hash format") return 1 # Parse url (comma-separated) url = [u.strip() for u in str(url_arg).split(',') if u.strip()] if not url: log("Error: No valid url provided") return 1 # Get backend and delete url try: storage = FileStorage(config) backend = storage[store_name] for url in url: backend.delete_url(file_hash, url) ctx.emit(f"Deleted URL: {url}") return 0 except KeyError: log(f"Error: Storage backend '{store_name}' not configured") return 1 except Exception as exc: log(f"Error deleting URL: {exc}", file=sys.stderr) return 1 # Register cmdlet register(["delete-url", "del-url", "delete_url"])(Delete_Url)