from ycode.tools.base import Tool, ToolRegistry, ToolResult, truncate_middle, validate_args

SCHEMA = {
    "type": "object",
    "properties": {"path": {"type": "string"}, "n": {"type": "integer"}, "mode": {"type": "string", "enum": ["a", "b"]}},
    "required": ["path"],
}


class Boom(Tool):
    name = "boom"
    description = "always fails"
    parameters = {"type": "object", "properties": {}, "required": []}

    def run(self, args, ctx):
        raise RuntimeError("kaboom")


class Echo(Tool):
    name = "echo"
    description = "echo"
    parameters = SCHEMA

    def run(self, args, ctx):
        return ToolResult(content=args["path"], summary="ok")


def test_validate_args():
    assert validate_args(SCHEMA, {"path": "x"}) is None
    assert "missing required" in validate_args(SCHEMA, {})
    assert "must be integer" in validate_args(SCHEMA, {"path": "x", "n": "3"})
    assert "must be integer" in validate_args(SCHEMA, {"path": "x", "n": True})
    assert "one of" in validate_args(SCHEMA, {"path": "x", "mode": "c"})
    assert "JSON object" in validate_args(SCHEMA, "not a dict")


def test_dispatch_routes_to_tool(workspace, make_ctx):
    registry = ToolRegistry([Echo()])
    result = registry.dispatch("echo", {"path": "hi"}, make_ctx(workspace))
    assert result.content == "hi" and not result.is_error
    assert result.duration >= 0


def test_dispatch_unknown_tool(workspace, make_ctx):
    result = ToolRegistry([Echo()]).dispatch("nope", {}, make_ctx(workspace))
    assert result.is_error and "unknown tool" in result.content and "echo" in result.content


def test_dispatch_malformed_arguments(workspace, make_ctx):
    registry = ToolRegistry([Echo()])
    assert registry.dispatch("echo", {"path": 5}, make_ctx(workspace)).is_error
    assert registry.dispatch("echo", None, make_ctx(workspace)).is_error


def test_dispatch_contains_unexpected_exceptions(workspace, make_ctx):
    result = ToolRegistry([Boom()]).dispatch("boom", {}, make_ctx(workspace))
    assert result.is_error and "RuntimeError: kaboom" in result.content


def test_duplicate_registration_rejected():
    import pytest

    with pytest.raises(ValueError):
        ToolRegistry([Echo(), Echo()])


def test_truncate_middle_keeps_head_and_tail():
    text = "HEAD" + "x" * 10000 + "TAIL"
    out = truncate_middle(text, 300)
    assert out.startswith("HEAD") and out.endswith("TAIL") and "omitted" in out
    assert truncate_middle("short", 300) == "short"