archondev 0.1.0 → 1.2.0

package/dist/chunk-IMZN36GC.js

@@ -0,0 +1,159 @@
+import {
+  KeyValidator,
+  keyManager
+} from "./chunk-SMR7JQK6.js";
+import {
+  loadConfig,
+  saveConfig
+} from "./chunk-WCCBJSNI.js";
+
+// src/cli/keys.ts
+import chalk from "chalk";
+import ora from "ora";
+import { createInterface } from "readline";
+var VALID_PROVIDERS = ["anthropic", "openai", "google"];
+function isValidProvider(provider) {
+  return VALID_PROVIDERS.includes(provider);
+}
+async function promptForKey(provider) {
+  const rl = createInterface({
+    input: process.stdin,
+    output: process.stdout
+  });
+  return new Promise((resolve) => {
+    const providerName = provider.charAt(0).toUpperCase() + provider.slice(1);
+    rl.question(`Enter your ${providerName} API key: `, (answer) => {
+      rl.close();
+      resolve(answer.trim());
+    });
+  });
+}
+async function addKey(provider, options = {}) {
+  if (!isValidProvider(provider)) {
+    console.error(chalk.red(`Invalid provider: ${provider}`));
+    console.log(chalk.dim(`Valid providers: ${VALID_PROVIDERS.join(", ")}`));
+    process.exit(1);
+  }
+  const label = options.label ?? "default";
+  const apiKey = await promptForKey(provider);
+  if (!apiKey) {
+    console.error(chalk.red("API key cannot be empty"));
+    process.exit(1);
+  }
+  const spinner = ora("Validating API key...").start();
+  const validator = new KeyValidator();
+  const validationResult = await validator.validateKey(provider, apiKey);
+  if (!validationResult.valid) {
+    spinner.fail(chalk.red(`API key validation failed: ${validationResult.error}`));
+    process.exit(1);
+  }
+  spinner.text = "Encrypting and storing API key...";
+  const result = await keyManager.addKey(provider, apiKey, label);
+  if (!result.success) {
+    spinner.fail(chalk.red(`Failed to store key: ${result.error}`));
+    process.exit(1);
+  }
+  const config = await loadConfig();
+  if (config.tier === "FREE") {
+    await saveConfig({ ...config, tier: "BYOK" });
+    spinner.succeed(chalk.green(`API key for ${provider} (${label}) added successfully!`));
+    console.log(chalk.blue("Your tier has been upgraded to BYOK (Bring Your Own Key)"));
+    console.log(chalk.dim("You now have access to all models and unlimited atoms."));
+  } else {
+    spinner.succeed(chalk.green(`API key for ${provider} (${label}) added successfully!`));
+  }
+  const adversarialStatus = await keyManager.getAdversarialStatus();
+  if (adversarialStatus.enabled) {
+    console.log(chalk.green(`\u2713 ${adversarialStatus.message}`));
+  } else {
+    console.log(chalk.yellow(`\u25CB ${adversarialStatus.message}`));
+  }
+}
+async function listKeys() {
+  const spinner = ora("Loading configured keys...").start();
+  const keys = await keyManager.listKeys();
+  spinner.stop();
+  if (keys.length === 0) {
+    console.log(chalk.yellow("No API keys configured"));
+    console.log(chalk.dim("Run `archon keys add <provider>` to add a key"));
+    console.log(chalk.dim(`Available providers: ${VALID_PROVIDERS.join(", ")}`));
+    return;
+  }
+  console.log(chalk.green("Configured API keys:"));
+  console.log();
+  const byProvider = {};
+  for (const key of keys) {
+    if (!byProvider[key.provider]) {
+      byProvider[key.provider] = [];
+    }
+    byProvider[key.provider]?.push(key);
+  }
+  for (const [provider, providerKeys] of Object.entries(byProvider)) {
+    console.log(chalk.blue(`  ${provider.toUpperCase()}`));
+    for (const key of providerKeys ?? []) {
+      const addedDate = new Date(key.createdAt).toLocaleDateString();
+      const primaryBadge = key.isPrimary ? chalk.green(" [primary]") : "";
+      console.log(`    \u2022 ${key.label}${primaryBadge} ${chalk.dim(`(added ${addedDate})`)}`);
+    }
+  }
+  console.log();
+  const adversarialStatus = await keyManager.getAdversarialStatus();
+  if (adversarialStatus.enabled) {
+    console.log(chalk.green(`\u2713 ${adversarialStatus.message}`));
+  } else {
+    console.log(chalk.yellow(`\u25CB ${adversarialStatus.message}`));
+  }
+  console.log();
+  console.log(chalk.dim("Note: API keys are stored locally with encryption"));
+}
+async function removeKey(provider, options = {}) {
+  if (!isValidProvider(provider)) {
+    console.error(chalk.red(`Invalid provider: ${provider}`));
+    console.log(chalk.dim(`Valid providers: ${VALID_PROVIDERS.join(", ")}`));
+    process.exit(1);
+  }
+  const label = options.label;
+  const labelMsg = label ? ` (${label})` : "";
+  const spinner = ora(`Removing API key for ${provider}${labelMsg}...`).start();
+  const result = await keyManager.removeKey(provider, label);
+  if (!result.success) {
+    spinner.fail(chalk.red(result.error ?? "Failed to remove key"));
+    process.exit(1);
+  }
+  const hasRemainingKeys = await keyManager.hasAnyKey();
+  if (!hasRemainingKeys) {
+    const config = await loadConfig();
+    if (config.tier === "BYOK") {
+      await saveConfig({ ...config, tier: "FREE" });
+      spinner.succeed(chalk.green(`API key for ${provider}${labelMsg} removed`));
+      console.log(chalk.yellow("No API keys remaining. Tier reverted to FREE."));
+      return;
+    }
+  }
+  spinner.succeed(chalk.green(`API key for ${provider}${labelMsg} removed`));
+  const adversarialStatus = await keyManager.getAdversarialStatus();
+  if (!adversarialStatus.enabled && adversarialStatus.providerCount > 0) {
+    console.log(chalk.yellow(`\u25CB ${adversarialStatus.message}`));
+  }
+}
+async function setPrimaryKey(provider, label) {
+  if (!isValidProvider(provider)) {
+    console.error(chalk.red(`Invalid provider: ${provider}`));
+    console.log(chalk.dim(`Valid providers: ${VALID_PROVIDERS.join(", ")}`));
+    process.exit(1);
+  }
+  const spinner = ora(`Setting primary key for ${provider} to "${label}"...`).start();
+  const result = await keyManager.setPrimary(provider, label);
+  if (!result.success) {
+    spinner.fail(chalk.red(result.error ?? "Failed to set primary key"));
+    process.exit(1);
+  }
+  spinner.succeed(chalk.green(`Primary key for ${provider} set to "${label}"`));
+}
+
+export {
+  addKey,
+  listKeys,
+  removeKey,
+  setPrimaryKey
+};