# 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, print_function, unicode_literals

import os
import sys
from argparse import Namespace

from mach.decorators import (
    CommandProvider,
    Command,
)
from mozbuild.base import MachCommandBase

here = os.path.abspath(os.path.dirname(__file__))
parser = None
logger = None


def run_gtest(context, **kwargs):
    from mozlog.commandline import setup_logging

    if not kwargs.get("log"):
        kwargs["log"] = setup_logging("gtest", kwargs, {"mach": sys.stdout})
    global logger
    logger = kwargs["log"]

    args = Namespace(**kwargs)

    import mozinfo

    if mozinfo.info.get("buildapp") == "mobile/android":
        return run_gtest_android(context, args)
    return run_gtest_desktop(context, args)


def run_gtest_desktop(context, args):
    prog = context.firefox_bin
    xre_path = os.path.dirname(context.firefox_bin)
    if sys.platform == "darwin":
        xre_path = os.path.join(xre_path, "Resources")
    utility_path = context.bin_dir
    cwd = os.path.join(context.package_root, "gtest")

    logger.info(
        "mach calling run_gtest with prog=%s xre_path=%s cwd=%s utility_path=%s"
        % (prog, xre_path, cwd, utility_path)
    )
    # The gtest option parser ignores some options normally passed to the mozharness
    # command, so some hacking is required, for now:
    extra_args = [arg for arg in args.args if not arg.startswith("-")]
    if extra_args:
        os.environ["GTEST_FILTER"] = extra_args[0]
        logger.info("GTEST_FILTER=%s" % extra_args[0])

    import rungtests

    tester = rungtests.GTests()
    return tester.run_gtest(
        prog,
        xre_path,
        cwd,
        utility_path=utility_path,
        enable_webrender=args.enable_webrender,
    )


def run_gtest_android(context, args):
    config = context.mozharness_config
    if config:
        args.adb_path = config["exes"]["adb"] % {
            "abs_work_dir": context.mozharness_workdir
        }
    cwd = os.path.join(context.package_root, "gtest")
    libxul_path = os.path.join(cwd, "gtest_bin", "gtest", "libxul.so")

    logger.info(
        "mach calling android run_gtest with package=%s cwd=%s libxul=%s"
        % (args.package, cwd, libxul_path)
    )
    # The remote gtest option parser ignores some options normally passed to the mozharness
    # command, so some hacking is required, for now:
    extra_args = [arg for arg in args.args if not arg.startswith("-")]
    test_filter = extra_args[0] if extra_args else None
    logger.info("test filter=%s" % test_filter)

    import remotegtests

    tester = remotegtests.RemoteGTests()
    return tester.run_gtest(
        cwd,
        args.shuffle,
        test_filter,
        args.package,
        args.adb_path,
        args.device_serial,
        args.remote_test_root,
        libxul_path,
        args.symbols_path,
        args.enable_webrender,
    )


def setup_argument_parser():
    import mozinfo

    mozinfo.find_and_update_from_json(here)
    global parser
    if mozinfo.info.get("buildapp") == "mobile/android":
        import remotegtests

        parser = remotegtests.remoteGtestOptions()
    else:
        import rungtests

        parser = rungtests.gtestOptions()
    return parser


@CommandProvider
class GtestCommands(MachCommandBase):
    @Command(
        "gtest",
        category="testing",
        description="Run the gtest harness.",
        parser=setup_argument_parser,
    )
    def gtest(self, **kwargs):
        self._mach_context.activate_mozharness_venv()
        result = run_gtest(self._mach_context, **kwargs)
        return 0 if result else 1