#[macro_use]
extern crate simple_log;

use std::env;
use std::path::PathBuf;

use simple_log::LogConfigBuilder;

use renice::renice;
use transcode::job::TranscodeJob;
use crate::configuration::Config;

mod configuration;
mod files;
mod ffmpeg;
mod renice;
mod transcode;
mod error;

fn main() {
    let config = configuration::Config::from_file(&env::var("CONFIG").unwrap_or(String::from("./config.toml")));
    setup_logger(&config);
    trace!("Config: {:#?}", &config);

    debug!("commit: {}", env!("GIT_COMMIT_HASH"));
    debug!("tag: {}", env!("GIT_TAG"));

    let input_files = files::get_files(&config.files.input_path)
        .into_iter()
        .filter(|path| {
            let path = path.to_str().unwrap();
            config.files.include.iter().any(|include| path.contains(include))
        })
        .collect::<Vec<_>>();
    info!("Found {} file(s) to be processed.", input_files.len());

    let success_count = start_transcode_run(&input_files, &config);

    info!("Finished processing {}/{} file(s).", success_count, input_files.len());
}

fn start_transcode_run(input_files: &Vec<PathBuf>, config: &Config) -> i32 {
    let mut success_count = 0;

    for file in input_files {
        let fail = |job: &TranscodeJob, remarks: &str| {
            error!("Failed to process file {}: {}", job.input, remarks);
        };

        let mut output_path = file.clone();
        output_path.set_extension(&config.ffmpeg.output.format);
        let output_path = PathBuf::from(&config.files.output_path).join(output_path.file_name().unwrap()); // TODO: This is a bit of a mess.

        let job = TranscodeJob::new(
            file.to_str().unwrap(),
            output_path.to_str().unwrap(),
        );

        // check if overwriting
        if job.check_if_exists() {
            if config.ffmpeg.overwrite.unwrap_or(false) {
                info!("Overwriting file {}.", job.input);
            } else {
                info!("Skipping file {} as it already exists.", job.input);
                continue;
            }
        }

        info!("Processing file {}.", job.input);
        let mut child = match job.run(&config.ffmpeg) {
            Ok(child) => child,
            Err(e) => {
                fail(&job, &format!("Failed to start ffmpeg process: {}", e));
                continue;
            }
        };
        if let Some(process) = &config.ffmpeg.process {
            if (process.niceness.unwrap_or(0)) > 0 {
                if let Err(e) = renice(child.id(), process.niceness.unwrap_or(0)) {
                    error!("Failed to renice ffmpeg process: {}", e);
                }
            }
        }

        if let Err(e) = child.wait() {
            fail(&job, &format!("Failed to wait for ffmpeg process: {}", e));
            continue;
        }

        // TODO: Cleanup

        info!("Finished processing file {}.", job.input);
        success_count += 1;
    }

    success_count
}

fn setup_logger(config: &configuration::Config) {
    let log_config = LogConfigBuilder::builder()
        .path("backups.log")
        .size(10 * 1000)
        .roll_count(10)
        .time_format("%Y-%m-%d %H:%M:%S")
        .level(if config.is_debug() { "trace" } else { "info" })
        .output_file()
        .output_console()
        .build();

    simple_log::new(log_config).unwrap();
}