from datetime import time, timedelta
import itertools
import logging
import os

from django.conf import settings
from django.contrib.auth.models import User
from django.test import RequestFactory

import pytest
from model_bakery import baker

from aircox import models
from aircox.test import Interface


req_factory = RequestFactory()
"""Request Factory used among different tests."""


settings.ALLOWED_HOSTS = list(settings.ALLOWED_HOSTS) + [
    "sub.server.org",
    "server.org",
]


@pytest.fixture
def staff_user():
    return baker.make(User, is_active=True, is_staff=True)


@pytest.fixture
def logger():
    logger = Interface(logging, {"info": None, "debug": None, "error": None, "warning": None})
    return logger


@pytest.fixture
def station():
    return baker.make(
        models.Station,
        hosts="server.org",
        audio_streams="stream 1\nstream 2",
        default=True,
    )


@pytest.fixture
def sub_station():
    """Non-default station."""
    return baker.make(models.Station, hosts="sub.server.org", default=False)


@pytest.fixture
def stations(station, sub_station):
    return [station, sub_station]


@pytest.fixture
def programs(stations):
    items = list(
        itertools.chain(
            *(
                baker.make(
                    models.Program,
                    station=station,
                    cover=None,
                    status=models.Program.STATUS_PUBLISHED,
                    _quantity=2,
                )
                for station in stations
            )
        )
    )
    for item in items:
        item.save()
    return items


@pytest.fixture
def program(programs):
    return programs[0]


@pytest.fixture
def sched_initials(programs):
    # use concrete class; timezone is provided in order to ensure DST
    items = [
        baker.prepare(
            models.Schedule,
            program=program,
            time=time(16, 00),
            timezone="Europe/Brussels",
        )
        for program in programs
    ]
    models.Schedule.objects.bulk_create(items)
    return items


@pytest.fixture
def sched_reruns(sched_initials):
    # use concrete class
    items = [
        baker.prepare(
            models.Schedule,
            initial=initial,
            program=initial.program,
            date=initial.date,
            time=(initial.start + timedelta(hours=1)).time(),
        )
        for initial in sched_initials
    ]
    models.Schedule.objects.bulk_create(items)
    return items


@pytest.fixture
def schedules(sched_initials, sched_reruns):
    return sched_initials + sched_reruns


@pytest.fixture
def episodes(programs):
    return [baker.make(models.Episode, parent=program, cover=None) for program in programs]


@pytest.fixture
def episode(episodes):
    return episodes[0]


@pytest.fixture
def sound(program):
    return baker.make(models.Sound, file="tmp/test.wav", program=program)


@pytest.fixture
def sounds(program):
    objs = [
        models.Sound(program=program, file=f"tmp/test-{i}.wav", broadcast=(i == 0), is_downloadable=(i == 1))
        for i in range(0, 3)
    ]
    models.Sound.objects.bulk_create(objs)
    return objs


@pytest.fixture
def podcasts(episode, sounds):
    objs = [
        models.EpisodeSound(
            episode=episode,
            sound=sound,
            broadcast=True,
        )
        for sound in sounds
    ]
    models.EpisodeSound.objects.bulk_create(objs)
    return objs


@pytest.fixture
def tracks(episode, sound):
    items = [baker.prepare(models.Track, episode=episode, position=i, timestamp=i * 60) for i in range(0, 3)]
    items += [baker.prepare(models.Track, sound=sound, position=i, timestamp=i * 60) for i in range(0, 3)]
    models.Track.objects.bulk_create(items)
    return items


@pytest.fixture
def user():
    return User.objects.create_user(username="user1", password="bar")


@pytest.fixture
def png_content():
    image_file = "{}/image.png".format(os.path.dirname(__file__))
    with open(image_file, "rb") as fh:
        return fh.read()