luxlabs 1.0.0

package/commands/pricing.js

@@ -0,0 +1,328 @@
+/**
+ * Lux CLI - Pricing Commands
+ *
+ * Manage AWS AppConfig pricing configuration for Lux credits.
+ *
+ * Commands:
+ *   lux pricing show                   - Show current pricing config
+ *   lux pricing export [file]          - Export pricing config to JSON file
+ *   lux pricing deploy [file]          - Deploy pricing config to AWS AppConfig
+ *   lux pricing set-secrets            - Set AWS credentials in Cloudflare Workers
+ */
+
+const chalk = require('chalk');
+const fs = require('fs');
+const path = require('path');
+
+// AppConfig details (from user's setup)
+const APPCONFIG = {
+  region: 'us-east-2',
+  applicationId: 'wgicy0c',
+  environmentId: 'wz6k5o6',
+  environmentName: 'development',
+  configProfileId: 'gagerlg',
+};
+
+// Default pricing configuration (matches lux-studio-api/src/lib/appconfig-pricing.ts)
+const DEFAULT_PRICING_CONFIG = {
+  usdToCredits: 100,     // $1 = 100 credits
+  luxMarkup: 1.2,        // 20% markup
+
+  sessionExpiryMinutes: 15,
+  defaultCreditReservation: 20,
+  lowBalanceThreshold: 100,
+
+  models: {
+    'claude-opus-4-5-20250514': {
+      input: 15.0,
+      output: 75.0,
+      cacheRead: 1.50,
+      cacheWrite: 18.75,
+    },
+    'claude-sonnet-4-5-20250514': {
+      input: 3.0,
+      output: 15.0,
+      cacheRead: 0.30,
+      cacheWrite: 3.75,
+    },
+    'claude-opus-4-20250514': {
+      input: 15.0,
+      output: 75.0,
+      cacheRead: 1.50,
+      cacheWrite: 18.75,
+    },
+    'claude-sonnet-4-20250514': {
+      input: 3.0,
+      output: 15.0,
+      cacheRead: 0.30,
+      cacheWrite: 3.75,
+    },
+    'claude-3-5-sonnet-20241022': {
+      input: 3.0,
+      output: 15.0,
+      cacheRead: 0.30,
+      cacheWrite: 3.75,
+    },
+    'claude-3-5-sonnet-20240620': {
+      input: 3.0,
+      output: 15.0,
+      cacheRead: 0.30,
+      cacheWrite: 3.75,
+    },
+    'default': {
+      input: 3.0,
+      output: 15.0,
+      cacheRead: 0.30,
+      cacheWrite: 3.75,
+    },
+  },
+
+  features: {
+    enableCaching: true,
+    enableDetailedUsageTracking: true,
+  },
+
+  version: '1.0.0',
+  updatedAt: new Date().toISOString(),
+};
+
+async function showHelp() {
+  console.log(`
+${chalk.bold('Lux Pricing Commands')}
+
+Manage AWS AppConfig pricing configuration for Lux credits.
+
+${chalk.yellow('Usage:')}
+  lux pricing <command> [options]
+
+${chalk.yellow('Commands:')}
+  show                    Show current/default pricing configuration
+  export [file]           Export pricing config to JSON file (default: pricing-config.json)
+  deploy <file>           Deploy pricing config to AWS AppConfig
+  set-secrets             Show commands to set AWS credentials in Cloudflare Workers
+
+${chalk.yellow('AppConfig Details:')}
+  Region:       ${APPCONFIG.region}
+  Application:  ${APPCONFIG.applicationId}
+  Environment:  ${APPCONFIG.environmentName} (${APPCONFIG.environmentId})
+  Profile:      ${APPCONFIG.configProfileId}
+
+${chalk.yellow('Examples:')}
+  ${chalk.gray('# View current pricing')}
+  lux pricing show
+
+  ${chalk.gray('# Export default config to edit')}
+  lux pricing export ./my-pricing.json
+
+  ${chalk.gray('# Deploy updated pricing')}
+  lux pricing deploy ./my-pricing.json
+`);
+}
+
+async function showPricing() {
+  console.log(chalk.bold('\nCurrent Pricing Configuration:\n'));
+
+  console.log(chalk.yellow('Credit Conversion:'));
+  console.log(`  USD to Credits: ${DEFAULT_PRICING_CONFIG.usdToCredits} credits per $1`);
+  console.log(`  Markup: ${((DEFAULT_PRICING_CONFIG.luxMarkup - 1) * 100).toFixed(0)}%`);
+
+  console.log(chalk.yellow('\nSession Settings:'));
+  console.log(`  Expiry: ${DEFAULT_PRICING_CONFIG.sessionExpiryMinutes} minutes`);
+  console.log(`  Default Reservation: ${DEFAULT_PRICING_CONFIG.defaultCreditReservation} credits`);
+  console.log(`  Low Balance Warning: ${DEFAULT_PRICING_CONFIG.lowBalanceThreshold} credits`);
+
+  console.log(chalk.yellow('\nModel Pricing (per 1M tokens):'));
+  for (const [model, pricing] of Object.entries(DEFAULT_PRICING_CONFIG.models)) {
+    if (model === 'default') continue;
+    console.log(`  ${chalk.cyan(model)}:`);
+    console.log(`    Input:  $${pricing.input.toFixed(2)}`);
+    console.log(`    Output: $${pricing.output.toFixed(2)}`);
+    if (pricing.cacheRead) {
+      console.log(`    Cache Read:  $${pricing.cacheRead.toFixed(2)}`);
+      console.log(`    Cache Write: $${pricing.cacheWrite.toFixed(2)}`);
+    }
+  }
+
+  console.log(chalk.yellow('\nDefault Fallback Pricing:'));
+  const defaultPricing = DEFAULT_PRICING_CONFIG.models.default;
+  console.log(`  Input:  $${defaultPricing.input.toFixed(2)}`);
+  console.log(`  Output: $${defaultPricing.output.toFixed(2)}`);
+
+  console.log(chalk.gray('\n(Deploy via AppConfig to customize these values)'));
+}
+
+async function exportConfig(args) {
+  const outputFile = args[0] || 'pricing-config.json';
+  const outputPath = path.resolve(process.cwd(), outputFile);
+
+  const config = {
+    ...DEFAULT_PRICING_CONFIG,
+    version: '1.0.0',
+    updatedAt: new Date().toISOString(),
+  };
+
+  fs.writeFileSync(outputPath, JSON.stringify(config, null, 2));
+  console.log(chalk.green(`\nPricing configuration exported to: ${outputPath}`));
+  console.log(chalk.gray('\nEdit this file and use "lux pricing deploy <file>" to update pricing.'));
+}
+
+async function deployConfig(args) {
+  if (!args[0]) {
+    console.log(chalk.red('Error: Please provide a configuration file path'));
+    console.log(chalk.gray('Usage: lux pricing deploy <file>'));
+    process.exit(1);
+  }
+
+  const configPath = path.resolve(process.cwd(), args[0]);
+
+  if (!fs.existsSync(configPath)) {
+    console.log(chalk.red(`Error: File not found: ${configPath}`));
+    process.exit(1);
+  }
+
+  let config;
+  try {
+    config = JSON.parse(fs.readFileSync(configPath, 'utf8'));
+  } catch (err) {
+    console.log(chalk.red('Error: Invalid JSON in configuration file'));
+    console.log(chalk.gray(err.message));
+    process.exit(1);
+  }
+
+  // Validate configuration
+  const requiredFields = ['usdToCredits', 'luxMarkup', 'models'];
+  for (const field of requiredFields) {
+    if (!(field in config)) {
+      console.log(chalk.red(`Error: Missing required field: ${field}`));
+      process.exit(1);
+    }
+  }
+
+  // Update version and timestamp
+  config.version = config.version || '1.0.0';
+  config.updatedAt = new Date().toISOString();
+
+  console.log(chalk.yellow('\nDeploying pricing configuration to AWS AppConfig...\n'));
+
+  // Check for AWS credentials
+  const awsAccessKey = process.env.AWS_ACCESS_KEY_ID;
+  const awsSecretKey = process.env.AWS_SECRET_ACCESS_KEY;
+
+  if (!awsAccessKey || !awsSecretKey) {
+    console.log(chalk.red('Error: AWS credentials not found in environment'));
+    console.log(chalk.gray('\nSet the following environment variables:'));
+    console.log(chalk.gray('  export AWS_ACCESS_KEY_ID="your-access-key"'));
+    console.log(chalk.gray('  export AWS_SECRET_ACCESS_KEY="your-secret-key"'));
+    process.exit(1);
+  }
+
+  try {
+    // Use AWS SDK to deploy
+    const { AppConfigClient, CreateHostedConfigurationVersionCommand } = require('@aws-sdk/client-appconfig');
+
+    const client = new AppConfigClient({
+      region: APPCONFIG.region,
+      credentials: {
+        accessKeyId: awsAccessKey,
+        secretAccessKey: awsSecretKey,
+      },
+    });
+
+    const command = new CreateHostedConfigurationVersionCommand({
+      ApplicationId: APPCONFIG.applicationId,
+      ConfigurationProfileId: APPCONFIG.configProfileId,
+      Content: Buffer.from(JSON.stringify(config)),
+      ContentType: 'application/json',
+      Description: `Pricing update - ${config.version}`,
+    });
+
+    const response = await client.send(command);
+
+    console.log(chalk.green('Pricing configuration deployed successfully!'));
+    console.log(chalk.gray(`  Version: ${response.VersionNumber}`));
+    console.log(chalk.gray(`  Application: ${APPCONFIG.applicationId}`));
+    console.log(chalk.gray(`  Environment: ${APPCONFIG.environmentName}`));
+    console.log(chalk.gray(`  Profile: ${APPCONFIG.configProfileId}`));
+
+    console.log(chalk.yellow('\nNext step: Start a deployment in AWS AppConfig console'));
+    console.log(chalk.gray('Or use AWS CLI:'));
+    console.log(chalk.cyan(`  aws appconfig start-deployment \\
+    --application-id ${APPCONFIG.applicationId} \\
+    --environment-id ${APPCONFIG.environmentId} \\
+    --deployment-strategy-id AppConfig.AllAtOnce \\
+    --configuration-profile-id ${APPCONFIG.configProfileId} \\
+    --configuration-version ${response.VersionNumber} \\
+    --region ${APPCONFIG.region}`));
+
+  } catch (err) {
+    if (err.code === 'MODULE_NOT_FOUND') {
+      console.log(chalk.red('Error: AWS SDK not installed'));
+      console.log(chalk.gray('\nInstall the AWS SDK:'));
+      console.log(chalk.gray('  npm install @aws-sdk/client-appconfig'));
+      process.exit(1);
+    }
+
+    console.log(chalk.red('Error deploying configuration:'));
+    console.log(chalk.gray(err.message));
+    process.exit(1);
+  }
+}
+
+async function setSecrets() {
+  console.log(chalk.bold('\nSet AWS credentials in Cloudflare Workers\n'));
+
+  console.log(chalk.yellow('Run these commands in the lux-studio-api directory:\n'));
+
+  const secrets = [
+    { name: 'AWS_ACCESS_KEY_ID', desc: 'IAM user access key' },
+    { name: 'AWS_SECRET_ACCESS_KEY', desc: 'IAM user secret key' },
+  ];
+
+  for (const secret of secrets) {
+    console.log(chalk.gray(`# ${secret.desc}`));
+    console.log(chalk.cyan(`npx wrangler secret put ${secret.name}`));
+    console.log();
+  }
+
+  console.log(chalk.yellow('Optional (defaults are already set in code):'));
+  console.log(chalk.gray(`# Region: ${APPCONFIG.region}`));
+  console.log(chalk.cyan(`echo "${APPCONFIG.region}" | npx wrangler secret put AWS_REGION`));
+  console.log();
+  console.log(chalk.gray(`# Application ID: ${APPCONFIG.applicationId}`));
+  console.log(chalk.cyan(`echo "${APPCONFIG.applicationId}" | npx wrangler secret put AWS_APPCONFIG_APPLICATION_ID`));
+  console.log();
+  console.log(chalk.gray(`# Environment ID: ${APPCONFIG.environmentId}`));
+  console.log(chalk.cyan(`echo "${APPCONFIG.environmentId}" | npx wrangler secret put AWS_APPCONFIG_ENVIRONMENT_ID`));
+  console.log();
+  console.log(chalk.gray(`# Profile ID: ${APPCONFIG.configProfileId}`));
+  console.log(chalk.cyan(`echo "${APPCONFIG.configProfileId}" | npx wrangler secret put AWS_APPCONFIG_CONFIG_PROFILE_ID`));
+}
+
+async function handlePricing(args) {
+  const [subcommand, ...rest] = args;
+
+  switch (subcommand) {
+    case 'show':
+      await showPricing();
+      break;
+    case 'export':
+      await exportConfig(rest);
+      break;
+    case 'deploy':
+      await deployConfig(rest);
+      break;
+    case 'set-secrets':
+      await setSecrets();
+      break;
+    case 'help':
+    case undefined:
+      await showHelp();
+      break;
+    default:
+      console.log(chalk.red(`Unknown command: ${subcommand}`));
+      await showHelp();
+      process.exit(1);
+  }
+}
+
+module.exports = { handlePricing };