import urllib
from typing import List
from app import models, functions, helpers
from app.lib import trackslib
from app import api


@helpers.background
def create_everything() -> List[models.Track]:
    """
    Creates album objects for all albums and returns
    a list of track objects
    """
    albums: list[models.Album] = functions.get_all_albums()

    api.ALBUMS.clear()
    api.ALBUMS.extend(albums)
    trackslib.create_all_tracks()



def get_album_duration(album: list) -> int:
    """
    Gets the duration of an album.
    """

    album_duration = 0

    for track in album:
        album_duration += track["length"]

    return album_duration


def get_album_image(album: list) -> str:
    """
    Gets the image of an album.
    """

    for track in album:
        img = functions.extract_thumb(track["filepath"])

        if img is not None:
            return img

    return functions.use_defaults()


def get_album_tracks(album: str, artist: str) -> List:
    return [
        track
        for track in api.DB_TRACKS
        if track["album"] == album and track["albumartist"] == artist
    ]


def create_album(track) -> models.Album:
    """
    Generates and returns an album object from a track object.
    """
    album = {
        "album": track["album"],
        "artist": track["albumartist"],
    }

    album_tracks = get_album_tracks(album["album"], album["artist"])

    album["count"] = len(album_tracks)
    album["duration"] = get_album_duration(album_tracks)
    album["date"] = album_tracks[0]["date"]
    album["artistimage"] = urllib.parse.quote_plus(
        album_tracks[0]["albumartist"] + ".webp"
    )

    album["image"] = get_album_image(album_tracks)

    return models.Album(album)


def find_album(albumtitle, artist):
    for album in api.ALBUMS:
        if album.album == albumtitle and album.artist == artist:
            return album


def search_albums_by_name(query: str) -> List[models.Album]:
    """
    Searches albums by album name.
    """
    title_albums: List[models.Album] = []
    artist_albums: List[models.Album] = []

    for album in api.ALBUMS:
        if query.lower() in album.album.lower():
            title_albums.append(album)

    for album in api.ALBUMS:
        if query.lower() in album.artist.lower():
            artist_albums.append(album)

    return [*title_albums, *artist_albums]