From 5d6b5049c2a34a3f8abe725d4a871ba84a0dd714 Mon Sep 17 00:00:00 2001
From: salmonik <salmonik@tuta.io>
Date: Mon, 24 Aug 2026 14:48:12 +0200
Subject: TLS transport and games framework on client and server

---
 src/bin/server/database.rs | 26 ++++++++++++++++++++------
 1 file changed, 20 insertions(+), 6 deletions(-)

(limited to 'src/bin/server/database.rs')

diff --git a/src/bin/server/database.rs b/src/bin/server/database.rs
index 0fac387..1653071 100644
--- a/src/bin/server/database.rs
+++ b/src/bin/server/database.rs
@@ -11,14 +11,22 @@ pub fn init_db() -> Connection {
             balance INTEGER DEFAULT 1000
         )",
         [],
-    ).unwrap();
+    )
+    .unwrap();
     // Add column if not exists (for migration)
     let _ = conn.execute("ALTER TABLE users ADD COLUMN account_id INTEGER UNIQUE", []);
-    let _ = conn.execute("ALTER TABLE users ADD COLUMN balance INTEGER DEFAULT 1000", []);
+    let _ = conn.execute(
+        "ALTER TABLE users ADD COLUMN balance INTEGER DEFAULT 1000",
+        [],
+    );
     conn
 }
 
-pub fn add_user(conn: &Connection, username: &str, hashed_password: &str) -> Result<(), Box<dyn std::error::Error>> {
+pub fn add_user(
+    conn: &Connection,
+    username: &str,
+    hashed_password: &str,
+) -> Result<(), Box<dyn std::error::Error>> {
     loop {
         let account_id: u32 = rand::random::<u32>();
         match conn.execute(
@@ -33,7 +41,9 @@ pub fn add_user(conn: &Connection, username: &str, hashed_password: &str) -> Res
 }
 
 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 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()
@@ -43,7 +53,9 @@ pub fn get_hashed_password(conn: &Connection, username: &str) -> Option<String>
 }
 
 pub fn get_account_id(conn: &Connection, username: &str) -> Option<u32> {
-    let mut stmt = conn.prepare("SELECT account_id FROM users WHERE username = ?1").unwrap();
+    let mut stmt = conn
+        .prepare("SELECT account_id FROM users WHERE username = ?1")
+        .unwrap();
     let mut rows = stmt.query([username]).unwrap();
     if let Ok(Some(row)) = rows.next() {
         Some(row.get::<_, u32>(0).ok()?)
@@ -53,7 +65,9 @@ pub fn get_account_id(conn: &Connection, username: &str) -> Option<u32> {
 }
 
 pub fn get_balance(conn: &Connection, username: &str) -> Option<i64> {
-    let mut stmt = conn.prepare("SELECT balance FROM users WHERE username = ?1").unwrap();
+    let mut stmt = conn
+        .prepare("SELECT balance FROM users WHERE username = ?1")
+        .unwrap();
     let mut rows = stmt.query([username]).unwrap();
     if let Ok(Some(row)) = rows.next() {
         Some(row.get::<_, i64>(0).ok()?)
-- 
cgit v1.2.3