from contextlib import asynccontextmanager
from fastapi import FastAPI, Depends, Request
from sqlalchemy.ext.asyncio import AsyncSession
from fastapi.responses import FileResponse
from fastapi.staticfiles import StaticFiles
import uvicorn
from telethon import TelegramClient
import aiohttp
import time
import jwt

from app.dependencies import get_session
from app.models.user import User
from config import config, OauthCreds
from app.MusicProvider.auth import get_oauth_creds

client = TelegramClient('nowplaying_callback', config.api_id, config.api_hash)


@asynccontextmanager
async def lifespan(app: FastAPI):
    await client.connect()
    await client.sign_in(bot_token=config.bot_token)
    yield


app = FastAPI(lifespan=lifespan)
app.mount('/static', StaticFiles(directory='app/static', html=True), name='static')


class LinkException(Exception):
    pass


@app.exception_handler(LinkException)
async def link_exception_handler(request: Request, exc: LinkException):
    return FileResponse('static/error.html', status_code=400)


async def code_to_token(code: str, uri: str, creds: OauthCreds) -> tuple[str, str, int]:
    token_headers = {
        "Authorization": "Basic " + creds.encoded,
        "Content-Type": "application/x-www-form-urlencoded"
    }
    token_data = {
        "grant_type": "authorization_code",
        "code": code,
        "redirect_uri": creds.redirect
    }
    async with aiohttp.ClientSession() as session:
        resp = await session.post(uri, data=token_data, headers=token_headers)
    resp = await resp.json()
    if 'access_token' not in resp:
        raise LinkException()
    return resp['access_token'], resp['refresh_token'], int(resp['expires_in'])


def get_decoded_id(string: str):
    try:
        return jwt.decode(string, config.jwt_secret, algorithms=['HS256'])['tg_id']
    except:
        raise LinkException()


@app.get('/spotify_callback')
async def spotify_callback(code: str, state: str, session: AsyncSession = Depends(get_session)):
    user_id = get_decoded_id(state)
    token, refresh_token, expires_in = await code_to_token(code, 'https://accounts.spotify.com/api/token', config.spotify)
    creds = get_oauth_creds(token, refresh_token, expires_in)
    user = await session.get(User, user_id)
    if user:
        user.spotify_auth = creds
        user.default = 'spotify'
    else:
        user = User(id=user_id,
                    spotify_auth=creds,
                    default='spotify'
                    )
        session.add(user)
    await session.commit()
    await client.send_message(user_id, "Account linked!")
    return FileResponse('static/success.html', media_type='text/html')


@app.get('/ym_callback')
async def ym_callback(state: str, code: str, cid: str, session: AsyncSession = Depends(get_session)):
    user_id = get_decoded_id(state)
    token, refresh_token, expires_in = await code_to_token(code, 'https://oauth.yandex.com/token', config.ymusic)
    creds = get_oauth_creds(token, refresh_token, expires_in)
    user = await session.get(User, user_id)
    if user:
        user.ymusic_auth = creds
        user.default = 'ymusic'
    else:
        user = User(id=user_id,
                    ymusic_auth=creds,
                    default='ymusic'
                    )
        session.add(user)
    await session.commit()
    await client.send_message(user_id, "Account linked!")
    return FileResponse('static/success.html', media_type='text/html')


if __name__ == '__main__':
    uvicorn.run(app, host='0.0.0.0', port=8080)