#!/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("--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)

  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())