from flask import Blueprint
from flask_apispec import doc, marshal_with, use_kwargs
from http import HTTPStatus

import src.schemas.playlists.common as common_serializers
import src.schemas.playlists.spotify as serializers
from src.db.playlists.spotify import get_playlist_streams_weekly_query, get_playlist_top_weekly_query, \
    get_playlists_data, get_top_playlist_history_dates_query
from src.domains.playlists.spotify import parse_playlists_data
from src.utils.query import get_query_asdict


blueprint = Blueprint("playlists_spotify", __name__)


@blueprint.route("/playlists/spotify/", methods=["GET"])
@doc("Get playlists data from DB.")
@use_kwargs(serializers.PlaylistsData.Request, location="query")
@marshal_with(serializers.PlaylistsData.Response, code=HTTPStatus.OK, description="Playlists data.")
@marshal_with(None, code=HTTPStatus.UNAUTHORIZED, description="Authentication failed.")
def get_playlists(**data):
    query = get_playlists_data(**data)
    result = get_query_asdict(query)
    return {"playlists": parse_playlists_data(result, with_positions=False)}


@blueprint.route("/playlists/spotify/top/weekly/", methods=["GET"])
@doc("")
@use_kwargs(common_serializers.PlaylistsTopWeekly.Request, location="query")
@marshal_with(
    common_serializers.PlaylistsTopWeekly.Response(many=True),
    code=HTTPStatus.OK,
    description="Latest top ranked playlists.",
)
@marshal_with(None, code=HTTPStatus.UNAUTHORIZED, description="Authentication failed.")
def get_playlists_top_weekly(**data):
    return get_query_asdict(get_playlist_top_weekly_query(**data))


@blueprint.route("/playlists/spotify/previous/top/dates/", methods=["POST"])
@doc("")
@use_kwargs(common_serializers.PlaylistsPreviousTopDates.RequestPostBody, location="json")
@marshal_with(common_serializers.PlaylistsPreviousTopDates.Response(many=True), code=HTTPStatus.OK, description="")
@marshal_with(None, code=HTTPStatus.UNAUTHORIZED, description="Authentication failed.")
def get_playlists_previous_top_dates(**data):
    return get_query_asdict(get_top_playlist_history_dates_query(**data))


@blueprint.route("/playlists/spotify/streams/weekly/", methods=["GET"])
@doc("")
@use_kwargs(common_serializers.PlaylistsStreamsWeekly.Request, location="query")
@marshal_with(common_serializers.PlaylistsStreamsWeekly.Response(many=True), code=HTTPStatus.OK, description="")
@marshal_with(None, code=HTTPStatus.UNAUTHORIZED, description="Authentication failed.")
def get_playlists_streams_weekly(**data):
    playlist_ids = data["playlist_ids"]
    market = data["market"]

    return get_query_asdict(get_playlist_streams_weekly_query(playlist_ids, market))