import random
from datetime import datetime

import pytest
from libretime_api_client.v2 import ApiClient

from libretime_playout.liquidsoap.models import StreamPreferences
from libretime_playout.player.events import (
    ActionEvent,
    EventKind,
    FileEvent,
    WebStreamEvent,
    event_isoparse,
)
from libretime_playout.player.schedule import (
    generate_file_events,
    generate_live_events,
    generate_webstream_events,
    get_schedule,
)


@pytest.fixture(name="api_client")
def _api_client_fixture():
    base_url = "http://localhost"
    return ApiClient(base_url=base_url, api_key="test_key")


SHOW_1 = {"id": 1, "name": "Show 1", "live_enabled": False}
SHOW_2 = {"id": 2, "name": "Show 2", "live_enabled": False}
SHOW_3 = {"id": 3, "name": "Show 3", "live_enabled": True}
SHOW_4 = {"id": 4, "name": "Show 4", "live_enabled": False}

SHOW_INSTANCE_1 = {
    "id": 1,
    "starts_at": "2022-09-05T11:00:00Z",
    "ends_at": "2022-09-05T11:10:00Z",
    "show": 1,
}
SHOW_INSTANCE_2 = {
    "id": 2,
    "starts_at": "2022-09-05T11:10:00Z",
    "ends_at": "2022-09-05T12:10:00Z",
    "show": 2,
}
SHOW_INSTANCE_3 = {
    "id": 3,
    "starts_at": "2022-09-05T12:10:00Z",
    "ends_at": "2022-09-05T13:00:00Z",
    "show": 3,
}
SHOW_INSTANCE_4 = {
    "id": 4,
    "starts_at": "2022-09-05T13:00:00Z",
    "ends_at": "2022-09-05T14:10:00Z",
    "show": 4,
}

FILE_1 = {
    "id": 1,
    "mime": "audio/flac",
    "length": "00:03:41.041723",
    "replay_gain": "4.52",
    "cue_in": "00:00:08.252450",
    "cue_out": "00:03:27.208000",
    "artist_name": "Nils Frahm",
    "album_title": "Tripping with Nils Frahm",
    "track_title": "The Dane",
    "url": None,
    "size": 10000,
}
FILE_2 = {
    "id": 2,
    "mime": "audio/flac",
    "length": "00:06:08.668798",
    "replay_gain": "11.46",
    "cue_in": "00:00:13.700800",
    "cue_out": "00:05:15.845000",
    "artist_name": "Nils Frahm",
    "album_title": "Tripping with Nils Frahm",
    "track_title": "My Friend the Forest",
    "url": None,
    "size": 10000,
}
FILE_3 = {
    "id": 3,
    "mime": "audio/flac",
    "length": "00:14:18.400000",
    "replay_gain": "-2.13",
    "cue_in": "00:00:55.121100",
    "cue_out": "00:14:18.400000",
    "artist_name": "Nils Frahm",
    "album_title": "Tripping with Nils Frahm",
    "track_title": "All Melody",
    "url": None,
    "size": 10000,
}
FILE_4 = {
    "id": 4,
    "mime": "audio/flac",
    "length": "00:10:45.472200",
    "replay_gain": "-1.65",
    "cue_in": "00:00:00",
    "cue_out": "00:10:26.891000",
    "artist_name": "Nils Frahm",
    "album_title": "Tripping with Nils Frahm",
    "track_title": "#2",
    "url": None,
    "size": 10000,
}
FILE_5 = {
    "id": 5,
    "mime": "audio/mp3",
    "length": "00:59:04.989000",
    "replay_gain": "-1.39",
    "cue_in": "00:00:00",
    "cue_out": "00:58:59.130000",
    "artist_name": "Democracy Now! Audio",
    "album_title": "Democracy Now! Audio",
    "track_title": "Democracy Now! 2022-09-05 Monday",
    "url": None,
    "size": 10000,
}

WEBSTREAM_1 = {
    "id": 1,
    "name": "External radio",
    "url": "http://stream.radio.org/main.ogg",
}

SCHEDULE_1 = {
    "id": 1,
    "starts_at": "2022-09-05T11:00:00Z",
    "ends_at": "2022-09-05T11:05:02.144200Z",
    "cue_in": "00:00:13.700800",
    "cue_out": "00:05:15.845000",
    "fade_in": "00:00:00.500000",
    "fade_out": "00:00:00.500000",
    "file": 2,
    "instance": 1,
    "length": "00:05:02.144200",
    "stream": None,
}
SCHEDULE_2 = {
    "id": 2,
    "starts_at": "2022-09-05T11:05:02.144200Z",
    "ends_at": "2022-09-05T11:10:00Z",
    "cue_in": "00:00:00",
    "cue_out": "00:04:57.855800",
    "fade_in": "00:00:00.500000",
    "fade_out": "00:00:00.500000",
    "file": 4,
    "instance": 1,
    "length": "00:10:26.891000",
    "stream": None,
}
SCHEDULE_3 = {
    "id": 3,
    "starts_at": "2022-09-05T11:10:00Z",
    "ends_at": "2022-09-05T12:08:59Z",
    "cue_in": "00:00:00",
    "cue_out": "00:58:59.130000",
    "fade_in": "00:00:00.500000",
    "fade_out": "00:00:00.500000",
    "file": 5,
    "instance": 2,
    "length": "00:58:59.130000",
    "stream": None,
}
SCHEDULE_4 = {
    "id": 4,
    "starts_at": "2022-09-05T12:08:59Z",
    "ends_at": "2022-09-05T12:10:00Z",
    "cue_in": "00:00:00",
    "cue_out": "00:01:01",
    "fade_in": "00:00:00.500000",
    "fade_out": "00:00:00.500000",
    "file": 4,
    "instance": 2,
    "length": "00:10:26.891000",
    "stream": None,
}
SCHEDULE_5 = {
    "id": 5,
    "starts_at": "2022-09-05T12:10:00Z",
    "ends_at": "2022-09-05T12:40:00Z",
    "cue_in": "00:00:00",
    "cue_out": "00:30:00",
    "fade_in": "00:00:00.500000",
    "fade_out": "00:00:00.500000",
    "file": None,
    "instance": 3,
    "length": "00:30:00",
    "stream": 1,
}
SCHEDULE_6 = {
    "id": 6,
    "starts_at": "2022-09-05T12:40:00Z",
    "ends_at": "2022-09-05T12:53:23Z",
    "cue_in": "00:00:55.121100",
    "cue_out": "00:14:18.400000",
    "fade_in": "00:00:00.500000",
    "fade_out": "00:00:00.500000",
    "file": 3,
    "instance": 3,
    "length": "00:13:23.278900",
    "stream": None,
}
SCHEDULE_7 = {
    "id": 7,
    "starts_at": "2022-09-05T12:53:23Z",
    "ends_at": "2022-09-05T12:58:25Z",
    "cue_in": "00:00:13.700800",
    "cue_out": "00:05:15.845000",
    "fade_in": "00:00:00.500000",
    "fade_out": "00:00:00.500000",
    "file": 2,
    "instance": 3,
    "length": "00:05:02.144200",
    "stream": None,
}
SCHEDULE_8 = {
    "id": 8,
    "starts_at": "2022-09-05T12:58:25Z",
    "ends_at": "2022-09-05T13:00:00Z",
    "cue_in": "00:00:08.252450",
    "cue_out": "00:01:35",
    "fade_in": "00:00:00.500000",
    "fade_out": "00:00:00.500000",
    "file": 1,
    "instance": 3,
    "length": "00:03:18.955550",
    "stream": None,
}
SCHEDULE_9 = {
    "id": 9,
    "starts_at": "2022-09-05T13:00:00Z",
    "ends_at": "2022-09-05T13:05:02.144200Z",
    "cue_in": "00:00:13.700800",
    "cue_out": "00:05:15.845000",
    "fade_in": "00:00:00.500000",
    "fade_out": "00:00:00.500000",
    "file": 2,
    "instance": 4,
    "length": "00:05:02.144200",
    "stream": None,
}
SCHEDULE_10 = {
    "id": 10,
    "starts_at": "2022-09-05T13:05:02.144200Z",
    "ends_at": "2022-09-05T13:10:00Z",
    "cue_in": "00:00:00",
    "cue_out": "00:04:57.855800",
    "fade_in": "00:00:00.500000",
    "fade_out": "00:00:00.500000",
    "file": 4,
    "instance": 4,
    "length": "00:10:26.891000",
    "stream": None,
}
SCHEDULE = [
    SCHEDULE_1,
    SCHEDULE_2,
    SCHEDULE_3,
    SCHEDULE_4,
    SCHEDULE_5,
    SCHEDULE_6,
    SCHEDULE_7,
    SCHEDULE_8,
    SCHEDULE_9,
    SCHEDULE_10,
]


def test_generate_live_events(stream_preferences: StreamPreferences):
    show_instance_3 = SHOW_INSTANCE_3.copy()
    show_instance_3["starts_at"] = event_isoparse(show_instance_3["starts_at"])
    show_instance_3["ends_at"] = event_isoparse(show_instance_3["ends_at"])

    result = {}
    generate_live_events(result, show_instance_3, stream_preferences)
    assert result == {
        "2022-09-05-13-00-00": ActionEvent(
            start=datetime(2022, 9, 5, 13, 0),
            end=datetime(2022, 9, 5, 13, 0),
            type=EventKind.ACTION,
            event_type="kick_out",
        ),
    }

    result = {}
    stream_preferences.input_fade_transition = 2.0
    generate_live_events(result, show_instance_3, stream_preferences)
    assert result == {
        "2022-09-05-12-59-58": ActionEvent(
            start=datetime(2022, 9, 5, 12, 59, 58),
            end=datetime(2022, 9, 5, 12, 59, 58),
            type=EventKind.ACTION,
            event_type="switch_off",
        ),
        "2022-09-05-13-00-00": ActionEvent(
            start=datetime(2022, 9, 5, 13, 0),
            end=datetime(2022, 9, 5, 13, 0),
            type=EventKind.ACTION,
            event_type="kick_out",
        ),
    }


def test_generate_file_events(stream_preferences: StreamPreferences):
    schedule_1 = SCHEDULE_1.copy()
    schedule_1["starts_at"] = event_isoparse(schedule_1["starts_at"])
    schedule_1["ends_at"] = event_isoparse(schedule_1["ends_at"])

    result = {}
    generate_file_events(result, schedule_1, FILE_2, SHOW_1, stream_preferences)
    assert result == {
        "2022-09-05-11-00-00": FileEvent(
            start=datetime(2022, 9, 5, 11, 0),
            end=datetime(2022, 9, 5, 11, 5, 2),
            type=EventKind.FILE,
            row_id=1,
            uri=None,
            id=2,
            show_name="Show 1",
            fade_in=500.0,
            fade_out=500.0,
            cue_in=13.7008,
            cue_out=315.845,
            track_title="My Friend the Forest",
            artist_name="Nils Frahm",
            mime="audio/flac",
            replay_gain=11.46 - 3.5,
            filesize=10000,
            file_ready=False,
        )
    }

    result = {}
    stream_preferences.replay_gain_enabled = False
    generate_file_events(result, schedule_1, FILE_2, SHOW_1, stream_preferences)
    assert result == {
        "2022-09-05-11-00-00": FileEvent(
            start=datetime(2022, 9, 5, 11, 0),
            end=datetime(2022, 9, 5, 11, 5, 2),
            type=EventKind.FILE,
            row_id=1,
            uri=None,
            id=2,
            show_name="Show 1",
            fade_in=500.0,
            fade_out=500.0,
            cue_in=13.7008,
            cue_out=315.845,
            track_title="My Friend the Forest",
            artist_name="Nils Frahm",
            mime="audio/flac",
            replay_gain=None,
            filesize=10000,
            file_ready=False,
        )
    }


def test_generate_webstream_events():
    schedule_5 = SCHEDULE_5.copy()
    schedule_5["starts_at"] = event_isoparse(schedule_5["starts_at"])
    schedule_5["ends_at"] = event_isoparse(schedule_5["ends_at"])

    result = {}
    generate_webstream_events(result, schedule_5, WEBSTREAM_1, SHOW_3)
    assert result == {
        "2022-09-05-12-10-00": WebStreamEvent(
            start=datetime(2022, 9, 5, 12, 9, 55),
            end=datetime(2022, 9, 5, 12, 9, 55),
            type=EventKind.WEB_STREAM_BUFFER_START,
            row_id=5,
            uri="http://stream.radio.org/main.ogg",
            id=1,
            show_name="Show 3",
        ),
        "2022-09-05-12-10-00_0": WebStreamEvent(
            start=datetime(2022, 9, 5, 12, 10),
            end=datetime(2022, 9, 5, 12, 40),
            type=EventKind.WEB_STREAM_OUTPUT_START,
            row_id=5,
            uri="http://stream.radio.org/main.ogg",
            id=1,
            show_name="Show 3",
        ),
        "2022-09-05-12-40-00": WebStreamEvent(
            start=datetime(2022, 9, 5, 12, 40),
            end=datetime(2022, 9, 5, 12, 40),
            type=EventKind.WEB_STREAM_BUFFER_END,
            row_id=5,
            uri="http://stream.radio.org/main.ogg",
            id=1,
            show_name="Show 3",
        ),
        "2022-09-05-12-40-00_0": WebStreamEvent(
            start=datetime(2022, 9, 5, 12, 40),
            end=datetime(2022, 9, 5, 12, 40),
            type=EventKind.WEB_STREAM_OUTPUT_END,
            row_id=5,
            uri="http://stream.radio.org/main.ogg",
            id=1,
            show_name="Show 3",
        ),
    }


@pytest.mark.parametrize(
    "schedule",
    [
        (SCHEDULE),
        (random.sample(SCHEDULE, len(SCHEDULE))),
    ],
)
def test_get_schedule(schedule, requests_mock, api_client: ApiClient):
    base_url = api_client.base_url

    requests_mock.get(
        f"{base_url}/api/v2/stream/preferences",
        json={
            "input_fade_transition": 2.0,
            "message_format": 0,
            "message_offline": "",
            "replay_gain_enabled": True,
            "replay_gain_offset": -3.5,
        },
    )

    requests_mock.get(f"{base_url}/api/v2/schedule", json=schedule)

    requests_mock.get(f"{base_url}/api/v2/shows/1", json=SHOW_1)
    requests_mock.get(f"{base_url}/api/v2/shows/2", json=SHOW_2)
    requests_mock.get(f"{base_url}/api/v2/shows/3", json=SHOW_3)
    requests_mock.get(f"{base_url}/api/v2/shows/4", json=SHOW_4)
    requests_mock.get(f"{base_url}/api/v2/show-instances/1", json=SHOW_INSTANCE_1)
    requests_mock.get(f"{base_url}/api/v2/show-instances/2", json=SHOW_INSTANCE_2)
    requests_mock.get(f"{base_url}/api/v2/show-instances/3", json=SHOW_INSTANCE_3)
    requests_mock.get(f"{base_url}/api/v2/show-instances/4", json=SHOW_INSTANCE_4)
    requests_mock.get(f"{base_url}/api/v2/files/1", json=FILE_1)
    requests_mock.get(f"{base_url}/api/v2/files/2", json=FILE_2)
    requests_mock.get(f"{base_url}/api/v2/files/3", json=FILE_3)
    requests_mock.get(f"{base_url}/api/v2/files/4", json=FILE_4)
    requests_mock.get(f"{base_url}/api/v2/files/5", json=FILE_5)
    requests_mock.get(f"{base_url}/api/v2/webstreams/1", json=WEBSTREAM_1)

    assert get_schedule(api_client) == {
        "2022-09-05-11-00-00": FileEvent(
            start=datetime(2022, 9, 5, 11, 0),
            end=datetime(2022, 9, 5, 11, 5, 2),
            type=EventKind.FILE,
            row_id=1,
            uri=None,
            id=2,
            show_name="Show 1",
            fade_in=500.0,
            fade_out=500.0,
            cue_in=13.7008,
            cue_out=315.845,
            track_title="My Friend the Forest",
            artist_name="Nils Frahm",
            mime="audio/flac",
            replay_gain=11.46 - 3.5,
            filesize=10000,
            file_ready=False,
        ),
        "2022-09-05-11-05-02": FileEvent(
            start=datetime(2022, 9, 5, 11, 5, 2),
            end=datetime(2022, 9, 5, 11, 10),
            type=EventKind.FILE,
            row_id=2,
            uri=None,
            id=4,
            show_name="Show 1",
            fade_in=500.0,
            fade_out=500.0,
            cue_in=0.0,
            cue_out=297.8558,
            track_title="#2",
            artist_name="Nils Frahm",
            mime="audio/flac",
            replay_gain=-1.65 - 3.5,
            filesize=10000,
            file_ready=False,
        ),
        "2022-09-05-11-10-00": FileEvent(
            start=datetime(2022, 9, 5, 11, 10),
            end=datetime(2022, 9, 5, 12, 8, 59),
            type=EventKind.FILE,
            row_id=3,
            uri=None,
            id=5,
            show_name="Show 2",
            fade_in=500.0,
            fade_out=500.0,
            cue_in=0.0,
            cue_out=3539.13,
            track_title="Democracy Now! 2022-09-05 Monday",
            artist_name="Democracy Now! Audio",
            mime="audio/mp3",
            replay_gain=-1.39 - 3.5,
            filesize=10000,
            file_ready=False,
        ),
        "2022-09-05-12-08-59": FileEvent(
            start=datetime(2022, 9, 5, 12, 8, 59),
            end=datetime(2022, 9, 5, 12, 10),
            type=EventKind.FILE,
            row_id=4,
            uri=None,
            id=4,
            show_name="Show 2",
            fade_in=500.0,
            fade_out=500.0,
            cue_in=0.0,
            cue_out=61.0,
            track_title="#2",
            artist_name="Nils Frahm",
            mime="audio/flac",
            replay_gain=-1.65 - 3.5,
            filesize=10000,
            file_ready=False,
        ),
        "2022-09-05-12-10-00": WebStreamEvent(
            start=datetime(2022, 9, 5, 12, 9, 55),
            end=datetime(2022, 9, 5, 12, 9, 55),
            type=EventKind.WEB_STREAM_BUFFER_START,
            row_id=5,
            uri="http://stream.radio.org/main.ogg",
            id=1,
            show_name="Show 3",
        ),
        "2022-09-05-12-10-00_0": WebStreamEvent(
            start=datetime(2022, 9, 5, 12, 10),
            end=datetime(2022, 9, 5, 12, 40),
            type=EventKind.WEB_STREAM_OUTPUT_START,
            row_id=5,
            uri="http://stream.radio.org/main.ogg",
            id=1,
            show_name="Show 3",
        ),
        "2022-09-05-12-40-00": WebStreamEvent(
            start=datetime(2022, 9, 5, 12, 40),
            end=datetime(2022, 9, 5, 12, 40),
            type=EventKind.WEB_STREAM_BUFFER_END,
            row_id=5,
            uri="http://stream.radio.org/main.ogg",
            id=1,
            show_name="Show 3",
        ),
        "2022-09-05-12-40-00_0": WebStreamEvent(
            start=datetime(2022, 9, 5, 12, 40),
            end=datetime(2022, 9, 5, 12, 40),
            type=EventKind.WEB_STREAM_OUTPUT_END,
            row_id=5,
            uri="http://stream.radio.org/main.ogg",
            id=1,
            show_name="Show 3",
        ),
        "2022-09-05-12-40-00_1": FileEvent(
            start=datetime(2022, 9, 5, 12, 40),
            end=datetime(2022, 9, 5, 12, 53, 23),
            type=EventKind.FILE,
            row_id=6,
            uri=None,
            id=3,
            show_name="Show 3",
            fade_in=500.0,
            fade_out=500.0,
            cue_in=55.1211,
            cue_out=858.4,
            track_title="All Melody",
            artist_name="Nils Frahm",
            mime="audio/flac",
            replay_gain=-2.13 - 3.5,
            filesize=10000,
            file_ready=False,
        ),
        "2022-09-05-12-53-23": FileEvent(
            start=datetime(2022, 9, 5, 12, 53, 23),
            end=datetime(2022, 9, 5, 12, 58, 25),
            type=EventKind.FILE,
            row_id=7,
            uri=None,
            id=2,
            show_name="Show 3",
            fade_in=500.0,
            fade_out=500.0,
            cue_in=13.7008,
            cue_out=315.845,
            track_title="My Friend the Forest",
            artist_name="Nils Frahm",
            mime="audio/flac",
            replay_gain=11.46 - 3.5,
            filesize=10000,
            file_ready=False,
        ),
        "2022-09-05-12-58-25": FileEvent(
            start=datetime(2022, 9, 5, 12, 58, 25),
            end=datetime(2022, 9, 5, 13, 0),
            type=EventKind.FILE,
            row_id=8,
            uri=None,
            id=1,
            show_name="Show 3",
            fade_in=500.0,
            fade_out=500.0,
            cue_in=8.25245,
            cue_out=95.0,
            track_title="The Dane",
            artist_name="Nils Frahm",
            mime="audio/flac",
            replay_gain=4.52 - 3.5,
            filesize=10000,
            file_ready=False,
        ),
        "2022-09-05-12-59-58": ActionEvent(
            start=datetime(2022, 9, 5, 12, 59, 58),
            end=datetime(2022, 9, 5, 12, 59, 58),
            type=EventKind.ACTION,
            event_type="switch_off",
        ),
        "2022-09-05-13-00-00": ActionEvent(
            start=datetime(2022, 9, 5, 13, 0),
            end=datetime(2022, 9, 5, 13, 0),
            type=EventKind.ACTION,
            event_type="kick_out",
        ),
        "2022-09-05-13-00-00_0": FileEvent(
            start=datetime(2022, 9, 5, 13, 0),
            end=datetime(2022, 9, 5, 13, 5, 2),
            type=EventKind.FILE,
            row_id=9,
            uri=None,
            id=2,
            show_name="Show 4",
            fade_in=500.0,
            fade_out=500.0,
            cue_in=13.7008,
            cue_out=315.845,
            track_title="My Friend the Forest",
            artist_name="Nils Frahm",
            mime="audio/flac",
            replay_gain=11.46 - 3.5,
            filesize=10000,
            file_ready=False,
        ),
        "2022-09-05-13-05-02": FileEvent(
            start=datetime(2022, 9, 5, 13, 5, 2),
            end=datetime(2022, 9, 5, 13, 10),
            type=EventKind.FILE,
            row_id=10,
            uri=None,
            id=4,
            show_name="Show 4",
            fade_in=500.0,
            fade_out=500.0,
            cue_in=0.0,
            cue_out=297.8558,
            track_title="#2",
            artist_name="Nils Frahm",
            mime="audio/flac",
            replay_gain=-1.65 - 3.5,
            filesize=10000,
            file_ready=False,
        ),
    }