//////////////////////////////////////////////////////////////////////
// This file is part of Remere's Map Editor
//////////////////////////////////////////////////////////////////////
// Remere's Map Editor is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Remere's Map Editor 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
//////////////////////////////////////////////////////////////////////

#include "main.h"
#include "lua_api_brush.h"
#include "../brush.h"
#include "../gui.h"

namespace LuaAPI {

	// Helper to get brush type as string
	static std::string getBrushTypeName(const Brush* brush) {
		if (!brush) {
			return "unknown";
		}

		if (brush->isRaw()) {
			return "raw";
		}
		if (brush->isDoodad()) {
			return "doodad";
		}
		if (brush->isGround()) {
			return "ground";
		}
		if (brush->isWall()) {
			return "wall";
		}
		if (brush->isWallDecoration()) {
			return "wall_decoration";
		}
		if (brush->isTable()) {
			return "table";
		}
		if (brush->isCarpet()) {
			return "carpet";
		}
		if (brush->isDoor()) {
			return "door";
		}
		if (brush->isOptionalBorder()) {
			return "optional_border";
		}
		if (brush->isMonster()) {
			return "creature";
		}
		if (brush->isSpawnMonster()) {
			return "spawn";
		}
		if (brush->isHouse()) {
			return "house";
		}
		if (brush->isHouseExit()) {
			return "house_exit";
		}
		if (brush->isWaypoint()) {
			return "waypoint";
		}
		if (brush->isFlag()) {
			return "flag";
		}
		if (brush->isEraser()) {
			return "eraser";
		}
		if (brush->isTerrain()) {
			return "terrain";
		}

		return "unknown";
	}

	void registerBrush(sol::state &lua) {
		// Register Brush usertype (read-only)
		lua.new_usertype<Brush>("Brush", sol::no_constructor,

								// Properties (read-only)
								"id", sol::property([](Brush* b) -> uint32_t {
									return b ? b->getID() : 0;
								}),
								"name", sol::property([](Brush* b) -> std::string {
									return b ? b->getName() : "";
								}),
								"lookId", sol::property([](Brush* b) -> int {
									return b ? b->getLookID() : 0;
								}),
								"type", sol::property([](Brush* b) -> std::string {
									return getBrushTypeName(b);
								}),

								// Capabilities (read-only)
								"canDrag", sol::property([](Brush* b) -> bool {
									return b && b->canDrag();
								}),
								"canSmear", sol::property([](Brush* b) -> bool {
									return b && b->canSmear();
								}),
								"needsBorders", sol::property([](Brush* b) -> bool {
									return b && b->needBorders();
								}),
								"oneSizeFitsAll", sol::property([](Brush* b) -> bool {
									return b && b->oneSizeFitsAll();
								}),
								"maxVariation", sol::property([](Brush* b) -> int {
									return b ? b->getMaxVariation() : 0;
								}),
								"visibleInPalette", sol::property([](Brush* b) -> bool {
									return b && b->visibleInPalette();
								}),

								// Type checks
								"isRaw", sol::property([](Brush* b) { return b && b->isRaw(); }), "isDoodad", sol::property([](Brush* b) { return b && b->isDoodad(); }), "isTerrain", sol::property([](Brush* b) { return b && b->isTerrain(); }), "isGround", sol::property([](Brush* b) { return b && b->isGround(); }), "isWall", sol::property([](Brush* b) { return b && b->isWall(); }), "isTable", sol::property([](Brush* b) { return b && b->isTable(); }), "isCarpet", sol::property([](Brush* b) { return b && b->isCarpet(); }), "isDoor", sol::property([](Brush* b) { return b && b->isDoor(); }), "isCreature", sol::property([](Brush* b) { return b && b->isMonster(); }), "isSpawn", sol::property([](Brush* b) { return b && b->isSpawnMonster(); }), "isHouse", sol::property([](Brush* b) { return b && b->isHouse(); }), "isEraser", sol::property([](Brush* b) { return b && b->isEraser(); }),

								// String representation
								sol::meta_function::to_string, [](Brush* b) -> std::string {
									if (!b) {
										return "Brush(invalid)";
									}
									return "Brush(\"" + b->getName() + "\", type=" + getBrushTypeName(b) + ")";
								});

		// Register Brushes namespace for brush lookup
		sol::table brushes = lua.create_named_table("Brushes");

		// Get brush by name
		brushes["get"] = [](const std::string &name) {
			return g_brushes.getBrush(name);
		};

		// Get all brush names (for iteration)
		brushes["getNames"] = [](sol::this_state ts) -> sol::table {
			sol::state_view lua(ts);
			sol::table names = lua.create_table();

			int idx = 1;
			for (const auto &pair : g_brushes.getMap()) {
				names[idx++] = pair.first;
			}
			return names;
		};

		// Extend the existing 'app' table with brush-related functions
		sol::table app = lua["app"];

		// Current brush
		app.set_function("getBrush", []() {
			return g_gui.GetCurrentBrush();
		});

		// Set brush by name
		app.set_function("setBrush", [](const std::string &name) {
			Brush* brush = g_brushes.getBrush(name);
			if (brush) {
				g_gui.SelectBrush(brush);
				return true;
			}
			return false;
		});

		// Register BrushShape enum
		lua.new_enum("BrushShape", "SQUARE", BRUSHSHAPE_SQUARE, "CIRCLE", BRUSHSHAPE_CIRCLE);
	}

} // namespace LuaAPI