feat: refactor rust commands

src-tauri/src/commands.rs

@@ -1,140 +1,3 @@
-use keyring::Entry;
-use std::process::Command;
-use std::time::Duration;
-use webpage::{Webpage, WebpageOptions};
-
-#[tauri::command]
-pub async fn show_in_folder(path: String) {
-  #[cfg(target_os = "windows")]
-  {
-    Command::new("explorer")
-      .args(["/select,", &path]) // The comma after select is not a typo
-      .spawn()
-      .unwrap();
-  }
-
-  #[cfg(target_os = "linux")]
-  {
-    use std::fs::metadata;
-    use std::path::PathBuf;
-    if path.contains(",") {
-      // see https://gitlab.freedesktop.org/dbus/dbus/-/issues/76
-      let new_path = match metadata(&path).unwrap().is_dir() {
-        true => path,
-        false => {
-          let mut path2 = PathBuf::from(path);
-          path2.pop();
-          path2.into_os_string().into_string().unwrap()
-        }
-      };
-      Command::new("xdg-open").arg(&new_path).spawn().unwrap();
-    } else {
-      Command::new("dbus-send")
-        .args([
-          "--session",
-          "--dest=org.freedesktop.FileManager1",
-          "--type=method_call",
-          "/org/freedesktop/FileManager1",
-          "org.freedesktop.FileManager1.ShowItems",
-          format!("array:string:file://{path}").as_str(),
-          "string:\"\"",
-        ])
-        .spawn()
-        .unwrap();
-    }
-  }
-
-  #[cfg(target_os = "macos")]
-  {
-    Command::new("open").args(["-R", &path]).spawn().unwrap();
-  }
-}
-
-#[derive(serde::Serialize)]
-pub struct OpenGraphResponse {
-  title: String,
-  description: String,
-  url: String,
-  image: String,
-}
-
-pub async fn fetch_opengraph(url: String) -> OpenGraphResponse {
-  let options = WebpageOptions {
-    allow_insecure: false,
-    max_redirections: 3,
-    timeout: Duration::from_secs(15),
-    useragent: "lume - desktop app".to_string(),
-    ..Default::default()
-  };
-
-  let result = match Webpage::from_url(&url, options) {
-    Ok(webpage) => webpage,
-    Err(_) => {
-      return OpenGraphResponse {
-        title: "".to_string(),
-        description: "".to_string(),
-        url: "".to_string(),
-        image: "".to_string(),
-      }
-    }
-  };
-
-  let html = result.html;
-
-  return OpenGraphResponse {
-    title: html
-      .opengraph
-      .properties
-      .get("title")
-      .cloned()
-      .unwrap_or_default(),
-    description: html
-      .opengraph
-      .properties
-      .get("description")
-      .cloned()
-      .unwrap_or_default(),
-    url: html
-      .opengraph
-      .properties
-      .get("url")
-      .cloned()
-      .unwrap_or_default(),
-    image: html
-      .opengraph
-      .images
-      .get(0)
-      .and_then(|i| Some(i.url.clone()))
-      .unwrap_or_default(),
-  };
-}
-
-#[tauri::command]
-pub async fn opengraph(url: String) -> OpenGraphResponse {
-  let result = fetch_opengraph(url).await;
-  return result;
-}
-
-#[tauri::command]
-pub fn secure_save(key: String, value: String) -> Result<(), ()> {
-  let entry = Entry::new("Lume", &key).expect("Failed to create entry");
-  let _ = entry.set_password(&value);
-  Ok(())
-}
-
-#[tauri::command]
-pub fn secure_load(key: String) -> Result<String, String> {
-  let entry = Entry::new("Lume", &key).expect("Failed to create entry");
-  if let Ok(password) = entry.get_password() {
-    Ok(password.into())
-  } else {
-    Err("Not found".into())
-  }
-}
-
-#[tauri::command]
-pub fn secure_remove(key: String) -> Result<(), ()> {
-  let entry = Entry::new("Lume", &key).expect("Failed to create entry");
-  let _ = entry.delete_password();
-  Ok(())
-}
+pub mod folder;
+pub mod opg;
+pub mod secret;

src-tauri/src/commands/folder.rs

@@ -0,0 +1,48 @@
+use std::process::Command;
+
+#[tauri::command]
+pub async fn show_in_folder(path: String) {
+  #[cfg(target_os = "windows")]
+  {
+    Command::new("explorer")
+      .args(["/select,", &path]) // The comma after select is not a typo
+      .spawn()
+      .unwrap();
+  }
+
+  #[cfg(target_os = "linux")]
+  {
+    use std::fs::metadata;
+    use std::path::PathBuf;
+    if path.contains(",") {
+      // see https://gitlab.freedesktop.org/dbus/dbus/-/issues/76
+      let new_path = match metadata(&path).unwrap().is_dir() {
+        true => path,
+        false => {
+          let mut path2 = PathBuf::from(path);
+          path2.pop();
+          path2.into_os_string().into_string().unwrap()
+        }
+      };
+      Command::new("xdg-open").arg(&new_path).spawn().unwrap();
+    } else {
+      Command::new("dbus-send")
+        .args([
+          "--session",
+          "--dest=org.freedesktop.FileManager1",
+          "--type=method_call",
+          "/org/freedesktop/FileManager1",
+          "org.freedesktop.FileManager1.ShowItems",
+          format!("array:string:file://{path}").as_str(),
+          "string:\"\"",
+        ])
+        .spawn()
+        .unwrap();
+    }
+  }
+
+  #[cfg(target_os = "macos")]
+  {
+    Command::new("open").args(["-R", &path]).spawn().unwrap();
+  }
+}

src-tauri/src/commands/opg.rs

@@ -0,0 +1,50 @@
+use std::time::Duration;
+use webpage::{Webpage, WebpageOptions};
+
+#[derive(serde::Serialize)]
+pub struct OpenGraphResponse {
+  title: String,
+  description: String,
+  url: String,
+  image: String,
+}
+
+#[tauri::command]
+pub fn fetch_opg(url: String) -> Result<OpenGraphResponse, ()> {
+  let mut options = WebpageOptions::default();
+  options.allow_insecure = true;
+  options.max_redirections = 3;
+  options.timeout = Duration::from_secs(15);
+
+  let info = Webpage::from_url(&url, options).expect("Failed");
+  let html = info.html;
+
+  let result = OpenGraphResponse {
+    title: html
+      .opengraph
+      .properties
+      .get("title")
+      .cloned()
+      .unwrap_or_default(),
+    description: html
+      .opengraph
+      .properties
+      .get("description")
+      .cloned()
+      .unwrap_or_default(),
+    url: html
+      .opengraph
+      .properties
+      .get("url")
+      .cloned()
+      .unwrap_or_default(),
+    image: html
+      .opengraph
+      .images
+      .get(0)
+      .and_then(|i| Some(i.url.clone()))
+      .unwrap_or_default(),
+  };
+
+  Ok(result.into())
+}

src-tauri/src/commands/secret.rs

@@ -0,0 +1,25 @@
+use keyring::Entry;
+
+#[tauri::command]
+pub fn secure_save(key: String, value: String) -> Result<(), ()> {
+  let entry = Entry::new("Lume", &key).expect("Failed to create entry");
+  let _ = entry.set_password(&value);
+  Ok(())
+}
+
+#[tauri::command]
+pub fn secure_load(key: String) -> Result<String, String> {
+  let entry = Entry::new("Lume", &key).expect("Failed to create entry");
+  if let Ok(password) = entry.get_password() {
+    Ok(password.into())
+  } else {
+    Err("Not found".into())
+  }
+}
+
+#[tauri::command]
+pub fn secure_remove(key: String) -> Result<(), ()> {
+  let entry = Entry::new("Lume", &key).expect("Failed to remove entry");
+  let _ = entry.delete_password();
+  Ok(())
+}

src-tauri/src/main.rs

@@ -3,7 +3,7 @@
   windows_subsystem = "windows"
 )]
 
-mod commands;
+pub mod commands;
 
 use tauri_plugin_autostart::MacosLauncher;
 use tauri_plugin_sql::{Migration, MigrationKind};
@@ -23,14 +23,12 @@ fn main() {
       tauri_plugin_sql::Builder::default()
         .add_migrations(
           "sqlite:lume_v3.db",
-          vec![
-            Migration {
-              version: 20230418013219,
-              description: "initial data",
-              sql: include_str!("../migrations/20230418013219_initial_data.sql"),
-              kind: MigrationKind::Up,
-            },
-          ],
+          vec![Migration {
+            version: 20230418013219,
+            description: "initial data",
+            sql: include_str!("../migrations/20230418013219_initial_data.sql"),
+            kind: MigrationKind::Up,
+          }],
         )
         .build(),
     )
@@ -50,11 +48,11 @@ fn main() {
       Some(vec![]),
     ))
     .invoke_handler(tauri::generate_handler![
-      commands::opengraph,
-      commands::secure_save,
-      commands::secure_load,
-      commands::secure_remove,
-      commands::show_in_folder,
+      commands::secret::secure_save,
+      commands::secret::secure_load,
+      commands::secret::secure_remove,
+      commands::folder::show_in_folder,
+      commands::opg::fetch_opg,
     ])
     .run(ctx)
     .expect("error while running tauri application");