mod collect_playlist_dates;
mod download_testfiles;
mod gen_dictionary;
mod gen_locales;
mod util;

use std::path::PathBuf;

use clap::{Parser, Subcommand};

#[derive(Parser)]
struct Cli {
    #[clap(subcommand)]
    command: Commands,
    #[clap(short = 'd', default_value = "..")]
    project_root: PathBuf,
    #[clap(short, default_value = "8")]
    concurrency: usize,
}

#[derive(Subcommand)]
enum Commands {
    CollectPlaylistDates,
    WritePlaylistDates,
    GenLocales,
    GenDict,
    DownloadTestfiles,
}

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

    match cli.command {
        Commands::CollectPlaylistDates => {
            collect_playlist_dates::collect_dates(&cli.project_root, cli.concurrency).await;
        }
        Commands::WritePlaylistDates => {
            collect_playlist_dates::write_samples_to_dict(&cli.project_root);
        }
        Commands::GenLocales => {
            gen_locales::generate_locales(&cli.project_root).await;
        }
        Commands::GenDict => gen_dictionary::generate_dictionary(&cli.project_root),
        Commands::DownloadTestfiles => {
            download_testfiles::download_testfiles(&cli.project_root).await
        }
    };
}