use std::fmt::{Display, Formatter};
use crate::cli::ArgumentsParser;
use crate::cli::CommandArguments::*;
use clap::Args;
use di::{injectable, Ref};
use serde::{Deserialize, Serialize};
use crate::options::{Options, OptionsProvider, ValueProvider};
#[derive(Args, Clone, Debug, Default, Deserialize, Serialize)]
pub struct RunnerOptions {
#[arg(long)]
pub cpus: Option<u16>,
}
#[injectable]
impl RunnerOptions {
fn new(provider: Ref<OptionsProvider>) -> Self {
provider.get()
}
}
impl Options for RunnerOptions {
fn get_name() -> String {
"Runner Options".to_owned()
}
fn get_value<TValue, F>(&self, select: F) -> TValue
where
F: FnOnce(&Self) -> Option<TValue>,
{
ValueProvider::get(self, select)
}
fn merge(&mut self, alternative: &Self) {
if self.cpus.is_none() {
self.cpus.clone_from(&alternative.cpus);
}
}
fn apply_defaults(&mut self) {
if self.cpus.is_none() {
self.cpus = Some(num_cpus::get() as u16);
}
}
#[must_use]
fn validate(&self) -> bool {
true
}
fn from_args() -> Option<Self> {
match ArgumentsParser::get() {
Some(Batch { runner, .. }) => Some(runner),
Some(Spectrogram { runner, .. }) => Some(runner),
Some(Transcode { runner, .. }) => Some(runner),
_ => None,
}
}
fn from_json(json: &str) -> Result<Self, serde_json::error::Error> {
serde_json::from_str(json)
}
}
impl Display for RunnerOptions {
fn fmt(&self, formatter: &mut Formatter<'_>) -> std::fmt::Result {
let output = if let Ok(json) = serde_json::to_string_pretty(self) {
json
} else {
format!("{self:?}")
};
output.fmt(formatter)
}
}