Merge pull request #45 from CodeMonkeyCybersecurity/claude/api-key-management-script-011CUrCr7tXgZkrecW3LCZmP

feat(moni): add API key rotation command

Author: CodeMonkeyCybersecurity

cmd/update/moni.go

@@ -7,6 +7,7 @@ package update
 import (
 	"fmt"
 
+	"github.com/CodeMonkeyCybersecurity/eos/pkg/bionicgpt"
 	"github.com/CodeMonkeyCybersecurity/eos/pkg/bionicgpt/apikeys"
 	"github.com/CodeMonkeyCybersecurity/eos/pkg/bionicgpt/postinstall"
 	"github.com/CodeMonkeyCybersecurity/eos/pkg/bionicgpt/refresh"
@@ -25,6 +26,12 @@ var (
 	moniRefreshValidateOnly bool
 	moniRefreshInstallDir   string
 
+	// Moni API key rotation flags
+	moniRotateAPIKeysDryRun      bool
+	moniRotateAPIKeysSkipBackup  bool
+	moniRotateAPIKeysSkipVerify  bool
+	moniRotateAPIKeysSkipRestart bool
+	moniRotateAPIKeysInstallDir  string
 	// Moni post-install and API key rotation flags
 	moniPostInstall   bool
 	moniRotateAPIKeys bool
@@ -117,6 +124,61 @@ Examples:
 	MoniCmd.Flags().BoolVar(&moniRefreshForce, "refresh", false,
 		"Refresh Moni configuration and restart services")
 
+	// Rotate API keys subcommand
+	rotateAPIKeysCmd := &cobra.Command{
+		Use:   "rotate-api-keys",
+		Short: "Rotate LiteLLM virtual API keys for Moni",
+		Long: `Rotate the virtual API keys used by Moni to access LiteLLM models.
+
+This operation regenerates the virtual key with access to all configured models:
+  • Moni (GPT-5-mini)
+  • Moni-4.1 (GPT-4.1-mini)
+  • Moni-o3 (o3-mini)
+  • nomic-embed-text (Ollama embeddings)
+
+The rotation process:
+  1. ASSESS: Check prerequisites (database, LiteLLM health, current keys)
+  2. INTERVENE: Generate new key, update .env, update database, restart app
+  3. EVALUATE: Verify new key works and is properly configured
+
+Safety features:
+  • Automatic backup of .env file before changes
+  • Transaction-like behavior with automatic rollback on failure
+  • Comprehensive verification tests after rotation
+  • Old keys are deleted after successful rotation
+
+Estimated downtime: ~30 seconds (during app restart)
+
+Examples:
+  # Full API key rotation with confirmation
+  eos update moni rotate-api-keys
+
+  # Dry run (show what would be done)
+  eos update moni rotate-api-keys --dry-run
+
+  # Skip backup (not recommended)
+  eos update moni rotate-api-keys --skip-backup
+
+  # Skip verification tests
+  eos update moni rotate-api-keys --skip-verify
+
+  # Custom installation directory
+  eos update moni rotate-api-keys --install-dir /opt/moni`,
+		RunE: eos.Wrap(runMoniRotateAPIKeys),
+	}
+
+	rotateAPIKeysCmd.Flags().BoolVar(&moniRotateAPIKeysDryRun, "dry-run", false,
+		"Show what would be done without making changes")
+	rotateAPIKeysCmd.Flags().BoolVar(&moniRotateAPIKeysSkipBackup, "skip-backup", false,
+		"Skip .env backup (not recommended)")
+	rotateAPIKeysCmd.Flags().BoolVar(&moniRotateAPIKeysSkipVerify, "skip-verify", false,
+		"Skip verification tests after rotation")
+	rotateAPIKeysCmd.Flags().BoolVar(&moniRotateAPIKeysSkipRestart, "skip-restart", false,
+		"Skip app restart after rotation")
+	rotateAPIKeysCmd.Flags().StringVar(&moniRotateAPIKeysInstallDir, "install-dir", "/opt/bionicgpt",
+		"Path to Moni installation directory")
+
+	MoniCmd.AddCommand(refreshCmd, rotateAPIKeysCmd)
 	// Add post-install flag
 	MoniCmd.Flags().BoolVar(&moniPostInstall, "post-install", false,
 		"Run post-installation configuration (upsert models, regenerate API keys)")
@@ -295,6 +357,53 @@ func runMoniRefresh(rc *eos_io.RuntimeContext, cmd *cobra.Command, args []string
 	return nil
 }
 
+// runMoniRotateAPIKeys handles the API key rotation operation
+// Orchestration layer: delegates to pkg/bionicgpt for business logic
+func runMoniRotateAPIKeys(rc *eos_io.RuntimeContext, cmd *cobra.Command, args []string) error {
+	logger := otelzap.Ctx(rc.Ctx)
+
+	logger.Info("Starting Moni API key rotation",
+		zap.String("install_dir", moniRotateAPIKeysInstallDir),
+		zap.Bool("dry_run", moniRotateAPIKeysDryRun),
+		zap.Bool("skip_backup", moniRotateAPIKeysSkipBackup),
+		zap.Bool("skip_verify", moniRotateAPIKeysSkipVerify),
+		zap.Bool("skip_restart", moniRotateAPIKeysSkipRestart))
+
+	// Build rotation configuration
+	config := &bionicgpt.RotateAPIKeysConfig{
+		InstallDir:  moniRotateAPIKeysInstallDir,
+		DryRun:      moniRotateAPIKeysDryRun,
+		SkipBackup:  moniRotateAPIKeysSkipBackup,
+		SkipVerify:  moniRotateAPIKeysSkipVerify,
+		SkipRestart: moniRotateAPIKeysSkipRestart,
+	}
+
+	// Execute rotation
+	if err := bionicgpt.RotateAPIKeys(rc, config); err != nil {
+		logger.Error("Moni API key rotation failed", zap.Error(err))
+		return fmt.Errorf("API key rotation failed: %w", err)
+	}
+
+	logger.Info("━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━")
+	logger.Info("✅ API KEY ROTATION COMPLETE")
+	logger.Info("━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━")
+	logger.Info("")
+	logger.Info("🔑 New virtual key has been generated and configured")
+	logger.Info("")
+	logger.Info("🤖 Authorized Models:")
+	logger.Info("   • Moni (GPT-5-mini)")
+	logger.Info("   • Moni-4.1 (GPT-4.1-mini)")
+	logger.Info("   • Moni-o3 (o3-mini)")
+	logger.Info("   • nomic-embed-text (Ollama)")
+	logger.Info("")
+	logger.Info("🧪 Test in Moni UI:")
+	logger.Info("   http://localhost:8513")
+	logger.Info("   Try: 'What is your name?'")
+	logger.Info("")
+	logger.Info("📝 Monitor logs:")
+	logger.Info("   docker compose -f /opt/bionicgpt/docker-compose.yml logs -f app litellm-proxy")
+	logger.Info("")
+
 // runMoniInit handles the Moni initialization worker
 // Orchestration layer: delegates to pkg/moni for business logic
 func runMoniInit(rc *eos_io.RuntimeContext, cmd *cobra.Command, args []string) error {