From a975e753b2547130abe67b2940a7be2caf8b0566 Mon Sep 17 00:00:00 2001
From: topty <topty@live.com>
Date: Sat, 16 Dec 2023 00:33:43 +0700
Subject: [PATCH] buddies version

---
 docs/game_specific_info.md |  1 +
 titles/cm/read.py          |  1 +
 titles/mai2/buddies.py     | 12 ++++++++++++
 titles/mai2/const.py       |  2 ++
 titles/mai2/index.py       |  6 +++++-
 5 files changed, 21 insertions(+), 1 deletion(-)
 create mode 100644 titles/mai2/buddies.py

diff --git a/docs/game_specific_info.md b/docs/game_specific_info.md
index 5810ff8..0bdc572 100644
--- a/docs/game_specific_info.md
+++ b/docs/game_specific_info.md
@@ -201,6 +201,7 @@ Config file is located in `config/cxb.yaml`.
 | SDEZ      | 18         | maimai DX UNiVERSE PLUS |
 | SDEZ      | 19         | maimai DX FESTiVAL      |
 | SDEZ      | 20         | maimai DX FESTiVAL PLUS |
+| SDEZ      | 21         | maimai DX BUDDiES       |
 
 ### Importer
 
diff --git a/titles/cm/read.py b/titles/cm/read.py
index 376789d..80684bd 100644
--- a/titles/cm/read.py
+++ b/titles/cm/read.py
@@ -206,6 +206,7 @@ class CardMakerReader(BaseReader):
             "1.25": Mai2Constants.VER_MAIMAI_DX_UNIVERSE_PLUS,
             "1.30": Mai2Constants.VER_MAIMAI_DX_FESTIVAL,
             "1.35": Mai2Constants.VER_MAIMAI_DX_FESTIVAL_PLUS,
+            "1.40": Mai2Constants.VER_MAIMAI_DX_BUDDIES,
         }
 
         for root, dirs, files in os.walk(base_dir):
diff --git a/titles/mai2/buddies.py b/titles/mai2/buddies.py
new file mode 100644
index 0000000..24e792a
--- /dev/null
+++ b/titles/mai2/buddies.py
@@ -0,0 +1,12 @@
+from typing import Dict
+
+from core.config import CoreConfig
+from titles.mai2.festivalplus import Mai2FestivalPlus
+from titles.mai2.const import Mai2Constants
+from titles.mai2.config import Mai2Config
+
+
+class Mai2Buddies(Mai2FestivalPlus):
+    def __init__(self, cfg: CoreConfig, game_cfg: Mai2Config) -> None:
+        super().__init__(cfg, game_cfg)
+        self.version = Mai2Constants.VER_MAIMAI_DX_BUDDIES
diff --git a/titles/mai2/const.py b/titles/mai2/const.py
index a4c29db..ce70d70 100644
--- a/titles/mai2/const.py
+++ b/titles/mai2/const.py
@@ -53,6 +53,7 @@ class Mai2Constants:
     VER_MAIMAI_DX_UNIVERSE_PLUS = 18
     VER_MAIMAI_DX_FESTIVAL = 19
     VER_MAIMAI_DX_FESTIVAL_PLUS = 20
+    VER_MAIMAI_DX_BUDDIES = 21
 
     VERSION_STRING = (
         "maimai",
@@ -76,6 +77,7 @@ class Mai2Constants:
         "maimai DX UNiVERSE PLUS",
         "maimai DX FESTiVAL",
         "maimai DX FESTiVAL PLUS",
+        "maimai DX BUDDiES",
     )
 
     @classmethod
diff --git a/titles/mai2/index.py b/titles/mai2/index.py
index 793aaef..b4cb228 100644
--- a/titles/mai2/index.py
+++ b/titles/mai2/index.py
@@ -25,6 +25,7 @@ from .universe import Mai2Universe
 from .universeplus import Mai2UniversePlus
 from .festival import Mai2Festival
 from .festivalplus import Mai2FestivalPlus
+from .buddies import Mai2Buddies
 
 
 class Mai2Servlet(BaseServlet):
@@ -58,6 +59,7 @@ class Mai2Servlet(BaseServlet):
             Mai2UniversePlus,
             Mai2Festival,
             Mai2FestivalPlus,
+            Mai2Buddies,
         ]
 
         self.logger = logging.getLogger("mai2")
@@ -252,8 +254,10 @@ class Mai2Servlet(BaseServlet):
             internal_ver = Mai2Constants.VER_MAIMAI_DX_UNIVERSE_PLUS
         elif version >= 130 and version < 135:  # FESTiVAL
             internal_ver = Mai2Constants.VER_MAIMAI_DX_FESTIVAL
-        elif version >= 135:  # FESTiVAL PLUS
+        elif version >= 135 and version < 140:  # FESTiVAL PLUS
             internal_ver = Mai2Constants.VER_MAIMAI_DX_FESTIVAL_PLUS
+        elif version >= 140:  # BUDDiES
+            internal_ver = Mai2Constants.VER_MAIMAI_DX_BUDDIES
 
         if (
             request.getHeader("Mai-Encoding") is not None