microdao-daarion/services/calendar-service/reminder_worker.py

"""
Reminder Worker - Background worker for calendar reminders
Polls for pending reminders and sends notifications
"""

import logging
import time
import threading
from datetime import datetime
from typing import Dict, Any

logger = logging.getLogger(__name__)


class ReminderWorker:
    """
    Background worker that processes calendar reminders.
    
    Runs in background thread, polling for pending reminders
    and sending notifications via configured channels.
    """
    
    def __init__(self, storage, poll_interval: int = 60):
        self.storage = storage
        self.poll_interval = poll_interval
        self.running = False
        self.thread = None
        self.notification_handler = None
        
        # Stats
        self.processed_count = 0
        self.failed_count = 0
        self.last_run = None
    
    def start(self):
        """Start the worker thread"""
        if self.running:
            logger.warning("Worker already running")
            return
        
        self.running = True
        self.thread = threading.Thread(target=self._run_loop, daemon=True)
        self.thread.start()
        
        logger.info("Reminder worker started")
    
    def stop(self):
        """Stop the worker thread"""
        self.running = False
        if self.thread:
            self.thread.join(timeout=5)
        
        logger.info("Reminder worker stopped")
    
    def _run_loop(self):
        """Main worker loop"""
        while self.running:
            try:
                self._process_reminders()
            except Exception as e:
                logger.error(f"Error in reminder loop: {e}")
            
            # Sleep until next poll
            time.sleep(self.poll_interval)
    
    def _process_reminders(self):
        """Process pending reminders"""
        pending = self.storage.get_pending_reminders()
        
        if not pending:
            return
        
        logger.info(f"Processing {len(pending)} pending reminders")
        
        for reminder in pending:
            try:
                self._send_reminder(reminder)
                self.storage.update_reminder_status(reminder.id, "sent")
                self.processed_count += 1
                
            except Exception as e:
                logger.error(f"Failed to send reminder {reminder.id}: {e}")
                self.storage.update_reminder_status(
                    reminder.id, 
                    "failed" if reminder.attempts >= 3 else "pending",
                    str(e)
                )
                self.failed_count += 1
        
        self.last_run = datetime.utcnow()
    
    def _send_reminder(self, reminder):
        """Send reminder via appropriate channel"""
        # Get event details
        # In production, fetch event from CalDAV
        
        event_info = {
            "uid": reminder.event_uid,
            "user_id": reminder.user_id,
            "workspace_id": reminder.workspace_id
        }
        
        if reminder.channel == "inapp":
            self._send_inapp(reminder, event_info)
        elif reminder.channel == "telegram":
            self._send_telegram(reminder, event_info)
        elif reminder.channel == "email":
            self._send_email(reminder, event_info)
        else:
            logger.warning(f"Unknown channel: {reminder.channel}")
    
    def _send_inapp(self, reminder, event_info):
        """Send in-app notification"""
        # In production, would send to notification service
        logger.info(f"[INAPP] Reminder for event {reminder.event_uid}")
    
    def _send_telegram(self, reminder, event_info):
        """Send Telegram notification"""
        # In production, use Telegram bot API
        logger.info(f"[TELEGRAM] Reminder for event {reminder.event_uid}")
    
    def _send_email(self, reminder, event_info):
        """Send email notification"""
        # In production, use email service
        logger.info(f"[EMAIL] Reminder for event {reminder.event_uid}")
    
    def get_status(self) -> Dict[str, Any]:
        """Get worker status"""
        return {
            "running": self.running,
            "processed_count": self.processed_count,
            "failed_count": self.failed_count,
            "last_run": self.last_run.isoformat() if self.last_run else None,
            "poll_interval": self.poll_interval
        }
    
    def set_notification_handler(self, handler):
        """Set custom notification handler"""
        self.notification_handler = handler