From 87567ddb9410cd5c118609ee4daa80f86e54f714 Mon Sep 17 00:00:00 2001
From: Mike Dilger <mike@mikedilger.com>
Date: Wed, 27 Sep 2023 13:22:04 +1300
Subject: [PATCH] storage: write_wizard_complete, read_wizard_complete

---
 src/storage/mod.rs | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/src/storage/mod.rs b/src/storage/mod.rs
index b89967a5..0376abf2 100644
--- a/src/storage/mod.rs
+++ b/src/storage/mod.rs
@@ -626,6 +626,43 @@ impl Storage {
         }
     }
 
+    pub fn write_wizard_complete<'a>(
+        &'a self,
+        wizard_complete: bool,
+        rw_txn: Option<&mut RwTxn<'a>>,
+    ) -> Result<(), Error> {
+        let bytes = wizard_complete.write_to_vec()?;
+
+        let f = |txn: &mut RwTxn<'a>| -> Result<(), Error> {
+            self.general.put(txn, b"wizard_complete", &bytes)?;
+            Ok(())
+        };
+
+        match rw_txn {
+            Some(txn) => f(txn)?,
+            None => {
+                let mut txn = self.env.write_txn()?;
+                f(&mut txn)?;
+                txn.commit()?;
+            }
+        };
+
+        Ok(())
+    }
+
+    pub fn read_wizard_complete(&self) -> bool {
+        let txn = match self.env.read_txn() {
+            Ok(txn) => txn,
+            Err(_) => return false,
+        };
+
+        match self.general.get(&txn, b"wizard_complete") {
+            Err(_) => false,
+            Ok(None) => false,
+            Ok(Some(bytes)) => bool::read_from_buffer(bytes).unwrap_or(false),
+        }
+    }
+
     // Settings ----------------------------------------------------------
 
     // This defines functions for read_{setting} and write_{setting} for each