# 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 subprocess

import buildconfig
import mozpack.path as mozpath
import six
import toml


# Try to read the package name or otherwise assume same name as the crate path.
def _get_crate_name(crate_path):
    try:
        with open(mozpath.join(crate_path, "Cargo.toml"), encoding="utf-8") as f:
            return toml.load(f)["package"]["name"]
    except Exception:
        return mozpath.basename(crate_path)


CARGO_LOCK = mozpath.join(buildconfig.topsrcdir, "Cargo.lock")
CARGO_TOML = mozpath.join(buildconfig.topsrcdir, "Cargo.toml")


def _run_process(args):
    env = os.environ.copy()
    env["CARGO"] = str(buildconfig.substs["CARGO"])
    env["RUSTC"] = str(buildconfig.substs["RUSTC"])

    p = subprocess.Popen(args, env=env, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

    stdout, stderr = p.communicate()
    stdout = six.ensure_text(stdout)
    stderr = six.ensure_text(stderr)
    if p.returncode != 0:
        print(stdout)
        print(stderr)
    return (stdout, p.returncode)


def generate_metadata(output, cargo_config):
    stdout, returncode = _run_process(
        [
            buildconfig.substs["CARGO"],
            "metadata",
            "--frozen",
            "--all-features",
            "--format-version",
            "1",
            "--manifest-path",
            CARGO_TOML,
        ]
    )

    if returncode != 0:
        return returncode

    output.write(stdout)

    # This is not quite accurate, but cbindgen only cares about a subset of the
    # data which, when changed, causes these files to change.
    return set([CARGO_LOCK, CARGO_TOML])


def generate(output, metadata_path, cbindgen_crate_path, *in_tree_dependencies):
    stdout, returncode = _run_process(
        [
            buildconfig.substs["CBINDGEN"],
            buildconfig.topsrcdir,
            "--lockfile",
            CARGO_LOCK,
            "--crate",
            _get_crate_name(cbindgen_crate_path),
            "--metadata",
            metadata_path,
            "--cpp-compat",
        ]
    )

    if returncode != 0:
        return returncode

    output.write(stdout)

    deps = set()
    deps.add(CARGO_LOCK)
    deps.add(mozpath.join(cbindgen_crate_path, "cbindgen.toml"))
    for directory in in_tree_dependencies + (cbindgen_crate_path,):
        for path, dirs, files in os.walk(directory):
            for file in files:
                if os.path.splitext(file)[1] == ".rs":
                    deps.add(mozpath.join(path, file))

    return deps
