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 { let results = self.meta_query("select * from meta where key='database_version'", &[])?;