import os

from pr.tools.base import get_tools_definition
from pr.tools.filesystem import list_directory, read_file, search_replace, write_file
from pr.tools.patch import apply_patch, create_diff


class TestFilesystemTools:

    def test_write_and_read_file(self, temp_dir):
        filepath = os.path.join(temp_dir, "test.txt")
        content = "Hello, World!"

        write_result = write_file(filepath, content)
        assert write_result["status"] == "success"

        read_result = read_file(filepath)
        assert read_result["status"] == "success"
        assert content in read_result["content"]

    def test_read_nonexistent_file(self):
        result = read_file("/nonexistent/path/file.txt")
        assert result["status"] == "error"

    def test_list_directory(self, temp_dir):
        test_file = os.path.join(temp_dir, "testfile.txt")
        with open(test_file, "w") as f:
            f.write("test")

        result = list_directory(temp_dir)
        assert result["status"] == "success"
        assert any(item["name"] == "testfile.txt" for item in result["items"])

    def test_search_replace(self, temp_dir):
        filepath = os.path.join(temp_dir, "test.txt")
        content = "Hello, World!"
        with open(filepath, "w") as f:
            f.write(content)

        result = search_replace(filepath, "World", "Universe")
        assert result["status"] == "success"

        read_result = read_file(filepath)
        assert "Hello, Universe!" in read_result["content"]


class TestPatchTools:

    def test_create_diff(self, temp_dir):
        file1 = os.path.join(temp_dir, "file1.txt")
        file2 = os.path.join(temp_dir, "file2.txt")
        with open(file1, "w") as f:
            f.write("line1\nline2\nline3\n")
        with open(file2, "w") as f:
            f.write("line1\nline2 modified\nline3\n")

        result = create_diff(file1, file2)
        assert result["status"] == "success"
        assert "line2" in result["diff"]
        assert "line2 modified" in result["diff"]

    def test_apply_patch(self, temp_dir):
        filepath = os.path.join(temp_dir, "file.txt")
        with open(filepath, "w") as f:
            f.write("line1\nline2\nline3\n")

        # Create a simple patch
        patch_content = """--- a/file.txt
+++ b/file.txt
@@ -1,3 +1,3 @@
 line1
-line2
+line2 modified
 line3
"""
        result = apply_patch(filepath, patch_content)
        assert result["status"] == "success"

        read_result = read_file(filepath)
        assert "line2 modified" in read_result["content"]


class TestToolDefinitions:

    def test_get_tools_definition_returns_list(self):
        tools = get_tools_definition()
        assert isinstance(tools, list)
        assert len(tools) > 0

    def test_all_tools_have_required_fields(self):
        tools = get_tools_definition()

        for tool in tools:
            assert "type" in tool
            assert tool["type"] == "function"
            assert "function" in tool

            func = tool["function"]
            assert "name" in func
            assert "description" in func
            assert "parameters" in func

    def test_filesystem_tools_present(self):
        tools = get_tools_definition()
        tool_names = [t["function"]["name"] for t in tools]

        assert "read_file" in tool_names
        assert "write_file" in tool_names
        assert "list_directory" in tool_names
        assert "search_replace" in tool_names

    def test_patch_tools_present(self):
        tools = get_tools_definition()
        tool_names = [t["function"]["name"] for t in tools]

        assert "apply_patch" in tool_names
        assert "create_diff" in tool_names