From 2a4c9ef6055bac0b51674bd35aa6a0370120a1e1 Mon Sep 17 00:00:00 2001
From: Nikolas Rimikis <leptopoda@users.noreply.github.com>
Date: Wed, 27 Sep 2023 10:03:28 +0200
Subject: [PATCH] fix(neon): cache initialization

Signed-off-by: Nikolas Rimikis <leptopoda@users.noreply.github.com>
---
 .../neon/lib/src/utils/request_manager.dart   | 23 +++++++++++++++----
 1 file changed, 18 insertions(+), 5 deletions(-)

diff --git a/packages/neon/neon/lib/src/utils/request_manager.dart b/packages/neon/neon/lib/src/utils/request_manager.dart
index ec1b96ed..d67dd3dc 100644
--- a/packages/neon/neon/lib/src/utils/request_manager.dart
+++ b/packages/neon/neon/lib/src/utils/request_manager.dart
@@ -28,8 +28,10 @@ class RequestManager {
   static RequestManager get instance => _requestManager ??= RequestManager();
 
   Future<void> initCache() async {
-    _cache = Cache();
-    await _cache!.init();
+    final cache = Cache();
+    await cache.init();
+
+    _cache = cache;
   }
 
   Cache? _cache;
@@ -209,13 +211,24 @@ class Cache {
     );
   }
 
+  Database get _requireDatabase {
+    final database = _database;
+    if (database == null) {
+      throw StateError(
+        'Cache has not been set up yet. Please make sure Cache.init() has been called before and completed.',
+      );
+    }
+
+    return database;
+  }
+
   Future<bool> has(final String key) async =>
-      Sqflite.firstIntValue(await _database!.rawQuery('SELECT COUNT(*) FROM cache WHERE key = ?', [key])) == 1;
+      Sqflite.firstIntValue(await _requireDatabase.rawQuery('SELECT COUNT(*) FROM cache WHERE key = ?', [key])) == 1;
 
   Future<String?> get(final String key) async =>
-      (await _database!.rawQuery('SELECT value FROM cache WHERE key = ?', [key]))[0]['value'] as String?;
+      (await _requireDatabase.rawQuery('SELECT value FROM cache WHERE key = ?', [key]))[0]['value'] as String?;
 
-  Future<void> set(final String key, final String value) async => _database!.rawQuery(
+  Future<void> set(final String key, final String value) async => _requireDatabase.rawQuery(
         'INSERT INTO cache (key, value) VALUES (?, ?) ON CONFLICT(key) DO UPDATE SET value = excluded.value',
         [key, value],
       );