building db support with diesel.rs

cookbook/src/lib.rs

@@ -1,7 +1,28 @@
+#[macro_use]
+extern crate diesel;
+extern crate dotenv;
+
+pub mod schema;
+pub mod models;
+
 mod recipe;
 mod storage;
 mod importer;
 
+use diesel::prelude::*;
+use dotenv::dotenv;
+use std::env;
+
+pub fn establish_connection() -> SqliteConnection {
+    dotenv().ok();
+
+    let db_url = env::var("DATABASE_URL").expect("DATABASE_URL must be set !");
+    SqliteConnection::establish(&db_url)
+        .expect(&format!("Error connecting to {}", db_url))
+}
+
+
+
 #[cfg(test)]
 mod tests {
     #[test]