implement saving mixes

+ add: get mixes
+ handle mixes on recently played
+ move modules around to fix circular deps

app/api/plugins/mixes.py

@@ -1,7 +1,9 @@
+from typing import Literal
 from flask_openapi3 import Tag
 from flask_openapi3 import APIBlueprint
 from pydantic import BaseModel, Field
 
+from app.db.userdata import MixTable
 from app.plugins.mixes import MixesPlugin
 from app.store.homepage import HomepageStore
 from app.store.tracks import TrackStore
@@ -13,37 +15,43 @@ api = APIBlueprint(
 )
 
 
-@api.post("/track")
-def get_track_mix():
-    """
-    Get a track mix
-    """
-    mixes = MixesPlugin()
-    track = TrackStore.trackhashmap["9eeee292264ad01b"].get_best()
-    tracks = mixes.get_track_mix([track])
-
-    return {
-        "total": len(tracks),
-        "tracks": tracks,
-    }
+class GetMixesBody(BaseModel):
+    mixtype: Literal["artists", "tracks"] = Field(description="The type of mix")
 
 
-@api.post("/artist")
-def get_artist_mix():
-    mixes = MixesPlugin()
-    # return mixes.create_artist_mixes()
-    # tracks = mixes.get_artist_mix("09306be8039b98ad")
+@api.get("/<mixtype>")
+def get_artist_mixes(path: GetMixesBody):
+    srcmixes = MixTable.get_all(with_userid=True)
+    mixes = []
 
-    # return {
-    #     "total": len(tracks),
-    #     "tracks": tracks,
-    # }
+    if path.mixtype == "artists":
+        mixes = [mix.to_dict(convert_timestamp=True) for mix in srcmixes]
+    elif path.mixtype == "tracks":
+        plugin = MixesPlugin()
 
-    return "hi"
+        for mix in srcmixes:
+            custom_mix = plugin.get_track_mix(mix)
+            if custom_mix:
+                mixes.append(custom_mix.to_dict(convert_timestamp=True))
+
+    seen_mixids = set()
+
+    # filter duplicates by trackshash
+    final_mixes = []
+    for mix in mixes:
+        # INFO: Ignore duplicates for artist mixes
+        if mix["id"] in seen_mixids and path.mixtype == "tracks":
+            continue
+
+        final_mixes.append(mix)
+        seen_mixids.add(mix["id"])
+
+    return final_mixes
 
 
 class MixQuery(BaseModel):
     mixid: str = Field(description="The mix id")
+    sourcehash: str = Field(description="The sourcehash of the mix")
 
 
 @api.get("/")
@@ -58,8 +66,45 @@ def get_mix(query: MixQuery):
         case _:
             return {"msg": "Invalid mix ID"}, 400
 
-    mix = HomepageStore.get_mix(mixtype, query.mixid[1:])
-    if mix:
+    # INFO: Check if the mix is already in the homepage store
+    mix = HomepageStore.get_mix(mixtype, query.mixid)
+    if mix and mix["sourcehash"] == query.sourcehash:
         return mix, 200
 
-    return {"msg": "Mix not found"}, 404
+    # INF0: Get the mix from the db
+    mix = MixTable.get_by_sourcehash(query.sourcehash)
+
+    if not mix:
+        return {"msg": "Mix not found"}, 404
+
+    if mixtype == "custom_mixes":
+        plugin = MixesPlugin()
+        mix = plugin.get_track_mix(mix)
+
+        if not mix:
+            return {"msg": "Mix not found"}, 404
+
+    return mix.to_full_dict(), 200
+
+
+class SaveMixRequest(BaseModel):
+    mixid: str = Field(description="The id of the mix")
+    type: str = Field(description="The type of mix")
+    sourcehash: str = Field(description="The sourcehash of the mix")
+
+
+@api.post("/save")
+def save_mix(body: SaveMixRequest):
+    mix_type = body.type
+    mix_sourcehash = body.sourcehash
+
+    if mix_type == "artist":
+        state = MixTable.save_artist_mix(mix_sourcehash)
+    elif mix_type == "track":
+        state = MixTable.save_track_mix(mix_sourcehash)
+
+    mix = HomepageStore.find_mix(body.mixid)
+
+    if mix:
+        mix.saved = state
+    return {"msg": "Mixes saved"}, 200