1 files changed, 31 insertions, 0 deletions

diff --git a/src/server/database.rs b/src/server/database.rs
new file mode 100644
index 0000000..9c51997
--- /dev/null
+++ b/src/server/database.rs
@@ -0,0 +1,31 @@
+use rusqlite::Connection;
+
+pub fn init_db() -> Connection {
+    let conn = Connection::open("server.db").unwrap();
+    conn.execute(
+        "CREATE TABLE IF NOT EXISTS users (
+            username TEXT PRIMARY KEY,
+            hashed_password TEXT NOT NULL
+        )",
+        [],
+    ).unwrap();
+    conn
+}
+
+pub fn add_user(conn: &Connection, username: &str, hashed_password: &str) -> Result<(), Box<dyn std::error::Error>> {
+    conn.execute(
+        "INSERT INTO users (username, hashed_password) VALUES (?1, ?2)",
+        [username, hashed_password],
+    )?;
+    Ok(())
+}
+
+pub fn get_hashed_password(conn: &Connection, username: &str) -> Option<String> {
+    let mut stmt = conn.prepare("SELECT hashed_password FROM users WHERE username = ?1").unwrap();
+    let mut rows = stmt.query([username]).unwrap();
+    if let Ok(Some(row)) = rows.next() {
+        row.get(0).ok()
+    } else {
+        None
+    }
+}