aidp 0.34.1 → 0.39.3

data/lib/aidp/cli/prompts_command.rb

@@ -0,0 +1,253 @@
+# frozen_string_literal: true
+
+require "tty-prompt"
+require "tty-table"
+require_relative "../prompts/prompt_template_manager"
+require_relative "../message_display"
+
+module Aidp
+  class CLI
+    # Command handler for `aidp prompts` subcommand
+    #
+    # Provides commands for managing prompt templates:
+    #   - List all available templates
+    #   - Show template details
+    #   - Customize templates for project
+    #   - Reset customized templates
+    #
+    # Usage:
+    #   aidp prompts list
+    #   aidp prompts show <template_id>
+    #   aidp prompts customize <template_id>
+    #   aidp prompts reset <template_id>
+    class PromptsCommand
+      include Aidp::MessageDisplay
+
+      def initialize(prompt: TTY::Prompt.new, project_dir: Dir.pwd)
+        @prompt = prompt
+        @project_dir = project_dir
+        @template_manager = Prompts::PromptTemplateManager.new(project_dir: project_dir)
+      end
+
+      # Main entry point for prompts command
+      def run(args)
+        subcommand = args.shift
+
+        case subcommand
+        when "list", "ls"
+          list_templates(args)
+        when "show", "view"
+          show_template(args)
+        when "customize", "edit"
+          customize_template(args)
+        when "reset"
+          reset_template(args)
+        when "-h", "--help", nil
+          display_usage
+        else
+          display_message("Unknown subcommand: #{subcommand}", type: :error)
+          display_usage
+        end
+      end
+
+      private
+
+      def list_templates(args)
+        category_filter = nil
+
+        # Parse options
+        until args.empty?
+          token = args.shift
+          case token
+          when "--category", "-c"
+            category_filter = args.shift
+          when "-h", "--help"
+            display_list_usage
+            return
+          end
+        end
+
+        templates = @template_manager.list_templates
+
+        if templates.empty?
+          display_message("No prompt templates found.", type: :info)
+          return
+        end
+
+        # Filter by category if specified
+        if category_filter
+          templates = templates.select { |t| t[:category] == category_filter }
+          if templates.empty?
+            display_message("No templates found in category: #{category_filter}", type: :info)
+            return
+          end
+        end
+
+        # Group by category
+        grouped = templates.group_by { |t| t[:category] }
+
+        display_message("\nAvailable Prompt Templates", type: :highlight)
+        display_message("=" * 50, type: :muted)
+
+        grouped.each do |category, category_templates|
+          display_message("\n#{category}/", type: :info)
+
+          table_data = category_templates.map do |t|
+            source_badge = case @template_manager.determine_source(t[:path])
+            when :project then "[project]"
+            when :user then "[user]"
+            else "[builtin]"
+            end
+            [
+              "  #{File.basename(t[:id])}",
+              t[:version] || "1.0.0",
+              source_badge,
+              truncate(t[:description] || "", 40)
+            ]
+          end
+
+          table = TTY::Table.new(
+            header: ["Template", "Version", "Source", "Description"],
+            rows: table_data
+          )
+          display_message(table.render(:unicode, padding: [0, 1]), type: :info)
+        end
+
+        display_message("\nUse 'aidp prompts show <template_id>' to view details", type: :muted)
+      end
+
+      def show_template(args)
+        template_id = args.shift
+
+        if template_id.nil? || template_id.start_with?("-")
+          display_message("Error: Template ID required", type: :error)
+          display_show_usage
+          return
+        end
+
+        info = @template_manager.template_info(template_id)
+
+        if info.nil?
+          display_message("Template not found: #{template_id}", type: :error)
+          display_message("Use 'aidp prompts list' to see available templates", type: :muted)
+          return
+        end
+
+        display_message("\nTemplate: #{info[:id]}", type: :highlight)
+        display_message("=" * 50, type: :muted)
+        display_message("Name: #{info[:name]}", type: :info)
+        display_message("Description: #{info[:description]}", type: :info) if info[:description]
+        display_message("Version: #{info[:version]}", type: :info) if info[:version]
+        display_message("Source: #{info[:source]}", type: :info)
+        display_message("Path: #{info[:path]}", type: :muted)
+
+        if info[:variables]&.any?
+          display_message("\nVariables:", type: :highlight)
+          info[:variables].each { |v| display_message("  - {{#{v}}}", type: :info) }
+        end
+
+        display_message("\nPrompt Preview:", type: :highlight)
+        display_message("-" * 50, type: :muted)
+        display_message(info[:prompt_preview] || "(empty)", type: :info)
+        display_message("-" * 50, type: :muted)
+
+        if info[:source] != :project
+          display_message("\nTo customize this template for your project:", type: :muted)
+          display_message("  aidp prompts customize #{template_id}", type: :muted)
+        end
+      end
+
+      def customize_template(args)
+        template_id = args.shift
+
+        if template_id.nil? || template_id.start_with?("-")
+          display_message("Error: Template ID required", type: :error)
+          display_customize_usage
+          return
+        end
+
+        begin
+          path = @template_manager.customize_template(template_id)
+          display_message("Template customized successfully!", type: :success)
+          display_message("Edit the template at: #{path}", type: :info)
+          display_message("\nThe template will now be loaded from your project directory.", type: :muted)
+          display_message("Use 'aidp prompts reset #{template_id}' to restore the default.", type: :muted)
+        rescue Prompts::TemplateNotFoundError
+          display_message("Template not found: #{template_id}", type: :error)
+          display_message("Use 'aidp prompts list' to see available templates", type: :muted)
+        rescue => e
+          display_message("Failed to customize template: #{e.message}", type: :error)
+        end
+      end
+
+      def reset_template(args)
+        template_id = args.shift
+
+        if template_id.nil? || template_id.start_with?("-")
+          display_message("Error: Template ID required", type: :error)
+          display_reset_usage
+          return
+        end
+
+        # Confirm reset
+        unless @prompt.yes?("Reset #{template_id} to default? This will delete your customizations.")
+          display_message("Reset cancelled", type: :info)
+          return
+        end
+
+        if @template_manager.reset_template(template_id)
+          display_message("Template reset to default successfully!", type: :success)
+        else
+          display_message("No customization found for: #{template_id}", type: :info)
+        end
+      end
+
+      def truncate(text, max_length)
+        return "" if text.nil?
+        return text if text.length <= max_length
+
+        "#{text[0, max_length - 3]}..."
+      end
+
+      def display_usage
+        display_message("\nUsage: aidp prompts <subcommand> [options]", type: :info)
+        display_message("\nSubcommands:", type: :info)
+        display_message("  list, ls              List all available prompt templates", type: :info)
+        display_message("  show, view <id>       Show template details", type: :info)
+        display_message("  customize, edit <id>  Copy template to project for customization", type: :info)
+        display_message("  reset <id>            Reset customized template to default", type: :info)
+        display_message("\nExamples:", type: :info)
+        display_message("  aidp prompts list", type: :info)
+        display_message("  aidp prompts list --category decision_engine", type: :info)
+        display_message("  aidp prompts show decision_engine/condition_detection", type: :info)
+        display_message("  aidp prompts customize decision_engine/condition_detection", type: :info)
+        display_message("  aidp prompts reset decision_engine/condition_detection", type: :info)
+      end
+
+      def display_list_usage
+        display_message("\nUsage: aidp prompts list [options]", type: :info)
+        display_message("\nOptions:", type: :info)
+        display_message("  --category, -c <name>  Filter by category", type: :info)
+        display_message("  -h, --help            Show this help", type: :info)
+      end
+
+      def display_show_usage
+        display_message("\nUsage: aidp prompts show <template_id>", type: :info)
+        display_message("\nExamples:", type: :info)
+        display_message("  aidp prompts show decision_engine/condition_detection", type: :info)
+      end
+
+      def display_customize_usage
+        display_message("\nUsage: aidp prompts customize <template_id>", type: :info)
+        display_message("\nExamples:", type: :info)
+        display_message("  aidp prompts customize decision_engine/condition_detection", type: :info)
+      end
+
+      def display_reset_usage
+        display_message("\nUsage: aidp prompts reset <template_id>", type: :info)
+        display_message("\nExamples:", type: :info)
+        display_message("  aidp prompts reset decision_engine/condition_detection", type: :info)
+      end
+    end
+  end
+end

data/lib/aidp/cli/security_command.rb

@@ -160,9 +160,7 @@ module Aidp
         # Check if env var exists
         unless ENV.key?(env_var)
           @prompt.warn("Warning: Environment variable '#{env_var}' is not currently set")
-          unless @prompt.yes?("Continue anyway?")
-            return 1
-          end
+          return 1 unless @prompt.yes?("Continue anyway?")
         end
 
         # Ask for optional description
@@ -189,9 +187,7 @@ module Aidp
           @prompt.say("  Env var: #{env_var}")
           @prompt.say("  Registered at: #{result[:registered_at]}")
 
-          if scope_list.any?
-            @prompt.say("  Scopes: #{scope_list.join(", ")}")
-          end
+          @prompt.say("  Scopes: #{scope_list.join(", ")}") if scope_list.any?
 
           @prompt.say("\nThe secret value will be proxied through short-lived tokens.")
           @prompt.say("Agent processes will not have direct access to '#{env_var}'.")
@@ -229,9 +225,7 @@ module Aidp
 
         if registry.unregister(name: secret_name)
           @prompt.ok("Secret '#{secret_name}' unregistered")
-          if revoked_count > 0
-            @prompt.say("  Revoked #{revoked_count} active token(s)")
-          end
+          @prompt.say("  Revoked #{revoked_count} active token(s)") if revoked_count > 0
           0
         else
           @prompt.error("Failed to unregister secret")
@@ -349,7 +343,7 @@ module Aidp
         end
 
         # Run RSpec with timeout protection
-        cmd = "bundle exec rspec #{rspec_path} --format documentation"
+        cmd = "bundle exec rspec #{rspec_path} --format failures"
         @prompt.say("$ #{cmd}\n")
         @prompt.say("(timeout: #{AUDIT_TIMEOUT_SECONDS / 60} minutes)\n")
 

data/lib/aidp/cli/storage_command.rb

@@ -0,0 +1,277 @@
+# frozen_string_literal: true
+
+require "tty-prompt"
+require_relative "../database/storage_migrator"
+
+module Aidp
+  class CLI
+    # Command handler for `aidp storage` subcommand
+    #
+    # Provides commands for managing AIDP storage:
+    #   - migrate: Migrate file-based storage to SQLite
+    #   - status: Show storage migration status
+    #   - cleanup: Remove old file-based storage after migration
+    #
+    # Usage:
+    #   aidp storage migrate
+    #   aidp storage migrate --dry-run
+    #   aidp storage migrate --no-backup
+    #   aidp storage status
+    #   aidp storage cleanup
+    class StorageCommand
+      include Aidp::MessageDisplay
+
+      def initialize(prompt: TTY::Prompt.new, project_dir: nil)
+        @prompt = prompt
+        @project_dir = project_dir || Dir.pwd
+      end
+
+      # Main entry point for storage command
+      def run(args)
+        subcommand = args.shift
+
+        case subcommand
+        when "migrate"
+          run_migrate(args)
+        when "status"
+          run_status
+        when "cleanup"
+          run_cleanup(args)
+        when "-h", "--help", nil
+          display_usage
+        else
+          display_message("Unknown subcommand: #{subcommand}", type: :error)
+          display_usage
+        end
+      end
+
+      private
+
+      def run_migrate(args)
+        dry_run = false
+        backup = true
+        force = false
+
+        until args.empty?
+          token = args.shift
+          case token
+          when "--dry-run"
+            dry_run = true
+          when "--no-backup"
+            backup = false
+          when "--force"
+            force = true
+          when "-h", "--help"
+            display_migrate_usage
+            return
+          else
+            display_message("Unknown option: #{token}", type: :error)
+            display_migrate_usage
+            return
+          end
+        end
+
+        migrator = Database::StorageMigrator.new(
+          project_dir: @project_dir,
+          dry_run: dry_run
+        )
+
+        # Check if migration is needed
+        unless migrator.migration_needed?
+          display_message("No file-based storage found. Nothing to migrate.", type: :info)
+          return
+        end
+
+        # Check if already migrated
+        if migrator.already_migrated? && !force
+          display_message("Database already contains migrated data.", type: :warning)
+          display_message("Use --force to migrate anyway (may create duplicates).", type: :info)
+          return
+        end
+
+        # Confirm migration
+        if dry_run
+          display_message("DRY RUN: No changes will be made.", type: :highlight)
+        else
+          unless force
+            confirmed = @prompt.yes?("Migrate file-based storage to SQLite?")
+            return unless confirmed
+          end
+        end
+
+        # Run migration
+        display_message("Starting storage migration...", type: :info)
+
+        result = migrator.migrate!(backup: backup)
+
+        # Display results
+        display_migration_results(result)
+      end
+
+      def run_status
+        migrator = Database::StorageMigrator.new(project_dir: @project_dir)
+
+        display_message("\nStorage Migration Status", type: :info)
+        display_message("=" * 40, type: :muted)
+
+        if migrator.migration_needed?
+          display_message("File-based storage: Found", type: :warning)
+          display_message("  Migration recommended", type: :info)
+        else
+          display_message("File-based storage: Not found", type: :success)
+        end
+
+        if migrator.already_migrated?
+          display_message("SQLite database: Contains data", type: :success)
+        else
+          display_message("SQLite database: Empty or not initialized", type: :info)
+        end
+
+        db_file = ConfigPaths.database_file(@project_dir)
+        if File.exist?(db_file)
+          size = File.size(db_file)
+          display_message("Database file: #{db_file}", type: :info)
+          display_message("Database size: #{format_size(size)}", type: :info)
+        end
+      end
+
+      def run_cleanup(args)
+        keep_config = true
+        force = false
+
+        until args.empty?
+          token = args.shift
+          case token
+          when "--include-config"
+            keep_config = false
+          when "--force"
+            force = true
+          when "-h", "--help"
+            display_cleanup_usage
+            return
+          else
+            display_message("Unknown option: #{token}", type: :error)
+            display_cleanup_usage
+            return
+          end
+        end
+
+        migrator = Database::StorageMigrator.new(project_dir: @project_dir)
+
+        # Verify migration was completed
+        unless migrator.already_migrated?
+          display_message("Cannot cleanup: No migrated data found in database.", type: :error)
+          display_message("Run 'aidp storage migrate' first.", type: :info)
+          return
+        end
+
+        unless migrator.migration_needed?
+          display_message("No file-based storage to clean up.", type: :info)
+          return
+        end
+
+        # Confirm cleanup
+        unless force
+          display_message("This will permanently delete old file-based storage.", type: :warning)
+          confirmed = @prompt.yes?("Proceed with cleanup?")
+          return unless confirmed
+        end
+
+        display_message("Cleaning up old storage...", type: :info)
+        migrator.cleanup_old_storage!(keep_config: keep_config)
+
+        display_message("Cleanup complete.", type: :success)
+        display_message("Files removed: #{migrator.stats[:files_removed]}", type: :info)
+      end
+
+      def display_migration_results(result)
+        display_message("\nMigration Results", type: :info)
+        display_message("=" * 40, type: :muted)
+
+        case result[:status]
+        when :success
+          display_message("Status: Success", type: :success)
+        when :partial
+          display_message("Status: Partial (some errors occurred)", type: :warning)
+        when :skipped
+          display_message("Status: Skipped - #{result[:reason]}", type: :info)
+          return
+        end
+
+        if result[:dry_run]
+          display_message("(Dry run - no changes made)", type: :highlight)
+        end
+
+        stats = result[:stats]
+        if stats && !stats.empty?
+          display_message("\nMigrated:", type: :info)
+          stats.each do |key, count|
+            next if count.zero?
+            label = key.to_s.tr("_", " ").capitalize
+            display_message("  #{label}: #{count}", type: :info)
+          end
+        end
+
+        errors = result[:errors]
+        if errors && !errors.empty?
+          display_message("\nErrors:", type: :error)
+          errors.each do |error|
+            display_message("  #{error[:type]}: #{error[:error]}", type: :error)
+          end
+        end
+
+        unless result[:dry_run]
+          display_message("\nNext steps:", type: :info)
+          display_message("  1. Verify data with 'aidp storage status'", type: :info)
+          display_message("  2. Test AIDP functionality", type: :info)
+          display_message("  3. Run 'aidp storage cleanup' to remove old files", type: :info)
+        end
+      end
+
+      def display_usage
+        display_message("\nUsage: aidp storage <subcommand> [options]", type: :info)
+        display_message("\nSubcommands:", type: :info)
+        display_message("  migrate    Migrate file-based storage to SQLite", type: :info)
+        display_message("  status     Show storage migration status", type: :info)
+        display_message("  cleanup    Remove old file-based storage", type: :info)
+        display_message("\nOptions:", type: :info)
+        display_message("  -h, --help    Show this help message", type: :info)
+        display_message("\nExamples:", type: :info)
+        display_message("  aidp storage migrate", type: :info)
+        display_message("  aidp storage migrate --dry-run", type: :info)
+        display_message("  aidp storage status", type: :info)
+        display_message("  aidp storage cleanup", type: :info)
+      end
+
+      def display_migrate_usage
+        display_message("\nUsage: aidp storage migrate [options]", type: :info)
+        display_message("\nOptions:", type: :info)
+        display_message("  --dry-run      Show what would be migrated without making changes", type: :info)
+        display_message("  --no-backup    Skip creating backup of .aidp directory", type: :info)
+        display_message("  --force        Migrate even if database already has data", type: :info)
+        display_message("  -h, --help     Show this help message", type: :info)
+      end
+
+      def display_cleanup_usage
+        display_message("\nUsage: aidp storage cleanup [options]", type: :info)
+        display_message("\nOptions:", type: :info)
+        display_message("  --include-config  Also remove aidp.yml config file", type: :info)
+        display_message("  --force           Skip confirmation prompt", type: :info)
+        display_message("  -h, --help        Show this help message", type: :info)
+      end
+
+      def format_size(bytes)
+        units = %w[B KB MB GB]
+        unit = 0
+        size = bytes.to_f
+
+        while size >= 1024 && unit < units.length - 1
+          size /= 1024
+          unit += 1
+        end
+
+        "%.1f %s" % [size, units[unit]]
+      end
+    end
+  end
+end