#!/usr/bin/env python3

import sys
from pathlib import Path
from datetime import datetime, date
from typing import Optional, NamedTuple
from dataclasses import dataclass

import requests
from bs4 import BeautifulSoup
from icalendar import Calendar
import recurring_ical_events
import locale

@dataclass
class EventDetails:
    weekday: str
    date: str
    time: str
    summary: str
    location: str

    def to_html(self) -> str:
        return (
            f'<div class="nextevent">{self.weekday} '
            f'<strong>{self.date}{self.time}, {self.summary}</strong>'
            f'{self.location}</div>'
        )

class EventProcessor:
    def __init__(self, ics_url: str, content_file: str):
        self.ics_url = ics_url
        self.content_file = Path(content_file)
        self.fallback_html = (
            '<div class="nextevent">Leider unbekannt, aber '
            '<strong>frag mal den Vorstand</strong> der müsste es wissen</div>'
        )
    
    @property
    def fallback_content(self) -> str:
        return f"""_model: htmlpage
---
title: Willkommen beim PC Stammertal
---
html:

<h3>Unser nächster Anlass: </h3><br/>
{self.fallback_html}
<div class="threecolumn">
  <div>
    <a href="termine/"><img src=" /images/termine_square.jpg" alt="Terminkalender"> All unsere Termine</a><!-- Fallback>
  </div>
  <div>
    <a href="about/"><img src="/images/about_square.jpg" alt="Buch"> Alle Infos über uns</a>
  </div>
  <div>
    <a href="kontakt/"><img src="/images/kontakt_square.jpg" alt="Briefe"> Kontaktiere uns</a>
  </div>
</div>
---
_template:

page.html
"""
        
    def setup_locale(self) -> None:
        try:
            locale.setlocale(locale.LC_TIME, 'de_DE.UTF-8')
        except locale.Error as e:
            print(f"Warning: Failed to set locale: {e}", file=sys.stderr)

    def fetch_calendar(self) -> Optional[Calendar]:
        try:
            response = requests.get(self.ics_url, timeout=10)
            response.raise_for_status()
            return Calendar.from_ical(response.content)
        except (requests.RequestException, ValueError) as e:
            print(f"Error fetching calendar: {e}", file=sys.stderr)
            return None

    def format_event_time(self, start: datetime | date) -> str:
        return "" if isinstance(start, date) and not isinstance(start, datetime) else start.strftime(" %-H:%M")

    def get_next_event(self, calendar: Calendar) -> Optional[EventDetails]:
        try:
            events = recurring_ical_events.of(calendar).after(datetime.now())
            event = next(events)
            start = event.get('dtstart').dt
            
            return EventDetails(
                weekday=start.strftime("%A"),
                date=start.strftime("%-d. %B"),
                time=self.format_event_time(start),
                summary=event.get('summary', ''),
                location=f", {event.get('location')}" if event.get('location') else ""
            )
            
        except (StopIteration, AttributeError) as e:
            print(f"No upcoming events found: {e}", file=sys.stderr)
            return None

    def read_current_content(self) -> tuple[str, str]:
        try:
            content = self.content_file.read_text()
            soup = BeautifulSoup(content, 'html.parser')
            events = soup.find_all('div', {'class': 'nextevent'})
            return content, str(events[0]) if len(events) == 1 else ""
            
        except (IOError, IndexError) as e:
            print(f"Error reading content file: {e}", file=sys.stderr)
            return "", ""

    def process(self) -> str:
        self.setup_locale()
        
        content, source_str = self.read_current_content()
        if not content:
            return self.fallback_content
            
        if len(source_str) > 0 and source_str in content:
            calendar = self.fetch_calendar()
            if not calendar:
                return content.replace(source_str, self.fallback_html).rstrip()
                
            event = self.get_next_event(calendar)
            if not event:
                return content.replace(source_str, self.fallback_html).rstrip()
                
            return content.replace(source_str, event.to_html()).rstrip()
            
        return self.fallback_content

def main() -> None:
    if len(sys.argv) != 2:
        print(f"Usage: {sys.argv[0]} <ICS_URL>", file=sys.stderr)
        sys.exit(1)
        
    processor = EventProcessor(
        ics_url=sys.argv[1],
        content_file="/opt/lektor/project/content/contents.lr"
    )
    
    print(processor.process(), end='')
    print("")
    print("")

if __name__ == "__main__":
    main()