rs-suno 0.34.0

A download-only command-line tool for mirroring your Suno.ai library.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

//! Config loading and the CLI-to-`FlagOverrides` mapping.

use std::path::Path;

use anyhow::{Context, Result};
use suno_core::{Config, FlagOverrides, Settings};

use crate::cli::args::{GlobalArgs, SyncArgs};
use crate::cli::desired::ExitCode;
use crate::cli::logs;

pub(crate) enum ConfigState {
    Loaded(Config),
    Absent,
    Error(String),
}

/// Load config, printing a config error and returning its exit code on failure.
pub(crate) fn load_config_reported(
    override_path: Option<&Path>,
) -> std::result::Result<Option<Config>, ExitCode> {
    match load_config(override_path) {
        Ok(ConfigState::Loaded(cfg)) => Ok(Some(cfg)),
        Ok(ConfigState::Absent) => Ok(None),
        Ok(ConfigState::Error(message)) => {
            eprintln!("error: {message}");
            Err(ExitCode::Config)
        }
        Err(err) => {
            eprintln!("error: {err:#}");
            Err(ExitCode::General)
        }
    }
}

/// Load config from the override or platform default. A missing default file is
/// `Absent`; a missing explicit `--config`, or a parse error, is an error.
pub(crate) fn load_config(override_path: Option<&Path>) -> Result<ConfigState> {
    let explicit = override_path.is_some();
    let Some(path) = logs::config_path(override_path) else {
        return Ok(ConfigState::Absent);
    };
    match std::fs::read_to_string(&path) {
        Ok(text) => match Config::from_toml(&text) {
            Ok(cfg) => Ok(ConfigState::Loaded(cfg)),
            Err(err) => Ok(ConfigState::Error(format!("{}: {err}", path.display()))),
        },
        Err(err) if err.kind() == std::io::ErrorKind::NotFound => {
            if explicit {
                Ok(ConfigState::Error(format!(
                    "config file not found: {}",
                    path.display()
                )))
            } else {
                Ok(ConfigState::Absent)
            }
        }
        Err(err) => Err(err).with_context(|| format!("could not read {}", path.display())),
    }
}

pub(crate) fn flag_overrides(global: &GlobalArgs, args: &SyncArgs) -> FlagOverrides {
    FlagOverrides {
        token: global.token.clone(),
        settings: Settings {
            format: args.format.map(Into::into),
            concurrency: args.concurrency,
            retries: args.retries,
            min_newest: args.min_newest,
            // A presence-only toggle can only enable; absence defers to config/env.
            animated_covers: args.animated_covers.then_some(true),
            video_cover_retention: args.video_cover_retention.map(Into::into),
            animated_cover_quality: args.animated_cover_quality,
            animated_cover_max_fps: args.animated_cover_max_fps,
            animated_cover_max_width: args.animated_cover_max_width,
            animated_cover_compression_level: args.animated_cover_compression_level,
            animated_cover_lossless: args.animated_cover_lossless.then_some(true),
            details_sidecar: args.details_sidecar.then_some(true),
            lyrics_sidecar: args.lyrics_sidecar.then_some(true),
            lrc_sidecar: args.lrc_sidecar.then_some(true),
            video_mp4: args.video_mp4.then_some(true),
            download_stems: args.download_stems.then_some(true),
            stem_format: args.stem_format.map(Into::into),
            naming_template: args.naming_template.clone(),
            character_set: args.character_set.map(Into::into),
            // There is no `--token-command` flag; `token_command` resolves only
            // from config/env, so it stays `None` here (see `Settings`).
            ..Default::default()
        },
    }
}