From dea5ba34792a2c204e9a3850bbafbc34314aa2ce Mon Sep 17 00:00:00 2001
From: Vishesh Handa <me@vhanda.in>
Date: Fri, 8 May 2020 17:15:02 +0200
Subject: [PATCH] HomeScreen: Simplify code

---
 lib/screens/home_screen.dart | 21 ++++++++++++---------
 1 file changed, 12 insertions(+), 9 deletions(-)

diff --git a/lib/screens/home_screen.dart b/lib/screens/home_screen.dart
index baea1a2a..50060c66 100644
--- a/lib/screens/home_screen.dart
+++ b/lib/screens/home_screen.dart
@@ -1,5 +1,6 @@
 import 'package:flutter/material.dart';
 import 'package:gitjournal/core/flattened_notes_folder.dart';
+import 'package:gitjournal/core/notes_folder.dart';
 import 'package:gitjournal/iap.dart';
 import 'package:provider/provider.dart';
 
@@ -13,29 +14,31 @@ class HomeScreen extends StatefulWidget {
 }
 
 class _HomeScreenState extends State<HomeScreen> {
-  FlattenedNotesFolder flattenedNotesFolder;
+  NotesFolder notesFolder;
 
   @override
   void initState() {
     super.initState();
 
     InAppPurchases.confirmProPurchase();
+    Future.delayed(Duration.zero, _initFolder);
+  }
 
-    Future.delayed(Duration.zero, () {
-      if (!mounted) return;
-      final rootFolder = Provider.of<NotesFolderFS>(context);
-      setState(() {
-        flattenedNotesFolder = FlattenedNotesFolder(rootFolder);
-      });
+  void _initFolder() async {
+    if (!mounted) return;
+
+    final rootFolder = Provider.of<NotesFolderFS>(context);
+    setState(() {
+      notesFolder = FlattenedNotesFolder(rootFolder);
     });
   }
 
   @override
   Widget build(BuildContext context) {
-    if (flattenedNotesFolder == null) {
+    if (notesFolder == null) {
       return Container();
     }
 
-    return FolderView(notesFolder: flattenedNotesFolder);
+    return FolderView(notesFolder: notesFolder);
   }
 }