cgol-tui 0.6.10

use app::{App, Universe};
use std::{io::Read, str::FromStr};

pub mod app;

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let arg_universes = parse_args()?;

    let mut app = App::default().with_universes(arg_universes);

    let mut terminal = ratatui::try_init()?;

    let res = app.run(&mut terminal);

    ratatui::try_restore()?;

    Ok(res?)
}

fn parse_args() -> Result<Vec<Universe>, Box<dyn std::error::Error>> {
    let args = std::env::args().skip(1).collect::<Vec<_>>();
    if args.contains(&"-h".into()) || args.contains(&"--help".into()) {
        println!(
            "A Conway's Game of Life viewer TUI.
            
USAGE: cgol-tui [<pattern>,...]

where <pattern> is either a .cells file, or - for stdin"
        );
        std::process::exit(0);
    }
    let piped_universe = {
        let mut univ = String::new();
        if args.len() == 1 && args[0] == "-" {
            std::io::stdin().read_to_string(&mut univ)?;
        }
        if univ.is_empty() {
            vec![]
        } else {
            vec![Universe::from_str(&univ)?]
        }
    };
    let universes = args
        .iter()
        .flat_map(std::fs::read_to_string)
        .flat_map(|s| Universe::from_str(&s))
        .collect::<Vec<_>>();

    Ok([universes, piped_universe].concat())
}