CM-58331: Add --ide all option for AI guardrails commands

Ilanlido · claude · Ilanlido · commit 5ad5aa5f4f36 · 2026-02-03T12:15:33.000+02:00
- Add support for `--ide all` to install/uninstall/status hooks for all IDEs at once
- Update validate_and_parse_ide to return None for "all"
- Split Claude Code PreToolUse into PreToolUse:Read and PreToolUse:mcp for better status visibility
- Report results separately for each IDE when using --ide all

Co-Authored-By: Claude Opus 4.5 &lt;noreply@anthropic.com&gt;
diff --git a/cycode/cli/apps/ai_guardrails/command_utils.py b/cycode/cli/apps/ai_guardrails/command_utils.py
@@ -12,24 +12,26 @@
 console = Console()
 
 
-def validate_and_parse_ide(ide: str) -> AIIDEType:
-    """Validate IDE parameter and convert to AIIDEType enum.
+def validate_and_parse_ide(ide: str) -> Optional[AIIDEType]:
+    """Validate IDE parameter, returning None for 'all'.
 
     Args:
-        ide: IDE name string (e.g., 'cursor')
+        ide: IDE name string (e.g., 'cursor', 'claude-code', 'all')
 
     Returns:
-        AIIDEType enum value
+        AIIDEType enum value, or None if 'all' was specified
 
     Raises:
         typer.Exit: If IDE is invalid
     """
+    if ide.lower() == 'all':
+        return None
     try:
         return AIIDEType(ide.lower())
     except ValueError:
         valid_ides = ', '.join([ide_type.value for ide_type in AIIDEType])
         console.print(
-            f'[red]Error:[/] Invalid IDE "{ide}". Supported IDEs: {valid_ides}',
+            f'[red]Error:[/] Invalid IDE "{ide}". Supported IDEs: {valid_ides}, all',
             style='bold red',
         )
         raise typer.Exit(1) from None
diff --git a/cycode/cli/apps/ai_guardrails/consts.py b/cycode/cli/apps/ai_guardrails/consts.py
@@ -60,7 +60,7 @@ def _get_claude_code_hooks_dir() -> Path:
         hooks_dir=_get_claude_code_hooks_dir(),
         repo_hooks_subdir='.claude',
         hooks_file_name='settings.json',
-        hook_events=['UserPromptSubmit', 'PreToolUse'],
+        hook_events=['UserPromptSubmit', 'PreToolUse:Read', 'PreToolUse:mcp'],
     ),
 }
 
diff --git a/cycode/cli/apps/ai_guardrails/hooks_manager.py b/cycode/cli/apps/ai_guardrails/hooks_manager.py
@@ -203,7 +203,15 @@ def get_hooks_status(scope: str = 'user', repo_path: Optional[Path] = None, ide:
     ide_config = IDE_CONFIGS[ide]
     has_cycode_hooks = False
     for event in ide_config.hook_events:
-        entries = existing.get('hooks', {}).get(event, [])
+        # Handle event:matcher format
+        if ':' in event:
+            actual_event, matcher_prefix = event.split(':', 1)
+            all_entries = existing.get('hooks', {}).get(actual_event, [])
+            # Filter entries by matcher
+            entries = [e for e in all_entries if e.get('matcher', '').startswith(matcher_prefix)]
+        else:
+            entries = existing.get('hooks', {}).get(event, [])
+
         cycode_entries = [e for e in entries if is_cycode_hook_entry(e)]
         if cycode_entries:
             has_cycode_hooks = True
diff --git a/cycode/cli/apps/ai_guardrails/install_command.py b/cycode/cli/apps/ai_guardrails/install_command.py
@@ -11,7 +11,7 @@
     validate_and_parse_ide,
     validate_scope,
 )
-from cycode.cli.apps.ai_guardrails.consts import IDE_CONFIGS
+from cycode.cli.apps.ai_guardrails.consts import IDE_CONFIGS, AIIDEType
 from cycode.cli.apps.ai_guardrails.hooks_manager import install_hooks
 from cycode.cli.utils.sentry import add_breadcrumb
 
@@ -30,7 +30,7 @@ def install_command(
         str,
         typer.Option(
             '--ide',
-            help='IDE to install hooks for (e.g., "cursor"). Defaults to cursor.',
+            help='IDE to install hooks for (e.g., "cursor", "claude-code", or "all" for all IDEs). Defaults to cursor.',
         ),
     ] = 'cursor',
     repo_path: Annotated[
@@ -54,6 +54,7 @@ def install_command(
         cycode ai-guardrails install                    # Install for all projects (user scope)
         cycode ai-guardrails install --scope repo       # Install for current repo only
         cycode ai-guardrails install --ide cursor       # Install for Cursor IDE
+        cycode ai-guardrails install --ide all          # Install for all supported IDEs
         cycode ai-guardrails install --scope repo --repo-path /path/to/repo
     """
     add_breadcrumb('ai-guardrails-install')
@@ -62,17 +63,35 @@ def install_command(
     validate_scope(scope)
     repo_path = resolve_repo_path(scope, repo_path)
     ide_type = validate_and_parse_ide(ide)
-    ide_name = IDE_CONFIGS[ide_type].name
-    success, message = install_hooks(scope, repo_path, ide=ide_type)
 
-    if success:
-        console.print(f'[green]✓[/] {message}')
+    ides_to_install: list[AIIDEType] = list(AIIDEType) if ide_type is None else [ide_type]
+
+    results: list[tuple[str, bool, str]] = []
+    for current_ide in ides_to_install:
+        ide_name = IDE_CONFIGS[current_ide].name
+        success, message = install_hooks(scope, repo_path, ide=current_ide)
+        results.append((ide_name, success, message))
+
+    # Report results for each IDE
+    any_success = False
+    all_success = True
+    for _ide_name, success, message in results:
+        if success:
+            console.print(f'[green]✓[/] {message}')
+            any_success = True
+        else:
+            console.print(f'[red]✗[/] {message}', style='bold red')
+            all_success = False
+
+    if any_success:
         console.print()
         console.print('[bold]Next steps:[/]')
-        console.print(f'1. Restart {ide_name} to activate the hooks')
+        successful_ides = [name for name, success, _ in results if success]
+        ide_list = ', '.join(successful_ides)
+        console.print(f'1. Restart {ide_list} to activate the hooks')
         console.print('2. (Optional) Customize policy in ~/.cycode/ai-guardrails.yaml')
         console.print()
         console.print('[dim]The hooks will scan prompts, file reads, and MCP tool calls for secrets.[/]')
-    else:
-        console.print(f'[red]✗[/] {message}', style='bold red')
+
+    if not all_success:
         raise typer.Exit(1)
diff --git a/cycode/cli/apps/ai_guardrails/status_command.py b/cycode/cli/apps/ai_guardrails/status_command.py
@@ -8,6 +8,7 @@
 from rich.table import Table
 
 from cycode.cli.apps.ai_guardrails.command_utils import console, validate_and_parse_ide, validate_scope
+from cycode.cli.apps.ai_guardrails.consts import IDE_CONFIGS, AIIDEType
 from cycode.cli.apps.ai_guardrails.hooks_manager import get_hooks_status
 from cycode.cli.utils.sentry import add_breadcrumb
 
@@ -26,7 +27,7 @@ def status_command(
         str,
         typer.Option(
             '--ide',
-            help='IDE to check status for (e.g., "cursor"). Defaults to cursor.',
+            help='IDE to check status for (e.g., "cursor", "claude-code", or "all" for all IDEs). Defaults to cursor.',
         ),
     ] = 'cursor',
     repo_path: Annotated[
@@ -50,6 +51,7 @@ def status_command(
         cycode ai-guardrails status --scope user   # Show only user-level status
         cycode ai-guardrails status --scope repo   # Show only repo-level status
         cycode ai-guardrails status --ide cursor   # Check status for Cursor IDE
+        cycode ai-guardrails status --ide all      # Check status for all supported IDEs
     """
     add_breadcrumb('ai-guardrails-status')
 
@@ -59,34 +61,41 @@ def status_command(
         repo_path = Path(os.getcwd())
     ide_type = validate_and_parse_ide(ide)
 
-    scopes_to_check = ['user', 'repo'] if scope == 'all' else [scope]
+    ides_to_check: list[AIIDEType] = list(AIIDEType) if ide_type is None else [ide_type]
 
-    for check_scope in scopes_to_check:
-        status = get_hooks_status(check_scope, repo_path if check_scope == 'repo' else None, ide=ide_type)
+    scopes_to_check = ['user', 'repo'] if scope == 'all' else [scope]
 
+    for current_ide in ides_to_check:
+        ide_name = IDE_CONFIGS[current_ide].name
         console.print()
-        console.print(f'[bold]{check_scope.upper()} SCOPE[/]')
-        console.print(f'Path: {status["hooks_path"]}')
+        console.print(f'[bold cyan]═══ {ide_name} ═══[/]')
+
+        for check_scope in scopes_to_check:
+            status = get_hooks_status(check_scope, repo_path if check_scope == 'repo' else None, ide=current_ide)
+
+            console.print()
+            console.print(f'[bold]{check_scope.upper()} SCOPE[/]')
+            console.print(f'Path: {status["hooks_path"]}')
 
-        if not status['file_exists']:
-            console.print('[dim]No hooks.json file found[/]')
-            continue
+            if not status['file_exists']:
+                console.print('[dim]No hooks file found[/]')
+                continue
 
-        if status['cycode_installed']:
-            console.print('[green]✓ Cycode AI guardrails: INSTALLED[/]')
-        else:
-            console.print('[yellow]○ Cycode AI guardrails: NOT INSTALLED[/]')
+            if status['cycode_installed']:
+                console.print('[green]✓ Cycode AI guardrails: INSTALLED[/]')
+            else:
+                console.print('[yellow]○ Cycode AI guardrails: NOT INSTALLED[/]')
 
-        # Show hook details
-        table = Table(show_header=True, header_style='bold')
-        table.add_column('Hook Event')
-        table.add_column('Cycode Enabled')
-        table.add_column('Total Hooks')
+            # Show hook details
+            table = Table(show_header=True, header_style='bold')
+            table.add_column('Hook Event')
+            table.add_column('Cycode Enabled')
+            table.add_column('Total Hooks')
 
-        for event, info in status['hooks'].items():
-            enabled = '[green]Yes[/]' if info['enabled'] else '[dim]No[/]'
-            table.add_row(event, enabled, str(info['total_entries']))
+            for event, info in status['hooks'].items():
+                enabled = '[green]Yes[/]' if info['enabled'] else '[dim]No[/]'
+                table.add_row(event, enabled, str(info['total_entries']))
 
-        console.print(table)
+            console.print(table)
 
     console.print()
diff --git a/cycode/cli/apps/ai_guardrails/uninstall_command.py b/cycode/cli/apps/ai_guardrails/uninstall_command.py
@@ -11,7 +11,7 @@
     validate_and_parse_ide,
     validate_scope,
 )
-from cycode.cli.apps.ai_guardrails.consts import IDE_CONFIGS
+from cycode.cli.apps.ai_guardrails.consts import IDE_CONFIGS, AIIDEType
 from cycode.cli.apps.ai_guardrails.hooks_manager import uninstall_hooks
 from cycode.cli.utils.sentry import add_breadcrumb
 
@@ -30,7 +30,7 @@ def uninstall_command(
         str,
         typer.Option(
             '--ide',
-            help='IDE to uninstall hooks from (e.g., "cursor"). Defaults to cursor.',
+            help='IDE to uninstall hooks from (e.g., "cursor", "claude-code", "all"). Defaults to cursor.',
         ),
     ] = 'cursor',
     repo_path: Annotated[
@@ -54,20 +54,39 @@ def uninstall_command(
         cycode ai-guardrails uninstall                    # Remove user-level hooks
         cycode ai-guardrails uninstall --scope repo       # Remove repo-level hooks
         cycode ai-guardrails uninstall --ide cursor       # Uninstall from Cursor IDE
+        cycode ai-guardrails uninstall --ide all          # Uninstall from all supported IDEs
     """
     add_breadcrumb('ai-guardrails-uninstall')
 
     # Validate inputs
     validate_scope(scope)
     repo_path = resolve_repo_path(scope, repo_path)
     ide_type = validate_and_parse_ide(ide)
-    ide_name = IDE_CONFIGS[ide_type].name
-    success, message = uninstall_hooks(scope, repo_path, ide=ide_type)
 
-    if success:
-        console.print(f'[green]✓[/] {message}')
+    ides_to_uninstall: list[AIIDEType] = list(AIIDEType) if ide_type is None else [ide_type]
+
+    results: list[tuple[str, bool, str]] = []
+    for current_ide in ides_to_uninstall:
+        ide_name = IDE_CONFIGS[current_ide].name
+        success, message = uninstall_hooks(scope, repo_path, ide=current_ide)
+        results.append((ide_name, success, message))
+
+    # Report results for each IDE
+    any_success = False
+    all_success = True
+    for _ide_name, success, message in results:
+        if success:
+            console.print(f'[green]✓[/] {message}')
+            any_success = True
+        else:
+            console.print(f'[red]✗[/] {message}', style='bold red')
+            all_success = False
+
+    if any_success:
         console.print()
-        console.print(f'[dim]Restart {ide_name} for changes to take effect.[/]')
-    else:
-        console.print(f'[red]✗[/] {message}', style='bold red')
+        successful_ides = [name for name, success, _ in results if success]
+        ide_list = ', '.join(successful_ides)
+        console.print(f'[dim]Restart {ide_list} for changes to take effect.[/]')
+
+    if not all_success:
         raise typer.Exit(1)
diff --git a/tests/cli/commands/ai_guardrails/test_command_utils.py b/tests/cli/commands/ai_guardrails/test_command_utils.py
@@ -15,6 +15,9 @@ def test_validate_and_parse_ide_valid() -> None:
     assert validate_and_parse_ide('cursor') == AIIDEType.CURSOR
     assert validate_and_parse_ide('CURSOR') == AIIDEType.CURSOR
     assert validate_and_parse_ide('CuRsOr') == AIIDEType.CURSOR
+    assert validate_and_parse_ide('claude-code') == AIIDEType.CLAUDE_CODE
+    assert validate_and_parse_ide('Claude-Code') == AIIDEType.CLAUDE_CODE
+    assert validate_and_parse_ide('all') is None
 
 
 def test_validate_and_parse_ide_invalid() -> None:

Original file line number	Diff line number	Diff line change
`@@ -60,7 +60,7 @@ def _get_claude_code_hooks_dir() -> Path:`
`60`	`60`	`hooks_dir=_get_claude_code_hooks_dir(),`
`61`	`61`	`repo_hooks_subdir='.claude',`
`62`	`62`	`hooks_file_name='settings.json',`
`63`		`- hook_events=['UserPromptSubmit', 'PreToolUse'],`
	`63`	`+ hook_events=['UserPromptSubmit', 'PreToolUse:Read', 'PreToolUse:mcp'],`
`64`	`64`	`),`
`65`	`65`	`}`
`66`	`66`