From a1d54efbace85066234e2ec90620947e0d086cad Mon Sep 17 00:00:00 2001
From: beerpsi <beerpsi@duck.com>
Date: Thu, 20 Jun 2024 20:28:52 +0700
Subject: [PATCH] add upserts

---
 titles/chuni/base.py | 27 ++++++++++++++++++++++++++-
 1 file changed, 26 insertions(+), 1 deletion(-)

diff --git a/titles/chuni/base.py b/titles/chuni/base.py
index 9e8a634..2a662d7 100644
--- a/titles/chuni/base.py
+++ b/titles/chuni/base.py
@@ -941,6 +941,31 @@ class ChuniBase:
                 rating_type,
                 upsert[rating_type],
             )
+        
+        # added in LUMINOUS
+        if "userCMissionList" in upsert:
+            for cmission in upsert["userCMissionList"]:
+                mission_id = cmission["missionId"]
+
+                await self.data.item.put_cmission(
+                    user_id,
+                    {
+                        "missionId": mission_id,
+                        "point": cmission["point"],
+                    },
+                )
+
+                for progress in cmission["userCMissionProgressList"]:
+                    await self.data.item.put_cmission_progress(user_id, mission_id, progress)
+
+        if "userNetBattleData" in upsert:
+            net_battle = upsert["userNetBattleData"][0]
+
+            # fix the boolean
+            net_battle["isRankUpChallengeFailed"] = (
+                False if net_battle["isRankUpChallengeFailed"] == "false" else True
+            )
+            await self.data.profile.put_net_battle(user_id, net_battle)
 
         return {"returnCode": "1"}
 
@@ -969,4 +994,4 @@ class ChuniBase:
         return {
             "userId": data["userId"],
             "userNetBattleData": {"recentNBSelectMusicList": []},
-        }
\ No newline at end of file
+        }