/*
    Hurry Curry! - a game about cooking
    Copyright 2024 metamuffin

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published by
    the Free Software Foundation, version 3 of the License only.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program.  If not, see <https://www.gnu.org/licenses/>.

*/
pub mod conveyor;
pub mod customers;
pub mod environment_effect;
pub mod item_portal;
pub mod player_portal;

use crate::{data::ItemTileRegistry, game::Game, interaction::Recipe};
use anyhow::{anyhow, Result};
use conveyor::Conveyor;
use customers::{demands::generate_demands, Customers};
use environment_effect::{EnvironmentController, EnvironmentEffect, EnvironmentEffectController};
use hurrycurry_protocol::{
    glam::{IVec2, Vec2},
    ItemIndex, PacketC, TileIndex,
};
use item_portal::ItemPortal;
use player_portal::PlayerPortal;
use serde::{Deserialize, Serialize};
use std::collections::{HashMap, HashSet, VecDeque};

pub trait EntityT: Clone {
    fn tick(&mut self, game: &mut Game, packet_out: &mut VecDeque<PacketC>, dt: f32) -> Result<()>;
}

macro_rules! entities {
    ($($e:ident),*) => {
        #[derive(Debug, Clone)]
        pub enum Entity { $($e($e)),* }
        impl EntityT for Entity {
            fn tick(&mut self, game: &mut Game, packet_out: &mut VecDeque<PacketC>, dt: f32) -> Result<()> {
                match self { $(Entity::$e(x) => x.tick(game, packet_out, dt)),*, }
            }
        }
    };
}

entities!(
    Conveyor,
    ItemPortal,
    PlayerPortal,
    Customers,
    EnvironmentEffectController,
    EnvironmentController
);

#[derive(Debug, Clone, Deserialize, Serialize)]
#[serde(rename_all = "snake_case")]
pub enum EntityDecl {
    Conveyor {
        from: Option<IVec2>,
        to: Option<IVec2>,
        filter_dir: Option<IVec2>,
        filter: Option<String>,
        dir: Option<IVec2>,
        speed: Option<f32>,
    },
    ItemPortal {
        from: Option<IVec2>,
        to: IVec2,
    },
    PlayerPortal {
        from: Option<Vec2>,
        to: Vec2,
    },
    Customers {},
    EnvironmentEffect(EnvironmentEffect),
    Environment(Vec<String>),
}

pub fn construct_entity(
    pos: Option<IVec2>,
    decl: &EntityDecl,
    reg: &ItemTileRegistry,
    tiles_used: &HashSet<TileIndex>,
    items_used: &HashSet<ItemIndex>,
    raw_demands: &[(ItemIndex, Option<ItemIndex>, f32)],
    recipes: &[Recipe],
    initial_map: &HashMap<IVec2, (TileIndex, Option<ItemIndex>)>,
) -> Result<Entity> {
    Ok(match decl.to_owned() {
        EntityDecl::ItemPortal { from, to } => Entity::ItemPortal(ItemPortal {
            from: from.or(pos).ok_or(anyhow!("no item portal start"))?,
            to,
        }),
        EntityDecl::PlayerPortal { from, to } => Entity::PlayerPortal(PlayerPortal {
            from: from
                .or(pos.map(|v| v.as_vec2()))
                .ok_or(anyhow!("no player portal start"))?,
            to,
        }),
        EntityDecl::Conveyor {
            from,
            to,
            speed,
            dir,
            filter,
            filter_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.),
                filter_tile: filter_dir.map(|o| to + o),
                filter_item: filter.map(|name| reg.register_item(name)),
                cooldown: 0.,
            })
        }
        EntityDecl::Customers {} => {
            let demands = generate_demands(tiles_used, items_used, raw_demands, recipes);
            let chair = reg.register_tile("chair".to_string());
            let chairs = initial_map
                .iter()
                .filter(|(_, (tile, _))| *tile == chair)
                .map(|(e, _)| (*e, true))
                .collect();
            Entity::Customers(Customers::new(chairs, demands)?)
        }
        EntityDecl::EnvironmentEffect(config) => {
            Entity::EnvironmentEffectController(EnvironmentEffectController::new(config))
        }
        EntityDecl::Environment(names) => {
            Entity::EnvironmentController(EnvironmentController(names))
        }
    })
}