# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, # You can obtain one at http://mozilla.org/MPL/2.0/.
from __future__ import absolute_import
import logging
import sys
import os
import tempfile

from mach.decorators import CommandArgument, CommandProvider, Command
from mozbuild.base import MachCommandBase, BinaryNotFoundException

requirements = os.path.join(os.path.dirname(__file__), "requirements", "base.txt")


@CommandProvider
class CondprofileCommandProvider(MachCommandBase):
    def _init(self):
        self.activate_virtualenv()
        self.virtualenv_manager.install_pip_requirements(
            requirements, require_hashes=False
        )

    @Command("fetch-condprofile", category="testing")
    @CommandArgument("--target-dir", default=None, help="Target directory")
    @CommandArgument("--platform", default=None, help="Platform")
    @CommandArgument("--scenario", default="full", help="Scenario")  # grab choices
    @CommandArgument("--customization", default="default", help="Customization")  # same
    @CommandArgument("--task-id", default=None, help="Task ID")
    @CommandArgument("--download-cache", action="store_true", default=True)
    @CommandArgument(
        "--repo",
        default="mozilla-central",
        choices=["mozilla-central", "try"],
        help="Repository",
    )
    def fetch(
        self,
        target_dir,
        platform,
        scenario,
        customization,
        task_id,
        download_cache,
        repo,
    ):
        self._init()
        from condprof.client import get_profile
        from condprof.util import get_current_platform

        if platform is None:
            platform = get_current_platform()

        if target_dir is None:
            target_dir = tempfile.mkdtemp()

        get_profile(
            target_dir, platform, scenario, customization, task_id, download_cache, repo
        )

    @Command("run-condprofile", category="testing")
    @CommandArgument("archive", help="Archives Dir", type=str, default=None)
    @CommandArgument("--firefox", help="Firefox Binary", type=str, default=None)
    @CommandArgument("--scenario", help="Scenario to use", type=str, default="all")
    @CommandArgument("--profile", help="Existing profile Dir", type=str, default=None)
    @CommandArgument(
        "--customization", help="Profile customization to use", type=str, default="all"
    )
    @CommandArgument(
        "--visible", help="Don't use headless mode", action="store_true", default=False
    )
    @CommandArgument(
        "--archives-dir", help="Archives local dir", type=str, default="/tmp/archives"
    )
    @CommandArgument(
        "--force-new", help="Create from scratch", action="store_true", default=False
    )
    @CommandArgument(
        "--strict",
        help="Errors out immediatly on a scenario failure",
        action="store_true",
        default=True,
    )
    @CommandArgument(
        "--geckodriver",
        help="Path to the geckodriver binary",
        type=str,
        default=sys.platform.startswith("win") and "geckodriver.exe" or "geckodriver",
    )
    @CommandArgument("--device-name", help="Name of the device", type=str, default=None)
    def run(self, **kw):
        os.environ["MANUAL_MACH_RUN"] = "1"
        self._init()

        if kw["firefox"] is None:
            try:
                kw["firefox"] = self.get_binary_path()
            except BinaryNotFoundException as e:
                self.log(
                    logging.ERROR,
                    "run-condprofile",
                    {"error": str(e)},
                    "ERROR: {error}",
                )
                self.log(logging.INFO, "run-condprofile", {"help": e.help()}, "{help}")
                return 1

        from condprof.runner import run

        run(**kw)