#!/usr/bin/env python3

import os
import re
import tempfile

import pytest
import toml
import yaml
from helpers import NonExistentPath, TempGitRepository, grm, shell


def test_repos_find_nonexistent():
    with NonExistentPath() as nonexistent_dir:
        cmd = grm(["repos", "find", "local", nonexistent_dir])
        assert "does not exist" in cmd.stderr.lower()
        assert cmd.returncode != 0
        assert not os.path.exists(nonexistent_dir)


def test_repos_find_file():
    with tempfile.NamedTemporaryFile() as tmpfile:
        cmd = grm(["repos", "find", "local", tmpfile.name])
        assert "not a directory" in cmd.stderr.lower()
        assert cmd.returncode != 0


def test_repos_find_empty():
    with tempfile.TemporaryDirectory() as tmpdir:
        cmd = grm(["repos", "find", "local", tmpdir])
        assert cmd.returncode == 0
        assert len(cmd.stdout) == 0
        assert len(cmd.stderr) != 0


def test_repos_find_invalid_format():
    with tempfile.TemporaryDirectory() as tmpdir:
        cmd = grm(
            ["repos", "find", "local", tmpdir, "--format", "invalidformat"],
            is_invalid=True,
        )
        assert cmd.returncode != 0
        assert len(cmd.stdout) == 0
        assert "invalid value 'invalidformat'" in cmd.stderr


def test_repos_find_non_git_repos():
    with tempfile.TemporaryDirectory() as tmpdir:
        shell(
            f"""
            cd {tmpdir}
            mkdir non_git
            (
                cd ./non_git
                echo test > test
            )
        """
        )

        cmd = grm(["repos", "find", "local", tmpdir])

        assert cmd.returncode == 0
        assert len(cmd.stdout) == 0
        assert len(cmd.stderr) != 0


@pytest.mark.parametrize("default_format", [True, False])
@pytest.mark.parametrize("configtype", ["toml", "yaml"])
@pytest.mark.parametrize("exclude", [None, "^.*/repo2$", "^not_matching$"])
def test_repos_find(configtype, exclude, default_format):
    with tempfile.TemporaryDirectory() as tmpdir:
        shell(
            f"""
            cd {tmpdir}
            mkdir repo1
            (
                cd ./repo1
                git -c init.defaultBranch=master init
                echo test > test
                git add test
                git commit -m "commit1"
                git remote add origin https://example.com/repo2.git
                git remote add someremote ssh://example.com/repo2.git
            )
            mkdir repo2
            (
                cd ./repo2
                git -c init.defaultBranch=master init
                git checkout -b main
                echo test > test
                git add test
                git commit -m "commit1"
                git remote add origin https://example.com/repo2.git
            )
            mkdir non_git
            (
                cd non_git
                echo test > test
            )
        """
        )

        args = ["repos", "find", "local", tmpdir]
        if not default_format:
            args += ["--format", configtype]
        if exclude:
            args += ["--exclude", exclude]
        cmd = grm(args)
        assert cmd.returncode == 0
        if exclude == "^.*/repo2$":
            assert re.match(r"^.*\[skipped\] .*\/repo2$", cmd.stderr.lower())
            assert "repo2" in cmd.stderr.lower()
        else:
            assert len(cmd.stderr) == 0

        if default_format or configtype == "toml":
            output = toml.loads(cmd.stdout)
        elif configtype == "yaml":
            output = yaml.safe_load(cmd.stdout)
        else:
            raise NotImplementedError()

        assert isinstance(output, dict)
        assert set(output.keys()) == {"trees"}
        assert isinstance(output["trees"], list)
        assert len(output["trees"]) == 1
        for tree in output["trees"]:
            assert set(tree.keys()) == {"root", "repos"}
            assert tree["root"] == tmpdir
            assert isinstance(tree["repos"], list)
            assert len(tree["repos"]) == (1 if exclude == "^.*/repo2$" else 2)

            repo1 = [r for r in tree["repos"] if r["name"] == "repo1"][0]
            assert repo1["worktree_setup"] is False
            assert isinstance(repo1["remotes"], list)
            assert len(repo1["remotes"]) == 2

            origin = [r for r in repo1["remotes"] if r["name"] == "origin"][0]
            assert set(origin.keys()) == {"name", "type", "url"}
            assert origin["type"] == "https"
            assert origin["url"] == "https://example.com/repo2.git"

            someremote = [r for r in repo1["remotes"] if r["name"] == "someremote"][0]
            assert set(origin.keys()) == {"name", "type", "url"}
            assert someremote["type"] == "ssh"
            assert someremote["url"] == "ssh://example.com/repo2.git"

            if exclude == "^.*/repo2$":
                assert [r for r in tree["repos"] if r["name"] == "repo2"] == []
            else:
                repo2 = [r for r in tree["repos"] if r["name"] == "repo2"][0]
                assert repo2["worktree_setup"] is False
                assert isinstance(repo1["remotes"], list)
                assert len(repo2["remotes"]) == 1

                origin = [r for r in repo2["remotes"] if r["name"] == "origin"][0]
                assert set(origin.keys()) == {"name", "type", "url"}
                assert origin["type"] == "https"
                assert origin["url"] == "https://example.com/repo2.git"


@pytest.mark.parametrize("default_format", [True, False])
@pytest.mark.parametrize("configtype", ["toml", "yaml"])
def test_repos_find_in_root(configtype, default_format):
    with TempGitRepository() as repo_dir:
        args = ["repos", "find", "local", repo_dir]
        if not default_format:
            args += ["--format", configtype]
        cmd = grm(args)
        assert cmd.returncode == 0
        assert len(cmd.stderr) == 0

        if default_format or configtype == "toml":
            output = toml.loads(cmd.stdout)
        elif configtype == "yaml":
            output = yaml.safe_load(cmd.stdout)
        else:
            raise NotImplementedError()

        assert isinstance(output, dict)
        assert set(output.keys()) == {"trees"}
        assert isinstance(output["trees"], list)
        assert len(output["trees"]) == 1
        for tree in output["trees"]:
            assert set(tree.keys()) == {"root", "repos"}
            assert tree["root"] == os.path.dirname(repo_dir)
            assert isinstance(tree["repos"], list)
            assert len(tree["repos"]) == 1

            repo1 = [
                r for r in tree["repos"] if r["name"] == os.path.basename(repo_dir)
            ][0]
            assert repo1["worktree_setup"] is False
            assert isinstance(repo1["remotes"], list)
            assert len(repo1["remotes"]) == 2

            origin = [r for r in repo1["remotes"] if r["name"] == "origin"][0]
            assert set(origin.keys()) == {"name", "type", "url"}
            assert origin["type"] == "file"

            someremote = [r for r in repo1["remotes"] if r["name"] == "otherremote"][0]
            assert set(origin.keys()) == {"name", "type", "url"}
            assert someremote["type"] == "file"


@pytest.mark.parametrize("configtype", ["toml", "yaml"])
@pytest.mark.parametrize("default_format", [True, False])
def test_repos_find_with_invalid_repo(configtype, default_format):
    with tempfile.TemporaryDirectory() as tmpdir:
        shell(
            f"""
            cd {tmpdir}
            mkdir repo1
            (
                cd ./repo1
                git -c init.defaultBranch=master init
                echo test > test
                git add test
                git commit -m "commit1"
                git remote add origin https://example.com/repo2.git
                git remote add someremote ssh://example.com/repo2.git
            )
            mkdir repo2
            (
                cd ./repo2
                git -c init.defaultBranch=master init
                git checkout -b main
                echo test > test
                git add test
                git commit -m "commit1"
                git remote add origin https://example.com/repo2.git
            )
            mkdir broken_repo
            (
                cd broken_repo
                echo "broken" > .git
            )
        """
        )

        args = ["repos", "find", "local", tmpdir]
        if not default_format:
            args += ["--format", configtype]
        cmd = grm(args)
        assert cmd.returncode == 0
        assert "broken" in cmd.stderr

        if default_format or configtype == "toml":
            output = toml.loads(cmd.stdout)
        elif configtype == "yaml":
            output = yaml.safe_load(cmd.stdout)
        else:
            raise NotImplementedError()

        assert isinstance(output, dict)
        assert set(output.keys()) == {"trees"}
        assert isinstance(output["trees"], list)
        assert len(output["trees"]) == 1
        for tree in output["trees"]:
            assert set(tree.keys()) == {"root", "repos"}
            assert tree["root"] == tmpdir
            assert isinstance(tree["repos"], list)
            assert len(tree["repos"]) == 2

            repo1 = [r for r in tree["repos"] if r["name"] == "repo1"][0]
            assert repo1["worktree_setup"] is False
            assert isinstance(repo1["remotes"], list)
            assert len(repo1["remotes"]) == 2

            origin = [r for r in repo1["remotes"] if r["name"] == "origin"][0]
            assert set(origin.keys()) == {"name", "type", "url"}
            assert origin["type"] == "https"
            assert origin["url"] == "https://example.com/repo2.git"

            someremote = [r for r in repo1["remotes"] if r["name"] == "someremote"][0]
            assert set(origin.keys()) == {"name", "type", "url"}
            assert someremote["type"] == "ssh"
            assert someremote["url"] == "ssh://example.com/repo2.git"

            repo2 = [r for r in tree["repos"] if r["name"] == "repo2"][0]
            assert repo2["worktree_setup"] is False
            assert isinstance(repo1["remotes"], list)
            assert len(repo2["remotes"]) == 1

            origin = [r for r in repo2["remotes"] if r["name"] == "origin"][0]
            assert set(origin.keys()) == {"name", "type", "url"}
            assert origin["type"] == "https"
            assert origin["url"] == "https://example.com/repo2.git"