fix: prepend urls starting with www with https (#45)

intuis · Jun 30, 2024 · 553d3b7 · 553d3b7
1 parent 8cfd72f
commit 553d3b7
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/rm-main/src/cli.rs b/rm-main/src/cli.rs
@@ -38,6 +38,11 @@ async fn add_torrent(config: &Config, torrent: String) -> Result<()> {
                 filename: Some(torrent),
                 ..Default::default()
             }
+        } else if torrent.starts_with("www") {
+            TorrentAddArgs {
+                filename: Some(format!("https://{torrent}")),
+                ..Default::default()
+            }
         } else {
             let mut torrent_file = File::open(torrent)?;
             let mut buf = vec![];

diff --git a/rm-main/src/transmission/action.rs b/rm-main/src/transmission/action.rs
@@ -25,8 +25,15 @@ pub async fn action_handler(ctx: app::Ctx, mut trans_rx: UnboundedReceiver<Torre
     while let Some(action) = trans_rx.recv().await {
         match action {
             TorrentAction::Add(ref url, directory) => {
+                let formatted = {
+                    if url.starts_with("www") {
+                        format!("https://{url}")
+                    } else {
+                        url.to_string()
+                    }
+                };
                 let args = TorrentAddArgs {
-                    filename: Some(url.clone()),
+                    filename: Some(formatted),
                     download_dir: directory,
                     ..Default::default()
                 };