1 files changed, 19 insertions, 7 deletions

diff --git a/src/bin/client/database.rs b/src/bin/client/database.rs
index da069a7..a274f31 100644
--- a/src/bin/client/database.rs
+++ b/src/bin/client/database.rs
@@ -8,7 +8,8 @@ pub fn init_db() -> Connection {
             last_used DATETIME DEFAULT CURRENT_TIMESTAMP
         )",
         [],
-    ).unwrap();
+    )
+    .unwrap();
     conn.execute(
         "CREATE TABLE IF NOT EXISTS credentials (
             host TEXT PRIMARY KEY,
@@ -16,12 +17,15 @@ pub fn init_db() -> Connection {
             password TEXT NOT NULL
         )",
         [],
-    ).unwrap();
+    )
+    .unwrap();
     conn
 }
 
 pub fn get_last_host(conn: &Connection) -> Option<String> {
-    let mut stmt = conn.prepare("SELECT host FROM hosts ORDER BY last_used DESC LIMIT 1").unwrap();
+    let mut stmt = conn
+        .prepare("SELECT host FROM hosts ORDER BY last_used DESC LIMIT 1")
+        .unwrap();
     let mut rows = stmt.query([]).unwrap();
     if let Ok(Some(row)) = rows.next() {
         row.get(0).ok()
@@ -34,10 +38,16 @@ pub fn save_host(conn: &Connection, host: &str) {
     conn.execute(
         "INSERT OR REPLACE INTO hosts (host, last_used) VALUES (?1, CURRENT_TIMESTAMP)",
         [host],
-    ).unwrap();
+    )
+    .unwrap();
 }
 
-pub fn save_credentials(conn: &Connection, host: &str, username: &str, password: &str) -> Result<(), Box<dyn std::error::Error>> {
+pub fn save_credentials(
+    conn: &Connection,
+    host: &str,
+    username: &str,
+    password: &str,
+) -> Result<(), Box<dyn std::error::Error>> {
     conn.execute(
         "INSERT OR REPLACE INTO credentials (host, username, password) VALUES (?1, ?2, ?3)",
         (host, username, password),
@@ -46,7 +56,9 @@ pub fn save_credentials(conn: &Connection, host: &str, username: &str, password:
 }
 
 pub fn get_credentials(conn: &Connection, host: &str) -> Option<(String, String)> {
-    let mut stmt = conn.prepare("SELECT username, password FROM credentials WHERE host = ?1").unwrap();
+    let mut stmt = conn
+        .prepare("SELECT username, password FROM credentials WHERE host = ?1")
+        .unwrap();
     let mut rows = stmt.query([host]).unwrap();
     if let Ok(Some(row)) = rows.next() {
         let username: String = row.get(0).unwrap();
@@ -55,4 +67,4 @@ pub fn get_credentials(conn: &Connection, host: &str) -> Option<(String, String)
     } else {
         None
     }
-}
\ No newline at end of file
+}