#include <platform.h>
#include "spells/shipcurse.h"
#include "monster.h"

#include <kernel/equipment.h>
#include <kernel/faction.h>
#include <kernel/race.h>
#include <kernel/ship.h>
#include <kernel/spellbook.h>
#include <kernel/unit.h>

#include <quicklist.h>

#include <tolua.h>
#include <stdlib.h>

static int tolua_levitate_ship(lua_State * L)
{
    ship *sh = (ship *)tolua_tousertype(L, 1, 0);
    unit *mage = (unit *)tolua_tousertype(L, 2, 0);
    float power = (float)tolua_tonumber(L, 3, 0);
    int duration = (int)tolua_tonumber(L, 4, 0);
    int cno = levitate_ship(sh, mage, power, duration);
    tolua_pushnumber(L, (lua_Number)cno);
    return 1;
}

extern void spawn_undead(void);
extern void spawn_dragons(void);
extern void plan_monsters(struct faction *f);

static int tolua_planmonsters(lua_State * L)
{
    faction *f = (faction *)tolua_tousertype(L, 1, get_monsters());
    if (f) {
        plan_monsters(f);
    }

    return 0;
}

static int tolua_spawn_dragons(lua_State * L)
{
    spawn_dragons();
    return 0;
}

static int tolua_spawn_undead(lua_State * L)
{
    spawn_undead();
    return 0;
}

static int fix_familiars(struct lua_State *L)
{
    faction *f;
    for (f = factions; f; f = f->next) {
        unit *u;
        for (u = f->units; u; u = u->nextF) {
            struct sc_mage *mage = get_mage(u);
            if (mage && is_familiar(u)) {
                if (mage->spellbook && mage->magietyp == M_GRAY) {
                    equipment *eq;
                    char buffer[64];

                    spellbook_clear(mage->spellbook);
                    free(mage->spellbook);
                    mage->spellbook = 0;

                    _snprintf(buffer, sizeof(buffer), "%s_familiar", u_race(u)->_name);
                    eq = get_equipment(buffer);
                    if (eq) {
                        equip_unit_mask(u, eq, EQUIP_SPELLS);
                    }
                }
            }
        }
    }
    return 0;
}

void bind_monsters(struct lua_State *L)
{
    tolua_module(L, NULL, 0);
    tolua_beginmodule(L, NULL);
    {
        tolua_function(L, TOLUA_CAST "levitate_ship", tolua_levitate_ship);
        tolua_function(L, TOLUA_CAST "plan_monsters", tolua_planmonsters);
        tolua_function(L, TOLUA_CAST "spawn_undead", tolua_spawn_undead);
        tolua_function(L, TOLUA_CAST "spawn_dragons", tolua_spawn_dragons);
        tolua_function(L, TOLUA_CAST "fix_familiars", &fix_familiars);
    }
    tolua_endmodule(L);
}