From e222c31fb71b06960c5e7be1a6dc0577733c3baf Mon Sep 17 00:00:00 2001
From: LmeSzinc <37934724+LmeSzinc@users.noreply.github.com>
Date: Thu, 27 Feb 2025 23:30:13 +0800
Subject: [PATCH] Add: Auto increase from chapter A to B

---
 module/config/config_manual.py | 3 +++
 module/handler/fast_forward.py | 5 ++++-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/module/config/config_manual.py b/module/config/config_manual.py
index bba88a081..f35d6c8f1 100644
--- a/module/config/config_manual.py
+++ b/module/config/config_manual.py
@@ -115,6 +115,9 @@ class ManualConfig:
     MAP_HAS_MODE_SWITCH = False  # event_20240725_cn has mode switch in map preparation
     # Events from 20240725 to 20241219 introduced new chapter switches
     MAP_CHAPTER_SWITCH_20241219 = False
+    # Since event_20241219_cn chapter B unlocks event startup
+    # which means chapter AB are continuous
+    STAGE_INCREASE_AB = False
     MAP_HAS_CLEAR_PERCENTAGE = True
     MAP_HAS_WALK_SPEEDUP = False
     MAP_HAS_AMBUSH = True
diff --git a/module/handler/fast_forward.py b/module/handler/fast_forward.py
index 4c2d3cbf3..06ded406e 100644
--- a/module/handler/fast_forward.py
+++ b/module/handler/fast_forward.py
@@ -332,8 +332,11 @@ class FastForwardHandler(AutoSearchHandler):
             str: Name of next stage in upper case,
                 or origin name if unable to increase.
         """
+        stage_increase = self.STAGE_INCREASE
+        if self.config.STAGE_INCREASE_AB:
+            stage_increase.insert(0, 'A1 > A2 > A3 > B1 > B2 > B3')
         name = to_map_input_name(name)
-        for increase in self.STAGE_INCREASE:
+        for increase in stage_increase:
             increase = [i.strip(' \t\r\n') for i in increase.split('>')]
             if name in increase:
                 index = increase.index(name) + 1