#!/usr/bin/env python
#
# 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 sys
import os
import pathlib
import shutil
import subprocess
import tempfile
import zipfile
import buildconfig


def main():
    if not buildconfig.substs.get("MOZ_CODE_COVERAGE") or not buildconfig.substs.get(
        "MOZ_RUST_TESTS"
    ):
        return

    assert (
        "GRCOV_PATH" in os.environ
    ), "The environment variable GRCOV_PATH should contain a path to grcov"
    grcov_path = os.environ["GRCOV_PATH"]
    assert os.path.exists(grcov_path), "grcov should exist"

    grcov_command = [
        grcov_path,
        "-t",
        "lcov",
        "-p",
        buildconfig.topsrcdir,
        buildconfig.topobjdir,
    ]

    # We want to ignore system headers in our reports.
    if buildconfig.substs["OS_TARGET"] == "WINNT":
        # We use WINDOWSSDKDIR to find the directory holding the system headers on Windows.
        windows_sdk_dir = None
        config_opts = buildconfig.substs["MOZ_CONFIGURE_OPTIONS"].split(" ")
        for opt in config_opts:
            if opt.startswith("WINDOWSSDKDIR="):
                windows_sdk_dir = opt[len("WINDOWSSDKDIR=") :]
                break

        assert (
            windows_sdk_dir is not None
        ), "WINDOWSSDKDIR should be in MOZ_CONFIGURE_OPTIONS"

        ignore_dir_abs = pathlib.Path(windows_sdk_dir).parent

        # globs passed to grcov must exist and must be relative to the source directory.
        # If it doesn't exist, maybe it has moved and we need to update the paths above.
        # If it is no longer relative to the source directory, we no longer need to ignore it and
        # this code can be removed.
        assert ignore_dir_abs.is_dir(), f"{ignore_dir_abs} is not a directory"
        ignore_dir_rel = ignore_dir_abs.relative_to(buildconfig.topsrcdir)

        grcov_command += [
            "--ignore",
            f"{ignore_dir_rel}*",
        ]

    if buildconfig.substs["OS_TARGET"] == "Linux":
        gcc_dir = os.path.join(os.environ["MOZ_FETCHES_DIR"], "gcc")
        if "LD_LIBRARY_PATH" in os.environ:
            os.environ["LD_LIBRARY_PATH"] = "{}/lib64/:{}".format(
                gcc_dir, os.environ["LD_LIBRARY_PATH"]
            )
        else:
            os.environ["LD_LIBRARY_PATH"] = "{}/lib64/".format(gcc_dir)

        os.environ["PATH"] = "{}/bin/:{}".format(gcc_dir, os.environ["PATH"])

    grcov_output = subprocess.check_output(grcov_command)

    grcov_zip_path = os.path.join(buildconfig.topobjdir, "code-coverage-grcov.zip")
    with zipfile.ZipFile(grcov_zip_path, "a", zipfile.ZIP_DEFLATED) as z:
        z.writestr("grcov_lcov_output.info", grcov_output)


if __name__ == "__main__":
    main()