swingmusic/server/app/api/search.py

"""
Contains all the search routes.
"""
from typing import List
from app import helpers
from app.lib import searchlib
from flask import Blueprint
from flask import request

from server.app import instances, models

search_bp = Blueprint("search", __name__, url_prefix="/")

SEARCH_RESULTS = {
    "tracks": [],
    "albums": [],
    "artists": [],
}


class SearchResults:
    """
    Holds all the search results.
    """

    query: str = ""

    class Tracks:
        """
        Holds all the tracks search results.
        """

        results: List[models.Track]

    class Albums:
        """
        Holds all the albums search results.
        """

        results: List[models.Album]

    class Artists:
        """
        Holds all the artists search results.
        """

        results: List[models.Artist]


class DoSearch:
    def __init__(self, query: str) -> None:
        self.query = query
        self.tracks = helpers.Get.get_all_tracks()
        self.albums = helpers.Get.get_all_albums()
        self.artists = helpers.Get.get_all_artists()
        self.playlists = helpers.Get.get_all_playlists()

    def search_tracks(self):
        results = searchlib.SearchTracks(self.tracks, self.query)

    def search_artists(self):
        SearchResults.Artists.results = searchlib.SearchArtists(
            self.artists, self.query
        )


@search_bp.route("/search/tracks", methods=["GET"])
def search_tracks():
    """
    Searches for tracks.
    """

    query = request.args.get("q")
    if not query:
        return {"error": "No query provided"}, 400

    results = searchlib.SearchTracks(query)()
    SEARCH_RESULTS["tracks"] = results

    return {
        "tracks": results[:5],
        "more": len(results) > 5,
    }, 200


@search_bp.route("/search/albums", methods=["GET"])
def search_albums():
    """
    Searches for albums.
    """

    query = request.args.get("q")
    if not query:
        return {"error": "No query provided"}, 400

    results = searchlib.SearchAlbums(query)()
    SEARCH_RESULTS["albums"] = results

    return {
        "albums": results[:6],
        "more": len(results) > 6,
    }, 200


@search_bp.route("/search/artists", methods=["GET"])
def search_artists():
    """
    Searches for artists.
    """

    query = request.args.get("q")
    if not query:
        return {"error": "No query provided"}, 400

    results = searchlib.SearchArtists(query)()
    SEARCH_RESULTS["artists"] = results

    return {
        "artists": results[:6],
        "more": len(results) > 6,
    }, 200


@search_bp.route("/search")
def search():
    """
    Returns a list of songs, albums and artists that match the search query.
    """
    query = request.args.get("q") or "Mexican girl"

    albums = searchlib.SearchAlbums(query)()
    artists_dicts = searchlib.SearchArtists(query)()

    tracks = searchlib.SearchTracks(query)()
    top_artist = artists_dicts[0]["name"]

    _tracks = searchlib.GetTopArtistTracks(top_artist)()
    tracks = [*tracks, *[t for t in _tracks if t not in tracks]]

    SEARCH_RESULTS.clear()
    SEARCH_RESULTS["tracks"] = tracks
    SEARCH_RESULTS["albums"] = albums
    SEARCH_RESULTS["artists"] = artists_dicts

    return {
        "data": [
            {"tracks": tracks[:5], "more": len(tracks) > 5},
            {"albums": albums[:6], "more": len(albums) > 6},
            {"artists": artists_dicts[:6], "more": len(artists_dicts) > 6},
        ]
    }


@search_bp.route("/search/loadmore")
def search_load_more():
    """
    Returns more songs, albums or artists from a search query.
    """
    type = request.args.get("type")
    index = int(request.args.get("index"))

    if type == "tracks":
        return {
            "tracks": SEARCH_RESULTS["tracks"][index : index + 5],
            "more": len(SEARCH_RESULTS["tracks"]) > index + 5,
        }

    elif type == "albums":
        return {
            "albums": SEARCH_RESULTS["albums"][index : index + 6],
            "more": len(SEARCH_RESULTS["albums"]) > index + 6,
        }

    elif type == "artists":
        return {
            "artists": SEARCH_RESULTS["artists"][index : index + 6],
            "more": len(SEARCH_RESULTS["artists"]) > index + 6,
        }