port search to stores

+ fix favorites

app/lib/mapstuff.py

@@ -0,0 +1,68 @@
+from app.db.userdata import FavoritesTable, ScrobbleTable
+from app.store.albums import AlbumStore
+from app.store.artists import ArtistStore
+from app.store.tracks import TrackStore
+
+
+from typing import Any
+
+
+def map_scrobble_data():
+    """
+    Maps scrobble data to the in-memory stores.
+
+    The scrobble data is loaded from the database and grouped by trackhash.
+    The album and artist scrobble data (for those tracks) are then incremented based on the data.
+    """
+    records = ScrobbleTable.get_all(0, None)
+
+    # group records by trackhash
+    grouped: dict[str, dict[str, Any]] = {}
+
+    for record in records:
+        # aggregate playcount, playduration and lastplayed
+        item = grouped.setdefault(record.trackhash, {})
+        item["playcount"] = item.get("playcount", 0) + 1
+        item["playduration"] = item.get("playduration", 0) + record.duration
+        item["lastplayed"] = max(item.get("lastplayed", 0), record.timestamp)
+
+    # increment playcount, playduration and lastplayed for albums and artists
+    for trackhash, data in grouped.items():
+        track = TrackStore.trackhashmap.get(trackhash)
+
+        if track is None:
+            continue
+
+        track.increment_playcount(data["playduration"], data["lastplayed"])
+
+        album = AlbumStore.albummap.get(track.tracks[0].albumhash)
+        if album:
+            album.increment_playcount(data["playduration"], data["lastplayed"])
+
+        for artisthash in track.tracks[0].artisthashes:
+            artist = ArtistStore.artistmap.get(artisthash)
+            if artist:
+                artist.increment_playcount(data["playduration"], data["lastplayed"])
+
+
+def map_favorites():
+    """
+    Maps favorites data to the in-memory stores.
+    """
+    favorites = FavoritesTable.get_all()
+
+    for entry in favorites:
+        if entry.type == "album":
+            album = AlbumStore.albummap.get(entry.hash)
+            if album:
+                album.toggle_favorite_user(entry.userid)
+
+        elif entry.type == "artist":
+            artist = ArtistStore.artistmap.get(entry.hash)
+            if artist:
+                artist.toggle_favorite_user(entry.userid)
+
+        elif entry.type == "track":
+            track = TrackStore.trackhashmap.get(entry.hash)
+            if track:
+                track.toggle_favorite_user(entry.userid)