refactor: Convert jamdl shell script to a Rust application

Co-authored-by: aider (gemini/gemini-2.5-pro-preview-05-06) <aider@aider.chat>
This commit is contained in:
Chaos Rogers 2025-08-17 14:01:03 +01:00
parent ac1f838c1c
commit 8a1d67166f
2 changed files with 111 additions and 0 deletions

11
Cargo.toml Normal file
View file

@ -0,0 +1,11 @@
[package]
name = "jamdl"
version = "0.1.0"
edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
anyhow = "1.0.82"
clap = { version = "4.5.4", features = ["derive"] }
tempfile = "3.10.1"

100
src/main.rs Normal file
View file

@ -0,0 +1,100 @@
use anyhow::{Context, Result};
use clap::Parser;
use std::fs;
use std::path::Path;
use std::process::Command;
use tempfile::TempDir;
// ==== CONFIGURATION ====
const NAS_HOST: &str = "(localaddress)";
const NAS_USER: &str = "ssh_user";
const NAS_PATH: &str = "/path/to/files/on/nas";
const COOKIES_FILE: &str = "/location/of/apple-music-cookies-file.txt";
#[derive(Parser, Debug)]
#[command(author, version, about, long_about = None)]
struct Args {
/// The URL of the video/music to download
#[arg(name = "URL")]
url: String,
}
fn main() -> Result<()> {
let args = Args::parse();
// Create temp directory
let temp_dir = TempDir::new().context("Failed to create temporary directory")?;
let temp_path = temp_dir.path();
println!(
"[INFO] Created temporary directory at: {}",
temp_path.display()
);
download_media(&args.url, temp_path)?;
transfer_files(temp_path)?;
// Cleanup is handled by TempDir's Drop trait
println!("[INFO] Cleaning up...");
Ok(())
}
fn download_media(video_url: &str, download_path: &Path) -> Result<()> {
let mut cmd: Command;
if video_url.contains("music.apple.com") {
println!("[INFO] Apple Music link detected. Using gamdl...");
cmd = Command::new("gamdl");
if Path::new(COOKIES_FILE).exists() {
cmd.args(["--cookies-path", COOKIES_FILE, video_url]);
} else {
println!(
"[WARN] cookies.txt not found at {} — running gamdl without it",
COOKIES_FILE
);
cmd.arg(video_url);
}
} else if video_url.contains("soundcloud.com") {
println!("[INFO] Soundcloud link detected. Using scdl...");
cmd = Command::new("scdl");
cmd.args(["-l", video_url]);
} else {
println!("[INFO] Non-Apple Music link. Using yt-dlp...");
cmd = Command::new("yt-dlp");
cmd.args(["-o", "%(title)s.%(ext)s", video_url]);
}
cmd.current_dir(download_path);
let status = cmd
.status()
.with_context(|| format!("Failed to execute command: {:?}", cmd))?;
if !status.success() {
anyhow::bail!("Download command failed with status: {}", status);
}
Ok(())
}
fn transfer_files(source_path: &Path) -> Result<()> {
let is_empty = fs::read_dir(source_path)?.next().is_none();
if !is_empty {
println!("[INFO] Transferring files to NAS...");
let mut scp_cmd = Command::new("scp");
let source = format!(
"{}/.",
source_path.to_str().context("Invalid source path")?
);
let destination = format!("{}@{}:{}", NAS_USER, NAS_HOST, NAS_PATH);
scp_cmd.args(["-r", &source, &destination]);
let status = scp_cmd.status().context("Failed to execute scp")?;
if !status.success() {
anyhow::bail!("scp failed with status: {}", status);
}
} else {
println!("[WARN] No files found to transfer.");
}
Ok(())
}