from ninja import Router, Schema
from typing import List, Optional
from core.models import ScheduleTemplate, Channel
from django.shortcuts import get_object_or_404
from datetime import date

router = Router(tags=["schedule"])

class ScheduleTemplateSchema(Schema):
    id: int
    name: str
    description: Optional[str] = None
    timezone_name: str
    valid_from_date: Optional[date] = None
    valid_to_date: Optional[date] = None
    priority: int
    is_active: bool
    channel_id: int

class ScheduleTemplateCreateSchema(Schema):
    name: str
    description: Optional[str] = None
    timezone_name: str
    valid_from_date: Optional[date] = None
    valid_to_date: Optional[date] = None
    priority: int = 0
    is_active: bool = True
    channel_id: int

@router.get("/template/", response=List[ScheduleTemplateSchema])
def list_schedule_templates(request):
    return ScheduleTemplate.objects.all()

@router.get("/template/{template_id}", response=ScheduleTemplateSchema)
def get_schedule_template(request, template_id: int):
    return get_object_or_404(ScheduleTemplate, id=template_id)

@router.post("/template/", response={201: ScheduleTemplateSchema})
def create_schedule_template(request, payload: ScheduleTemplateCreateSchema):
    channel = get_object_or_404(Channel, id=payload.channel_id)
    
    template = ScheduleTemplate.objects.create(
        channel=channel,
        name=payload.name,
        description=payload.description,
        timezone_name=payload.timezone_name,
        valid_from_date=payload.valid_from_date,
        valid_to_date=payload.valid_to_date,
        priority=payload.priority,
        is_active=payload.is_active
    )
    return 201, template

class GenerateScheduleSchema(Schema):
    target_date: date

@router.post("/generate/{channel_id}")
def generate_schedule(request, channel_id: int, payload: GenerateScheduleSchema):
    channel = get_object_or_404(Channel, id=channel_id)
    # Inline import to prevent circular dependency issues during initialization
    from core.services.scheduler import ScheduleGenerator
    
    generator = ScheduleGenerator(channel=channel)
    airings_created = generator.generate_for_date(payload.target_date)
    return {"status": "success", "airings_created": airings_created}