snakepipe 2.2.0

use serde::{Deserialize, Serialize};
use std::fmt;
use std::io::{stdin, BufRead, Lines};

// options

#[derive(Debug, Serialize, Deserialize, Clone, Copy, Default)]
pub struct SizeOption {
    pub width: u32,
    pub height: u32,
}

/// Holds the options that were passed to the cli with a flag
/// that are relevent for rendering the game.
#[derive(Debug, Serialize, Deserialize, Clone, Default)]
#[serde(rename_all = "camelCase")]
pub struct InitOptions {
    pub frame_duration: u32,
    #[serde(skip)]
    pub snake_length: u32,
    pub size: SizeOption,
    pub features_with_version: std::collections::HashMap<String, String>,
    pub metadatas: std::collections::HashMap<String, String>,
}

// gamestate

#[derive(Debug, Deserialize, Serialize, Clone)]
#[serde(rename_all = "lowercase")]
pub enum Direction {
    Up,
    Right,
    Down,
    Left,
}

#[derive(Debug, Deserialize, Serialize, Clone)]
pub struct Position {
    pub x: i32,
    pub y: i32,
}

#[derive(Debug, Deserialize, Serialize, Clone)]
pub struct Snake {
    pub direction: Direction,
    pub head: Position,
    pub tail: Vec<Position>,
}

#[derive(Debug, Deserialize, Serialize, PartialEq, Clone)]
#[serde(rename_all = "lowercase")]
pub enum GameState {
    Paused,
    Over,
    Running,
}

impl fmt::Display for GameState {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        let str = match self {
            Self::Over => "Game Over",
            Self::Paused => "Paused",
            Self::Running => "Running",
        };
        write!(f, "{}", str)
    }
}

/// Holds the state of the game at any time
#[derive(Debug, Deserialize, Serialize, Clone)]
pub struct Game {
    pub snake: Snake,
    pub fruit: Position,
    pub score: u32,
    pub state: GameState,
}

/// Accepts the iterator from [`std::io::stdin()`]`.line()`
/// - parses the first line into `options` as [`InitOptions`]
/// - returns an iterator of [`Game`] inside `lines` (already parsed)
///
/// Used by [`parse_gamestate`] under the hood.
pub struct Input {
    pub options: InitOptions,
    pub lines: Box<dyn Iterator<Item = Game>>, // std::io::Lines<T>, //Lines<T>,
}

impl Input {
    /// Creates a input from a buffer (could be from [`std::io::stdin()`]`.line()`)
    pub fn new<T: BufRead + 'static>(
        mut lines: Lines<T>,
    ) -> Result<Input, Box<dyn std::error::Error>> {
        match lines.next() {
            Some(Ok(first_line)) => {
                let options: InitOptions = serde_json::from_str(&first_line)?;
                // flat_map keeps Some and extracts their values while removing Err - we ignore parse errors on lines / we dont panic on it
                let parsed_lines = lines.flat_map(|result_line| match result_line {
                    Ok(line) => match serde_json::from_str::<Game>(&line) {
                        Ok(parsed_line) => Some(parsed_line),
                        Err(_) => None,
                    },
                    Err(_) => None,
                });
                Ok(Self {
                    options,
                    lines: Box::new(parsed_lines),
                })
            }
            None => Err("Buffer is empty".into()),
            Some(Err(e)) => Err(e.into()),
        }
    }
}

/// Parses the stdin containing the gamestate
///
/// Example:
/// ```
/// use snakepipe::input::{parse_gamestate, Game};
///
/// fn main() -> () {
///     match parse_gamestate() {
///         Ok(input) => {
///             println!(
///                 "Frame duration {}, Snake length {}, Level {}x{}",
///                 input.options.frame_duration,
///                 input.options.snake_length,
///                 input.options.size.width,
///                 input.options.size.height
///             );
///             for parsed_line in input.lines {
///                 do_something(parsed_line);
///             }
///         }
///         Err(e) => {
///             eprintln!("Error occurred while parsing stdin: \"{}\"", e);
///         }
///     }
/// }
///
/// fn do_something(parsed_line: Game) {
///     println!("Snake head position {:?}", parsed_line.snake.head)
/// }
/// ```
///
/// If you want to parse from elsewhere than stdin, you can use [Input]
pub fn parse_gamestate() -> Result<Input, Box<dyn std::error::Error>> {
    // todo couldn't find how to peek into the input (to know if it comes from `snake gamestate` or `cat /some-file`), without consuming it
    // so we'll show a "Replay" message when `gamestate throttle` is used in the pipeline (even if it could only be used to throttle directly `gamestate`)
    let lines = stdin().lines();
    Input::new(lines)
}