Add rating loop

src/main.rs

@@ -2,24 +2,69 @@ mod database;
 mod player;
 mod userinterface;
 
+use std::env::args;
 use tokio;
 
 #[tokio::main]
 async fn main() {
+    // Initialize the mpris player
     let player = player::MprisPlayer::new().expect("Could not create player");
 
+    // Initialize the database
     let db = database::Database::new()
         .await
         .expect("Could not create database");
+
+    // Create the tables on an empty database
     db.create_tables().await.unwrap();
 
-    let (usernumber, userrating) = userinterface::get_user_rating(&db).await.expect("Lala");
+    if args().len() > 1 {
+        // Add a new user
+        let username = args().nth(1).unwrap();
+        if args().len() == 3 {
+            match args().nth(2).unwrap().parse::<i64>() {
+                Ok(id) => {
+                    let user_id = db
+                        .add_user_id(id, &username)
+                        .await
+                        .expect("Could not add user");
+                    println!("UserID for {username}: {user_id}");
+                }
+                Err(e) => eprintln!("{e}"),
+            }
+        } else {
+            let user_id = db.add_user(&username).await.expect("Could not add user");
+            println!("UserID for {username}: {user_id}");
+        }
+        return;
+    }
 
-    let (interpret, song) = player
-        .get_interpret_and_track()
-        .expect("Could not read track");
-
-    db.user_add_rating(usernumber, &interpret, &song, userrating)
+    let is_empty = db
+        .users_empty()
         .await
-        .expect("Can not add rating");
+        .expect("Could not read from database");
+
+    if is_empty {
+        println!("Add an user first. This can be achieved by using rate_music [name] [user_id]");
+        return;
+    }
+
+    loop {
+        let (usernumber, userrating) = userinterface::get_user_rating(&db).await.expect("Lala");
+
+        match player.get_interpret_and_track() {
+            Ok((interpret, song)) => {
+                if let Err(e) = db
+                    .user_add_rating(usernumber, &interpret, &song, userrating)
+                    .await
+                {
+                    eprintln!("{e}");
+                }
+            }
+            Err(e) => {
+                eprintln!("{e}");
+                continue;
+            }
+        }
+    }
 }