use std::io;
use std::process::exit;

use day49::is_legitimate;

fn read_row() -> Result<Vec<u8>, Box<dyn std::error::Error>> {
    let mut buffer = String::new();
    io::stdin().read_line(&mut buffer)?;

    let row: Vec<u8> = buffer
        .split_whitespace()
        .map(|x: &str| match x.parse::<u8>() {
            Ok(x) => match x {
                0 | 1 => Ok(x),
                _ => Err("Only 0 and 1 are accepted".to_string()),
            },
            Err(_) => Err("Only numbers are accepted".to_string()),
        })
        .collect::<Result<Vec<u8>, String>>()?;

    if row.is_empty() {
        Err("Empty row")?
    }

    Ok(row)
}

fn read_pool() -> Result<Vec<Vec<u8>>, Box<dyn std::error::Error>> {
    let mut pool = vec![read_row()?];
    while let Ok(row) = read_row() {
        if row.len() != pool.first().unwrap().len() {
            Err("Invalid row length")?
        }
        pool.push(row);
    }
    Ok(pool)
}

fn main() {
    println!("Insert the pool (space-separated 0s and 1s, one row per line)");
    println!("End input with an empty line");

    let pool = match read_pool() {
        Ok(x) => x,
        Err(e) => {
            eprintln!("Error: {e}");
            exit(1);
        }
    };

    if let Some(legit) = is_legitimate(&pool) {
        if legit {
            println!("The pool is legitimate");
        } else {
            println!("The pool is not legitimate");
        }
    } else {
        println!("Invalid pool");
    }
}