From ee4eddd63962993e17d4efe89cc7bf038478a5b3 Mon Sep 17 00:00:00 2001
From: ppc <albie@ppc.moe>
Date: Sun, 15 Sep 2024 19:22:39 +0000
Subject: [PATCH] add buddies plus support

---
 titles/mai2/buddiesplus.py | 36 ++++++++++++++++++++++++++++++++++++
 titles/mai2/const.py       |  4 +++-
 titles/mai2/index.py       |  8 ++++++--
 3 files changed, 45 insertions(+), 3 deletions(-)
 create mode 100644 titles/mai2/buddiesplus.py

diff --git a/titles/mai2/buddiesplus.py b/titles/mai2/buddiesplus.py
new file mode 100644
index 0000000..0467b03
--- /dev/null
+++ b/titles/mai2/buddiesplus.py
@@ -0,0 +1,36 @@
+from typing import Dict
+
+from core.config import CoreConfig
+from titles.mai2.buddies import Mai2Buddies
+from titles.mai2.const import Mai2Constants
+from titles.mai2.config import Mai2Config
+
+
+class Mai2BuddiesPlus(Mai2Buddies):
+    def __init__(self, cfg: CoreConfig, game_cfg: Mai2Config) -> None:
+        super().__init__(cfg, game_cfg)
+        self.version = Mai2Constants.VER_MAIMAI_DX_BUDDIES_PLUS
+
+    async def handle_cm_get_user_preview_api_request(self, data: Dict) -> Dict:
+        user_data = await super().handle_cm_get_user_preview_api_request(data)
+
+        # hardcode lastDataVersion for CardMaker
+        user_data["lastDataVersion"] = "1.45.00"
+        return user_data
+
+    async def handle_get_game_weekly_data_api_request(self, data: Dict) -> Dict:
+        return {
+            "gameWeeklyData": {
+                "missionCategory": 0,
+                "updateDate": "2024-03-21 09:00:00.0",
+                "beforeDate": "2099-12-31 00:00:00.0"
+            }
+        }
+    
+    async def handle_create_token_api_request(self, data: Dict) -> Dict:
+        return {
+            "Bearer": "ARTEMiSTOKEN" # duplicate of handle_user_login_api_request from Mai2Festival
+        }
+    
+    async def handle_remove_token_api_request(self, data: Dict) -> Dict:
+        return {}
diff --git a/titles/mai2/const.py b/titles/mai2/const.py
index 4dc10ce..0d13a0d 100644
--- a/titles/mai2/const.py
+++ b/titles/mai2/const.py
@@ -55,6 +55,7 @@ class Mai2Constants:
     VER_MAIMAI_DX_FESTIVAL = 19
     VER_MAIMAI_DX_FESTIVAL_PLUS = 20
     VER_MAIMAI_DX_BUDDIES = 21
+    VER_MAIMAI_DX_BUDDIES_PLUS = 22
 
     VERSION_STRING = (
         "maimai",
@@ -78,7 +79,8 @@ class Mai2Constants:
         "maimai DX UNiVERSE PLUS",
         "maimai DX FESTiVAL",
         "maimai DX FESTiVAL PLUS",
-        "maimai DX BUDDiES"
+        "maimai DX BUDDiES",
+        "maimai DX BUDDiES PLUS"
     )
 
     @classmethod
diff --git a/titles/mai2/index.py b/titles/mai2/index.py
index ad02648..e8b88ec 100644
--- a/titles/mai2/index.py
+++ b/titles/mai2/index.py
@@ -30,6 +30,7 @@ from .universeplus import Mai2UniversePlus
 from .festival import Mai2Festival
 from .festivalplus import Mai2FestivalPlus
 from .buddies import Mai2Buddies
+from .buddiesplus import Mai2BuddiesPlus
 
 
 class Mai2Servlet(BaseServlet):
@@ -64,7 +65,8 @@ class Mai2Servlet(BaseServlet):
             Mai2UniversePlus,
             Mai2Festival,
             Mai2FestivalPlus,
-            Mai2Buddies
+            Mai2Buddies,
+            Mai2BuddiesPlus
         ]
 
         self.logger = logging.getLogger("mai2")
@@ -302,8 +304,10 @@ class Mai2Servlet(BaseServlet):
               internal_ver = Mai2Constants.VER_MAIMAI_DX_FESTIVAL
           elif version >= 135 and version < 140:  # FESTiVAL PLUS
               internal_ver = Mai2Constants.VER_MAIMAI_DX_FESTIVAL_PLUS
-          elif version >= 140:  # BUDDiES
+          elif version >= 140 and version < 145:  # BUDDiES
               internal_ver = Mai2Constants.VER_MAIMAI_DX_BUDDIES
+          elif version >= 145:  # BUDDiES PLUS
+              internal_ver = Mai2Constants.VER_MAIMAI_DX_BUDDIES_PLUS
         elif game_code == "SDGA": # Int
           if version < 105:  # 1.0
               internal_ver = Mai2Constants.VER_MAIMAI_DX