import itertools
import random

import pytest

from commitizen.version_schemes import Pep440, VersionProtocol

simple_flow = [
    (("0.1.0", "PATCH", None, 0, None), "0.1.1"),
    (("0.1.0", "PATCH", None, 0, 1), "0.1.1.dev1"),
    (("0.1.1", "MINOR", None, 0, None), "0.2.0"),
    (("0.2.0", "MINOR", None, 0, None), "0.3.0"),
    (("0.2.0", "MINOR", None, 0, 1), "0.3.0.dev1"),
    (("0.3.0", "PATCH", None, 0, None), "0.3.1"),
    (("0.3.0", "PATCH", "alpha", 0, None), "0.3.1a0"),
    (("0.3.1a0", None, "alpha", 0, None), "0.3.1a1"),
    (("0.3.0", "PATCH", "alpha", 1, None), "0.3.1a1"),
    (("0.3.1a0", None, "alpha", 1, None), "0.3.1a1"),
    (("0.3.1a0", None, None, 0, None), "0.3.1"),
    (("0.3.1", "PATCH", None, 0, None), "0.3.2"),
    (("0.4.2", "MAJOR", "alpha", 0, None), "1.0.0a0"),
    (("1.0.0a0", None, "alpha", 0, None), "1.0.0a1"),
    (("1.0.0a1", None, "alpha", 0, None), "1.0.0a2"),
    (("1.0.0a1", None, "alpha", 0, 1), "1.0.0a2.dev1"),
    (("1.0.0a2.dev0", None, "alpha", 0, 1), "1.0.0a3.dev1"),
    (("1.0.0a2.dev0", None, "alpha", 0, 0), "1.0.0a3.dev0"),
    (("1.0.0a1", None, "beta", 0, None), "1.0.0b0"),
    (("1.0.0b0", None, "beta", 0, None), "1.0.0b1"),
    (("1.0.0b1", None, "rc", 0, None), "1.0.0rc0"),
    (("1.0.0rc0", None, "rc", 0, None), "1.0.0rc1"),
    (("1.0.0rc0", None, "rc", 0, 1), "1.0.0rc1.dev1"),
    (("1.0.0rc0", "PATCH", None, 0, None), "1.0.0"),
    (("1.0.0a3.dev0", None, "beta", 0, None), "1.0.0b0"),
    (("1.0.0", "PATCH", None, 0, None), "1.0.1"),
    (("1.0.1", "PATCH", None, 0, None), "1.0.2"),
    (("1.0.2", "MINOR", None, 0, None), "1.1.0"),
    (("1.1.0", "MINOR", None, 0, None), "1.2.0"),
    (("1.2.0", "PATCH", None, 0, None), "1.2.1"),
    (("1.2.1", "MAJOR", None, 0, None), "2.0.0"),
]

local_versions = [
    (("4.5.0+0.1.0", "PATCH", None, 0, None), "4.5.0+0.1.1"),
    (("4.5.0+0.1.1", "MINOR", None, 0, None), "4.5.0+0.2.0"),
    (("4.5.0+0.2.0", "MAJOR", None, 0, None), "4.5.0+1.0.0"),
]

# never bump backwards on pre-releases
linear_prerelease_cases = [
    (("0.1.1b1", None, "alpha", 0, None), "0.1.1b2"),
    (("0.1.1rc0", None, "alpha", 0, None), "0.1.1rc1"),
    (("0.1.1rc0", None, "beta", 0, None), "0.1.1rc1"),
]

weird_cases = [
    (("1.1", "PATCH", None, 0, None), "1.1.1"),
    (("1", "MINOR", None, 0, None), "1.1.0"),
    (("1", "MAJOR", None, 0, None), "2.0.0"),
    (("1a0", None, "alpha", 0, None), "1.0.0a1"),
    (("1a0", None, "alpha", 1, None), "1.0.0a1"),
    (("1", None, "beta", 0, None), "1.0.0b0"),
    (("1", None, "beta", 1, None), "1.0.0b1"),
    (("1beta", None, "beta", 0, None), "1.0.0b1"),
    (("1.0.0alpha1", None, "alpha", 0, None), "1.0.0a2"),
    (("1", None, "rc", 0, None), "1.0.0rc0"),
    (("1.0.0rc1+e20d7b57f3eb", "PATCH", None, 0, None), "1.0.0"),
]

# test driven development
tdd_cases = [
    (("0.1.1", "PATCH", None, 0, None), "0.1.2"),
    (("0.1.1", "MINOR", None, 0, None), "0.2.0"),
    (("2.1.1", "MAJOR", None, 0, None), "3.0.0"),
    (("0.9.0", "PATCH", "alpha", 0, None), "0.9.1a0"),
    (("0.9.0", "MINOR", "alpha", 0, None), "0.10.0a0"),
    (("0.9.0", "MAJOR", "alpha", 0, None), "1.0.0a0"),
    (("0.9.0", "MAJOR", "alpha", 1, None), "1.0.0a1"),
    (("1.0.0a2", None, "beta", 0, None), "1.0.0b0"),
    (("1.0.0a2", None, "beta", 1, None), "1.0.0b1"),
    (("1.0.0beta1", None, "rc", 0, None), "1.0.0rc0"),
    (("1.0.0rc1", None, "rc", 0, None), "1.0.0rc2"),
]

# additional pre-release tests run through various release scenarios
prerelease_cases = [
    #
    (("3.3.3", "PATCH", "alpha", 0, None), "3.3.4a0"),
    (("3.3.4a0", "PATCH", "alpha", 0, None), "3.3.4a1"),
    (("3.3.4a1", "MINOR", "alpha", 0, None), "3.4.0a0"),
    (("3.4.0a0", "PATCH", "alpha", 0, None), "3.4.0a1"),
    (("3.4.0a1", "MINOR", "alpha", 0, None), "3.4.0a2"),
    (("3.4.0a2", "MAJOR", "alpha", 0, None), "4.0.0a0"),
    (("4.0.0a0", "PATCH", "alpha", 0, None), "4.0.0a1"),
    (("4.0.0a1", "MINOR", "alpha", 0, None), "4.0.0a2"),
    (("4.0.0a2", "MAJOR", "alpha", 0, None), "4.0.0a3"),
    #
    (("1.0.0", "PATCH", "alpha", 0, None), "1.0.1a0"),
    (("1.0.1a0", "PATCH", "alpha", 0, None), "1.0.1a1"),
    (("1.0.1a1", "MINOR", "alpha", 0, None), "1.1.0a0"),
    (("1.1.0a0", "PATCH", "alpha", 0, None), "1.1.0a1"),
    (("1.1.0a1", "MINOR", "alpha", 0, None), "1.1.0a2"),
    (("1.1.0a2", "MAJOR", "alpha", 0, None), "2.0.0a0"),
    #
    (("1.0.0", "MINOR", "alpha", 0, None), "1.1.0a0"),
    (("1.1.0a0", "PATCH", "alpha", 0, None), "1.1.0a1"),
    (("1.1.0a1", "MINOR", "alpha", 0, None), "1.1.0a2"),
    (("1.1.0a2", "PATCH", "alpha", 0, None), "1.1.0a3"),
    (("1.1.0a3", "MAJOR", "alpha", 0, None), "2.0.0a0"),
    #
    (("1.0.0", "MAJOR", "alpha", 0, None), "2.0.0a0"),
    (("2.0.0a0", "MINOR", "alpha", 0, None), "2.0.0a1"),
    (("2.0.0a1", "PATCH", "alpha", 0, None), "2.0.0a2"),
    (("2.0.0a2", "MAJOR", "alpha", 0, None), "2.0.0a3"),
    (("2.0.0a3", "MINOR", "alpha", 0, None), "2.0.0a4"),
    (("2.0.0a4", "PATCH", "alpha", 0, None), "2.0.0a5"),
    (("2.0.0a5", "MAJOR", "alpha", 0, None), "2.0.0a6"),
    #
    (("2.0.0b0", "MINOR", "alpha", 0, None), "2.0.0b1"),
    (("2.0.0b0", "PATCH", "alpha", 0, None), "2.0.0b1"),
    #
    (("1.0.1a0", "PATCH", None, 0, None), "1.0.1"),
    (("1.0.1a0", "MINOR", None, 0, None), "1.1.0"),
    (("1.0.1a0", "MAJOR", None, 0, None), "2.0.0"),
    #
    (("1.1.0a0", "PATCH", None, 0, None), "1.1.0"),
    (("1.1.0a0", "MINOR", None, 0, None), "1.1.0"),
    (("1.1.0a0", "MAJOR", None, 0, None), "2.0.0"),
    #
    (("2.0.0a0", "MINOR", None, 0, None), "2.0.0"),
    (("2.0.0a0", "MAJOR", None, 0, None), "2.0.0"),
    (("2.0.0a0", "PATCH", None, 0, None), "2.0.0"),
    #
    (("3.0.0a1", None, None, 0, None), "3.0.0"),
    (("3.0.0b1", None, None, 0, None), "3.0.0"),
    (("3.0.0rc1", None, None, 0, None), "3.0.0"),
    #
    (("3.1.4", None, "alpha", 0, None), "3.1.4a0"),
    (("3.1.4", None, "beta", 0, None), "3.1.4b0"),
    (("3.1.4", None, "rc", 0, None), "3.1.4rc0"),
    #
    (("3.1.4", None, "alpha", 0, None), "3.1.4a0"),
    (("3.1.4a0", "PATCH", "alpha", 0, None), "3.1.4a1"),  # UNEXPECTED!
    (("3.1.4a0", "MINOR", "alpha", 0, None), "3.2.0a0"),
    (("3.1.4a0", "MAJOR", "alpha", 0, None), "4.0.0a0"),
]

excact_cases = [
    (("1.0.0", "PATCH", None, 0, None), "1.0.1"),
    (("1.0.0", "MINOR", None, 0, None), "1.1.0"),
    # with exact_increment=False: "1.0.0b0"
    (("1.0.0a1", "PATCH", "beta", 0, None), "1.0.1b0"),
    # with exact_increment=False: "1.0.0b1"
    (("1.0.0b0", "PATCH", "beta", 0, None), "1.0.1b0"),
    # with exact_increment=False: "1.0.0rc0"
    (("1.0.0b1", "PATCH", "rc", 0, None), "1.0.1rc0"),
    # with exact_increment=False: "1.0.0-rc1"
    (("1.0.0rc0", "PATCH", "rc", 0, None), "1.0.1rc0"),
    # with exact_increment=False: "1.0.0rc1-dev1"
    (("1.0.0rc0", "PATCH", "rc", 0, 1), "1.0.1rc0.dev1"),
    # with exact_increment=False: "1.0.0b0"
    (("1.0.0a1", "MINOR", "beta", 0, None), "1.1.0b0"),
    # with exact_increment=False: "1.0.0b1"
    (("1.0.0b0", "MINOR", "beta", 0, None), "1.1.0b0"),
    # with exact_increment=False: "1.0.0b1"
    (("1.0.0b0", "MINOR", "alpha", 0, None), "1.1.0a0"),
    # with exact_increment=False: "1.0.0rc0"
    (("1.0.0b1", "MINOR", "rc", 0, None), "1.1.0rc0"),
    # with exact_increment=False: "1.0.0rc1"
    (("1.0.0rc0", "MINOR", "rc", 0, None), "1.1.0rc0"),
    # with exact_increment=False: "1.0.0rc1-dev1"
    (("1.0.0rc0", "MINOR", "rc", 0, 1), "1.1.0rc0.dev1"),
    # with exact_increment=False: "2.0.0"
    (("2.0.0b0", "MAJOR", None, 0, None), "3.0.0"),
    # with exact_increment=False: "2.0.0"
    (("2.0.0b0", "MINOR", None, 0, None), "2.1.0"),
    # with exact_increment=False: "2.0.0"
    (("2.0.0b0", "PATCH", None, 0, None), "2.0.1"),
    # same with exact_increment=False
    (("2.0.0b0", "MAJOR", "alpha", 0, None), "3.0.0a0"),
    # with exact_increment=False: "2.0.0b1"
    (("2.0.0b0", "MINOR", "alpha", 0, None), "2.1.0a0"),
    # with exact_increment=False: "2.0.0b1"
    (("2.0.0b0", "PATCH", "alpha", 0, None), "2.0.1a0"),
]


@pytest.mark.parametrize(
    "test_input,expected",
    itertools.chain(
        tdd_cases,
        weird_cases,
        simple_flow,
        linear_prerelease_cases,
        prerelease_cases,
    ),
)
def test_bump_pep440_version(test_input, expected):
    current_version = test_input[0]
    increment = test_input[1]
    prerelease = test_input[2]
    prerelease_offset = test_input[3]
    devrelease = test_input[4]
    assert (
        str(
            Pep440(current_version).bump(
                increment=increment,
                prerelease=prerelease,
                prerelease_offset=prerelease_offset,
                devrelease=devrelease,
            )
        )
        == expected
    )


@pytest.mark.parametrize("test_input, expected", excact_cases)
def test_bump_pep440_version_force(test_input, expected):
    current_version = test_input[0]
    increment = test_input[1]
    prerelease = test_input[2]
    prerelease_offset = test_input[3]
    devrelease = test_input[4]
    assert (
        str(
            Pep440(current_version).bump(
                increment=increment,
                prerelease=prerelease,
                prerelease_offset=prerelease_offset,
                devrelease=devrelease,
                exact_increment=True,
            )
        )
        == expected
    )


@pytest.mark.parametrize("test_input,expected", local_versions)
def test_bump_pep440_version_local(test_input, expected):
    current_version = test_input[0]
    increment = test_input[1]
    prerelease = test_input[2]
    prerelease_offset = test_input[3]
    devrelease = test_input[4]
    is_local_version = True
    assert (
        str(
            Pep440(current_version).bump(
                increment=increment,
                prerelease=prerelease,
                prerelease_offset=prerelease_offset,
                devrelease=devrelease,
                is_local_version=is_local_version,
            )
        )
        == expected
    )


def test_pep440_scheme_property():
    version = Pep440("0.0.1")
    assert version.scheme is Pep440


def test_pep440_implement_version_protocol():
    assert isinstance(Pep440("0.0.1"), VersionProtocol)


def test_pep440_sortable():
    test_input = [x[0][0] for x in simple_flow]
    test_input.extend([x[1] for x in simple_flow])
    # randomize
    random_input = [Pep440(x) for x in random.sample(test_input, len(test_input))]
    assert len(random_input) == len(test_input)
    sorted_result = [str(x) for x in sorted(random_input)]
    assert sorted_result == [
        "0.1.0",
        "0.1.0",
        "0.1.1.dev1",
        "0.1.1",
        "0.1.1",
        "0.2.0",
        "0.2.0",
        "0.2.0",
        "0.3.0.dev1",
        "0.3.0",
        "0.3.0",
        "0.3.0",
        "0.3.0",
        "0.3.1a0",
        "0.3.1a0",
        "0.3.1a0",
        "0.3.1a0",
        "0.3.1a1",
        "0.3.1a1",
        "0.3.1a1",
        "0.3.1",
        "0.3.1",
        "0.3.1",
        "0.3.2",
        "0.4.2",
        "1.0.0a0",
        "1.0.0a0",
        "1.0.0a1",
        "1.0.0a1",
        "1.0.0a1",
        "1.0.0a1",
        "1.0.0a2.dev0",
        "1.0.0a2.dev0",
        "1.0.0a2.dev1",
        "1.0.0a2",
        "1.0.0a3.dev0",
        "1.0.0a3.dev0",
        "1.0.0a3.dev1",
        "1.0.0b0",
        "1.0.0b0",
        "1.0.0b0",
        "1.0.0b1",
        "1.0.0b1",
        "1.0.0rc0",
        "1.0.0rc0",
        "1.0.0rc0",
        "1.0.0rc0",
        "1.0.0rc1.dev1",
        "1.0.0rc1",
        "1.0.0",
        "1.0.0",
        "1.0.1",
        "1.0.1",
        "1.0.2",
        "1.0.2",
        "1.1.0",
        "1.1.0",
        "1.2.0",
        "1.2.0",
        "1.2.1",
        "1.2.1",
        "2.0.0",
    ]