#!/usr/bin/env python3
# Copyright 2023 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""This script is used to configure siso."""

import argparse
import os
import shutil
import sys

THIS_DIR = os.path.abspath(os.path.dirname(__file__))

SISO_PROJECT_CFG = "SISO_PROJECT"
SISO_ENV = os.path.join(THIS_DIR, ".sisoenv")

_BACKEND_STAR = os.path.join(THIS_DIR, "backend_config", "backend.star")

_GOOGLE_STAR = os.path.join(THIS_DIR, "backend_config", "google.star")
_KNOWN_GOOGLE_PROJECTS = (
    'goma-foundry-experiments',
    'rbe-chrome-official',
    'rbe-chrome-trusted',
    'rbe-chrome-untrusted',
    'rbe-chromium-trusted',
    'rbe-chromium-trusted-test',
    'rbe-chromium-untrusted',
    'rbe-chromium-untrusted-test',
    'rbe-webrtc-developer',
    'rbe-webrtc-trusted',
    'rbe-webrtc-untrusted',
)

_GOOGLE_CROS_STAR = os.path.join(
    THIS_DIR, "backend_config", "google_cros_chroot.star"
)


def ReadConfig():
    entries = {}
    if not os.path.isfile(SISO_ENV):
        print('The .sisoenv file has not been generated yet')
        return entries
    with open(SISO_ENV, 'r') as f:
        for line in f:
            parts = line.strip().split('=')
            if len(parts) > 1:
                entries[parts[0].strip()] = parts[1].strip()
        return entries


def main():
    parser = argparse.ArgumentParser(description="configure siso")
    parser.add_argument(
        "--rbe_instance",
        help="RBE instance to use for Siso."
        + "Ignored if --reapi_address is set for non-RBE address.",
    )
    parser.add_argument(
        "--reapi_instance", default="", help="REAPI instance to use for Siso."
    )
    parser.add_argument(
        "--reapi_address",
        help="REAPI address to use for "
        + "Siso. If set for non-RBE address, rbe_instance is "
        + "ignored.",
    )
    parser.add_argument(
        "--reapi_backend_config_path",
        help="REAPI backend config path. relative to "
        + "backend_config dir or absolute path.",
    )
    parser.add_argument(
        "--credential-helper",
        help="The credential helper to use for authenticating. "
        + "May be a binary on the PATH or an absolute path.",
    )
    parser.add_argument(
        "--get-siso-project",
        help="Print the currently configured siso project to stdout",
        action="store_true",
    )
    args = parser.parse_args()

    if args.get_siso_project:
        config = ReadConfig()
        if not SISO_PROJECT_CFG in config:
            return 1
        print(config[SISO_PROJECT_CFG])
        return 0

    project = None
    reapi_instance = args.reapi_instance
    reapi_address = args.reapi_address
    if (
        not reapi_address
        or reapi_address.endswith("remotebuildexecution.googleapis.com:443")
        and args.rbe_instance
    ):
        elems = args.rbe_instance.split("/")
        if len(elems) == 4 and elems[0] == "projects":
            project = elems[1]
            reapi_instance = elems[-1]

    with open(SISO_ENV, "w") as f:
        if project:
            f.write("%s=%s\n" % (SISO_PROJECT_CFG, project))
        f.write("SISO_REAPI_INSTANCE=%s\n" % reapi_instance)
        if reapi_address:
            f.write("SISO_REAPI_ADDRESS=%s\n" % reapi_address)
        if args.credential_helper:
            f.write("SISO_CREDENTIAL_HELPER=%s\n" % args.credential_helper)

    reapi_backend_config_path = args.reapi_backend_config_path
    if project and not reapi_backend_config_path:
        if project in _KNOWN_GOOGLE_PROJECTS:
            reapi_backend_config_path = _GOOGLE_STAR
        elif project.startswith('chromeos-') and project.endswith('-bot'):
            reapi_backend_config_path = _GOOGLE_CROS_STAR

    if reapi_backend_config_path:
        if not os.path.isabs(reapi_backend_config_path):
            reapi_backend_config_path = os.path.join(
                THIS_DIR, "backend_config", reapi_backend_config_path
            )
        if os.path.exists(_BACKEND_STAR):
            os.remove(_BACKEND_STAR)
        shutil.copy2(reapi_backend_config_path, _BACKEND_STAR)

    if not os.path.exists(_BACKEND_STAR):
        print(
            (
                'Need to provide {} (by reapi_backend_config_path custom_var) '
                + 'for your backend {} instance={}'
            ).format(_BACKEND_STAR, reapi_address, reapi_instance),
            file=sys.stderr,
        )
        return 1
    return 0


if __name__ == "__main__":
    sys.exit(main())