Medios-Macina/cmdlet/delete_url.py

from __future__ import annotations

from typing import Any, Dict, Sequence
import sys

import pipeline as ctx
from . import _shared as sh

Cmdlet, CmdletArg, SharedArgs, parse_cmdlet_args, get_field, normalize_hash = (
    sh.Cmdlet,
    sh.CmdletArg,
    sh.SharedArgs,
    sh.parse_cmdlet_args,
    sh.get_field,
    sh.normalize_hash,
)
from SYS.logger import log
from Store import Store


class Delete_Url(Cmdlet):
    """Delete URL associations from files via hash+store."""

    def __init__(self) -> None:
        super().__init__(
            name="delete-url",
            summary="Remove a URL association from a file",
            usage="@1 | delete-url <url>",
            arg=[
                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",
            ],
            exec=self.run,
        )
        self.register()
    
    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)
        urls = [u.strip() for u in str(url_arg).split(',') if u.strip()]
        if not urls:
            log("Error: No valid url provided")
            return 1
        
        # Get backend and delete url
        try:
            storage = Store(config)
            backend = storage[store_name]

            backend.delete_url(file_hash, urls)
            for u in urls:
                ctx.emit(f"Deleted URL: {u}")
            
            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


CMDLET = Delete_Url()