import os
import shutil

from pipenv.utils import temp_environ

import pytest


@pytest.mark.run
@pytest.mark.uninstall
@pytest.mark.install
def test_uninstall(PipenvInstance, pypi):
    with PipenvInstance(pypi=pypi) as p:
        c = p.pipenv("install requests")
        assert c.return_code == 0
        assert "requests" in p.pipfile["packages"]
        assert "requests" in p.lockfile["default"]
        assert "chardet" in p.lockfile["default"]
        assert "idna" in p.lockfile["default"]
        assert "urllib3" in p.lockfile["default"]
        assert "certifi" in p.lockfile["default"]

        c = p.pipenv("uninstall requests")
        assert c.return_code == 0
        assert "requests" not in p.pipfile["dev-packages"]
        assert "requests" not in p.lockfile["develop"]
        assert "chardet" not in p.lockfile["develop"]
        assert "idna" not in p.lockfile["develop"]
        assert "urllib3" not in p.lockfile["develop"]
        assert "certifi" not in p.lockfile["develop"]

        c = p.pipenv("run python -m requests.help")
        assert c.return_code > 0


@pytest.mark.run
@pytest.mark.uninstall
@pytest.mark.install
def test_mirror_uninstall(PipenvInstance, pypi):
    with temp_environ(), PipenvInstance(chdir=True) as p:

        mirror_url = os.environ.pop(
            "PIPENV_TEST_INDEX", "https://pypi.python.org/simple"
        )
        assert "pypi.org" not in mirror_url

        c = p.pipenv("install requests --pypi-mirror {0}".format(mirror_url))
        assert c.return_code == 0
        assert "requests" in p.pipfile["packages"]
        assert "requests" in p.lockfile["default"]
        assert "chardet" in p.lockfile["default"]
        assert "idna" in p.lockfile["default"]
        assert "urllib3" in p.lockfile["default"]
        assert "certifi" in p.lockfile["default"]
        # Ensure the --pypi-mirror parameter hasn't altered the Pipfile or Pipfile.lock sources
        assert len(p.pipfile["source"]) == 1
        assert len(p.lockfile["_meta"]["sources"]) == 1
        assert "https://pypi.org/simple" == p.pipfile["source"][0]["url"]
        assert "https://pypi.org/simple" == p.lockfile["_meta"]["sources"][0]["url"]

        c = p.pipenv("uninstall requests --pypi-mirror {0}".format(mirror_url))
        assert c.return_code == 0
        assert "requests" not in p.pipfile["dev-packages"]
        assert "requests" not in p.lockfile["develop"]
        assert "chardet" not in p.lockfile["develop"]
        assert "idna" not in p.lockfile["develop"]
        assert "urllib3" not in p.lockfile["develop"]
        assert "certifi" not in p.lockfile["develop"]
        # Ensure the --pypi-mirror parameter hasn't altered the Pipfile or Pipfile.lock sources
        assert len(p.pipfile["source"]) == 1
        assert len(p.lockfile["_meta"]["sources"]) == 1
        assert "https://pypi.org/simple" == p.pipfile["source"][0]["url"]
        assert "https://pypi.org/simple" == p.lockfile["_meta"]["sources"][0]["url"]

        c = p.pipenv("run python -m requests.help")
        assert c.return_code > 0


@pytest.mark.files
@pytest.mark.uninstall
@pytest.mark.install
def test_uninstall_all_local_files(PipenvInstance, testsroot):
    file_name = "requests-2.19.1.tar.gz"
    # Not sure where travis/appveyor run tests from
    source_path = os.path.abspath(os.path.join(testsroot, "test_artifacts", file_name))

    with PipenvInstance(chdir=True) as p:
        shutil.copy(source_path, os.path.join(p.path, file_name))
        os.mkdir(os.path.join(p.path, "requests"))
        c = p.pipenv("install {}".format(file_name))
        assert c.return_code == 0
        c = p.pipenv("uninstall --all")
        assert c.return_code == 0
        assert "requests" in c.out
        # Uninstall --all is not supposed to remove things from the pipfile
        # Note that it didn't before, but that instead local filenames showed as hashes
        assert "requests" in p.pipfile["packages"]


@pytest.mark.run
@pytest.mark.uninstall
@pytest.mark.install
def test_uninstall_all_dev(PipenvInstance, pypi):
    with PipenvInstance(pypi=pypi) as p:
        c = p.pipenv("install --dev requests six")
        assert c.return_code == 0

        c = p.pipenv("install pytz")
        assert c.return_code == 0

        assert "pytz" in p.pipfile["packages"]
        assert "requests" in p.pipfile["dev-packages"]
        assert "six" in p.pipfile["dev-packages"]
        assert "pytz" in p.lockfile["default"]
        assert "requests" in p.lockfile["develop"]
        assert "six" in p.lockfile["develop"]

        c = p.pipenv("uninstall --all-dev")
        assert c.return_code == 0
        assert "requests" not in p.pipfile["dev-packages"]
        assert "six" not in p.pipfile["dev-packages"]
        assert "requests" not in p.lockfile["develop"]
        assert "six" not in p.lockfile["develop"]
        assert "pytz" in p.pipfile["packages"]
        assert "pytz" in p.lockfile["default"]

        c = p.pipenv("run python -m requests.help")
        assert c.return_code > 0

        c = p.pipenv('run python -c "import pytz"')
        assert c.return_code == 0


@pytest.mark.uninstall
@pytest.mark.run
def test_normalize_name_uninstall(PipenvInstance, pypi):
    with PipenvInstance(pypi=pypi) as p:
        with open(p.pipfile_path, "w") as f:
            contents = """
# Pre comment
[packages]
Requests = "*"
python_DateUtil = "*"   # Inline comment
"""
            f.write(contents)

        c = p.pipenv("install")
        assert c.return_code == 0

        c = p.pipenv("uninstall python_dateutil")
        assert "Requests" in p.pipfile["packages"]
        assert "python_DateUtil" not in p.pipfile["packages"]
        with open(p.pipfile_path) as f:
            contents = f.read()
            assert "# Pre comment" in contents
            assert "# Inline comment" in contents