host: add persistence

host/Cargo.toml

@@ -6,6 +6,7 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+directories = "5.0.1"
 regex = "1.10.2"
 rusqlite = { version = "0.30.0", features = ["bundled"] }
 semver = { version = "1.0.20", features = ["serde"] }

host/src/db.rs

@@ -22,7 +22,7 @@ SELECT EXISTS
 const SELECT_LATEST_VERSION: &str = "\
 SELECT major, minor, patch
 FROM schema_version
-ORDER BY date_applied DESC
+ORDER BY applied_at DESC
 LIMIT 1
 ";
 

host/src/lib.rs

@@ -1,6 +1,8 @@
 mod db;
 pub mod message;
 
+use std::path::Path;
+
 use regex::Regex;
 use rusqlite::Connection;
 use serde_json::Value;
@@ -85,8 +87,8 @@ fn make_process(re: Regex) -> impl Fn(Query) -> String {
 }
 
 impl Context {
-    pub fn new() -> Result<Self, Error> {
-        let mut connection = Connection::open_in_memory()?;
+    pub fn new(db_path: impl AsRef<Path>) -> Result<Self, Error> {
+        let mut connection = Connection::open(db_path)?;
         db::init_tables(&mut connection)?;
         let process_regex = Regex::new(r"\W+").unwrap();
         let process = Box::new(make_process(process_regex));

host/src/main.rs

@@ -1,5 +1,6 @@
 use std::io::{self, BufReader, BufWriter, Read, Write};
 
+use directories::ProjectDirs;
 use serde_json::Value;
 
 use noematic::{
@@ -14,6 +15,7 @@ enum Error {
     Io(io::Error),
     Json(serde_json::Error),
     Noematic(noematic::Error),
+    MissingHomeDir,
     UnsupportedVersion,
     UnsupportedLength,
 }
@@ -24,6 +26,7 @@ impl std::fmt::Display for Error {
             Error::Io(e) => write!(f, "IO error: {}", e),
             Error::Json(e) => write!(f, "JSON error: {}", e),
             Error::Noematic(e) => write!(f, "{}", e),
+            Error::MissingHomeDir => write!(f, "Missing home directory"),
             Error::UnsupportedVersion => write!(f, "Unsupported version"),
             Error::UnsupportedLength => write!(f, "Unsupported length"),
         }
@@ -93,11 +96,22 @@ fn write_response(writer: &mut impl Write, response: Response) -> Result<(), Err
     Ok(())
 }
 
+fn get_project_dirs() -> Result<ProjectDirs, Error> {
+    ProjectDirs::from("com.github", "henrytill", "noematic").ok_or(Error::MissingHomeDir)
+}
+
 fn main() -> Result<(), Error> {
     let mut reader = BufReader::new(io::stdin());
     let mut writer = BufWriter::new(io::stdout());
 
-    let mut context = Context::new()?;
+    let db_path = {
+        let project_dirs: ProjectDirs = get_project_dirs()?;
+        let db_dir = project_dirs.data_dir();
+        std::fs::create_dir_all(&db_dir)?;
+        db_dir.join("db.sqlite3")
+    };
+
+    let mut context = Context::new(db_path)?;
 
     while let Some(message) = read(&mut reader)? {
         let json: Value = serde_json::from_slice(&message)?;