#!/usr/bin/env node
/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

import { fileURLToPath } from 'node:url';

const commands = {
  prepare: { module: 'prepare.mjs' },
  'real-ax': {
    module: 'real-ax-launcher.mjs',
    options: {
      '--scenario': 'MAKA_CU_AX_MODEL_SCENARIO',
      '--provider': 'MAKA_CU_MODEL_PROVIDER',
    },
  },
  'real-model': {
    module: 'real-model.mjs',
    options: {
      '--scenario': 'MAKA_CU_E2E_SCENARIO',
      '--provider': 'MAKA_CU_PROVIDER',
    },
  },
  'restart-soak': { module: 'process-restart-launcher.mjs' },
  'provider-matrix': { module: 'provider-matrix.mjs' },
};

const [commandName, ...commandArgs] = process.argv.slice(2);
if (!commandName || commandName === 'help' || commandName === '--help' || commandName === '-h') {
  process.stdout.write(
    `Usage: node scripts/computer-use.mjs <command> [options]\n\nCommands:\n${Object.keys(commands)
      .map((name) => `  ${name}`)
      .join('\n')}\n`,
  );
  process.exit(0);
}

const command = commands[commandName];
if (!command) {
  process.stderr.write(`Unknown Computer Use command: ${commandName}\n`);
  process.exit(1);
}

const forwardedArgs = [];
for (let index = 0; index < commandArgs.length; index += 1) {
  const argument = commandArgs[index];
  const [optionName, inlineValue] = argument.split('=', 2);
  const environmentName = command.options?.[optionName];
  if (!environmentName) {
    forwardedArgs.push(argument);
    continue;
  }
  const value = inlineValue ?? commandArgs[++index];
  if (!value || value.startsWith('--')) {
    throw new Error(`${optionName} requires a value`);
  }
  process.env[environmentName] = value;
}

process.argv = [
  process.argv[0],
  fileURLToPath(new URL(`./computer-use/${command.module}`, import.meta.url)),
  ...forwardedArgs,
];
await import(`./computer-use/${command.module}`);