cycode 3.11.3__tar.gz → 3.11.5.dev2__tar.gz

{cycode-3.11.3 → cycode-3.11.5.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cycode
-Version: 3.11.3
+Version: 3.11.5.dev2
 Summary: Boost security in your dev lifecycle via SAST, SCA, Secrets & IaC scanning.
 License-Expression: MIT
 License-File: LICENCE

cycode-3.11.5.dev2/cycode/__init__.py

@@ -0,0 +1 @@
+__version__ = '3.11.5.dev2'  # DON'T TOUCH. Placeholder. Will be filled automatically on poetry build from Git Tag

{cycode-3.11.3 → cycode-3.11.5.dev2}/cycode/cli/apps/ai_guardrails/__init__.py

@@ -1,5 +1,6 @@
 import typer
 
+from cycode.cli.apps.ai_guardrails.ensure_auth_command import ensure_auth_command
 from cycode.cli.apps.ai_guardrails.install_command import install_command
 from cycode.cli.apps.ai_guardrails.scan.scan_command import scan_command
 from cycode.cli.apps.ai_guardrails.status_command import status_command
@@ -17,3 +18,6 @@ app.command(
     name='scan',
     short_help='Scan content from AI IDE hooks for secrets (reads JSON from stdin).',
 )(scan_command)
+app.command(hidden=True, name='ensure-auth', short_help='Ensure authentication, triggering auth if needed.')(
+    ensure_auth_command
+)

{cycode-3.11.3 → cycode-3.11.5.dev2}/cycode/cli/apps/ai_guardrails/consts.py

@@ -6,6 +6,7 @@ Currently supports:
 """
 
 import platform
+from copy import deepcopy
 from enum import Enum
 from pathlib import Path
 from typing import NamedTuple
@@ -25,6 +26,13 @@ class PolicyMode(str, Enum):
     WARN = 'warn'
 
 
+class InstallMode(str, Enum):
+    """Installation mode for ai-guardrails install command."""
+
+    REPORT = 'report'
+    BLOCK = 'block'
+
+
 class IDEConfig(NamedTuple):
     """Configuration for an AI IDE."""
 
@@ -76,12 +84,15 @@ DEFAULT_IDE = AIIDEType.CURSOR
 
 # Command used in hooks
 CYCODE_SCAN_PROMPT_COMMAND = 'cycode ai-guardrails scan'
+CYCODE_ENSURE_AUTH_COMMAND = 'cycode ai-guardrails ensure-auth'
 
 
-def _get_cursor_hooks_config() -> dict:
+def _get_cursor_hooks_config(async_mode: bool = False) -> dict:
     """Get Cursor-specific hooks configuration."""
     config = IDE_CONFIGS[AIIDEType.CURSOR]
-    hooks = {event: [{'command': CYCODE_SCAN_PROMPT_COMMAND}] for event in config.hook_events}
+    command = f'{CYCODE_SCAN_PROMPT_COMMAND} &' if async_mode else CYCODE_SCAN_PROMPT_COMMAND
+    hooks = {event: [{'command': command}] for event in config.hook_events}
+    hooks['sessionStart'] = [{'command': CYCODE_ENSURE_AUTH_COMMAND}]
 
     return {
         'version': 1,
@@ -89,7 +100,7 @@ def _get_cursor_hooks_config() -> dict:
     }
 
 
-def _get_claude_code_hooks_config() -> dict:
+def _get_claude_code_hooks_config(async_mode: bool = False) -> dict:
     """Get Claude Code-specific hooks configuration.
 
     Claude Code uses a different hook format with nested structure:
@@ -98,36 +109,48 @@ def _get_claude_code_hooks_config() -> dict:
     """
     command = f'{CYCODE_SCAN_PROMPT_COMMAND} --ide claude-code'
 
+    hook_entry = {'type': 'command', 'command': command}
+    if async_mode:
+        hook_entry['async'] = True
+        hook_entry['timeout'] = 20
+
     return {
         'hooks': {
+            'SessionStart': [
+                {
+                    'matcher': 'startup',
+                    'hooks': [{'type': 'command', 'command': CYCODE_ENSURE_AUTH_COMMAND}],
+                }
+            ],
             'UserPromptSubmit': [
                 {
-                    'hooks': [{'type': 'command', 'command': command}],
+                    'hooks': [deepcopy(hook_entry)],
                 }
             ],
             'PreToolUse': [
                 {
                     'matcher': 'Read',
-                    'hooks': [{'type': 'command', 'command': command}],
+                    'hooks': [deepcopy(hook_entry)],
                 },
                 {
                     'matcher': 'mcp__.*',
-                    'hooks': [{'type': 'command', 'command': command}],
+                    'hooks': [deepcopy(hook_entry)],
                 },
             ],
         },
     }
 
 
-def get_hooks_config(ide: AIIDEType) -> dict:
+def get_hooks_config(ide: AIIDEType, async_mode: bool = False) -> dict:
     """Get the hooks configuration for a specific IDE.
 
     Args:
         ide: The AI IDE type
+        async_mode: If True, hooks run asynchronously (non-blocking)
 
     Returns:
         Dict with hooks configuration for the specified IDE
     """
     if ide == AIIDEType.CLAUDE_CODE:
-        return _get_claude_code_hooks_config()
-    return _get_cursor_hooks_config()
+        return _get_claude_code_hooks_config(async_mode=async_mode)
+    return _get_cursor_hooks_config(async_mode=async_mode)

cycode-3.11.5.dev2/cycode/cli/apps/ai_guardrails/ensure_auth_command.py

@@ -0,0 +1,21 @@
+import typer
+
+from cycode.cli.apps.auth.auth_common import get_authorization_info
+from cycode.cli.apps.auth.auth_manager import AuthManager
+from cycode.cli.exceptions.handle_auth_errors import handle_auth_exception
+from cycode.cli.logger import logger
+
+
+def ensure_auth_command(ctx: typer.Context) -> None:
+    """Ensure the user is authenticated, triggering authentication if needed."""
+    auth_info = get_authorization_info(ctx)
+    if auth_info is not None:
+        logger.debug('Already authenticated')
+        return
+
+    logger.debug('Not authenticated, starting authentication')
+    try:
+        auth_manager = AuthManager()
+        auth_manager.authenticate()
+    except Exception as err:
+        handle_auth_exception(ctx, err)

{cycode-3.11.3 → cycode-3.11.5.dev2}/cycode/cli/apps/ai_guardrails/hooks_manager.py

@@ -5,17 +5,21 @@ Handles installation, removal, and status checking of AI IDE hooks.
 Supports multiple IDEs: Cursor, Claude Code (future).
 """
 
+import copy
 import json
 from pathlib import Path
 from typing import Optional
 
+import yaml
+
 from cycode.cli.apps.ai_guardrails.consts import (
-    CYCODE_SCAN_PROMPT_COMMAND,
     DEFAULT_IDE,
     IDE_CONFIGS,
     AIIDEType,
+    PolicyMode,
     get_hooks_config,
 )
+from cycode.cli.apps.ai_guardrails.scan.consts import DEFAULT_POLICY, POLICY_FILE_NAME
 from cycode.logger import get_logger
 
 logger = get_logger('AI Guardrails Hooks')
@@ -58,6 +62,13 @@ def save_hooks_file(hooks_path: Path, hooks_config: dict) -> bool:
         return False
 
 
+_CYCODE_COMMAND_MARKERS = ('cycode ai-guardrails',)
+
+
+def _is_cycode_command(command: str) -> bool:
+    return any(marker in command for marker in _CYCODE_COMMAND_MARKERS)
+
+
 def is_cycode_hook_entry(entry: dict) -> bool:
     """Check if a hook entry is from cycode-cli.
 
@@ -68,7 +79,7 @@ def is_cycode_hook_entry(entry: dict) -> bool:
     """
     # Check Cursor format (flat command)
     command = entry.get('command', '')
-    if CYCODE_SCAN_PROMPT_COMMAND in command:
+    if _is_cycode_command(command):
         return True
 
     # Check Claude Code format (nested hooks array)
@@ -76,14 +87,58 @@ def is_cycode_hook_entry(entry: dict) -> bool:
     for hook in hooks:
         if isinstance(hook, dict):
             hook_command = hook.get('command', '')
-            if CYCODE_SCAN_PROMPT_COMMAND in hook_command:
+            if _is_cycode_command(hook_command):
                 return True
 
     return False
 
 
+def _load_policy(policy_path: Path) -> dict:
+    """Load existing policy file merged with defaults, or return defaults if not found."""
+    if not policy_path.exists():
+        return copy.deepcopy(DEFAULT_POLICY)
+    try:
+        existing = yaml.safe_load(policy_path.read_text(encoding='utf-8')) or {}
+    except Exception:
+        existing = {}
+    return {**copy.deepcopy(DEFAULT_POLICY), **existing}
+
+
+def create_policy_file(scope: str, mode: PolicyMode, repo_path: Optional[Path] = None) -> tuple[bool, str]:
+    """Create or update the ai-guardrails.yaml policy file.
+
+    If the file already exists, only the mode field is updated.
+    If it doesn't exist, a new file is created from the default policy.
+
+    Args:
+        scope: 'user' for user-level, 'repo' for repository-level
+        mode: The policy mode to set
+        repo_path: Repository path (required if scope is 'repo')
+
+    Returns:
+        Tuple of (success, message)
+    """
+    config_dir = repo_path / '.cycode' if scope == 'repo' and repo_path else Path.home() / '.cycode'
+    policy_path = config_dir / POLICY_FILE_NAME
+
+    policy = _load_policy(policy_path)
+
+    policy['mode'] = mode.value
+
+    try:
+        config_dir.mkdir(parents=True, exist_ok=True)
+        policy_path.write_text(yaml.dump(policy, default_flow_style=False, sort_keys=False), encoding='utf-8')
+        return True, f'AI guardrails policy ({mode.value} mode) set: {policy_path}'
+    except Exception as e:
+        logger.error('Failed to create policy file', exc_info=e)
+        return False, f'Failed to create policy file: {policy_path}'
+
+
 def install_hooks(
-    scope: str = 'user', repo_path: Optional[Path] = None, ide: AIIDEType = DEFAULT_IDE
+    scope: str = 'user',
+    repo_path: Optional[Path] = None,
+    ide: AIIDEType = DEFAULT_IDE,
+    report_mode: bool = False,
 ) -> tuple[bool, str]:
     """
     Install Cycode AI guardrails hooks.
@@ -92,6 +147,7 @@ def install_hooks(
         scope: 'user' for user-level hooks, 'repo' for repository-level hooks
         repo_path: Repository path (required if scope is 'repo')
         ide: The AI IDE type (default: Cursor)
+        report_mode: If True, install hooks in async mode (non-blocking)
 
     Returns:
         Tuple of (success, message)
@@ -104,7 +160,7 @@ def install_hooks(
     existing.setdefault('hooks', {})
 
     # Get IDE-specific hooks configuration
-    hooks_config = get_hooks_config(ide)
+    hooks_config = get_hooks_config(ide, async_mode=report_mode)
 
     # Add/update Cycode hooks
     for event, entries in hooks_config['hooks'].items():

{cycode-3.11.3 → cycode-3.11.5.dev2}/cycode/cli/apps/ai_guardrails/install_command.py

@@ -11,8 +11,8 @@ from cycode.cli.apps.ai_guardrails.command_utils import (
     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 install_hooks
+from cycode.cli.apps.ai_guardrails.consts import IDE_CONFIGS, AIIDEType, InstallMode, PolicyMode
+from cycode.cli.apps.ai_guardrails.hooks_manager import create_policy_file, install_hooks
 
 
 def install_command(
@@ -43,6 +43,15 @@ def install_command(
             resolve_path=True,
         ),
     ] = None,
+    mode: Annotated[
+        InstallMode,
+        typer.Option(
+            '--mode',
+            '-m',
+            help='Installation mode: "report" for async non-blocking hooks with warn policy, '
+            '"block" for sync blocking hooks.',
+        ),
+    ] = InstallMode.REPORT,
 ) -> None:
     """Install AI guardrails hooks for supported IDEs.
 
@@ -50,7 +59,8 @@ def install_command(
     and MCP tool calls for secrets before they are sent to AI models.
 
     Examples:
-        cycode ai-guardrails install                    # Install for all projects (user scope)
+        cycode ai-guardrails install                    # Install in report mode (default)
+        cycode ai-guardrails install --mode block       # Install in block mode
         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
@@ -66,7 +76,8 @@ def install_command(
     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)
+        report_mode = mode == InstallMode.REPORT
+        success, message = install_hooks(scope, repo_path, ide=current_ide, report_mode=report_mode)
         results.append((ide_name, success, message))
 
     # Report results for each IDE
@@ -81,14 +92,31 @@ def install_command(
             all_success = False
 
     if any_success:
-        console.print()
-        console.print('[bold]Next steps:[/]')
-        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.[/]')
+        policy_mode = PolicyMode.WARN if mode == InstallMode.REPORT else PolicyMode.BLOCK
+        _install_policy(scope, repo_path, policy_mode)
+        _print_next_steps(results, mode)
 
     if not all_success:
         raise typer.Exit(1)
+
+
+def _install_policy(scope: str, repo_path: Optional[Path], policy_mode: PolicyMode) -> None:
+    policy_success, policy_message = create_policy_file(scope, policy_mode, repo_path)
+    if policy_success:
+        console.print(f'[green]✓[/] {policy_message}')
+    else:
+        console.print(f'[red]✗[/] {policy_message}', style='bold red')
+
+
+def _print_next_steps(results: list[tuple[str, bool, str]], mode: InstallMode) -> None:
+    console.print()
+    console.print('[bold]Next steps:[/]')
+    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()
+    if mode == InstallMode.REPORT:
+        console.print('[dim]Report mode: hooks run async (non-blocking) and policy is set to warn.[/]')
+    else:
+        console.print('[dim]The hooks will scan prompts, file reads, and MCP tool calls for secrets.[/]')

{cycode-3.11.3 → cycode-3.11.5.dev2}/cycode/cli/apps/ai_guardrails/scan/handlers.py

@@ -345,12 +345,10 @@ def _scan_path_for_secrets(ctx: typer.Context, file_path: str, policy: dict) ->
     if not file_path or not os.path.exists(file_path):
         return None, None
 
-    with open(file_path, encoding='utf-8', errors='replace') as f:
-        content = f.read()
-
-    # Truncate content based on policy max_bytes
     max_bytes = get_policy_value(policy, 'secrets', 'max_bytes', default=200000)
-    content = truncate_utf8(content, max_bytes)
+
+    with open(file_path, encoding='utf-8', errors='replace') as f:
+        content = f.read(max_bytes)
 
     # Get timeout from policy
     timeout_ms = get_policy_value(policy, 'secrets', 'timeout_ms', default=30000)

{cycode-3.11.3 → cycode-3.11.5.dev2}/pyproject.toml

@@ -21,7 +21,7 @@ classifiers = [
     "Programming Language :: Python :: 3.14",
 ]
 dynamic = ["dependencies"]
-version = "3.11.3"
+version = "3.11.5.dev2"
 
 [project.scripts]
 cycode = "cycode.cli.app:app"

cycode-3.11.3/cycode/__init__.py

@@ -1 +0,0 @@
-__version__ = '3.11.3'  # DON'T TOUCH. Placeholder. Will be filled automatically on poetry build from Git Tag