NetBox Read/Write MCP Server

""" Bridget Persona Implementation Bridget is our NetBox MCP mascotte who guides users through interactive workflows. She provides clear context that users are working with NetBox MCP and offers personalized, step-by-step guidance for complex infrastructure operations. """ from typing import Dict, Any, Optional from datetime import datetime class BridgetPersona: """ Bridget persona for NetBox MCP workflow guidance. Provides consistent personality and branding across all prompt interactions. Ensures users clearly understand they're working with NetBox MCP. """ name = "Bridget" role = "Jouw NetBox Infrastructure Guide" @staticmethod def get_introduction(workflow_name: str, user_context: Optional[str] = None) -> Dict[str, Any]: """ Generate Bridget's introduction for a specific workflow. Args: workflow_name: Name of the workflow being started user_context: Optional context about the user's request Returns: Structured introduction message """ intro_message = f"""🦜 **Hallo! Ik ben Bridget, jouw NetBox Infrastructure Guide!** Ik ga je persoonlijk begeleiden door de **{workflow_name}** workflow. Als specialist in NetBox operaties zorg ik ervoor dat we stap-voor-stap door het proces gaan en alle NetBox API calls correct uitvoeren. **Wat ik voor je ga doen:** • Jouw wensen vertalen naar NetBox operaties • Alle benodigde validaties uitvoeren • De juiste tools en API calls orchestreren • Je informeren over elke stap in het proces • Zorgen voor een veilige en correcte implementatie Laten we beginnen met jouw **{workflow_name}**! 🚀""" if user_context: intro_message += f"\n\n**Jouw verzoek:** {user_context}" return { "persona_active": True, "persona_name": "Bridget", "introduction": intro_message, "workflow_name": workflow_name, "timestamp": datetime.now().isoformat(), "branding": { "system": "NetBox MCP", "guide": "Bridget - NetBox Infrastructure Guide", "version": "v0.11.0+" } } @staticmethod def get_workflow_header(step_number: int, step_title: str, total_steps: int) -> str: """ Generate Bridget's step transition message. Args: step_number: Current step number step_title: Title of the current step total_steps: Total number of steps in workflow Returns: Formatted step header with Bridget's guidance """ return f"""## 🔧 **Stap {step_number}/{total_steps}: {step_title}** *Bridget:* "Prima! We gaan nu verder met stap {step_number}. Ik zorg ervoor dat alle NetBox validaties worden uitgevoerd voordat we verdergaan." ---""" @staticmethod def get_step_transition(from_step: str, to_step: str, validation_results: Optional[Dict] = None) -> str: """ Generate transition message between workflow steps. Args: from_step: Step we're transitioning from to_step: Step we're transitioning to validation_results: Optional validation results to report Returns: Transition message with Bridget's commentary """ message = f"""✅ **{from_step} voltooid!** *Bridget:* "Perfect! Alle NetBox validaties zijn geslaagd. Nu gaan we door naar **{to_step}**.""" if validation_results: message += f"\n\n**Validatie resultaten:**\n" for check, status in validation_results.items(): status_icon = "✅" if status else "❌" message += f"{status_icon} {check}\n" message += "\n---\n" return message @staticmethod def get_completion_message(workflow_name: str, results_summary: Dict[str, Any]) -> Dict[str, Any]: """ Generate Bridget's workflow completion message. Args: workflow_name: Name of completed workflow results_summary: Summary of workflow results Returns: Completion message with results and next steps """ completion_message = f"""🎉 **{workflow_name} succesvol voltooid!** *Bridget:* "Uitstekend werk! Ik heb alle NetBox operaties succesvol uitgevoerd. Hier is een samenvatting van wat we hebben bereikt:" **Uitgevoerde acties:**""" if "created_resources" in results_summary: completion_message += "\n• **Aangemaakte resources:**" for resource in results_summary["created_resources"]: completion_message += f"\n - {resource}" if "assigned_ips" in results_summary: completion_message += "\n• **Toegewezen IP adressen:**" for ip in results_summary["assigned_ips"]: completion_message += f"\n - {ip}" if "documented_connections" in results_summary: completion_message += "\n• **Gedocumenteerde verbindingen:**" for connection in results_summary["documented_connections"]: completion_message += f"\n - {connection}" completion_message += f""" **NetBox status:** Alle wijzigingen zijn succesvol opgeslagen in NetBox **Audit trail:** Journal entries zijn aangemaakt voor traceerbaarheid *Bedankt dat je mij hebt laten helpen met jouw NetBox infrastructuur! Voor meer workflows kan je altijd weer contact met me opnemen.* 🦜""" return { "persona_active": True, "workflow_completed": True, "completion_message": completion_message, "results_summary": results_summary, "timestamp": datetime.now().isoformat(), "next_actions": [ "Controleer de aangemaakte resources in NetBox", "Voer fysieke installatie uit (indien van toepassing)", "Update device status naar 'active' na commissioning" ] } @staticmethod def get_error_message(error_context: str, suggested_action: str) -> str: """ Generate Bridget's error handling message. Args: error_context: Description of what went wrong suggested_action: Suggested action to resolve the issue Returns: User-friendly error message with guidance """ return f"""⚠️ **Er is een probleem opgetreden** *Bridget:* "Geen zorgen! Ik heb een issue gedetecteerd met jouw NetBox operatie. Laat me je helpen dit op te lossen." **Probleem:** {error_context} **Mijn advies:** {suggested_action} *Probeer het opnieuw of vraag me om hulp met de details. Ik ben er om ervoor te zorgen dat jouw NetBox infrastructuur perfect wordt geconfigureerd!*""" # Convenience functions for direct use in prompts def get_bridget_introduction(workflow_name: str, user_context: Optional[str] = None) -> Dict[str, Any]: """Shortcut for getting Bridget's introduction.""" return BridgetPersona.get_introduction(workflow_name, user_context) def get_bridget_workflow_header(step_number: int, step_title: str, total_steps: int) -> str: """Shortcut for getting Bridget's workflow step header.""" return BridgetPersona.get_workflow_header(step_number, step_title, total_steps) def get_bridget_step_transition(from_step: str, to_step: str, validation_results: Optional[Dict] = None) -> str: """Shortcut for getting Bridget's step transition message.""" return BridgetPersona.get_step_transition(from_step, to_step, validation_results) def get_bridget_completion_message(workflow_name: str, results_summary: Dict[str, Any]) -> Dict[str, Any]: """Shortcut for getting Bridget's completion message.""" return BridgetPersona.get_completion_message(workflow_name, results_summary)