From 93f6e7083af1db40b16db049b62025f8cdcbbc5a Mon Sep 17 00:00:00 2001 From: Jesse Luehrs Date: Mon, 5 Nov 2018 01:48:16 -0500 Subject: start sketching out some db stuff --- src/db.rs | 43 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 43 insertions(+) create mode 100644 src/db.rs (limited to 'src/db.rs') diff --git a/src/db.rs b/src/db.rs new file mode 100644 index 0000000..ef5a58f --- /dev/null +++ b/src/db.rs @@ -0,0 +1,43 @@ +use error::Result; + +const SCHEMA: &'static str = " + CREATE TABLE `tracks` ( + artist varchar(1024) NOT NULL, + album varchar(1024) DEFAULT NULL, + name varchar(1024) NOT NULL, + timestamp integer(11) NOT NULL + ); +"; + +pub struct DB { + conn: rusqlite::Connection, +} + +impl DB { + pub fn new>(path: &P) -> Result { + if !path.as_ref().exists() { + Self::create(path)?; + } + + return Ok(DB { + conn: rusqlite::Connection::open(path)?, + }) + } + + fn create>(path: &P) -> Result<()> { + println!( + "Initializing database at {}", + path.as_ref().to_string_lossy(), + ); + + if let Some(parent) = path.as_ref().parent() { + std::fs::create_dir_all(parent)?; + let conn = rusqlite::Connection::open(path)?; + conn.execute(SCHEMA, rusqlite::NO_PARAMS)?; + Ok(()) + } + else { + unimplemented!(); + } + } +} -- cgit v1.2.3-54-g00ecf