From b531f8ca20ee20daf8cf3ea599d0ff590c6b3079 Mon Sep 17 00:00:00 2001
From: Abraham Toriz <categulario@gmail.com>
Date: Thu, 15 Jul 2021 11:48:49 -0500
Subject: [PATCH] database set current sheet

---
 src/commands/list.rs | 1 +
 src/database.rs      | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/src/commands/list.rs b/src/commands/list.rs
index 3a4b0f0..3003fbb 100644
--- a/src/commands/list.rs
+++ b/src/commands/list.rs
@@ -117,6 +117,7 @@ mod tests {
         let mut err = Vec::new();
         let config = Default::default();
         db.init().unwrap();
+        db.set_current_sheet("sheet2").unwrap();
 
         db.entry_insert(Utc.ymd(2021, 1, 1).and_hms(0, 0, 0), None, None, "_archived".into()).unwrap();
         db.entry_insert(Utc.ymd(2021, 1, 1).and_hms(0, 0, 0), None, None, "sheet1".into()).unwrap();
diff --git a/src/database.rs b/src/database.rs
index 642e43a..41fcbb2 100644
--- a/src/database.rs
+++ b/src/database.rs
@@ -147,6 +147,12 @@ pub trait Database {
         Ok(results.into_iter().next().map(|m| m.value))
     }
 
+    fn set_current_sheet(&mut self, sheet: &str) -> Result<()> {
+        self.execute("INSERT INTO meta (key, value) VALUES ('current_sheet', ?1)", &[&sheet])?;
+
+        Ok(())
+    }
+
     fn version(&self) -> Result<DBVersion> {
         let results = self.meta_query("select * from meta where key='database_version'", &[])?;