86 lines
2.4 KiB
Python
86 lines
2.4 KiB
Python
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 SYS.logger import log
|
|
from Store import Store
|
|
|
|
|
|
class Add_Url(Cmdlet):
|
|
"""Add URL associations to files via hash+store."""
|
|
|
|
NAME = "add-url"
|
|
SUMMARY = "Associate a URL with a file"
|
|
USAGE = "@1 | add-url <url>"
|
|
ARGS = [
|
|
SharedArgs.HASH,
|
|
SharedArgs.STORE,
|
|
CmdletArg("url", required=True, description="URL to associate"),
|
|
]
|
|
DETAIL = [
|
|
"- Associates URL with file identified by hash+store",
|
|
"- Multiple url can be comma-separated",
|
|
]
|
|
|
|
def run(self, result: Any, args: Sequence[str], config: Dict[str, Any]) -> int:
|
|
"""Add URL to 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 add url
|
|
try:
|
|
storage = Store(config)
|
|
backend = storage[store_name]
|
|
|
|
backend.add_url(file_hash, urls)
|
|
for u in urls:
|
|
ctx.emit(f"Added URL: {u}")
|
|
|
|
return 0
|
|
|
|
except KeyError:
|
|
log(f"Error: Storage backend '{store_name}' not configured")
|
|
return 1
|
|
except Exception as exc:
|
|
log(f"Error adding URL: {exc}", file=sys.stderr)
|
|
return 1
|
|
|
|
|
|
# Register cmdlet
|
|
register(["add-url", "add_url"])(Add_Url)
|
|
|
|
|
|
|