From f3bea2db3ca27582dc17b5abfa1e87102ff63d6c Mon Sep 17 00:00:00 2001
From: Moe <moe@chocola.dev>
Date: Fri, 29 Apr 2022 17:49:37 +0700
Subject: [PATCH] Add initial screen support in menu

---
 lib/components/menu.dart | 16 ++++++++++++++--
 lib/main.dart            |  2 +-
 2 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/lib/components/menu.dart b/lib/components/menu.dart
index 987e0a5..b98eaa8 100644
--- a/lib/components/menu.dart
+++ b/lib/components/menu.dart
@@ -8,7 +8,9 @@ import 'package:nekoya_flutter/screens/cart.dart';
 import 'package:nekoya_flutter/screens/transactions.dart';
 
 class Menu extends StatefulWidget {
-  const Menu({Key? key}) : super(key: key);
+  const Menu({Key? key, required this.initialScreen}) : super(key: key);
+
+  final int initialScreen;
 
   @override
   State<Menu> createState() => _MenuState();
@@ -20,7 +22,17 @@ class _MenuState extends State<Menu> {
 
   @override
   void initState() {
-    _selectedWidget = const Products();
+    if (widget.initialScreen == 0) {
+      _selectedWidget = const Checkout();
+    } else if (widget.initialScreen == 1) {
+      _selectedWidget = const Payment();
+    } else if (widget.initialScreen == 2) {
+      _selectedWidget = const Products();
+    } else if (widget.initialScreen == 3) {
+      _selectedWidget = const Cart();
+    } else if (widget.initialScreen == 4) {
+      _selectedWidget = const Transactions();
+    }
     super.initState();
   }
 
diff --git a/lib/main.dart b/lib/main.dart
index bf20507..115536d 100644
--- a/lib/main.dart
+++ b/lib/main.dart
@@ -27,7 +27,7 @@ class MyApp extends StatelessWidget {
         pageTransitionType: PageTransitionType.fade,
         backgroundColor: const Color(0xff1b1c1e),
         splashIconSize: 150,
-        nextScreen: const Menu(),
+        nextScreen: const Menu(initialScreen: 2,),
       ),
       debugShowCheckedModeBanner: false,
     );