modify return values further

src/main.rs

@@ -1,6 +1,7 @@
 use postgres::{Client, NoTls};
 use std::{
     env::args,
+    error::Error,
     fs::{self, File},
     io::{self, BufRead, BufReader, Write},
 };
@@ -189,12 +190,12 @@ fn import_file(client: &mut postgres::Client, file: Option<String>) {
             continue;
         }
 
-        let output = client.query_one("SELECT insert_url($1)", &[&line]).unwrap();
-
-        let result: i32 = output.get(0);
-        if result == 1 {
-            writeln!(fucked_file, "{}", line).unwrap();
-        }
+        match client.query_one("SELECT insert_url($1);", &[&line]) {
+            Ok(_) => break,
+            Err(_) => {
+                writeln!(fucked_file, "{}", line).unwrap();
+            }
+        };
     }
 }
 
@@ -210,12 +211,19 @@ fn add_url(client: &mut postgres::Client, uri: Option<String>) {
         }
     };
 
-    let output = client.query_one("SELECT insert_url($1);", &[&url]).unwrap();
+    let url: &str = url.trim();
 
-    let result: i32 = output.get(0);
-    if result == 1 {
-        eprintln!("Database didn\'t like your input");
-    }
+    let output = match client.query_one("SELECT insert_url($1);", &[&url]) {
+        Ok(val) => val,
+        Err(val) => {
+            eprintln!("{}", val);
+            return;
+        }
+    };
+
+    let something: String = output.get(0);
+
+    println!("{}", something);
 }
 
 fn add_artist_with_custom_name(client: &mut postgres::Client, uri_and_artist: Option<Vec<String>>) {
@@ -238,17 +246,23 @@ fn add_artist_with_custom_name(client: &mut postgres::Client, uri_and_artist: Op
         }
     };
 
-    let output = client
-        .query_one(
-            "SELECT insert_url_with_custom_artist($1, $2);",
-            &[&url, &artist_name],
-        )
-        .unwrap();
+    let url: &str = url.trim();
+    let artist_name: &str = artist_name.trim();
 
-    let result: i32 = output.get(0);
-    if result == 1 {
-        eprintln!("Database didn\'t like your input");
-    }
+    let output = match client.query_one(
+        "SELECT insert_url_with_custom_artist($1, $2);",
+        &[&url, &artist_name],
+    ) {
+        Ok(val) => val,
+        Err(val) => {
+            eprintln!("{}", val);
+            return;
+        }
+    };
+
+    let something: String = output.get(0);
+
+    println!("{}", something);
 }
 
 fn main() {