from datetime import datetime
from typing import List
from uuid import uuid4, UUID

from fastapi import APIRouter, status, HTTPException
from sqlalchemy import select, Sequence

from app.models.media import MediaFileResponse, Link
from app.schema import MediaFile, SessionDep

router = APIRouter(
    prefix="/media",
    tags=["media"]
)

@router.get("/update-titles")
def update_titles(db: SessionDep) -> list[MediaFileResponse]:
    results: list[MediaFileResponse] = []
    files = db.query(MediaFile).filter(MediaFile.review == 1).all()
    for mediafile in files:
        mediafile.update_title()
        db.add(mediafile)
        response = MediaFileResponse(id=mediafile.id,
                                     title=mediafile.title,
                                     file_name=mediafile.file_name,
                                     cloud_link=mediafile.cloud_link,
                                     url=str(mediafile.url),
                                     review=(mediafile.review == 1),
                                     should_download=(mediafile.should_download == 1))
        results.append(response)
    db.commit()
    return results


@router.get("/files", response_model=List[MediaFileResponse])
def get_files(db: SessionDep, review: bool = False, download: bool = False) -> List[MediaFileResponse]:
    results: list[MediaFileResponse] = []
    files: Sequence[MediaFile]
    if review:
        files = db.query(MediaFile).filter(MediaFile.review == 1).all()
    elif download:
        files = db.query(MediaFile).filter(MediaFile.should_download == 1).all()
    else:
        files = db.scalars(select(MediaFile)).all()
    for mediafile in files:
        response = MediaFileResponse(id=mediafile.id,
                                     title=mediafile.title,
                                     file_name=mediafile.file_name,
                                     cloud_link=mediafile.cloud_link,
                                     url=str(mediafile.url),
                                     review=(mediafile.review == 1),
                                     should_download=(mediafile.should_download == 1))
        results.append(response)
    return results

@router.put("/files/{file_id}", response_model=MediaFileResponse)
def update_file(file_id: UUID, db: SessionDep, info: MediaFileResponse) -> MediaFileResponse:
    mediaFile = db.get(MediaFile, file_id)
    if not mediaFile:
        raise HTTPException(status_code=404, detail="MediaFile could not be found")
    mediaFile.file_name = info.file_name
    mediaFile.cloud_link = info.cloud_link
    mediaFile.url = info.url
    mediaFile.title = info.title
    mediaFile.last_modified_date = datetime.now()
    mediaFile.review = info.review
    mediaFile.should_download = info.should_download
    db.add(mediaFile)
    db.commit()
    return info


@router.post("/files", status_code=status.HTTP_201_CREATED)
def add_file(new_link: Link, db: SessionDep) -> MediaFileResponse:
    print(new_link.url)
    try:
        mediaFile: MediaFile = MediaFile()
        setattr(mediaFile, "url", new_link.url)
        setattr(mediaFile, "review", True)
        setattr(mediaFile, "should_download", True)
        db.add(mediaFile)
        db.commit()
    except:
        raise HTTPException(status_code=409, detail="Link duplicate")
    response = MediaFileResponse(id=uuid4(),
                                 title=mediaFile.title,
                                 file_name=mediaFile.file_name,
                                 cloud_link=mediaFile.cloud_link,
                                 url=new_link.url,
                                 review=(mediaFile.review == 1),
                                 should_download=(mediaFile.should_download == 1))
    return response