/**
 * @license
 * Copyright 2025 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */

import yargs from 'yargs/yargs';
import { hideBin } from 'yargs/helpers';
import process from 'node:process';
import {
  Config,
  loadEnvironment,
  createServerConfig,
} from '@gemini-code/server';
import { Settings } from './settings.js';

const DEFAULT_GEMINI_MODEL = 'gemini-2.5-flash-preview-04-17';

// Keep CLI-specific argument parsing
interface CliArgs {
  model: string | undefined;
  sandbox: boolean | string | undefined;
  debug_mode: boolean | undefined;
  question: string | undefined;
  full_context: boolean | undefined;
}

async function parseArguments(): Promise<CliArgs> {
  const argv = await yargs(hideBin(process.argv))
    .option('model', {
      alias: 'm',
      type: 'string',
      description: `The Gemini model to use. Defaults to ${DEFAULT_GEMINI_MODEL}.`,
      default: process.env.GEMINI_CODE_MODEL || DEFAULT_GEMINI_MODEL,
    })
    .option('sandbox', {
      alias: 's',
      type: 'boolean',
      description: 'Whether to run in sandbox mode. Defaults to false.',
    })
    .option('debug_mode', {
      alias: 'z',
      type: 'boolean',
      description: 'Whether to run in debug mode. Defaults to false.',
      default: false,
    })
    .option('question', {
      alias: 'q',
      type: 'string',
      description:
        'The question to pass to the command when using piped input.',
    })
    .option('full_context', {
      alias: 'f',
      type: 'boolean',
      description:
        'Recursively include all files within the current directory as context.',
      default: false,
    })
    .help()
    .alias('h', 'help')
    .strict().argv;
  return argv;
}

// Renamed function for clarity
export async function loadCliConfig(settings: Settings): Promise<Config> {
  // Load .env file using logic from server package
  loadEnvironment();

  // Check API key (CLI responsibility)
  if (!process.env.GEMINI_API_KEY) {
    console.log(
      'GEMINI_API_KEY is not set. See https://ai.google.dev/gemini-api/docs/api-key to obtain one. ' +
        'Please set it in your .env file or as an environment variable.',
    );
    process.exit(1);
  }

  // Parse CLI arguments
  const argv = await parseArguments();

  // Create config using factory from server package
  return createServerConfig(
    process.env.GEMINI_API_KEY,
    argv.model || DEFAULT_GEMINI_MODEL,
    argv.sandbox ?? settings.sandbox ?? false,
    process.cwd(),
    argv.debug_mode || false,
    argv.question || '',
    argv.full_context || false,
    settings.toolDiscoveryCommand,
    settings.toolCallCommand,
    settings.mcpServerCommand,
  );
}