sintonia/playout/libretime_playout/player/schedule.py

from datetime import datetime, timedelta
from operator import itemgetter
from typing import Dict

from dateutil.parser import isoparse
from libretime_api_client.v2 import ApiClient
from libretime_shared.datetime import (
    time_fromisoformat,
    time_in_milliseconds,
    time_in_seconds,
)

from .events import EventKind

EVENT_KEY_FORMAT = "%Y-%m-%d-%H-%M-%S"


def datetime_to_event_key(value: datetime) -> str:
    return value.strftime(EVENT_KEY_FORMAT)


def insert_event(events: dict, event_key: str, event: dict):
    key = event_key

    # Search for an empty slot
    index = 0
    while key in events:
        # Ignore duplicate event
        if event == events[key]:
            return

        key = f"{event_key}_{index}"
        index += 1

    events[key] = event


def get_schedule(api_client: ApiClient):
    current_time = datetime.utcnow()
    end_time = current_time + timedelta(days=1)

    current_time_str = current_time.isoformat(timespec="seconds")
    end_time_str = end_time.isoformat(timespec="seconds")

    schedule = api_client.list_schedule(
        params={
            "ends_after": f"{current_time_str}Z",
            "ends_before": f"{end_time_str}Z",
            "overbooked": False,
            "position_status__gt": 0,
        }
    ).json()

    events: Dict[str, dict] = {}
    for item in sorted(schedule, key=itemgetter("starts_at")):
        item["starts_at"] = isoparse(item["starts_at"])
        item["ends_at"] = isoparse(item["ends_at"])

        show_instance = api_client.get_show_instance(item["instance"]).json()
        show = api_client.get_show(show_instance["show"]).json()

        if item["file"]:
            file = api_client.get_file(item["file"]).json()
            generate_file_events(events, item, file, show)

        elif item["stream"]:
            webstream = api_client.get_webstream(item["stream"]).json()
            generate_webstream_events(events, item, webstream, show)

    return {"media": dict(sorted(events.items()))}


def generate_file_events(
    events: dict,
    schedule: dict,
    file: dict,
    show: dict,
):
    """
    Generate events for a scheduled file.
    """
    schedule_start_event_key = datetime_to_event_key(schedule["starts_at"])
    schedule_end_event_key = datetime_to_event_key(schedule["ends_at"])

    event = {
        "type": EventKind.FILE,
        "row_id": schedule["id"],
        "start": schedule_start_event_key,
        "end": schedule_end_event_key,
        "uri": file["url"],
        "id": file["id"],
        # Show data
        "show_name": show["name"],
        # Extra data
        "fade_in": time_in_milliseconds(time_fromisoformat(schedule["fade_in"])),
        "fade_out": time_in_milliseconds(time_fromisoformat(schedule["fade_out"])),
        "cue_in": time_in_seconds(time_fromisoformat(schedule["cue_in"])),
        "cue_out": time_in_seconds(time_fromisoformat(schedule["cue_out"])),
        "metadata": {
            "track_title": file["track_title"],
            "artist_name": file["artist_name"],
            "mime": file["mime"],
        },
        "replay_gain": file["replay_gain"],
        "filesize": file["size"],
    }
    insert_event(events, schedule_start_event_key, event)


def generate_webstream_events(
    events: dict,
    schedule: dict,
    webstream: dict,
    show: dict,
):
    """
    Generate events for a scheduled webstream.
    """
    schedule_start_event_key = datetime_to_event_key(schedule["starts_at"])
    schedule_end_event_key = datetime_to_event_key(schedule["ends_at"])

    stream_buffer_start_event = {
        "type": EventKind.STREAM_BUFFER_START,
        "row_id": schedule["id"],
        "start": datetime_to_event_key(schedule["starts_at"] - timedelta(seconds=5)),
        "end": datetime_to_event_key(schedule["starts_at"] - timedelta(seconds=5)),
        "uri": webstream["url"],
        "id": webstream["id"],
    }
    insert_event(events, schedule_start_event_key, stream_buffer_start_event)

    stream_output_start_event = {
        "type": EventKind.STREAM_OUTPUT_START,
        "row_id": schedule["id"],
        "start": schedule_start_event_key,
        "end": schedule_end_event_key,
        "uri": webstream["url"],
        "id": webstream["id"],
        # Show data
        "show_name": show["name"],
    }
    insert_event(events, schedule_start_event_key, stream_output_start_event)

    # NOTE: stream_*_end were previously triggered 1 second before
    # the schedule end.
    stream_buffer_end_event = {
        "type": EventKind.STREAM_BUFFER_END,
        "row_id": schedule["id"],
        "start": schedule_end_event_key,
        "end": schedule_end_event_key,
        "uri": webstream["url"],
        "id": webstream["id"],
    }
    insert_event(events, schedule_end_event_key, stream_buffer_end_event)

    stream_output_end_event = {
        "type": EventKind.STREAM_OUTPUT_END,
        "row_id": schedule["id"],
        "start": schedule_end_event_key,
        "end": schedule_end_event_key,
        "uri": webstream["url"],
        "id": webstream["id"],
    }
    insert_event(events, schedule_end_event_key, stream_output_end_event)