[chunithm] support luminous+

titles/chuni/base.py

@@ -8,7 +8,7 @@ import pytz
 
 from core.config import CoreConfig
 from titles.chuni.config import ChuniConfig
-from titles.chuni.const import ChuniConstants, ItemKind
+from titles.chuni.const import ChuniConstants, FavoriteItemKind, ItemKind
 from titles.chuni.database import ChuniData
 
 
@@ -1014,6 +1014,28 @@ class ChuniBase:
             )
             await self.data.profile.put_net_battle(user_id, net_battle)
 
+        # New in LUMINOUS PLUS
+        if "userFavoriteMusicList" in upsert:
+            # musicId, orderId
+            music_ids = set(int(m["musicId"]) for m in upsert["userFavoriteMusicList"])
+            current_favorites = await self.data.item.get_all_favorites(
+                user_id, self.version, fav_kind=FavoriteItemKind.MUSIC
+            )
+
+            if current_favorites is None:
+                current_favorites = []
+
+            current_favorite_ids = set(x.favId for x in current_favorites)
+            keep_ids = current_favorite_ids.intersection(music_ids)
+            deleted_ids = current_favorite_ids - keep_ids
+            added_ids = music_ids - keep_ids
+
+            for fav_id in deleted_ids:
+                await self.data.item.delete_favorite_music(user_id, self.version, fav_id)
+            
+            for fav_id in added_ids:
+                await self.data.item.put_favorite_music(user_id, self.version, fav_id)
+
         return {"returnCode": "1"}
 
     async def handle_upsert_user_chargelog_api_request(self, data: Dict) -> Dict: