import pytest
import mock
import platform
import json

from voluptuous import Schema, Any

from dvc import analytics
from dvc.cli import parse_args
from dvc.utils.compat import str, builtin_str


string = Any(str, builtin_str)


@pytest.fixture
def tmp_global_config(tmp_path):
    """
    Fixture to prevent modifying the actual global config
    """
    with mock.patch(
        "dvc.config.Config.get_global_config_dir", return_value=str(tmp_path)
    ):
        yield


@mock.patch("dvc.daemon._spawn")
@mock.patch("json.dump")
def test_collect_and_send_report(mock_json, mock_daemon, tmp_global_config):
    analytics.collect_and_send_report()
    report = mock_json.call_args[0][0]

    with pytest.raises(KeyError):
        report["cmd_class"]

    with pytest.raises(KeyError):
        report["cmd_return_code"]

    args = parse_args(["add", "foo"])
    return_code = 0

    analytics.collect_and_send_report(args, return_code)
    report = mock_json.call_args[0][0]

    assert report["cmd_class"] == "CmdAdd"
    assert report["cmd_return_code"] == return_code

    assert mock_daemon.call_count == 2


def test_runtime_info(tmp_global_config):
    schema = Schema(
        {
            "dvc_version": string,
            "is_binary": bool,
            "scm_class": Any("Git", None),
            "user_id": string,
            "system_info": dict,
        },
        required=True,
    )

    assert schema(analytics._runtime_info())


@mock.patch("requests.post")
def test_send(mock_post, tmp_path):
    url = "https://analytics.dvc.org"
    report = {"name": "dummy report"}
    fname = str(tmp_path / "report")

    with open(fname, "w") as fobj:
        json.dump(report, fobj)

    analytics.send(fname)
    assert mock_post.called
    assert mock_post.call_args.args[0] == url


@pytest.mark.parametrize(
    "config, result",
    [
        ({}, True),
        ({"analytics": "false"}, False),
        ({"analytics": "true"}, True),
        ({"unknown": "broken"}, True),
        ({"analytics": "false", "unknown": "broken"}, False),
    ],
)
def test_is_enabled(dvc_repo, config, result, monkeypatch, tmp_global_config):
    configobj = dvc_repo.config._repo_config
    configobj["core"] = config
    configobj.write()

    # reset DVC_TEST env var, which affects `is_enabled()`
    monkeypatch.delenv("DVC_TEST")

    assert result == analytics.is_enabled()


def test_system_info():
    schema = Schema({"os": Any("windows", "mac", "linux")}, required=True)

    system = platform.system()

    if system == "Windows":
        schema = schema.extend(
            {
                "windows_version_build": int,
                "windows_version_major": int,
                "windows_version_minor": int,
                "windows_version_service_pack": string,
            }
        )

    if system == "Darwin":
        schema = schema.extend({"mac_version": string})

    if system == "Linux":
        schema = schema.extend(
            {
                "linux_distro": string,
                "linux_distro_like": string,
                "linux_distro_version": string,
            }
        )

    assert schema(analytics._system_info())


def test_find_or_create_user_id(tmp_global_config):
    created = analytics._find_or_create_user_id()
    found = analytics._find_or_create_user_id()

    assert created == found