import { readConfig, saveConfig, resolveApiKey, DEFAULT_PROMPT } from './config.js';
import { streamChat } from './api.js';
import { printInfo, printError } from './output.js';

export async function setKey(args) {
  if (args.length < 1) {
    printError('请输入密钥');
    process.exit(1);
  }
  const cfg = readConfig();
  saveConfig({ apiKey: args[0], prompt: cfg.prompt });
  printInfo('API 密钥已保存。');
}

export async function setPrompt(args) {
  if (args.length < 1) {
    printError('请输入新的 system prompt');
    process.exit(1);
  }
  const cfg = readConfig();
  const prompt = args.join(' ');
  saveConfig({ apiKey: cfg.apiKey, prompt });
  printInfo('system prompt 已更新。');
}

export function viewPrompt() {
  const cfg = readConfig();
  printInfo('当前 system prompt:');
  printInfo(cfg.prompt || DEFAULT_PROMPT);
}

async function runChat(model, args) {
  if (args.length < 1) {
    printError('请输入要发送的内容');
    process.exit(1);
  }
  const userInput = args.join(' ');
  const cfg = readConfig();
  const { apiKey, source } = resolveApiKey(cfg);
  if (!apiKey) {
    printError('未设置 API 密钥，请先运行: ds k <key> 或设置环境变量 DEEPSEEK_API_KEY');
    process.exit(1);
  }
  if (source === 'env') {
    printInfo('使用环境变量 DEEPSEEK_API_KEY');
  }
  await streamChat({ model, apiKey, prompt: cfg.prompt, userInput });
}

export async function chat(args) {
  await runChat('deepseek-chat', args);
}

export async function reason(args) {
  await runChat('deepseek-reasoner', args);
}