use super::Command;

use crate::spatial::{Vec3, Area, area_rel_block_overlap, area_contains_block};
use crate::instance::{ArgType, InstBundle};
use crate::map_block::{MapBlock};
use crate::block_utils::clean_name_id_map;
use crate::utils::{query_keys, fmt_big_num};


fn fill_area(block: &mut MapBlock, area: Area, id: u16) {
	let nd = block.node_data.get_mut();
	for z in area.min.z ..= area.max.z {
		let z_start = z * 256;
		for y in area.min.y ..= area.max.y {
			let zy_start = z_start + y * 16;
			for x in area.min.x ..= area.max.x {
				nd.nodes[(zy_start + x) as usize] = id;
			}
		}
	}
}


fn fill(inst: &mut InstBundle) {
	let area = inst.args.area.unwrap();
	let node = inst.args.new_node.as_ref().unwrap().as_bytes().to_owned();

	let keys = query_keys(&mut inst.db, &mut inst.status,
		None, Some(area), false, true);

	inst.status.begin_editing();

	let mut count: u64 = 0;
	for key in keys {
		let pos = Vec3::from_block_key(key);
		let data = inst.db.get_block(key).unwrap();
		let mut block = MapBlock::deserialize(&data).unwrap();

		if area_contains_block(&area, pos) {
			let nd = block.node_data.get_mut();
			for x in &mut nd.nodes {
				*x = 0;
			}
			block.nimap.map.clear();
			block.nimap.insert(0, &node);
			count += nd.nodes.len() as u64;
		} else {
			let slice = area_rel_block_overlap(&area, pos).unwrap();
			let fill_id = block.nimap.get_id(&node).unwrap_or_else(|| {
				let next = block.nimap.get_max_id().unwrap() + 1;
				block.nimap.insert(next, &node);
				next
			});
			fill_area(&mut block, slice, fill_id);
			clean_name_id_map(&mut block);
			count += slice.volume();
		}

		inst.db.set_block(key, &block.serialize()).unwrap();
		inst.status.inc_done();
	}

	inst.status.end_editing();
	inst.status.log_info(
		format!("{} nodes filled.", fmt_big_num(count)).as_str());
}


pub fn get_command() -> Command {
	Command {
		func: fill,
		verify_args: None,
		args: vec![
			(ArgType::Area(true), "Area to fill"),
			(ArgType::NewNode(true), "Node to fill area with")
		],
		help: "Fill the entire area with one node."
	}
}