use std::path::PathBuf;
use anyhow::Result;
use clap::Args;
use libsail::collection::{Indexable, Iterable};
use crate::cli::FormatArg;
use crate::input::{Inputs, dispatch};
use crate::output::{emit, writer};
#[derive(Args)]
pub struct ShuffleArgs {
#[arg(default_value = "-")]
pub input: PathBuf,
#[arg(long)]
pub seed: Option<u64>,
#[arg(long, value_enum)]
pub format: Option<FormatArg>,
#[arg(short, long)]
pub output: Option<PathBuf>,
}
impl ShuffleArgs {
pub fn run(self) -> Result<()> {
let inputs = Inputs::read(std::slice::from_ref(&self.input), self.format)?;
let entry = &inputs.entries()[0];
let out = writer(self.output.as_deref())?;
let seed = self.seed.unwrap_or_else(rand::random);
dispatch!(inputs.format(), entry, |collection, _size, _name, write| {
emit(collection.permute(seed).iter(), write, out)
})
}
}