Compare commits
3 commits
ee23385af9
...
53bbfd9501
| Author | SHA1 | Date | |
|---|---|---|---|
| 53bbfd9501 | |||
| 5631a12c7c | |||
| c1013feda6 |
2 changed files with 43 additions and 5 deletions
1
.gitignore
vendored
1
.gitignore
vendored
|
|
@ -17,3 +17,4 @@ target
|
|||
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
|
||||
#.idea/
|
||||
Cargo.lock
|
||||
.DS_Store
|
||||
|
|
|
|||
47
src/main.rs
47
src/main.rs
|
|
@ -1,12 +1,13 @@
|
|||
use anyhow::{Context, Result};
|
||||
use clap::Parser;
|
||||
use serde::Deserialize;
|
||||
use std::collections::HashMap;
|
||||
use std::fs;
|
||||
use std::path::Path;
|
||||
use std::process::Command;
|
||||
use tempfile::TempDir;
|
||||
|
||||
#[derive(Debug, Deserialize)]
|
||||
#[derive(Debug, Deserialize, Clone)]
|
||||
struct Settings {
|
||||
nas_host: String,
|
||||
nas_user: String,
|
||||
|
|
@ -14,12 +15,21 @@ struct Settings {
|
|||
cookies_file: String,
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize)]
|
||||
struct AppConfig {
|
||||
profiles: HashMap<String, Settings>,
|
||||
}
|
||||
|
||||
#[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,
|
||||
|
||||
/// Profile name to use from config (falls back to 'default')
|
||||
#[arg(short, long, default_value = "default")]
|
||||
profile: String,
|
||||
}
|
||||
|
||||
fn main() -> Result<()> {
|
||||
|
|
@ -48,10 +58,37 @@ fn main() -> Result<()> {
|
|||
|
||||
println!("[INFO] Using config file at: {}", config_path.display());
|
||||
|
||||
let settings = config::Config::builder()
|
||||
.add_source(config::File::from(config_path))
|
||||
.build()?
|
||||
.try_deserialize::<Settings>()?;
|
||||
let cfg = config::Config::builder()
|
||||
.add_source(config::File::from(config_path.clone()))
|
||||
.build()?;
|
||||
|
||||
let selected_profile = args.profile.clone();
|
||||
let settings: Settings = match cfg.clone().try_deserialize::<AppConfig>() {
|
||||
Ok(app_cfg) => {
|
||||
if let Some(s) = app_cfg.profiles.get(&selected_profile).cloned() {
|
||||
println!("[INFO] Using profile: {}", selected_profile);
|
||||
s
|
||||
} else if let Some(s) = app_cfg.profiles.get("default").cloned() {
|
||||
if selected_profile != "default" {
|
||||
println!("[WARN] Profile '{}' not found. Falling back to 'default'.", selected_profile);
|
||||
let available = app_cfg.profiles.keys().cloned().collect::<Vec<_>>().join(", ");
|
||||
println!("[INFO] Available profiles: {}", available);
|
||||
} else {
|
||||
println!("[INFO] Using profile: default");
|
||||
}
|
||||
s
|
||||
} else {
|
||||
anyhow::bail!("No profile '{}' found and no 'default' profile defined in config.", selected_profile);
|
||||
}
|
||||
}
|
||||
Err(_) => {
|
||||
if selected_profile != "default" {
|
||||
println!("[WARN] Requested profile '{}' but config file uses single-profile format. Using that single profile.", selected_profile);
|
||||
}
|
||||
println!("[INFO] Using legacy single-profile configuration.");
|
||||
cfg.try_deserialize::<Settings>()?
|
||||
}
|
||||
};
|
||||
|
||||
// Create temp directory
|
||||
let temp_dir = TempDir::new().context("Failed to create temporary directory")?;
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue