Marc Plano-Lesay
c3764c125a
Tags are attached to all magnet links provided by a given source, and passed to actions. This allows for e.g. better categorization in Bitmagnet.
86 lines
2.8 KiB
Rust
86 lines
2.8 KiB
Rust
use crate::actions::bitmagnet::BitmagnetConfig;
|
|
use crate::actions::transmission::TransmissionConfig;
|
|
use crate::args::Args;
|
|
use crate::notifications::ntfy::NtfyConfig;
|
|
use color_eyre::eyre::{eyre, Result, WrapErr};
|
|
use directories::ProjectDirs;
|
|
use figment::providers::Env;
|
|
use figment::{
|
|
providers::{Format, Toml},
|
|
Figment,
|
|
};
|
|
use figment_file_provider_adapter::FileAdapter;
|
|
use log::debug;
|
|
use serde::{Deserialize, Serialize};
|
|
use std::collections::HashMap;
|
|
use std::path::{Path, PathBuf};
|
|
|
|
/// Configuration for a Reddit source
|
|
#[derive(Debug, Serialize, Deserialize)]
|
|
pub struct SourceConfig {
|
|
pub username: String,
|
|
pub title_filter: Option<String>,
|
|
pub imdb_id: Option<String>,
|
|
#[serde(default)]
|
|
pub tags: Vec<String>,
|
|
}
|
|
|
|
/// Main application configuration
|
|
#[derive(Debug, Serialize, Deserialize)]
|
|
pub struct Config {
|
|
#[serde(default)]
|
|
pub bitmagnet: Option<BitmagnetConfig>,
|
|
|
|
#[serde(default)]
|
|
pub transmission: Option<TransmissionConfig>,
|
|
|
|
#[serde(default)]
|
|
pub ntfy: Option<NtfyConfig>,
|
|
|
|
#[serde(default)]
|
|
pub sources: HashMap<String, SourceConfig>,
|
|
}
|
|
|
|
/// Loads the configuration from the specified file or default location
|
|
pub fn load_config(args: &Args) -> Result<Config> {
|
|
let mut conf_extractor = Figment::new();
|
|
let config_file_path: Option<PathBuf> = match &args.config {
|
|
Some(path) => Some(Path::new(path).to_path_buf()),
|
|
None => ProjectDirs::from("fr", "enoent", "reddit-magnet")
|
|
.map(|p| p.config_dir().join("config.toml")),
|
|
};
|
|
match config_file_path {
|
|
Some(path) => {
|
|
if path.exists() {
|
|
debug!("Reading configuration from {:?}", path);
|
|
conf_extractor = conf_extractor.merge(FileAdapter::wrap(Toml::file_exact(path)));
|
|
} else {
|
|
debug!("Configuration file doesn't exist at {:?}", path);
|
|
}
|
|
}
|
|
None => {
|
|
debug!("No configuration file specified, using default configuration");
|
|
}
|
|
}
|
|
|
|
let conf: Config = conf_extractor
|
|
.merge(FileAdapter::wrap(Env::prefixed("REDDIT_MAGNET_")))
|
|
.extract()
|
|
.wrap_err_with(|| "Invalid configuration or insufficient command line arguments")?;
|
|
|
|
if conf.sources.is_empty() {
|
|
return Err(eyre!("No sources found in configuration. Please add at least one source to your configuration file."));
|
|
}
|
|
|
|
Ok(conf)
|
|
}
|
|
|
|
/// Gets the database path from the command line arguments or default location
|
|
pub fn get_db_path(args: &Args) -> Result<PathBuf> {
|
|
match &args.db {
|
|
Some(path) => Ok(PathBuf::from(path)),
|
|
None => ProjectDirs::from("fr", "enoent", "reddit-magnet")
|
|
.map(|p| p.data_dir().join("reddit-magnet.db"))
|
|
.ok_or_else(|| eyre!("Could not determine data directory")),
|
|
}
|
|
}
|