import pandas as pd
import numpy as np
import re
import requests
import os
from fastapi import APIRouter, Security, Depends, HTTPException, Response
from fastapi.responses import JSONResponse
from sqlalchemy.orm import Session

from app.database import get_db, engine, Base
from app.security import get_current_user
from app.crud import create_song, get_setting, set_setting, get_all_songs_and_votes
from typing import Any

router = APIRouter(
    prefix="/admin",
    dependencies=[Security(get_current_user, scopes=["admin"])],
    responses={404: {"description": "Not found"}},
)


def get_main_category(categories) -> int:
    if np.sum(categories != None) == 1:
        return np.argmax(categories != None, axis=0)
    elif "h" in categories:
        return np.argmax(categories == "h", axis=0)
    else:
        return np.argmax(categories != None, axis=0)


def get_youtube_id(url):
    if url is None:
        return None

    youtube_regex = (
        r'(https?://)?(www\.)?'
        '(youtube|youtu|youtube-nocookie)\.(com|be)/'
        '(watch\?v=|embed/|shorts/|v/|.+\?v=)?([^&=%\?]{11})')

    youtube_regex_match = re.match(youtube_regex, url)
    if youtube_regex_match:
        return youtube_regex_match.group(6)

    return None


def get_thumbnail(url):
    if url is None:
        return "/static/cover.jpg"

    m = get_youtube_id(url)
    if m:
        thumbnail_url = "https://img.youtube.com/vi/" + m + "/mqdefault.jpg"
        return thumbnail_url
    elif "spotify" in url:
        return re.findall(r'(https?://i.scdn.co/image[^"]+)', requests.get(url).text)[0]
    else:
        return "/static/cover.jpg"


def get_spotify_id(url):
    if url is None:
        return None
    if "spotify" in url:
        return url.split("/track/")[1]
    else:
        return None


@router.post("/load_list")
async def create_upload_file(include_non_singable: bool = False, db: Session = Depends(get_db)):

    Base.metadata.drop_all(engine)
    Base.metadata.create_all(engine)

    song_list = pd.read_excel(os.environ['LIST_URL'])

    song_list = song_list.replace({np.nan: None})
    song_list = song_list.replace({"n/a": None})

    category_names = list(song_list.iloc[0][8:17])

    for i, row in song_list[1:].iterrows():

        row = np.array(row)

        if (row[17] == "nein") and not include_non_singable:
            continue

        if not row[2]: # no title
            continue

        yt_id = get_youtube_id(row[3])
        spfy_id = get_spotify_id(row[3])

        categories = {n: v for n, v in zip(
            category_names, row[8:17] != None)}

        if (not np.any(list(categories.values()))):
            main_category = None
        else:
            main_category = category_names[get_main_category(row[8:17])]

        create_song(db,
                    og_artist=row[0],
                    aca_artist=row[1],
                    title=row[2],
                    url=row[3],
                    source=row[4],
                    yt_id=yt_id,
                    spfy_id=spfy_id,
                    thumbnail=get_thumbnail(row[3]),
                    is_current=row[5] == "ja",
                    is_aca=row[6] == "ja",
                    arng_url=row[7],
                    categories=categories,
                    main_category=main_category,
                    singable=row[17] != "nein",
                    comment=row[18]
                    )


@router.post("/toggle_veto_mode")
async def toggle_veto_mode(db: Session = Depends(get_db)) -> bool:
    veto_setting = get_setting(db, "veto_mode")
    if veto_setting:
        set_setting(db, "veto_mode", False)
        return False
    else:
        set_setting(db, "veto_mode", True)
        return True


@router.get("/results")
async def get_evaluation(format: str = "json", db = Depends(get_db)) -> Any:
    res = get_all_songs_and_votes(db)
    if format == "json":
        return JSONResponse(content=res)
    elif format == "csv":
        df = pd.json_normalize(res)
        df.columns = [c.split(".")[-1] for c in df.columns]
        return Response(content=df.to_csv(), media_type="application/csv", headers={'Content-Disposition': 'attachment; filename="results.csv"'})
    else:
        raise HTTPException(status_code=404, detail="format must be json or csv")