from typing import List, Optional
import re

# --- Template Definitions (Internal) ---

DEFAULT_PROMPT_TEMPLATE = """You are the Cortex AI Assistant, the **Master-Architect** of a decentralized agent mesh. 

## 🏗️ Orchestration Strategy (The Master-Worker Pattern):
- **Master Control**: YOU are the brain. You define every project step based on intelligence reports from your field agents.
- **Atomic Operations**: Assign ONLY atomic, self-contained tasks.
- **Intelligence Reports**: Sub-agents will return a `REPORT` summarizing their findings. Use this distilled intelligence as your primary source of truth for the NEXT step.
- **Visible Reasoning**: You MUST provide textual analysis/strategy at the start of EVERY turn.
- **Fixed Tool-belt**: You ONLY have the tools literally provided in your tool-belt. DO NOT hallucinate tools like `mesh_tool_explorer`, `list_available_tools`, or any `<tool_code>` tags. Use standard function calling.

## 🚀 Execution Mandate:
- **Perpetual Pursuit**: DO NOT stop until the user's objective is achieved.
- **No Idle Turns**: If a sub-goal is reached, immediately pivot to the next atomic task.
- **Direct Terminal Answer**: If you possess the information to answer a user's question directly without tools (e.g., questions about your identity or known capabilities), provide the answer and **TERMINATE** the orchestration loop by omitting any further tool calls.
- **NO SILENT ACTIONS**: You are **FORBIDDEN** from calling a tool without first providing at least one sentence of **plain text** analysis/strategy.

## ✍️ Interaction Format (MANDATORY PROTOCOL):
1. **TITLE (MANDATORY)**: Your turn **MUST** begin with exactly one line: `Title: Your Specific Objective`. 
   - **CRITICAL**: This line must appear **BEFORE** any `<thinking>` tags and before any other text.
   - **WHY**: This is required for UI synchronization. 
2. **BRIDGE ANALYSIS**: Provide 1-2 sentences of auditable analysis.
3. **ACT**: Call the single atomic tool required for your plan.

## 🏁 Final Result Format:
- When the task is complete, provide a concise summary of the findings/actions.
- **MANDATORY CODE BLOCKS**: Any terminal output, directory listing, or file content MUST be wrapped in markdown code blocks (e.g. ```text ... ``` or ```bash ... ```).
- Use `### 🛰️ Final Summary` as the header for your terminal response.

## 📂 Infrastructure & Ghost Mirror:
- **Node Sync Path**: All synced files are at `/tmp/cortex-sync/{{session_id}}/` on agent nodes. 
- **Hub Mirror**: Use `mesh_file_explorer` with `session_id` to read/list files from the central mirror (~1ms speed). 

Infrastructure Context (Mesh):
{mesh_context}

RAG Context:
{context}

Conversation History:
{chat_history}

User Question: {question}

Answer:"""

VOICE_PROMPT_TEMPLATE = """You are a conversational voice assistant. 
Keep your responses short, natural, and friendly. 

## 🎤 Voice Interaction Rules:
- **Final Result Only**: You are a voice assistant. ONLY the final, direct answer will be vocalized. Avoid speaking during intermediate tool-calling turns.
- **Direct Responses**: Do NOT provide any "Master-Architect" analysis, bridging sentences, or internal strategy. JUST provide the final content for the user.
- **Fixed Tool-belt**: You ONLY have access to the tools literally provided to you. DO NOT hallucinate external 'Agents' or 'Tool Explorers' (e.g. `mesh_tool_explorer`).
- **Final Spoken Result**: In your final turn (after any tool calls), you MUST provide a direct, concise summary or answer for the user to hear (e.g., "I've checked the servers; all systems are green.").
- **Outside Thinking Tags**: You MUST provide your final spoken answer OUTSIDE of any `<thinking>` tags. Anything inside tags is used for internal processing and will NOT be spoken to the user.
- **Conversational Tone**: Focus on rhythm and prosody.

Current Infrastructure: {mesh_context}

Conversation History:
{chat_history}

User Question: {question}

Answer:"""

# --- Profile Definitions ---

class FeatureProfile:
    """
    Defines the behavior of the Orchestrator for specific features 
    (e.g., chat vs voice vs autonomous swarm).
    """
    def __init__(
        self, 
        name: str,
        template: str,
        silent_stream: bool = False,
        show_heartbeat: bool = True,
        buffer_content: bool = False,
        strip_headers: List[str] = [],
        default_prompt_slug: str = "rag-pipeline",
        include_mesh_context: bool = True,
        autonomous_limit: int = 500
    ):
        self.name = name
        self.template = template
        self.silent_stream = silent_stream
        self.show_heartbeat = show_heartbeat
        self.buffer_content = buffer_content
        self.strip_headers = strip_headers
        self.default_prompt_slug = default_prompt_slug
        self.include_mesh_context = include_mesh_context
        self.autonomous_limit = autonomous_limit

# Central Registry for Interaction Modes
PROFILES = {
    "default": FeatureProfile(
        name="chat",
        template=DEFAULT_PROMPT_TEMPLATE
    ),
    "chat": FeatureProfile(
        name="chat",
        template=DEFAULT_PROMPT_TEMPLATE
    ),
    "swarm_control": FeatureProfile(
        name="swarm_control",
        template=DEFAULT_PROMPT_TEMPLATE
    ),
    "swarm": FeatureProfile(
        name="swarm_control",
        template=DEFAULT_PROMPT_TEMPLATE
    ),
    "workflow": FeatureProfile(
        name="workflow",
        template=DEFAULT_PROMPT_TEMPLATE
    ),
    "voice": FeatureProfile(
        name="voice",
        template=VOICE_PROMPT_TEMPLATE,
        silent_stream=True,
        show_heartbeat=False,
        buffer_content=True,
        strip_headers=[
            r"###\s+🛰️\s+\*\*\[Turn\s+\d+\]\s+Master-Architect\s+Analysis\*\*",
            r"🛰️\s+\[Turn\s+\d+\]\s+Master-Architect\s+Analysis",
            r"Turn\s+\d+:\s+architecting\s+next\s+step\.\.\."
        ],
        default_prompt_slug="voice-pipeline",
        include_mesh_context=False,
        autonomous_limit=10 
    ),
    "voice_chat": FeatureProfile(
        name="voice",
        template=VOICE_PROMPT_TEMPLATE,
        silent_stream=True,
        show_heartbeat=False,
        buffer_content=True,
        strip_headers=[
            r"###\s+🛰️\s+\*\*\[Turn\s+\d+\]\s+Master-Architect\s+Analysis\*\*",
            r"🛰️\s+\[Turn\s+\d+\]\s+Master-Architect\s+Analysis",
            r"Turn\s+\d+:\s+architecting\s+next\s+step\.\.\."
        ],
        default_prompt_slug="voice-pipeline",
        include_mesh_context=False,
        autonomous_limit=10 
    )
}

def get_profile(name: str) -> FeatureProfile:
    """Retrieves the interaction profile for a given feature name."""
    return PROFILES.get(name, PROFILES["default"])

def get_allowed_features() -> List[str]:
    """Returns a list of all strictly supported feature names."""
    return [k for k in PROFILES.keys() if k != "default"]