# Copyright 2022 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from PB.recipes.crosvm.build_chromeos import BuildChromeOsProperties

PYTHON_VERSION_COMPATIBILITY = "PY3"

DEPS = [
    "crosvm",
    "depot_tools/depot_tools",
    "recipe_engine/buildbucket",
    "recipe_engine/context",
    "recipe_engine/file",
    "recipe_engine/path",
    "recipe_engine/properties",
    "recipe_engine/step",
]

PROPERTIES = BuildChromeOsProperties

PACKAGE_LIST = [
    "chromeos-base/crosvm",
]


def SetupSource(api, workspace):
    repo = api.depot_tools.repo_resource("repo")
    gitilies = api.buildbucket.build.input.gitiles_commit
    upstream_url = "https://chromium.googlesource.com/crosvm/crosvm"
    revision = gitilies.id or "HEAD"

    # Initialize git user name to make repo happy.
    api.crosvm.prepare_git()

    # Init and sync the ChromeOS checkout
    api.step(
        "Init repo",
        [
            repo,
            "init",
            "--manifest-url=https://chromium.googlesource.com/chromiumos/manifest",
            "--manifest-branch=stable",
            "--depth=1",
            "--current-branch",
            "--groups=minilayout,crosvm",
        ],
    )
    api.step(
        "Sync repo",
        [
            repo,
            "sync",
            "--current-branch",
        ],
    )

    # Overwrite crosvm with the upstream revision we need to test
    with api.context(cwd=workspace.join("src/platform/crosvm")):
        api.step("Fetch upstream crosvm", ["git", "fetch", upstream_url])
        api.step("Checkout upstream revision", ["git", "checkout", revision])


def PrepareBuild(api):
    # Uprev crosvm related ebuild files
    api.step(
        "Uprev packages",
        [
            "./chromite/scripts/cros_uprev",
            "--package=%s" % ",".join(PACKAGE_LIST),
            "--overlay-type=public",
        ],
    )
    # Create chroot as a separate step to document the runtime
    cros_sdk = api.depot_tools.repo_resource("cros_sdk")
    api.step("Create SDK chroot", [cros_sdk, "--create"])


def BuildAndTest(api, board):
    cros_sdk = api.depot_tools.repo_resource("cros_sdk")
    # TODO: We currently build crosvm twice. Once with build_packages, once to run tests.
    api.step(
        "Build packages",
        [cros_sdk, "build_packages", "--board=%s" % board, "implicit-system"] + PACKAGE_LIST,
    )
    api.step(
        "Run unit tests",
        [cros_sdk, "cros_run_unit_tests", "--board=%s" % board, "--packages", "implicit-system"]
        + PACKAGE_LIST,
    )


def CleanUp(api):
    cros_sdk = api.depot_tools.repo_resource("cros_sdk")
    api.step("Deleting SDK chroot", [cros_sdk, "--delete"])


def RunSteps(api, properties):
    workspace = api.path["cleanup"].join("workspace")
    api.file.ensure_directory("Ensure workspace exists", workspace)

    with api.context(cwd=workspace, env={"DEPOT_TOOLS_UPDATE": "0"}):
        try:
            SetupSource(api, workspace)
            PrepareBuild(api)
            BuildAndTest(api, properties.board or "amd64-generic")
        finally:
            CleanUp(api)


def GenTests(api):
    yield api.test("build_default")