Move database into the main app

I was having issues including the lib as a crate, its fine just being a
module for now.

src/database/mod.rs

@@ -0,0 +1,59 @@
+use bcrypt::{hash, verify, DEFAULT_COST};
+use diesel::prelude::*;
+use diesel::sqlite::SqliteConnection;
+use dotenv::dotenv;
+
+use crate::database::models::{InsertUser, User};
+
+mod models;
+mod schema;
+
+fn connect() -> SqliteConnection {
+    dotenv().ok();
+
+    let db_url = dotenv::var("DATABASE_URL").expect("DATABASE_URL must be set");
+    SqliteConnection::establish(&db_url).expect("Error connecting to DB")
+}
+
+// TODO: Should probably use Result here
+pub fn create_user(user: &str, pass: &str) -> Option<User> {
+    use schema::users::dsl::*;
+
+    let hashed = hash(pass, DEFAULT_COST);
+    if let Ok(hash) = hashed {
+        let connection = connect();
+        diesel::insert_into(users)
+            .values(InsertUser {
+                username: user,
+                password: &hash,
+            })
+            .execute(&connection)
+            .unwrap();
+
+        match users
+            .filter(username.eq(user))
+            .load::<User>(&connection)
+            .unwrap()
+            .first()
+        {
+            Some(u) => Some(u.clone()),
+            None => None,
+        }
+    } else {
+        None
+    }
+}
+
+pub fn get_user(user: &str, pass: &str) -> Option<User> {
+    use schema::users::dsl::*;
+
+    match users
+        .filter(username.eq(user))
+        .load::<User>(&connect())
+        .unwrap_or(Vec::<User>::new())
+        .first()
+    {
+        Some(u) if verify(pass, &u.password).unwrap_or(false) => Some(u.clone()),
+        _ => None,
+    }
+}