From 5050020287024de32da482179b70870540df8fce Mon Sep 17 00:00:00 2001
From: Liplum <Li_plum@outlook.com>
Date: Tue, 24 Sep 2024 11:34:49 +0800
Subject: [PATCH] [timetable] push "/timetable/palettes/custom" after adding
 generated palette

---
 lib/timetable/p13n/page/background.dart | 6 ++++--
 lib/timetable/p13n/utils/generate.dart  | 8 ++++----
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/lib/timetable/p13n/page/background.dart b/lib/timetable/p13n/page/background.dart
index 718a05d8..dbc1faf7 100644
--- a/lib/timetable/p13n/page/background.dart
+++ b/lib/timetable/p13n/page/background.dart
@@ -139,12 +139,14 @@ class _TimetableBackgroundEditorState extends ConsumerState<TimetableBackgroundE
               icon: const Icon(Icons.generating_tokens_outlined),
               label: i18n.p13n.background.generatePalette.text(),
               onPressed: () async {
-                await addPaletteFromImageByGenerator(
+                final added = await addPaletteFromImageByGenerator(
                   context,
                   FileImage(renderImageFile),
                 );
+                if (added) {
                 if (!context.mounted) return;
-                context.push("/timetable/palettes/custom");
+                  context.push("/timetable/palettes/custom");
+                }
               },
             ),
         ],
diff --git a/lib/timetable/p13n/utils/generate.dart b/lib/timetable/p13n/utils/generate.dart
index a5199026..43e3685c 100644
--- a/lib/timetable/p13n/utils/generate.dart
+++ b/lib/timetable/p13n/utils/generate.dart
@@ -38,7 +38,7 @@ Future<TimetablePalette> generatePaletteFromImage(
   return palette;
 }
 
-Future<void> addPaletteFromImageByGenerator(
+Future<bool> addPaletteFromImageByGenerator(
   BuildContext context,
   ImageProvider img, {
   int maxColorCount = 12,
@@ -47,13 +47,13 @@ Future<void> addPaletteFromImageByGenerator(
     img,
     maxColorCount: maxColorCount,
   );
-  if (!context.mounted) return;
+  if (!context.mounted) return false;
   final newPalette = await context.showSheet<TimetablePalette>(
     (ctx) => TimetablePaletteEditorPage(palette: palette),
     dismissible: false,
   );
-  if (newPalette == null) return;
+  if (newPalette == null) return false;
   TimetableInit.storage.palette.add(newPalette);
   await HapticFeedback.mediumImpact();
-  if (!context.mounted) return;
+  return true;
 }