import argparse
import sys
import pathlib
import random
from unittest import mock

import pytest
from _repobee import plugin
import repobee_plug as plug

from repobee_feedback import feedback

ASSIGNMENT_NAMES = ("task-1", "task-2")
STUDENT_TEAMS = tuple(
    [
        plug.StudentTeam(members=members)
        for members in (["slarse"], ["glassey"], ["grundb", "glennol"])
    ]
)
STUDENT_TEAM_NAMES = tuple(map(str, STUDENT_TEAMS))

PASS_ISSUE = plug.Issue(title="Pass", body="Well done!\nAbsolutely flawless!")
KOMP_ISSUE = plug.Issue(
    title="Komplettering", body="Not perfect, you need to fix this."
)
FAIL_ISSUE = plug.Issue(
    title="Fail", body="Unfortunately, there are severe errors."
)
ISSUES = (PASS_ISSUE, KOMP_ISSUE, FAIL_ISSUE)

random.seed(512)


def _write_issue(issue: plug.Issue, path: pathlib.Path):
    text = "{}\n{}".format(issue.title, issue.body)
    path.write_text(text, encoding=sys.getdefaultencoding())


def _write_multi_issues_file(repos_and_issues, path):
    with open(str(path), mode="w", encoding=sys.getdefaultencoding()) as file:
        cur = 0
        for repo_name, issue in repos_and_issues:
            if cur:
                file.write("\n")
            file.write("#ISSUE#{}#{}\n".format(repo_name, issue.title))
            file.write(issue.body)
            cur += 1


def test_register():
    """Just test that there is no crash"""
    plugin.register_plugins([feedback])


@pytest.fixture
def parsed_args_issues_dir(tmp_path):
    return argparse.Namespace(
        students=list(STUDENT_TEAMS),
        assignments=list(ASSIGNMENT_NAMES),
        batch_mode=True,
        issues_dir=str(tmp_path),
        multi_issues_file=None,
        truncation_length=50,
        allow_missing=False,
    )


@pytest.fixture
def parsed_args_multi_issues_file(with_multi_issues_file):
    issues_file, _ = with_multi_issues_file
    return argparse.Namespace(
        students=list(STUDENT_TEAMS),
        assignments=list(ASSIGNMENT_NAMES),
        batch_mode=True,
        issues_dir=None,
        multi_issues_file=str(issues_file),
        truncation_length=50,
        allow_missing=False,
    )


@pytest.fixture
def api_mock():
    return mock.MagicMock(spec=plug.PlatformAPI)


@pytest.fixture
def with_issues(tmp_path):
    """Create issue files in a temporary directory and return a list of (team,
    issue) tuples.
    """
    repo_names = plug.generate_repo_names(STUDENT_TEAM_NAMES, ASSIGNMENT_NAMES)
    existing_issues = []
    for repo_name in repo_names:
        issue_file = tmp_path / "{}.md".format(repo_name)
        issue = random.choice(ISSUES)
        _write_issue(issue, issue_file)
        existing_issues.append((repo_name, issue))
    return existing_issues


@pytest.fixture
def with_multi_issues_file(tmp_path):
    """Create the multi issues file."""
    repo_names = plug.generate_repo_names(STUDENT_TEAM_NAMES, ASSIGNMENT_NAMES)
    repos_and_issues = [
        (repo_name, random.choice(ISSUES)) for repo_name in repo_names
    ]
    issues_file = tmp_path / "issues.md"
    _write_multi_issues_file(repos_and_issues, issues_file)
    return issues_file, repos_and_issues


class TestCallback:
    """Tests for the primary callback."""

    def test_opens_issues_from_issues_dir(
        self, with_issues, parsed_args_issues_dir, api_mock
    ):
        """Test that the callback calls the API.open_issue for the expected
        repos and issues, when the issues all exist and are well formed.
        """
        expected_calls = [
            mock.call(issue.title, issue.body, mock.ANY)
            for repo_name, issue in with_issues
        ]

        feedback.callback(args=parsed_args_issues_dir, api=api_mock)

        api_mock.create_issue.assert_has_calls(expected_calls, any_order=True)

    def test_aborts_if_issue_is_missing(
        self, with_issues, parsed_args_issues_dir, api_mock, tmp_path
    ):
        """Test that the callback exits with a plug.PlugError if any of the
        expected issues is not found.
        """
        repo_without_issue = plug.generate_repo_name(
            STUDENT_TEAM_NAMES[-1], ASSIGNMENT_NAMES[0]
        )
        missing_file = tmp_path / "{}.md".format(repo_without_issue)
        missing_file.unlink()

        with pytest.raises(plug.PlugError) as exc_info:
            feedback.callback(args=parsed_args_issues_dir, api=api_mock)

        assert repo_without_issue in str(exc_info.value)
        assert not api_mock.create_issue.called

    def test_ignores_missing_issue_if_allow_missing(
        self, with_issues, parsed_args_issues_dir, api_mock, tmp_path
    ):
        """Test that missing issues are ignored if --allow-mising is set."""
        repo_without_issue = plug.generate_repo_name(
            STUDENT_TEAM_NAMES[-1], ASSIGNMENT_NAMES[0]
        )
        (tmp_path / "{}.md".format(repo_without_issue)).unlink()
        expected_calls = [
            mock.call(issue.title, issue.body, mock.ANY)
            for repo_name, issue in with_issues
            if repo_name != repo_without_issue
        ]
        args_dict = vars(parsed_args_issues_dir)
        args_dict["allow_missing"] = True
        args = argparse.Namespace(**args_dict)

        feedback.callback(args=args, api=api_mock)

        api_mock.create_issue.assert_has_calls(expected_calls, any_order=True)

    def test_opens_nothing_if_open_prompt_returns_false(
        self, with_issues, parsed_args_issues_dir, api_mock
    ):
        """Test that the callback does not attempt to open any issues if the
        'may I open' prompt returns false.
        """
        args_dict = vars(parsed_args_issues_dir)
        args_dict["batch_mode"] = False
        parsed_args_interactive = argparse.Namespace(**args_dict)

        with mock.patch("builtins.input", return_value="n", autospec=True):
            feedback.callback(args=parsed_args_interactive, api=api_mock)

        assert not api_mock.create_issue.called

    def test_opens_issues_from_multi_issues_file(
        self, with_multi_issues_file, api_mock, parsed_args_multi_issues_file
    ):
        """Test that the callback opens issues correctly when they are all
        contained in a multi issues file.
        """
        issues_file, repos_and_issues = with_multi_issues_file
        expected_calls = [
            mock.call(issue.title, issue.body, mock.ANY)
            for repo_name, issue in repos_and_issues
        ]

        feedback.callback(args=parsed_args_multi_issues_file, api=api_mock)

        api_mock.create_issue.assert_has_calls(expected_calls)

    def test_skips_unexpected_issues_in_multi_issues_file(
        self, with_multi_issues_file, parsed_args_multi_issues_file, api_mock
    ):
        """Test that an exception is raised if one or more issues are found
        relating to student repos that ar not in prod(assignments, students).
        """
        student_teams = parsed_args_multi_issues_file.students
        args_dict = vars(parsed_args_multi_issues_file)
        args_dict["students"] = student_teams[:-1]
        args = argparse.Namespace(**args_dict)
        unexpected_repos = plug.generate_repo_names(
            student_teams[-1:], ASSIGNMENT_NAMES
        )

        _, repos_and_issues = with_multi_issues_file
        expected_calls = [
            mock.call(issue.title, issue.body, mock.ANY)
            for repo_name, issue in repos_and_issues
            if repo_name not in unexpected_repos
        ]

        feedback.callback(args=args, api=api_mock)

        api_mock.create_issue.assert_has_calls(expected_calls, any_order=True)
