Fresh start - excluded large ROM JSON files

agents/calendar_agent.py

@@ -0,0 +1,118 @@
+#!/usr/bin/env python3
+"""
+Calendar Agent for Discord Channel 1474636036905631867
+
+Simple agent to add/view Home Assistant calendar events via natural language.
+"""
+
+import sys
+import os
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+from api.ha_calendar import (
+    add_calendar_event,
+    get_todays_events,
+    get_upcoming_events,
+    format_events,
+    format_upcoming,
+)
+from datetime import datetime
+import re
+
+
+class CalendarAgent:
+    """Lightweight calendar agent for Discord."""
+    
+    def __init__(self):
+        self.channel_id = "1474636036905631867"
+    
+    def handle_message(self, message: str) -> str:
+        """Process incoming message and return response."""
+        msg = message.lower().strip()
+        
+        # Add event patterns
+        add_patterns = [
+            r"add\s+(.+?)\s+on\s+(\d{4}-\d{2}-\d{2})\s*(?:at\s*)?(\d{1,2}:?\d{2})?\s*(am|pm)?",
+            r"schedule\s+(.+?)\s+for\s+(\d{4}-\d{2}-\d{2})\s*(?:at\s*)?(\d{1,2}:?\d{2})?\s*(am|pm)?",
+            r"new\s+event:\s*(.+?)\s+(\d{4}-\d{2}-\d{2})\s*(\d{1,2}:?\d{2})?",
+        ]
+        
+        for pattern in add_patterns:
+            match = re.search(pattern, msg, re.IGNORECASE)
+            if match:
+                return self._handle_add_event(match)
+        
+        # Today's events
+        if any(p in msg for p in ["today's events", "what do i have today", "today's schedule", "today schedule"]):
+            return self._handle_today()
+        
+        # Upcoming events
+        if any(p in msg for p in ["upcoming", "what's coming", "whats coming", "show me", "this week", "this month"]):
+            return self._handle_upcoming()
+        
+        return None  # Not a calendar command
+    
+    def _handle_add_event(self, match) -> str:
+        """Handle adding a new event."""
+        groups = match.groups()
+        summary = groups[0].strip()
+        
+        date = groups[1]
+        time = groups[2] if len(groups) > 2 else None
+        ampm = groups[3] if len(groups) > 3 else None
+        
+        # Build datetime string
+        if time:
+            time = time.replace(":", "")
+            if ampm and ampm.lower() == "pm" and len(time) <= 2:
+                # Convert 3pm to 15:00
+                hour = int(time)
+                if hour < 12:
+                    hour += 12
+                    time = f"{hour:02d}:00"
+            elif ampm and ampm.lower() == "am":
+                time = f"{int(time):02d}:00"
+            start_dt = f"{date} {time}"
+        else:
+            # Default to 9 AM
+            start_dt = f"{date} 09:00"
+        
+        result = add_calendar_event(summary, start_dt)
+        
+        if result.get("success"):
+            return f"[OK] Added: **{summary}** on {date} at {time or '9:00 AM'}"
+        else:
+            return f"[ERROR] Failed to add event: {result.get('error', 'Unknown error')}"
+    
+    def _handle_today(self) -> str:
+        """Get today's events."""
+        events = get_todays_events()
+        return format_events(events)
+    
+    def _handle_upcoming(self, days: int = 7) -> str:
+        """Get upcoming events."""
+        events = get_upcoming_events(days)
+        return format_upcoming(events)
+
+
+def main():
+    """CLI entrypoint for testing."""
+    agent = CalendarAgent()
+    print("Calendar Agent ready. Type 'quit' to exit.")
+    print("Try: 'Add meeting on 2026-02-22 at 3pm' or 'What do I have today?'")
+    print()
+    
+    while True:
+        user_input = input("You: ").strip()
+        if user_input.lower() in ["quit", "exit", "q"]:
+            break
+        
+        response = agent.handle_message(user_input)
+        if response:
+            print(f"Agent: {response}")
+        else:
+            print("Agent: I'm a calendar agent. Try: 'Add event on [date] at [time]' or 'Show upcoming events'")
+
+
+if __name__ == "__main__":
+    main()