use std::{fmt::Display, path::PathBuf};

use cap_export::{gif::GifExportSettings, mp4::Mp4ExportSettings, ExporterBase, ExporterBuilder};
use clap::{Parser, ValueEnum};

#[derive(Parser, Debug)]
struct Cli {
    path: PathBuf,
    #[arg(value_enum)]
    format: Option<ExportFormat>,
    settings: Option<String>,
}

#[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, ValueEnum, Debug)]
enum ExportFormat {
    MP4,
    GIF,
}

impl Display for ExportFormat {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Self::GIF => write!(f, "gif"),
            Self::MP4 => write!(f, "mp4"),
        }
    }
}

#[tokio::main]
async fn main() {
    let cli = Cli::parse();

    let path: PathBuf = std::env::args().collect::<Vec<_>>().swap_remove(1).into();

    let base = ExporterBase::builder(path).build().await.unwrap();

    let format = cli.format.unwrap_or_else(|| {
        inquire::Select::new(
            "Select export format",
            vec![ExportFormat::MP4, ExportFormat::GIF],
        )
        .prompt()
        .unwrap()
    });

    let settings_str = cli
        .settings
        .unwrap_or_else(|| inquire::Text::new("Export settings JSON").prompt().unwrap());

    match format {
        ExportFormat::GIF => {
            let settings: GifExportSettings = serde_json::from_str(&settings_str).unwrap();
            let total_frames = base.total_frames(settings.fps);
            settings
                .export(base, move |progress| {
                    print!("Exporting frame {} of {}\r", progress, total_frames);
                })
                .await
                .unwrap();
        }
        ExportFormat::MP4 => {
            let settings: Mp4ExportSettings = serde_json::from_str(&settings_str).unwrap();
            let total_frames = base.total_frames(settings.fps);
            settings
                .export(base, move |progress| {
                    print!("Exporting frame {} of {}\r", progress, total_frames);
                })
                .await
                .unwrap();
        }
    }
}
