admin account registration as part of server init

1 files changed, 5 insertions, 2 deletions

diff --git a/src/database/database.rs b/src/database/database.rs
index ca0e018..7444779 100644
--- a/src/database/database.rs
+++ b/src/database/database.rs
@@ -33,7 +33,7 @@ impl Database {
         Ok(Database { pool })
     }
 
-    pub fn insert_user(&self, email: &str, username: &str, password_hash: &str) -> Result<User, DatabaseError> {
+    pub fn insert_user(&self, email: &str, username: &str, password_hash: &str, is_admin: bool) -> Result<User, DatabaseError> {
         static QUERY: &str = include_str!("sql/insert_user.sql");
         let conn = self.pool
             .get()
@@ -42,12 +42,13 @@ impl Database {
             .map_err(|e| DatabaseError::Query(e.to_string()))?;
 
         Ok(statement
-            .query_one((email, username, password_hash), |row| {
+            .query_one((email, username, password_hash, is_admin), |row| {
                 Ok(User::new(
                     row.get("id")?,
                     email.to_owned(),
                     username.to_owned(),
                     password_hash.to_owned(),
+                    is_admin
                 ))
             })
             .map_err(|e| DatabaseError::Query(e.to_string()))?
@@ -69,6 +70,7 @@ impl Database {
                     row.get("email")?,
                     row.get("username")?,
                     row.get("password_hash")?,
+                    row.get("is_admin")?,
                 ))
             })
             .optional()
@@ -91,6 +93,7 @@ impl Database {
                     row.get("email")?,
                     row.get("username")?,
                     row.get("password_hash")?,
+                    row.get("is_admin")?
                 ))
             })
             .optional()