# 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/.

import os
import unittest
from collections import defaultdict
from shutil import rmtree
from tempfile import mkdtemp

import mozpack.path as mozpath
from mach.logging import LoggingManager

from mozbuild.backend.configenvironment import ConfigEnvironment
from mozbuild.frontend.emitter import TreeMetadataEmitter
from mozbuild.frontend.reader import BuildReader

log_manager = LoggingManager()
log_manager.add_terminal_logging()


test_data_path = mozpath.abspath(mozpath.dirname(__file__))
test_data_path = mozpath.join(test_data_path, "data")


CONFIGS = defaultdict(
    lambda: {
        "defines": {},
        "substs": {"OS_TARGET": "WINNT"},
    },
    {
        "binary-components": {
            "defines": {},
            "substs": {
                "LIB_PREFIX": "lib",
                "LIB_SUFFIX": "a",
                "COMPILE_ENVIRONMENT": "1",
            },
        },
        "database": {
            "defines": {},
            "substs": {
                "CC": ["clang"],
                "CXX": ["clang++"],
                "LIB_PREFIX": "lib",
                "LIB_SUFFIX": "a",
            },
        },
        "database-compiler-wrapper": {
            "defines": {},
            "substs": {
                "CC": ["/usr/bin/kache", "clang"],
                "CXX": ["/usr/bin/kache", "clang++"],
                "COMPILER_WRAPPER": ["/usr/bin/kache"],
                "LIB_PREFIX": "lib",
                "LIB_SUFFIX": "a",
            },
        },
        "database-ccache": {
            "defines": {},
            "substs": {
                "CC": ["/usr/bin/ccache", "clang"],
                "CXX": ["/usr/bin/ccache", "clang++"],
                "COMPILER_WRAPPER": ["/usr/bin/ccache"],
                "LIB_PREFIX": "lib",
                "LIB_SUFFIX": "a",
            },
        },
        "rust-library": {
            "defines": {},
            "substs": {
                "COMPILE_ENVIRONMENT": "1",
                "RUST_TARGET": "x86_64-unknown-linux-gnu",
                "LIB_PREFIX": "lib",
                "LIB_SUFFIX": "a",
            },
        },
        "host-rust-library": {
            "defines": {},
            "substs": {
                "COMPILE_ENVIRONMENT": "1",
                "RUST_HOST_TARGET": "x86_64-unknown-linux-gnu",
                "RUST_TARGET": "armv7-linux-androideabi",
                "LIB_PREFIX": "lib",
                "LIB_SUFFIX": "a",
            },
        },
        "host-rust-library-features": {
            "defines": {},
            "substs": {
                "COMPILE_ENVIRONMENT": "1",
                "RUST_HOST_TARGET": "x86_64-unknown-linux-gnu",
                "RUST_TARGET": "armv7-linux-androideabi",
                "LIB_PREFIX": "lib",
                "LIB_SUFFIX": "a",
            },
        },
        "rust-library-features": {
            "defines": {},
            "substs": {
                "COMPILE_ENVIRONMENT": "1",
                "RUST_TARGET": "x86_64-unknown-linux-gnu",
                "LIB_PREFIX": "lib",
                "LIB_SUFFIX": "a",
            },
        },
        "rust-program-features": {
            "defines": {},
            "substs": {
                "COMPILE_ENVIRONMENT": "1",
                "RUST_TARGET": "i686-pc-windows-msvc",
                "BIN_SUFFIX": ".exe",
            },
        },
        "host-rust-program-features": {
            "defines": {},
            "substs": {
                "COMPILE_ENVIRONMENT": "1",
                "RUST_TARGET": "i686-pc-windows-msvc",
                "RUST_HOST_TARGET": "i686-pc-windows-msvc",
                "BIN_SUFFIX": ".exe",
                "HOST_BIN_SUFFIX": ".exe",
            },
        },
        "host-rust-program-output-category": {
            "defines": {},
            "substs": {
                "COMPILE_ENVIRONMENT": "1",
                "RUST_TARGET": "i686-pc-windows-msvc",
                "RUST_HOST_TARGET": "i686-pc-windows-msvc",
                "BIN_SUFFIX": ".exe",
                "HOST_BIN_SUFFIX": ".exe",
            },
        },
        "rust-programs": {
            "defines": {},
            "substs": {
                "COMPILE_ENVIRONMENT": "1",
                "RUST_TARGET": "i686-pc-windows-msvc",
                "RUST_HOST_TARGET": "i686-pc-windows-msvc",
                "BIN_SUFFIX": ".exe",
                "HOST_BIN_SUFFIX": ".exe",
            },
        },
        "test-support-binaries-tracked": {
            "defines": {},
            "substs": {
                "COMPILE_ENVIRONMENT": "1",
                "LIB_SUFFIX": "dll",
                "BIN_SUFFIX": ".exe",
            },
        },
        "sources": {
            "defines": {},
            "substs": {
                "LIB_PREFIX": "lib",
                "LIB_SUFFIX": "a",
            },
        },
        "stub0": {
            "defines": {
                "MOZ_TRUE_1": "1",
                "MOZ_TRUE_2": "1",
            },
            "substs": {
                "MOZ_FOO": "foo",
                "MOZ_BAR": "bar",
            },
        },
        "substitute_config_files": {
            "defines": {},
            "substs": {
                "MOZ_FOO": "foo",
                "MOZ_BAR": "bar",
            },
        },
        "test_config": {
            "defines": {
                "foo": "baz qux",
                "baz": 1,
            },
            "substs": {
                "foo": "bar baz",
            },
        },
        "visual-studio": {
            "defines": {},
            "substs": {
                "MOZ_APP_NAME": "my_app",
            },
        },
        "prog-lib-c-only": {
            "defines": {},
            "substs": {
                "COMPILE_ENVIRONMENT": "1",
                "LIB_SUFFIX": ".a",
                "BIN_SUFFIX": "",
            },
        },
        "gn-processor": {
            "defines": {},
            "substs": {
                "BUILD_BACKENDS": [
                    "GnMozbuildWriter",
                    "RecursiveMake",
                ],
                "COMPILE_ENVIRONMENT": "1",
                "STL_FLAGS": [],
                "RUST_TARGET": "x86_64-unknown-linux-gnu",
                "LIB_PREFIX": "lib",
                "LIB_SUFFIX": "a",
                "OS_TARGET": "Darwin",
            },
        },
        "ipdl_sources": {
            "defines": {},
            "substs": {
                "COMPILE_ENVIRONMENT": "1",
                "LIB_SUFFIX": ".a",
                "BIN_SUFFIX": "",
            },
        },
        "program-paths": {
            "defines": {},
            "substs": {
                "COMPILE_ENVIRONMENT": "1",
                "BIN_SUFFIX": ".prog",
            },
        },
        "extra-link-deps": {
            "defines": {},
            "substs": {
                "COMPILE_ENVIRONMENT": "1",
                "BIN_SUFFIX": "",
            },
        },
        "shared-lib-paths": {
            "defines": {},
            "substs": {
                "COMPILE_ENVIRONMENT": "1",
                "LIB_SUFFIX": "a",
                "DLL_PREFIX": "lib",
                "DLL_SUFFIX": ".so",
            },
        },
        "shared-library-output-category": {
            "defines": {},
            "substs": {
                "COMPILE_ENVIRONMENT": "1",
                "LIB_SUFFIX": "a",
                "DLL_PREFIX": "lib",
                "DLL_SUFFIX": ".so",
            },
        },
        "linkage": {
            "defines": {},
            "substs": {
                "CC_TYPE": "clang",
                "COMPILE_ENVIRONMENT": "1",
                "LIB_SUFFIX": "a",
                "BIN_SUFFIX": ".exe",
                "DLL_SUFFIX": ".so",
                "OBJ_SUFFIX": "o",
                "EXPAND_LIBS_LIST_STYLE": "list",
            },
        },
        "rust-library-archive-dep": {
            "defines": {},
            "substs": {
                "COMPILE_ENVIRONMENT": "1",
                "RUST_TARGET": "x86_64-unknown-linux-gnu",
                "LIB_PREFIX": "lib",
                "LIB_SUFFIX": "a",
                "OBJ_SUFFIX": "o",
            },
        },
    },
)


class BackendTester(unittest.TestCase):
    def setUp(self):
        self._old_env = dict(os.environ)
        os.environ.pop("MOZ_OBJDIR", None)

    def tearDown(self):
        os.environ.clear()
        os.environ.update(self._old_env)

    def _get_environment(self, name, srcdir_name=None):
        """Obtain a new instance of a ConfigEnvironment for a known profile.

        A new temporary object directory is created for the environment. The
        environment is cleaned up automatically when the test finishes.

        `srcdir_name` selects the test data directory, defaulting to `name`. It
        can differ from `name` to run a different config profile against an
        existing data directory.
        """
        config = CONFIGS[name]
        config["substs"]["MOZ_UI_LOCALE"] = "en-US"

        srcdir = mozpath.join(test_data_path, srcdir_name or name)
        config["substs"]["top_srcdir"] = srcdir

        # Create the objdir in the srcdir to ensure that they share the
        # same drive on Windows.
        objdir = mkdtemp(dir=srcdir)
        self.addCleanup(rmtree, objdir)

        return ConfigEnvironment(srcdir, objdir, **config)

    def _emit(self, name, env=None):
        env = env or self._get_environment(name)
        reader = BuildReader(env)
        emitter = TreeMetadataEmitter(env)

        return env, emitter.emit(reader.read_topsrcdir())

    def _consume(self, name, cls, env=None):
        env, objs = self._emit(name, env=env)
        backend = cls(env)
        backend.consume(objs)

        return env

    def _tree_paths(self, topdir, filename):
        for dirpath, dirnames, filenames in os.walk(topdir):
            for f in filenames:
                if f == filename:
                    yield mozpath.relpath(mozpath.join(dirpath, f), topdir)

    def _mozbuild_paths(self, env):
        return self._tree_paths(env.topsrcdir, "moz.build")

    def _makefile_in_paths(self, env):
        return self._tree_paths(env.topsrcdir, "Makefile.in")


__all__ = ["BackendTester"]
