"""
Response builders for different AI IDE hooks.
Each IDE has its own response format for hooks. This module provides
an abstract interface and concrete implementations for each supported IDE.
"""
from abc import ABC, abstractmethod
class IDEResponseBuilder(ABC):
"""Abstract base class for IDE-specific response builders."""
@abstractmethod
def allow_permission(self) -> dict:
"""Build response to allow file read or MCP execution."""
@abstractmethod
def deny_permission(self, user_message: str, agent_message: str) -> dict:
"""Build response to deny file read or MCP execution."""
@abstractmethod
def ask_permission(self, user_message: str, agent_message: str) -> dict:
"""Build response to ask user for permission (warn mode)."""
@abstractmethod
def allow_prompt(self) -> dict:
"""Build response to allow prompt submission."""
@abstractmethod
def deny_prompt(self, user_message: str) -> dict:
"""Build response to deny prompt submission."""
class CursorResponseBuilder(IDEResponseBuilder):
"""Response builder for Cursor IDE hooks.
Cursor hook response formats:
- beforeSubmitPrompt: {"continue": bool, "user_message": str}
- beforeReadFile: {"permission": str, "user_message": str, "agent_message": str}
- beforeMCPExecution: {"permission": str, "user_message": str, "agent_message": str}
"""
def allow_permission(self) -> dict:
"""Allow file read or MCP execution."""
return {'permission': 'allow'}
def deny_permission(self, user_message: str, agent_message: str) -> dict:
"""Deny file read or MCP execution."""
return {'permission': 'deny', 'user_message': user_message, 'agent_message': agent_message}
def ask_permission(self, user_message: str, agent_message: str) -> dict:
"""Ask user for permission (warn mode)."""
return {'permission': 'ask', 'user_message': user_message, 'agent_message': agent_message}
def allow_prompt(self) -> dict:
"""Allow prompt submission."""
return {'continue': True}
def deny_prompt(self, user_message: str) -> dict:
"""Deny prompt submission."""
return {'continue': False, 'user_message': user_message}
# Registry of response builders by IDE name
_RESPONSE_BUILDERS: dict[str, IDEResponseBuilder] = {
'cursor': CursorResponseBuilder(),
}
def get_response_builder(ide: str = 'cursor') -> IDEResponseBuilder:
"""Get the response builder for a specific IDE.
Args:
ide: The IDE name (e.g., 'cursor', 'claude-code')
Returns:
IDEResponseBuilder instance for the specified IDE
Raises:
ValueError: If the IDE is not supported
"""
builder = _RESPONSE_BUILDERS.get(ide.lower())
if not builder:
raise ValueError(f'Unsupported IDE: {ide}. Supported IDEs: {list(_RESPONSE_BUILDERS.keys())}')
return builder
