import { chat, reason, setKey, setPrompt, viewPrompt } from './commands.js';
import { printInfo } from './output.js';

function usage(exit = true) {
  printInfo(`用法:
  ds k <key>        设置/替换 API 密钥
  ds s <prompt>     设置系统提示词
  ds v              查看当前提示词
  ds c <text>       使用 deepseek-chat（流式）
  ds r <text>       使用 deepseek-reasoner（流式，含思考）
环境优先级: 命令行 > 环境变量 DEEPSEEK_API_KEY > 配置文件
默认模型: deepseek-reasoner`);
  if (exit) process.exit(1);
}

export async function main(argv) {
  if (!argv || argv.length === 0 || ['-h', '--help'].includes(argv[0])) {
    usage(false);
    return;
  }

  const [cmd, ...rest] = argv;

  switch (cmd) {
    case 'k':
      await setKey(rest);
      break;
    case 's':
      await setPrompt(rest);
      break;
    case 'v':
      viewPrompt();
      break;
    case 'c':
      await chat(rest);
      break;
    case 'r':
      await reason(rest);
      break;
    default:
      usage();
  }
}

// 若作为独立模块直接调用
if (process.argv[1] === new URL(import.meta.url).pathname) {
  main(process.argv.slice(2));
}