# SPDX-License-Identifier: MIT
"""
Hash Plugin for TrashClaw

Compute cryptographic hashes (MD5, SHA-1, SHA-256, SHA-512) of text or files.

Plugin contract:
  TOOL_DEF = dict with name, description, parameters (OpenAI function schema)
  run(**kwargs) -> str  (called with the parsed arguments)
"""

import hashlib
import os

TOOL_DEF = {
    "name": "hash",
    "description": "Compute cryptographic hash (MD5, SHA-1, SHA-256, SHA-512) of text or a file.",
    "parameters": {
        "type": "object",
        "properties": {
            "text": {
                "type": "string",
                "description": "Text to hash"
            },
            "file": {
                "type": "string",
                "description": "Path to file to hash (optional)"
            },
            "algorithm": {
                "type": "string",
                "description": "Hash algorithm: md5, sha1, sha256, sha512 (default: sha256)"
            }
        },
        "required": []
    }
}


def _hash_text(text: str, algorithm: str) -> str:
    try:
        h = hashlib.new(algorithm)
        h.update(text.encode("utf-8"))
        return h.hexdigest()
    except ValueError:
        return f"Error: unsupported algorithm '{algorithm}'"


def _hash_file(path: str, algorithm: str) -> str:
    if not os.path.exists(path):
        return f"Error: file not found: {path}"
    try:
        h = hashlib.new(algorithm)
        with open(path, "rb") as f:
            for chunk in iter(lambda: f.read(8192), b""):
                h.update(chunk)
        return h.hexdigest()
    except ValueError:
        return f"Error: unsupported algorithm '{algorithm}'"
    except Exception as e:
        return f"Error: {e}"


def run(text: str = "", file: str = "", algorithm: str = "sha256", **kwargs) -> str:
    algorithm = algorithm.lower().replace("-", "")
    supported = ["md5", "sha1", "sha256", "sha512"]

    if algorithm not in supported:
        return f"Unsupported algorithm '{algorithm}'. Use: {', '.join(supported)}"

    results = []

    if file:
        digest = _hash_file(file, algorithm)
        results.append(f"{algorithm.upper()}({file}):\n{digest}")

    if text:
        digest = _hash_text(text, algorithm)
        results.append(f"{algorithm.upper()}(text):\n{digest}")

    if not text and not file:
        return "Error: provide 'text' or 'file' to hash."

    return "\n\n".join(results)