work on logger
This commit is contained in:
parent
e84aa7e4e3
commit
6946b747ee
2
Cargo.lock
generated
2
Cargo.lock
generated
@ -1322,6 +1322,7 @@ dependencies = [
|
||||
"home",
|
||||
"jsonwebtoken",
|
||||
"lazy_static",
|
||||
"lettre",
|
||||
"lexical-sort",
|
||||
"local-ip-address",
|
||||
"log",
|
||||
@ -1337,6 +1338,7 @@ dependencies = [
|
||||
"sanitize-filename",
|
||||
"serde",
|
||||
"serde_json",
|
||||
"shlex",
|
||||
"signal-child",
|
||||
"simplelog",
|
||||
"sqlx",
|
||||
|
@ -31,6 +31,7 @@ futures-util = { version = "0.3", default-features = false, features = ["std"] }
|
||||
home = "0.5"
|
||||
jsonwebtoken = "9"
|
||||
lazy_static = "1.4"
|
||||
lettre = { version = "0.11", features = ["builder", "rustls-tls", "smtp-transport"], default-features = false }
|
||||
lexical-sort = "0.3"
|
||||
local-ip-address = "0.6"
|
||||
log = { version = "0.4", features = ["std", "serde", "kv", "kv_std", "kv_sval", "kv_serde"] }
|
||||
@ -47,6 +48,7 @@ sanitize-filename = "0.5"
|
||||
serde = { version = "1.0", features = ["derive"] }
|
||||
serde_json = "1.0"
|
||||
simplelog = { version = "0.12", features = ["paris"] }
|
||||
shlex = "1.1"
|
||||
static-files = "0.2"
|
||||
sysinfo ={ version = "0.30", features = ["linux-netdevs"] }
|
||||
sqlx = { version = "0.7", features = ["runtime-tokio", "sqlite"] }
|
||||
|
@ -150,7 +150,7 @@ pub fn file_formatter(
|
||||
fn main() {
|
||||
let to_file = true;
|
||||
|
||||
Logger::try_with_str("trace")
|
||||
Logger::try_with_str("WARN")
|
||||
.expect("LogSpecification String has errors")
|
||||
.format(console_formatter)
|
||||
.print_message()
|
||||
|
@ -6,10 +6,12 @@ use sysinfo::{Disks, Networks, System};
|
||||
|
||||
pub mod api;
|
||||
pub mod db;
|
||||
pub mod macros;
|
||||
pub mod player;
|
||||
pub mod sse;
|
||||
pub mod utils;
|
||||
|
||||
use utils::advanced_config::AdvancedConfig;
|
||||
use utils::args_parse::Args;
|
||||
|
||||
lazy_static! {
|
||||
|
6
ffplayout/src/macros/mod.rs
Normal file
6
ffplayout/src/macros/mod.rs
Normal file
@ -0,0 +1,6 @@
|
||||
#[macro_export]
|
||||
macro_rules! vec_strings {
|
||||
($($str:expr),*) => ({
|
||||
vec![$($str.to_string(),)*] as Vec<String>
|
||||
});
|
||||
}
|
98
ffplayout/src/utils/advanced_config.rs
Normal file
98
ffplayout/src/utils/advanced_config.rs
Normal file
@ -0,0 +1,98 @@
|
||||
use std::{fs::File, io::Read, path::PathBuf};
|
||||
|
||||
use serde::{Deserialize, Serialize};
|
||||
use shlex::split;
|
||||
|
||||
#[derive(Debug, Default, Serialize, Deserialize, Clone)]
|
||||
pub struct AdvancedConfig {
|
||||
pub decoder: DecoderConfig,
|
||||
pub encoder: EncoderConfig,
|
||||
pub ingest: IngestConfig,
|
||||
}
|
||||
|
||||
#[derive(Debug, Default, Serialize, Deserialize, Clone)]
|
||||
pub struct DecoderConfig {
|
||||
pub input_param: Option<String>,
|
||||
pub output_param: Option<String>,
|
||||
pub filters: Filters,
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub input_cmd: Option<Vec<String>>,
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub output_cmd: Option<Vec<String>>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Default, Serialize, Deserialize, Clone)]
|
||||
pub struct EncoderConfig {
|
||||
pub input_param: Option<String>,
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub input_cmd: Option<Vec<String>>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Default, Serialize, Deserialize, Clone)]
|
||||
pub struct IngestConfig {
|
||||
pub input_param: Option<String>,
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub input_cmd: Option<Vec<String>>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Default, Serialize, Deserialize, Clone)]
|
||||
pub struct Filters {
|
||||
pub deinterlace: Option<String>,
|
||||
pub pad_scale_w: Option<String>,
|
||||
pub pad_scale_h: Option<String>,
|
||||
pub pad_video: Option<String>,
|
||||
pub fps: Option<String>,
|
||||
pub scale: Option<String>,
|
||||
pub set_dar: Option<String>,
|
||||
pub fade_in: Option<String>,
|
||||
pub fade_out: Option<String>,
|
||||
pub overlay_logo_scale: Option<String>,
|
||||
pub overlay_logo_fade_in: Option<String>,
|
||||
pub overlay_logo_fade_out: Option<String>,
|
||||
pub overlay_logo: Option<String>,
|
||||
pub tpad: Option<String>,
|
||||
pub drawtext_from_file: Option<String>,
|
||||
pub drawtext_from_zmq: Option<String>,
|
||||
pub aevalsrc: Option<String>,
|
||||
pub afade_in: Option<String>,
|
||||
pub afade_out: Option<String>,
|
||||
pub apad: Option<String>,
|
||||
pub volume: Option<String>,
|
||||
pub split: Option<String>,
|
||||
}
|
||||
|
||||
impl AdvancedConfig {
|
||||
pub fn new(cfg_path: PathBuf) -> Self {
|
||||
let mut config: AdvancedConfig = Default::default();
|
||||
|
||||
if let Ok(mut file) = File::open(cfg_path) {
|
||||
let mut contents = String::new();
|
||||
|
||||
if let Err(e) = file.read_to_string(&mut contents) {
|
||||
eprintln!("Read advanced config file: {e}")
|
||||
};
|
||||
|
||||
if let Ok(tm) = toml_edit::de::from_str(&contents) {
|
||||
config = tm
|
||||
};
|
||||
|
||||
if let Some(input_parm) = &config.decoder.input_param {
|
||||
config.decoder.input_cmd = split(input_parm);
|
||||
}
|
||||
|
||||
if let Some(output_param) = &config.decoder.output_param {
|
||||
config.decoder.output_cmd = split(output_param);
|
||||
}
|
||||
|
||||
if let Some(input_param) = &config.encoder.input_param {
|
||||
config.encoder.input_cmd = split(input_param);
|
||||
}
|
||||
|
||||
if let Some(input_param) = &config.ingest.input_param {
|
||||
config.ingest.input_cmd = split(input_param);
|
||||
}
|
||||
};
|
||||
|
||||
config
|
||||
}
|
||||
}
|
585
ffplayout/src/utils/config.rs
Normal file
585
ffplayout/src/utils/config.rs
Normal file
@ -0,0 +1,585 @@
|
||||
use std::{
|
||||
env, fmt,
|
||||
fs::File,
|
||||
io::Read,
|
||||
path::{Path, PathBuf},
|
||||
process,
|
||||
str::FromStr,
|
||||
};
|
||||
|
||||
use chrono::NaiveTime;
|
||||
use log::LevelFilter;
|
||||
use serde::{de, Deserialize, Deserializer, Serialize, Serializer};
|
||||
use shlex::split;
|
||||
|
||||
use crate::AdvancedConfig;
|
||||
|
||||
use crate::utils::{free_tcp_socket, time_to_sec};
|
||||
use crate::vec_strings;
|
||||
|
||||
pub const DUMMY_LEN: f64 = 60.0;
|
||||
pub const IMAGE_FORMAT: [&str; 21] = [
|
||||
"bmp", "dds", "dpx", "exr", "gif", "hdr", "j2k", "jpg", "jpeg", "pcx", "pfm", "pgm", "phm",
|
||||
"png", "psd", "ppm", "sgi", "svg", "tga", "tif", "webp",
|
||||
];
|
||||
|
||||
// Some well known errors can be safely ignore
|
||||
pub const FFMPEG_IGNORE_ERRORS: [&str; 11] = [
|
||||
"ac-tex damaged",
|
||||
"codec s302m, is muxed as a private data stream",
|
||||
"corrupt decoded frame in stream",
|
||||
"corrupt input packet in stream",
|
||||
"end mismatch left",
|
||||
"Packet corrupt",
|
||||
"Referenced QT chapter track not found",
|
||||
"skipped MB in I-frame at",
|
||||
"Thread message queue blocking",
|
||||
"Warning MVs not available",
|
||||
"frame size not set",
|
||||
];
|
||||
|
||||
pub const FFMPEG_UNRECOVERABLE_ERRORS: [&str; 5] = [
|
||||
"Address already in use",
|
||||
"Invalid argument",
|
||||
"Numerical result",
|
||||
"Error initializing complex filters",
|
||||
"Error while decoding stream #0:0: Invalid data found when processing input",
|
||||
];
|
||||
|
||||
#[derive(Debug, Serialize, Deserialize, Clone, Eq, PartialEq)]
|
||||
#[serde(rename_all = "lowercase")]
|
||||
pub enum OutputMode {
|
||||
Desktop,
|
||||
HLS,
|
||||
Null,
|
||||
Stream,
|
||||
}
|
||||
|
||||
impl FromStr for OutputMode {
|
||||
type Err = String;
|
||||
|
||||
fn from_str(input: &str) -> Result<Self, Self::Err> {
|
||||
match input {
|
||||
"desktop" => Ok(Self::Desktop),
|
||||
"hls" => Ok(Self::HLS),
|
||||
"null" => Ok(Self::Null),
|
||||
"stream" => Ok(Self::Stream),
|
||||
_ => Err("Use 'desktop', 'hls', 'null' or 'stream'".to_string()),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Serialize, Deserialize, Clone, Eq, PartialEq)]
|
||||
#[serde(rename_all = "lowercase")]
|
||||
pub enum ProcessMode {
|
||||
Folder,
|
||||
Playlist,
|
||||
}
|
||||
|
||||
impl fmt::Display for ProcessMode {
|
||||
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||||
match *self {
|
||||
ProcessMode::Folder => write!(f, "folder"),
|
||||
ProcessMode::Playlist => write!(f, "playlist"),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl FromStr for ProcessMode {
|
||||
type Err = String;
|
||||
|
||||
fn from_str(input: &str) -> Result<Self, Self::Err> {
|
||||
match input {
|
||||
"folder" => Ok(Self::Folder),
|
||||
"playlist" => Ok(Self::Playlist),
|
||||
_ => Err("Use 'folder' or 'playlist'".to_string()),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub fn string_to_log_level<'de, D>(deserializer: D) -> Result<LevelFilter, D::Error>
|
||||
where
|
||||
D: Deserializer<'de>,
|
||||
{
|
||||
let s: String = Deserialize::deserialize(deserializer)?;
|
||||
|
||||
match s.to_lowercase().as_str() {
|
||||
"debug" => Ok(LevelFilter::Debug),
|
||||
"error" => Ok(LevelFilter::Error),
|
||||
"info" => Ok(LevelFilter::Info),
|
||||
"trace" => Ok(LevelFilter::Trace),
|
||||
"warn" => Ok(LevelFilter::Warn),
|
||||
"off" => Ok(LevelFilter::Off),
|
||||
_ => Err(de::Error::custom("Error level not exists!")),
|
||||
}
|
||||
}
|
||||
|
||||
fn log_level_to_string<S>(l: &LevelFilter, s: S) -> Result<S::Ok, S::Error>
|
||||
where
|
||||
S: Serializer,
|
||||
{
|
||||
match l {
|
||||
LevelFilter::Debug => s.serialize_str("DEBUG"),
|
||||
LevelFilter::Error => s.serialize_str("ERROR"),
|
||||
LevelFilter::Info => s.serialize_str("INFO"),
|
||||
LevelFilter::Trace => s.serialize_str("TRACE"),
|
||||
LevelFilter::Warn => s.serialize_str("WARNING"),
|
||||
LevelFilter::Off => s.serialize_str("OFF"),
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||||
pub struct Template {
|
||||
pub sources: Vec<Source>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||||
pub struct Source {
|
||||
pub start: NaiveTime,
|
||||
pub duration: NaiveTime,
|
||||
pub shuffle: bool,
|
||||
pub paths: Vec<PathBuf>,
|
||||
}
|
||||
|
||||
/// Global Config
|
||||
///
|
||||
/// This we init ones, when ffplayout is starting and use them globally in the hole program.
|
||||
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||||
pub struct PlayoutConfig {
|
||||
#[serde(default, skip_serializing, skip_deserializing)]
|
||||
pub advanced: Option<AdvancedConfig>,
|
||||
pub general: General,
|
||||
pub rpc_server: RpcServer,
|
||||
pub mail: Mail,
|
||||
pub logging: Logging,
|
||||
pub processing: Processing,
|
||||
pub ingest: Ingest,
|
||||
pub playlist: Playlist,
|
||||
pub storage: Storage,
|
||||
pub text: Text,
|
||||
#[serde(default)]
|
||||
pub task: Task,
|
||||
pub out: Out,
|
||||
}
|
||||
|
||||
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||||
pub struct General {
|
||||
pub help_text: String,
|
||||
pub stop_threshold: f64,
|
||||
|
||||
#[serde(default, skip_serializing, skip_deserializing)]
|
||||
pub config_path: String,
|
||||
|
||||
#[serde(default)]
|
||||
pub stat_file: String,
|
||||
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub generate: Option<Vec<String>>,
|
||||
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub ffmpeg_filters: Vec<String>,
|
||||
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub ffmpeg_libs: Vec<String>,
|
||||
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub template: Option<Template>,
|
||||
|
||||
#[serde(default, skip_serializing, skip_deserializing)]
|
||||
pub skip_validation: bool,
|
||||
|
||||
#[serde(default, skip_serializing, skip_deserializing)]
|
||||
pub validate: bool,
|
||||
}
|
||||
|
||||
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||||
pub struct RpcServer {
|
||||
pub help_text: String,
|
||||
pub enable: bool,
|
||||
pub address: String,
|
||||
pub authorization: String,
|
||||
}
|
||||
|
||||
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||||
pub struct Mail {
|
||||
pub help_text: String,
|
||||
pub subject: String,
|
||||
pub smtp_server: String,
|
||||
pub starttls: bool,
|
||||
pub sender_addr: String,
|
||||
pub sender_pass: String,
|
||||
pub recipient: String,
|
||||
pub mail_level: String,
|
||||
pub interval: u64,
|
||||
}
|
||||
|
||||
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||||
pub struct Logging {
|
||||
pub help_text: String,
|
||||
pub log_to_file: bool,
|
||||
pub backup_count: usize,
|
||||
pub local_time: bool,
|
||||
pub timestamp: bool,
|
||||
#[serde(alias = "log_path")]
|
||||
pub path: PathBuf,
|
||||
#[serde(
|
||||
alias = "log_level",
|
||||
serialize_with = "log_level_to_string",
|
||||
deserialize_with = "string_to_log_level"
|
||||
)]
|
||||
pub level: LevelFilter,
|
||||
pub ffmpeg_level: String,
|
||||
pub ingest_level: Option<String>,
|
||||
#[serde(default)]
|
||||
pub detect_silence: bool,
|
||||
#[serde(default)]
|
||||
pub ignore_lines: Vec<String>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||||
pub struct Processing {
|
||||
pub help_text: String,
|
||||
pub mode: ProcessMode,
|
||||
#[serde(default)]
|
||||
pub audio_only: bool,
|
||||
#[serde(default = "default_track_index")]
|
||||
pub audio_track_index: i32,
|
||||
#[serde(default)]
|
||||
pub copy_audio: bool,
|
||||
#[serde(default)]
|
||||
pub copy_video: bool,
|
||||
pub width: i64,
|
||||
pub height: i64,
|
||||
pub aspect: f64,
|
||||
pub fps: f64,
|
||||
pub add_logo: bool,
|
||||
pub logo: String,
|
||||
pub logo_scale: String,
|
||||
pub logo_opacity: f32,
|
||||
pub logo_position: String,
|
||||
#[serde(default = "default_tracks")]
|
||||
pub audio_tracks: i32,
|
||||
#[serde(default = "default_channels")]
|
||||
pub audio_channels: u8,
|
||||
pub volume: f64,
|
||||
#[serde(default)]
|
||||
pub custom_filter: String,
|
||||
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub cmd: Option<Vec<String>>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||||
pub struct Ingest {
|
||||
pub help_text: String,
|
||||
pub enable: bool,
|
||||
input_param: String,
|
||||
#[serde(default)]
|
||||
pub custom_filter: String,
|
||||
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub input_cmd: Option<Vec<String>>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||||
pub struct Playlist {
|
||||
pub help_text: String,
|
||||
pub path: PathBuf,
|
||||
pub day_start: String,
|
||||
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub start_sec: Option<f64>,
|
||||
|
||||
pub length: String,
|
||||
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub length_sec: Option<f64>,
|
||||
|
||||
pub infinit: bool,
|
||||
}
|
||||
|
||||
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||||
pub struct Storage {
|
||||
pub help_text: String,
|
||||
pub path: PathBuf,
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub paths: Vec<PathBuf>,
|
||||
#[serde(alias = "filler_clip")]
|
||||
pub filler: PathBuf,
|
||||
pub extensions: Vec<String>,
|
||||
pub shuffle: bool,
|
||||
}
|
||||
|
||||
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||||
pub struct Text {
|
||||
pub help_text: String,
|
||||
pub add_text: bool,
|
||||
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub node_pos: Option<usize>,
|
||||
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub zmq_stream_socket: Option<String>,
|
||||
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub zmq_server_socket: Option<String>,
|
||||
|
||||
pub fontfile: String,
|
||||
pub text_from_filename: bool,
|
||||
pub style: String,
|
||||
pub regex: String,
|
||||
}
|
||||
|
||||
#[derive(Debug, Default, Serialize, Deserialize, Clone)]
|
||||
pub struct Task {
|
||||
pub help_text: String,
|
||||
pub enable: bool,
|
||||
pub path: PathBuf,
|
||||
}
|
||||
|
||||
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||||
pub struct Out {
|
||||
pub help_text: String,
|
||||
pub mode: OutputMode,
|
||||
pub output_param: String,
|
||||
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub output_count: usize,
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub output_filter: Option<String>,
|
||||
#[serde(skip_serializing, skip_deserializing)]
|
||||
pub output_cmd: Option<Vec<String>>,
|
||||
}
|
||||
|
||||
fn default_track_index() -> i32 {
|
||||
-1
|
||||
}
|
||||
|
||||
fn default_tracks() -> i32 {
|
||||
1
|
||||
}
|
||||
|
||||
fn default_channels() -> u8 {
|
||||
2
|
||||
}
|
||||
|
||||
impl PlayoutConfig {
|
||||
/// Read config from YAML file, and set some extra config values.
|
||||
pub fn new(cfg_path: Option<PathBuf>, advanced_path: Option<PathBuf>) -> Self {
|
||||
let mut config_path = PathBuf::from("/etc/ffplayout/ffplayout.toml");
|
||||
|
||||
if let Some(cfg) = cfg_path {
|
||||
config_path = cfg;
|
||||
}
|
||||
|
||||
if !config_path.is_file() {
|
||||
if Path::new("./assets/ffplayout.toml").is_file() {
|
||||
config_path = PathBuf::from("./assets/ffplayout.toml")
|
||||
} else if let Some(p) = env::current_exe().ok().as_ref().and_then(|op| op.parent()) {
|
||||
config_path = p.join("ffplayout.toml")
|
||||
};
|
||||
}
|
||||
|
||||
let mut file = match File::open(&config_path) {
|
||||
Ok(file) => file,
|
||||
Err(_) => {
|
||||
eprintln!(
|
||||
"ffplayout.toml not found!\nPut \"ffplayout.toml\" in \"/etc/playout/\" or beside the executable!"
|
||||
);
|
||||
process::exit(1);
|
||||
}
|
||||
};
|
||||
|
||||
let mut contents = String::new();
|
||||
|
||||
if let Err(e) = file.read_to_string(&mut contents) {
|
||||
eprintln!("Read config file: {e}")
|
||||
};
|
||||
|
||||
let mut config: PlayoutConfig = toml_edit::de::from_str(&contents).unwrap();
|
||||
|
||||
if let Some(adv_path) = advanced_path {
|
||||
config.advanced = Some(AdvancedConfig::new(adv_path))
|
||||
}
|
||||
|
||||
config.general.generate = None;
|
||||
|
||||
config.general.config_path = config_path.to_string_lossy().to_string();
|
||||
|
||||
config.general.stat_file = home::home_dir()
|
||||
.unwrap_or_else(env::temp_dir)
|
||||
.join(if config.general.stat_file.is_empty() {
|
||||
".ffp_status"
|
||||
} else {
|
||||
&config.general.stat_file
|
||||
})
|
||||
.display()
|
||||
.to_string();
|
||||
|
||||
if config.logging.ingest_level.is_none() {
|
||||
config.logging.ingest_level = Some(config.logging.ffmpeg_level.clone())
|
||||
}
|
||||
|
||||
config.playlist.start_sec = Some(time_to_sec(&config.playlist.day_start));
|
||||
|
||||
if config.playlist.length.contains(':') {
|
||||
config.playlist.length_sec = Some(time_to_sec(&config.playlist.length));
|
||||
} else {
|
||||
config.playlist.length_sec = Some(86400.0);
|
||||
}
|
||||
|
||||
if config.processing.add_logo && !Path::new(&config.processing.logo).is_file() {
|
||||
config.processing.add_logo = false;
|
||||
}
|
||||
|
||||
config.processing.logo_scale = config
|
||||
.processing
|
||||
.logo_scale
|
||||
.trim_end_matches('~')
|
||||
.to_string();
|
||||
|
||||
if config.processing.audio_tracks < 1 {
|
||||
config.processing.audio_tracks = 1
|
||||
}
|
||||
|
||||
let mut process_cmd = vec_strings![];
|
||||
let advanced_output_cmd = config
|
||||
.advanced
|
||||
.as_ref()
|
||||
.and_then(|a| a.decoder.output_cmd.clone());
|
||||
|
||||
if config.processing.audio_only {
|
||||
process_cmd.append(&mut vec_strings!["-vn"]);
|
||||
} else if config.processing.copy_video {
|
||||
process_cmd.append(&mut vec_strings!["-c:v", "copy"]);
|
||||
} else if let Some(decoder_cmd) = &advanced_output_cmd {
|
||||
process_cmd.append(&mut decoder_cmd.clone());
|
||||
} else {
|
||||
let bitrate = format!(
|
||||
"{}k",
|
||||
config.processing.width * config.processing.height / 16
|
||||
);
|
||||
|
||||
let buff_size = format!(
|
||||
"{}k",
|
||||
(config.processing.width * config.processing.height / 16) / 2
|
||||
);
|
||||
|
||||
process_cmd.append(&mut vec_strings![
|
||||
"-pix_fmt",
|
||||
"yuv420p",
|
||||
"-r",
|
||||
&config.processing.fps,
|
||||
"-c:v",
|
||||
"mpeg2video",
|
||||
"-g",
|
||||
"1",
|
||||
"-b:v",
|
||||
&bitrate,
|
||||
"-minrate",
|
||||
&bitrate,
|
||||
"-maxrate",
|
||||
&bitrate,
|
||||
"-bufsize",
|
||||
&buff_size
|
||||
]);
|
||||
}
|
||||
|
||||
if config.processing.copy_audio {
|
||||
process_cmd.append(&mut vec_strings!["-c:a", "copy"]);
|
||||
} else if advanced_output_cmd.is_none() {
|
||||
process_cmd.append(&mut pre_audio_codec(
|
||||
&config.processing.custom_filter,
|
||||
&config.ingest.custom_filter,
|
||||
config.processing.audio_channels,
|
||||
));
|
||||
}
|
||||
|
||||
process_cmd.append(&mut vec_strings!["-f", "mpegts", "-"]);
|
||||
|
||||
config.processing.cmd = Some(process_cmd);
|
||||
|
||||
config.ingest.input_cmd = split(config.ingest.input_param.as_str());
|
||||
|
||||
config.out.output_count = 1;
|
||||
config.out.output_filter = None;
|
||||
|
||||
if config.out.mode == OutputMode::Null {
|
||||
config.out.output_cmd = Some(vec_strings!["-f", "null", "-"]);
|
||||
} else if let Some(mut cmd) = split(config.out.output_param.as_str()) {
|
||||
// get output count according to the var_stream_map value, or by counting output parameters
|
||||
if let Some(i) = cmd.clone().iter().position(|m| m == "-var_stream_map") {
|
||||
config.out.output_count = cmd[i + 1].split_whitespace().count();
|
||||
} else {
|
||||
config.out.output_count = cmd
|
||||
.iter()
|
||||
.enumerate()
|
||||
.filter(|(i, p)| i > &0 && !p.starts_with('-') && !cmd[i - 1].starts_with('-'))
|
||||
.count();
|
||||
}
|
||||
|
||||
if let Some(i) = cmd.clone().iter().position(|r| r == "-filter_complex") {
|
||||
config.out.output_filter = Some(cmd[i + 1].clone());
|
||||
cmd.remove(i);
|
||||
cmd.remove(i);
|
||||
}
|
||||
|
||||
config.out.output_cmd = Some(cmd);
|
||||
}
|
||||
|
||||
// when text overlay without text_from_filename is on, turn also the RPC server on,
|
||||
// to get text messages from it
|
||||
if config.text.add_text && !config.text.text_from_filename {
|
||||
config.rpc_server.enable = true;
|
||||
config.text.zmq_stream_socket = free_tcp_socket(String::new());
|
||||
config.text.zmq_server_socket =
|
||||
free_tcp_socket(config.text.zmq_stream_socket.clone().unwrap_or_default());
|
||||
config.text.node_pos = Some(2);
|
||||
} else {
|
||||
config.text.zmq_stream_socket = None;
|
||||
config.text.zmq_server_socket = None;
|
||||
config.text.node_pos = None;
|
||||
}
|
||||
|
||||
config
|
||||
}
|
||||
}
|
||||
|
||||
impl Default for PlayoutConfig {
|
||||
fn default() -> Self {
|
||||
Self::new(None, None)
|
||||
}
|
||||
}
|
||||
|
||||
/// When custom_filter contains loudnorm filter use a different audio encoder,
|
||||
/// s302m has higher quality, but is experimental
|
||||
/// and works not well together with the loudnorm filter.
|
||||
fn pre_audio_codec(proc_filter: &str, ingest_filter: &str, channel_count: u8) -> Vec<String> {
|
||||
let mut codec = vec_strings![
|
||||
"-c:a",
|
||||
"s302m",
|
||||
"-strict",
|
||||
"-2",
|
||||
"-sample_fmt",
|
||||
"s16",
|
||||
"-ar",
|
||||
"48000",
|
||||
"-ac",
|
||||
channel_count
|
||||
];
|
||||
|
||||
if proc_filter.contains("loudnorm") || ingest_filter.contains("loudnorm") {
|
||||
codec = vec_strings![
|
||||
"-c:a",
|
||||
"mp2",
|
||||
"-b:a",
|
||||
"384k",
|
||||
"-ar",
|
||||
"48000",
|
||||
"-ac",
|
||||
channel_count
|
||||
];
|
||||
}
|
||||
|
||||
codec
|
||||
}
|
163
ffplayout/src/utils/logging.rs
Normal file
163
ffplayout/src/utils/logging.rs
Normal file
@ -0,0 +1,163 @@
|
||||
extern crate log;
|
||||
extern crate simplelog;
|
||||
|
||||
use std::{
|
||||
collections::HashMap,
|
||||
io::{self, ErrorKind, Write},
|
||||
path::PathBuf,
|
||||
sync::{atomic::Ordering, Arc, Mutex},
|
||||
thread::{self, sleep},
|
||||
time::Duration,
|
||||
};
|
||||
|
||||
use chrono::prelude::*;
|
||||
use flexi_logger::writers::{FileLogWriter, LogWriter};
|
||||
use flexi_logger::{Age, Cleanup, Criterion, DeferredNow, FileSpec, Logger, Naming};
|
||||
use lettre::{
|
||||
message::header, transport::smtp::authentication::Credentials, Message, SmtpTransport,
|
||||
Transport,
|
||||
};
|
||||
use log::{kv::Value, Level, LevelFilter, Log, Metadata, Record};
|
||||
use paris::formatter::colorize_string;
|
||||
|
||||
use crate::utils::{
|
||||
config::{Logging, PlayoutConfig},
|
||||
control::ProcessControl,
|
||||
};
|
||||
|
||||
pub struct LogConsole;
|
||||
|
||||
impl LogWriter for LogConsole {
|
||||
fn write(&self, now: &mut DeferredNow, record: &Record<'_>) -> std::io::Result<()> {
|
||||
console_formatter(&mut std::io::stderr(), now, record)?;
|
||||
|
||||
println!();
|
||||
Ok(())
|
||||
}
|
||||
fn flush(&self) -> std::io::Result<()> {
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
struct MultiFileLogger {
|
||||
config: Logging,
|
||||
writers: Arc<Mutex<HashMap<String, Arc<Mutex<FileLogWriter>>>>>,
|
||||
}
|
||||
|
||||
impl MultiFileLogger {
|
||||
pub fn new(config: &Logging) -> Self {
|
||||
MultiFileLogger {
|
||||
config: config.clone(),
|
||||
writers: Arc::new(Mutex::new(HashMap::new())),
|
||||
}
|
||||
}
|
||||
|
||||
fn get_writer(&self, channel: &str) -> io::Result<Arc<Mutex<FileLogWriter>>> {
|
||||
let mut writers = self.writers.lock().unwrap();
|
||||
if !writers.contains_key(channel) {
|
||||
let writer = FileLogWriter::builder(
|
||||
FileSpec::default()
|
||||
.suppress_timestamp()
|
||||
.directory(self.config.path.clone())
|
||||
.basename("ffplayout")
|
||||
.discriminant(channel),
|
||||
)
|
||||
.format(file_formatter)
|
||||
.append()
|
||||
.rotate(
|
||||
Criterion::Age(Age::Day),
|
||||
Naming::TimestampsDirect,
|
||||
Cleanup::KeepLogFiles(self.config.backup_count),
|
||||
)
|
||||
.try_build()
|
||||
.map_err(|e| io::Error::new(io::ErrorKind::Other, e.to_string()))?;
|
||||
writers.insert(channel.to_string(), Arc::new(Mutex::new(writer)));
|
||||
}
|
||||
Ok(writers.get(channel).unwrap().clone())
|
||||
}
|
||||
}
|
||||
|
||||
impl LogWriter for MultiFileLogger {
|
||||
fn write(&self, now: &mut DeferredNow, record: &Record) -> io::Result<()> {
|
||||
let channel = record
|
||||
.key_values()
|
||||
.get("channel".into())
|
||||
.unwrap_or(Value::null())
|
||||
.to_string();
|
||||
let writer = self.get_writer(&channel);
|
||||
let w = writer?.lock().unwrap().write(now, record);
|
||||
|
||||
w
|
||||
}
|
||||
|
||||
fn flush(&self) -> io::Result<()> {
|
||||
let writers = self.writers.lock().unwrap();
|
||||
for writer in writers.values() {
|
||||
writer.lock().unwrap().flush()?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
fn console_formatter(w: &mut dyn Write, _now: &mut DeferredNow, record: &Record) -> io::Result<()> {
|
||||
let level = match record.level() {
|
||||
Level::Debug => colorize_string("<bright magenta>[DEBUG]</>"),
|
||||
Level::Error => colorize_string("<bright red>[ERROR]</>"),
|
||||
Level::Info => colorize_string("<bright green>[ INFO]</>"),
|
||||
Level::Trace => colorize_string("<bright yellow>[TRACE]</>"),
|
||||
Level::Warn => colorize_string("<yellow>[ WARN]</>"),
|
||||
};
|
||||
|
||||
write!(
|
||||
w,
|
||||
"{} {}",
|
||||
level,
|
||||
colorize_string(record.args().to_string()),
|
||||
)
|
||||
}
|
||||
|
||||
fn file_formatter(
|
||||
w: &mut dyn Write,
|
||||
now: &mut DeferredNow,
|
||||
record: &Record,
|
||||
) -> std::io::Result<()> {
|
||||
write!(
|
||||
w,
|
||||
"[{}] {} {}",
|
||||
now.now().format("%Y-%m-%d %H:%M:%S%.6f"),
|
||||
record.level(),
|
||||
record.args()
|
||||
)
|
||||
}
|
||||
|
||||
fn file_logger(config: &Logging) -> Box<dyn LogWriter> {
|
||||
if config.log_to_file {
|
||||
let logger = MultiFileLogger::new(config);
|
||||
|
||||
Box::new(logger)
|
||||
} else {
|
||||
Box::new(LogConsole)
|
||||
}
|
||||
}
|
||||
|
||||
/// Initialize our logging, to have:
|
||||
///
|
||||
/// - console logger
|
||||
/// - file logger
|
||||
/// - mail logger
|
||||
pub fn init_logging(
|
||||
config: &PlayoutConfig,
|
||||
proc_ctl: Option<ProcessControl>,
|
||||
messages: Option<Arc<Mutex<Vec<String>>>>,
|
||||
) -> io::Result<()> {
|
||||
Logger::try_with_str(config.logging.level.as_str())
|
||||
.map_err(|e| io::Error::new(ErrorKind::Other, e.to_string()))?
|
||||
.format(console_formatter)
|
||||
.log_to_stderr()
|
||||
.add_writer("file", file_logger(&config.logging))
|
||||
// .add_writer("Mail", Box::new(LogMailer))
|
||||
.start()
|
||||
.map_err(|e| io::Error::new(ErrorKind::Other, e.to_string()))?;
|
||||
|
||||
Ok(())
|
||||
}
|
@ -4,6 +4,7 @@ use std::{
|
||||
fmt,
|
||||
fs::{self, metadata, File},
|
||||
io::{stdin, stdout, Read, Write},
|
||||
net::TcpListener,
|
||||
path::{Path, PathBuf},
|
||||
str::FromStr,
|
||||
};
|
||||
@ -12,6 +13,7 @@ use chrono::{format::ParseErrorKind, prelude::*};
|
||||
use faccess::PathExt;
|
||||
use once_cell::sync::OnceCell;
|
||||
use path_clean::PathClean;
|
||||
use rand::Rng;
|
||||
use rpassword::read_password;
|
||||
use serde::{de, Deserialize, Deserializer, Serialize};
|
||||
use simplelog::*;
|
||||
@ -19,11 +21,14 @@ use sqlx::{sqlite::SqliteRow, FromRow, Pool, Row, Sqlite};
|
||||
|
||||
use crate::ARGS;
|
||||
|
||||
pub mod advanced_config;
|
||||
pub mod args_parse;
|
||||
pub mod channels;
|
||||
pub mod config;
|
||||
pub mod control;
|
||||
pub mod errors;
|
||||
pub mod files;
|
||||
pub mod logging;
|
||||
pub mod playlist;
|
||||
pub mod system;
|
||||
|
||||
@ -388,3 +393,17 @@ where
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// get a free tcp socket
|
||||
pub fn free_tcp_socket(exclude_socket: String) -> Option<String> {
|
||||
for _ in 0..100 {
|
||||
let port = rand::thread_rng().gen_range(45321..54268);
|
||||
let socket = format!("127.0.0.1:{port}");
|
||||
|
||||
if socket != exclude_socket && TcpListener::bind(("127.0.0.1", port)).is_ok() {
|
||||
return Some(socket);
|
||||
}
|
||||
}
|
||||
|
||||
None
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user