memo-cli/src/storage.rs

use chrono::{DateTime, Utc};
use rusqlite::{params, Connection, Error};
use std::fmt;

pub struct Memo {
	pub id: u32,
	pub body: String,
	pub due: Option<DateTime<Utc>>,
}

impl fmt::Display for Memo {
	fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
		let mut due_str = "null".to_string();
		if self.due.is_some() {
			due_str = self.due.unwrap().to_string();
		}
		return write!(
			f,
			"Memo(id={id}, body={body}, due={due})",
			id = self.id,
			body = self.body,
			due = due_str,
		);
	}
}

pub trait MemoStorage {
	fn all(&self) -> Result<Vec<Memo>, Error>;
	fn add(&self, body: &str, due: Option<DateTime<Utc>>) -> Result<(), Error>;
	fn del(&self, id: u32) -> Result<bool, Error>;
	fn get(&self, id: u32) -> Result<Memo, Error>;
}

// SQLiteStorage

pub struct SQLiteStorage {
	conn: Connection,
}

pub fn open_sqlite_storage(path: &str) -> Result<SQLiteStorage, Error> {
	let connection = Connection::open(path)?;
	// TODO check if table exist and is valid
	connection.execute(
		"CREATE TABLE IF NOT EXISTS memo (
			id INTEGER PRIMARY KEY,
			body TEXT NOT NULL,
			due DATETIME
		);",
		[],
	)?;
	return Ok(SQLiteStorage { conn: connection });
}

impl MemoStorage for SQLiteStorage {
	fn all(&self) -> Result<Vec<Memo>, Error> {
		let mut results = Vec::new();

		/*
		 * SQLite considers NULL as smallest value, so we will always get events with no due date
		 * first. To circumvent this, we first query all memos with a due date, and then all
		 * others. This is kinda jank but will do for now.
		 */

		{
			let mut statement = self.conn.prepare("SELECT * FROM memo WHERE due IS NOT NULL ORDER BY due, id")?;
			let mut rows = statement.query([])?;

			while let Some(row) = rows.next()? {
				results.push(Memo {
					id: row.get(0)?,
					body: row.get(1)?,
					due: row.get(2)?,
				});
			}
		}

		{
			let mut statement = self.conn.prepare("SELECT * FROM memo WHERE due IS NULL ORDER BY due, id")?;
			let mut rows = statement.query([])?;

			while let Some(row) = rows.next()? {
				results.push(Memo {
					id: row.get(0)?,
					body: row.get(1)?,
					due: row.get(2)?,
				});
			}
		}

		return Ok(results);
	}

	fn add(&self, body: &str, due: Option<DateTime<Utc>>) -> Result<(), Error> {
		// TODO join these 2 ifs?
		if due.is_some() {
			self.conn.execute(
				"INSERT INTO memo (body, due) VALUES (?, ?)",
				params![body, due],
			)?;
		} else {
			self.conn
				.execute("INSERT INTO memo (body) VALUES (?)", params![body])?;
		}
		return Ok(());
	}

	fn del(&self, id: u32) -> Result<bool, Error> {
		let count = self
			.conn
			.execute("DELETE FROM memo WHERE id = ?", params![id])?;
		if count > 0 {
			return Ok(true);
		} else {
			return Ok(false);
		}
	}

	fn get(&self, id: u32) -> Result<Memo, Error> {
		return Ok(self
			.conn
			.query_row("SELECT * FROM memo WHERE id = ?", params![id], |row| {
				return Ok(Memo {
					id: row.get(0)?,
					body: row.get(1)?,
					due: row.get(2).unwrap_or(None),
				});
			})?);
	}
}