"""
This library contains all the functions related to albums.
"""
import random
import urllib
from pprint import pprint
from typing import List

from app import api
from app import functions
from app import instances
from app import models
from app.lib import trackslib
from progress.bar import Bar


def get_all_albums() -> List[models.Album]:
    """
    Returns a list of album objects for all albums in the database.
    """
    print("Getting all albums...")

    albums: List[models.Album] = []
    db_albums = instances.album_instance.get_all_albums()

    _bar = Bar("Creating albums", max=len(db_albums))
    for album in db_albums:
        aa = models.Album(album)
        albums.append(aa)
        _bar.next()

    _bar.finish()

    return albums


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

    api.ALBUMS = albums
    api.ALBUMS.sort(key=lambda x: x.title)

    tracks = trackslib.create_all_tracks()

    api.TRACKS.clear()
    api.TRACKS.extend(tracks)
    api.TRACKS.sort(key=lambda x: x.title)


def find_album(albumtitle: str, artist: str) -> models.Album:
    """
    Finds an album by album title and artist.
    """
    left = 0
    right = len(api.ALBUMS) - 1
    iter = 0

    while left <= right:
        iter += 1
        mid = (left + right) // 2

        if api.ALBUMS[mid].title == albumtitle and api.ALBUMS[
                mid].artist == artist:
            return mid

        if api.ALBUMS[mid].title < albumtitle:
            left = mid + 1
        else:
            right = mid - 1

    return None


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 use_defaults() -> str:
    """
    Returns a path to a random image in the defaults directory.
    """
    path = "defaults/" + str(random.randint(0, 20)) + ".webp"
    return path


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

    for track in album:
        img_p = (track["album"] + track["albumartist"] + ".webp").replace(
            "/", "::")
        img = functions.extract_thumb(track["filepath"], webp_path=img_p)

        if img is not None:
            return img

    return use_defaults()


def get_album_tracks(album: str, artist: str) -> List:
    tracks = []

    for track in api.DB_TRACKS:
        try:
            if track["album"] == album and track["albumartist"] == artist:
                tracks.append(track)
        except TypeError:
            pprint(track, indent=4)
            print(album, artist)

    return tracks


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 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.title.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]