# Copyright (C) 2010-2023 RhodeCode GmbH
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License, version 3
# (only), as published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
# This program is dual-licensed. If you wish to learn more about the
# RhodeCode Enterprise Edition, including its added features, Support services,
# and proprietary license terms, please see https://rhodecode.com/licenses/

import stat

import pytest

from rhodecode.lib.str_utils import safe_bytes
from rhodecode.lib.vcs.nodes import DirNode
from rhodecode.lib.vcs.nodes import FileNode
from rhodecode.lib.vcs.nodes import Node
from rhodecode.lib.vcs.nodes import NodeError
from rhodecode.lib.vcs.nodes import NodeKind
from rhodecode.tests.vcs.conftest import BackendTestMixin


@pytest.fixture()
def binary_filenode():
    def node_maker(filename):
        data = (
            b"\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00\x10\x00\x00\x00"
            b"\x10\x08\x06\x00\x00\x00\x1f??a\x00\x00\x00\x04gAMA\x00\x00\xaf?7"
            b"\x05\x8a?\x00\x00\x00\x19tEXtSoftware\x00Adobe ImageReadyq?e<\x00"
            b"\x00\x025IDAT8?\xa5\x93?K\x94Q\x14\x87\x9f\xf7?Q\x1bs4?\x03\x9a"
            b'\xa8?B\x02\x8b$\x10[U;i\x13?6h?&h[?"\x14j?\xa2M\x7fB\x14F\x9aQ?&'
            b'\x842?\x0b\x89"\x82??!?\x9c!\x9c2l??{N\x8bW\x9dY\xb4\t/\x1c?='
            b"\x9b?}????\xa9*;9!?\x83\x91?[?\\v*?D\x04'`EpNp\xa2X'U?pVq\"Sw."
            b"\x1e?\x08\x01D?jw????\xbc??7{|\x9b?\x89$\x01??W@\x15\x9c\x05q`Lt/"
            b"\x97?\x94\xa1d?\x18~?\x18?\x18W[%\xb0?\x83??\x14\x88\x8dB?\xa6H"
            b'\tL\tl\x19>/\x01`\xac\xabx?\x9cl\nx\xb0\x98\x07\x95\x88D$"q['
            b"\x19?d\x00(o\n\xa0??\x7f\xb9\xa4?\x1bF\x1f\x8e\xac\xa8?j??eUU}?.?"
            b'\x9f\x8cE??x\x94??\r\xbdtoJU5"0N\x10U?\x00??V\t\x02\x9f\x81?U?'
            b'\x00\x9eM\xae2?r\x9b7\x83\x82\x8aP3????.?&"?\xb7ZP \x0c<?O'
            b"\xa5\t}\xb8?\x99\xa6?\x87?\x1di|/\xa0??0\xbe\x1fp?d&\x1a\xad"
            b"\x95\x8a\x07?\t*\x10??b:?d?.\x13C\x8a?\x12\xbe\xbf\x8e?{???"
            b"\x08?\x80\xa7\x13+d\x13>J?\x80\x15T\x95\x9a\x00??S\x8c\r?\xa1"
            b"\x03\x07?\x96\x9b\xa7\xab=E??\xa4\xb3?\x19q??B\x91=\x8d??k?J"
            b'\x0bV"??\xf7x?\xa1\x00?\\.\x87\x87???\x02F@D\x99],??\x10#?X'
            b"\xb7=\xb9\x10?Z\x1by???cI??\x1ag?\x92\xbc?T?t[\x92\x81?<_\x17~"
            b"\x92\x88?H%?\x10Q\x02\x9f\n\x81qQ\x0bm?\x1bX?\xb1AK\xa6\x9e\xb9?u"
            b'\xb2?1\xbe|/\x92M@\xa2!F?\xa9>"\r<DT?>\x92\x8e?>\x9a9Qv\x127?a'
            b"\xac?Y?8?:??]X???9\x80\xb7?u?\x0b#BZ\x8d=\x1d?p\x00\x00\x00\x00"
            b"IEND\xaeB`\x82"
        )
        return FileNode(filename, content=data)

    return node_maker


class TestNodeBasics:
    @pytest.mark.parametrize("path", ["/foo", "/foo/bar"])
    @pytest.mark.parametrize("kind", [NodeKind.FILE, NodeKind.DIR], ids=["FILE", "DIR"])
    def test_init_wrong_paths(self, path, kind):
        """
        Cannot initialize Node objects with path with slash at the beginning.
        """
        path = safe_bytes(path)
        with pytest.raises(NodeError):
            Node(path, kind)

    @pytest.mark.parametrize("path", ["path", "some/path"])
    @pytest.mark.parametrize("kind", [NodeKind.FILE, NodeKind.DIR], ids=["FILE", "DIR"])
    def test_name(self, path, kind):
        path = safe_bytes(path)
        node = Node(path, kind)
        assert node.name == "path"

    def test_name_root(self):
        node = Node(b"", NodeKind.DIR)
        assert node.name == ""

    def test_root_node_cannot_be_file(self):
        with pytest.raises(NodeError):
            Node(b"", NodeKind.FILE)

    def test_kind_setter(self):
        node = Node(b"", NodeKind.DIR)
        with pytest.raises(NodeError):
            node.kind = NodeKind.FILE

    def test_compare_equal(self):
        node1 = FileNode(b"test", content=b"")
        node2 = FileNode(b"test", content=b"")
        assert node1 == node2
        assert not node1 != node2

    def test_compare_unequal(self):
        node1 = FileNode(b"test", content=b"a")
        node2 = FileNode(b"test", content=b"b")
        assert node1 != node2
        assert not node1 == node2

    @pytest.mark.parametrize(
        "node_path, expected_parent_path",
        [
            ("", b""),
            ("some/path/", b"some/"),
            ("some/longer/path/", b"some/longer/"),
        ],
    )
    def test_parent_path_new(self, node_path, expected_parent_path):
        """
        Tests if node's parent path are properly computed.
        """
        node_path = safe_bytes(node_path)
        node = Node(node_path, NodeKind.DIR)
        parent_path = node.get_parent_path()
        assert parent_path.endswith(b"/") or node.is_root() and parent_path == b""
        assert parent_path == expected_parent_path

    """
    def _test_trailing_slash(self, path):
        if not path.endswith('/'):
            pytest.fail("Trailing slash tests needs paths to end with slash")
        for kind in NodeKind.FILE, NodeKind.DIR:
            with pytest.raises(NodeError):
                Node(path=path, kind=kind)

    def test_trailing_slash(self):
        for path in ('/', 'foo/', 'foo/bar/', 'foo/bar/biz/'):
            self._test_trailing_slash(path)
    """

    def test_is_file(self):
        node = Node(b"any", NodeKind.FILE)
        assert node.is_file()

        node = FileNode(b"any")
        assert node.is_file()
        with pytest.raises(AttributeError):
            node.nodes  # noqa

    def test_is_dir(self):
        node = Node(b"any_dir", NodeKind.DIR)
        assert node.is_dir()

        node = DirNode(b"any_dir")

        assert node.is_dir()
        with pytest.raises(NodeError):
            node.content  # noqa

    def test_dir_node_iter(self):
        nodes = [
            DirNode(b"docs"),
            DirNode(b"tests"),
            FileNode(b"bar"),
            FileNode(b"foo"),
            FileNode(b"readme.txt"),
            FileNode(b"setup.py"),
        ]
        dirnode = DirNode(b"", nodes=nodes)
        for node in dirnode:
            assert node == dirnode.get_node(node.path)

    def test_node_state(self):
        """
        Without link to commit nodes should raise NodeError.
        """
        node = FileNode(b"anything")
        with pytest.raises(NodeError):
            node.state  # noqa
        node = DirNode(b"anything")
        with pytest.raises(NodeError):
            node.state  # noqa

    def test_file_node_stat(self):
        node = FileNode(b"foobar", b"empty... almost")
        mode = node.mode  # default should be 0100644
        assert mode & stat.S_IRUSR
        assert mode & stat.S_IWUSR
        assert mode & stat.S_IRGRP
        assert mode & stat.S_IROTH
        assert not mode & stat.S_IWGRP
        assert not mode & stat.S_IWOTH
        assert not mode & stat.S_IXUSR
        assert not mode & stat.S_IXGRP
        assert not mode & stat.S_IXOTH

    def test_file_node_is_executable(self):
        node = FileNode(b"foobar", b"empty... almost", mode=0o100755)
        assert node.is_executable

        node = FileNode(b"foobar", b"empty... almost", mode=0o100500)
        assert node.is_executable

        node = FileNode(b"foobar", b"empty... almost", mode=0o100644)
        assert not node.is_executable

    def test_file_node_is_not_symlink(self):
        node = FileNode(b"foobar", b"empty...")
        assert not node.is_link()

    def test_mimetype(self):
        py_node = FileNode(b"test.py")
        tar_node = FileNode(b"test.tar.gz")

        ext = "CustomExtension"

        my_node2 = FileNode(b"myfile2")
        my_node2._mimetype = [ext]

        my_node3 = FileNode(b"myfile3")
        my_node3._mimetype = [ext, ext]

        assert py_node.mimetype == "text/x-python"
        assert py_node.get_mimetype() == ("text/x-python", None)

        assert tar_node.mimetype == "application/x-tar"
        assert tar_node.get_mimetype() == ("application/x-tar", "gzip")

        with pytest.raises(NodeError):
            my_node2.get_mimetype()

        assert my_node3.mimetype == ext
        assert my_node3.get_mimetype() == [ext, ext]

    def test_lines_counts(self):
        lines = [
            b"line1\n",
            b"line2\n",
            b"line3\n",
            b"\n",
            b"\n",
            b"line4\n",
        ]
        py_node = FileNode(b"test.py", b"".join(lines))

        assert (len(lines), len(lines)) == py_node.lines()
        assert (len(lines), len(lines) - 2) == py_node.lines(count_empty=True)

    def test_lines_no_newline(self):
        py_node = FileNode(b"test.py", b"oneline")

        assert (1, 1) == py_node.lines()
        assert (1, 1) == py_node.lines(count_empty=True)


class TestNodeContent(object):
    def test_if_binary(self, binary_filenode):
        filenode = binary_filenode(b"calendar.jpg")
        assert filenode.is_binary

    def test_binary_line_counts(self, binary_filenode):
        tar_node = binary_filenode(b"archive.tar.gz")
        assert (0, 0) == tar_node.lines(count_empty=True)

    def test_binary_mimetype(self, binary_filenode):
        tar_node = binary_filenode(b"archive.tar.gz")
        assert tar_node.mimetype == "application/x-tar"


@pytest.mark.usefixtures("vcs_repository_support")
class TestNodesCommits(BackendTestMixin):
    def test_node_last_commit(self, generate_repo_with_commits):
        repo = generate_repo_with_commits(20)
        last_commit = repo.get_commit()

        for x in range(3):
            node = last_commit.get_node(f"file_{x}.txt")
            assert node.last_commit == repo[x]