pub mod conveyor;
use crate::{data::Gamedata, game::Tile, protocol::PacketC};
use anyhow::{anyhow, Result};
use conveyor::Conveyor;
use glam::IVec2;
use serde::{Deserialize, Serialize};
use std::collections::{HashMap, VecDeque};

pub trait EntityT: Clone {
    fn tick(
        &mut self,
        data: &Gamedata,
        points: &mut i64,
        packet_out: &mut VecDeque<PacketC>,
        tiles: &mut HashMap<IVec2, Tile>,
        dt: f32,
    ) -> Result<()>;
}

#[derive(Debug, Clone)]
pub enum Entity {
    Conveyor(Conveyor),
}
impl EntityT for Entity {
    fn tick(
        &mut self,
        data: &Gamedata,
        points: &mut i64,
        packet_out: &mut VecDeque<PacketC>,
        tiles: &mut HashMap<IVec2, Tile>,
        dt: f32,
    ) -> Result<()> {
        match self {
            Entity::Conveyor(x) => x.tick(data, points, packet_out, tiles, dt),
        }
    }
}

#[derive(Debug, Clone, Deserialize, Serialize)]
#[serde(rename_all = "snake_case")]
pub enum EntityDecl {
    Conveyor {
        from: Option<IVec2>,
        to: Option<IVec2>,
        dir: Option<IVec2>,
        speed: Option<f32>,
    },
}

pub fn construct_entity(pos: Option<IVec2>, decl: &EntityDecl) -> Result<Entity> {
    Ok(match decl.to_owned() {
        EntityDecl::Conveyor {
            from,
            to,
            speed,
            dir,
        } => {
            let from = from.or(pos).ok_or(anyhow!("conveyor has no start"))?;
            let to = to
                .or(dir.map(|s| s + from))
                .ok_or(anyhow!("conveyor has no destination"))?;
            Entity::Conveyor(Conveyor {
                from,
                to,
                max_cooldown: 1. / speed.unwrap_or(2.),
                ..Default::default()
            })
        }
    })
}