import time
from urllib.parse import urljoin
from datetime import datetime, timedelta
import requests as RQ

from utils import timed_cache


class Sonarr(object):
    def __init__(self, url, api_key):
        self.url = url
        self.api_key = api_key
        self.root_folder = self.get("api/v3/rootFolder")[0]["path"]
        self.quality_profile = self.get("api/v3/qualityprofile")[0]
        self.language_profile = self.get("api/v3/languageprofile")[0]

    def get(self, url, **kwargs):
        kwargs["apikey"] = self.api_key
        kwargs["_"] = str(int(time.time()))
        res = RQ.get(urljoin(self.url, url), params=kwargs)
        res.raise_for_status()
        try:
            return res.json()
        except Exception:
            return res.text

    def search(self, query):
        return self.get("api/v3/series/lookup", term=query)

    def status(self):
        return self.get("api/v3/system/status")

    @timed_cache()
    def history(self, pageSize=500):
        return self.get(
            "api/v3/history",
            page=1,
            pageSize=500,
            sortDirection="descending",
            sortKey="date",
        )

    @timed_cache()
    def calendar(self, days=30):
        today = datetime.today()
        start = today - timedelta(days=days)
        end = today + timedelta(days=days)
        return self.get(
            "api/v3/calendar",
            unmonitored=False,
            start=start.isoformat(),
            end=end.isoformat(),
        )

    @timed_cache()
    def series(self, series_id=None, keys=None):
        if series_id is None:
            return self.get("api/v3/series")
        ret = {}
        ret["series"] = self.get("api/v3/series/{}".format(series_id))
        ret["episodes"] = self.get("api/v3/episode", seriesId=series_id)
        ret["episodeFile"] = self.get("api/v3/episodeFile", seriesId=series_id)
        ret["queue"] = self.get("api/v3/queue/details", seriesId=series_id)
        return ret

    @timed_cache(seconds=60)
    def queue(self, **kwargs):
        data = []
        page = 1
        while True:
            res = self.get("api/v3/queue", page=page, pageSize=100, **kwargs)
            data = res.get("records", [])
            page += 1
            if len(data) >= res.get("totalRecords", 0):
                break
        return data

    @timed_cache()
    def details(self, episode_id):
        return self.get("api/v3/queue/details", episodeIds=episode_id)

    @timed_cache()
    def episodes(self, series_id):
        return self.get("api/v3/episode", seriesId=series_id)

    def add(self, data):
        data["qualityProfileId"] = self.quality_profile["id"]
        data["languageProfileId"] = self.language_profile["id"]
        data["rootFolderPath"] = self.root_folder
        data["addOptions"] = {
            "ignoreEpisodesWithoutFiles": False,
            "ignoreEpisodesWithFiles": True,
            "searchForMissingEpisodes": True,
        }
        data["seasonFolder"] = True
        params = dict(apikey=self.api_key)
        res = RQ.post(
            urljoin(
                self.url,
                "api/v3/series"),
            json=data,
            params=params)
        return res.json()

    def log(self, limit=0):
        return self.get(
            "api/v3/log",
            page=1,
            pageSize=(limit or 1024),
            sortDirection="descending",
            sortKey="time",
        )

    def test(self):
        self.status()
        return {}