(*

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2004-2015 Andrey Korotaev <unC0Rr@gmail.com>

 *

 * This program 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; version 2 of the License

 *

 * 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 General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

 *)

{$INCLUDE "options.inc"}

unit uScript;

(*

 * This unit defines, implements and registers functions and

 * variables/constants bindings for usage in Lua scripts.

 *

 * Please keep https://hedgewars.org/kb/LuaAPI up to date!

 *

 * Note: If you add a new function, make sure to test if _all_ parameters

 *       work as intended! (Especially conversions errors can sneak in

 *       unnoticed and render the parameter useless!)

 *)

interface

procedure ScriptPrintStack;

procedure ScriptClearStack;

function ScriptLoad(name : shortstring; mustExist : boolean): boolean;

procedure ScriptOnPreviewInit;

procedure ScriptOnGameInit;

procedure ScriptOnScreenResize;

procedure ScriptSetInteger(name : shortstring; value : LongInt);

procedure ScriptSetString(name : shortstring; value : shortstring);

procedure ScriptSetMapGlobals;

procedure ScriptCall(fname : shortstring);

function ScriptCall(fname : shortstring; par1: LongInt) : LongInt;

function ScriptCall(fname : shortstring; par1, par2: LongInt) : LongInt;

function ScriptCall(fname : shortstring; par1, par2, par3: LongInt) : LongInt;

function ScriptCall(fname : shortstring; par1, par2, par3, par4 : LongInt) : LongInt;

function ScriptExists(fname : shortstring) : boolean;

procedure LuaParseString(s: shortString);

//function ParseCommandOverride(key, value : shortstring) : shortstring;  This did not work out well

procedure initModule;

procedure freeModule;

implementation

uses LuaPas,

    uConsole,

    uConsts,

    uGears,

    uGearsList,

    uGearsUtils,

    uFloat,

    uWorld,

    uAmmos,

    uTeams,

    uSound,

    uChat,

    uStats,

    uStore,

    uRandom,

    uTypes,

    uVariables,

    uCommands,

    uCaptions,

    uDebug,

    uCollisions,

    uRenderUtils,

    uTextures,

    uLandGraphics,

    uUtils,

    uIO,

    uVisualGearsList,

    uGearsHandlersMess,

    uPhysFSLayer,

    uLocale,

    SDLh

{$IFNDEF PAS2C}

    , typinfo

{$ENDIF}

    ;

var luaState : Plua_State;

    ScriptAmmoLoadout : shortstring;

    ScriptAmmoProbability : shortstring;

    ScriptAmmoDelay : shortstring;

    ScriptAmmoReinforcement : shortstring;

    ScriptLoaded : boolean;

    PointsBuffer: shortstring;

    PrevCursorX, PrevCursorY: LongInt;

    PendingTurnTimeLeft, PendingReadyTimeLeft: LongWord;

    isPendingTurnTimeLeft, isPendingReadyTimeLeft: boolean;

{$IFDEF USE_LUA_SCRIPT}

procedure ScriptPrepareAmmoStore; forward;

procedure ScriptApplyAmmoStore; forward;

procedure ScriptSetAmmo(ammo : TAmmoType; count, probability, delay, reinforcement: Byte); forward;

procedure ScriptSetAmmoDelay(ammo : TAmmoType; delay: LongWord); forward;

var LuaDebugInfo: lua_Debug;

procedure SetGlobals; forward;

procedure GetGlobals; forward;

procedure LuaParseString(s: shortString);

begin

    SetGlobals;

    AddFileLog('[Lua] input string: ' + s);

    AddChatString(#3 + '[Lua] > ' + s);

    if luaL_dostring(luaState, Str2PChar(s)) <> 0 then

        begin

        AddFileLog('[Lua] input string parsing error!');

        AddChatString(#5 + '[Lua] Error while parsing!');

        end

    else

        GetGlobals();

end;

function LuaUpdateDebugInfo(): Boolean;

begin

    FillChar(LuaDebugInfo, sizeof(LuaDebugInfo), 0);

    if lua_getstack(luaState, 1, @LuaDebugInfo) = 0 then

        exit(false); // stack not deep enough

    // get source name and line count

    lua_getinfo(luaState, PChar('Sl'), @LuaDebugInfo);

    exit(true);

end;

procedure LuaErrorOrWarning(s: shortstring; isWarning: boolean);

var src, intro: shortstring;

const

    maxsrclen = 20;

begin

    if isWarning then

        intro:= 'LUA WARNING'

    else

        intro:= 'LUA ERROR';

    if LuaUpdateDebugInfo() then

        begin

        src:= StrPas(LuaDebugInfo.source);

        s:= intro + ': [ ... '

            + copy(src, Length(src) - maxsrclen, maxsrclen - 3) + ':'

            + inttostr(LuaDebugInfo.currentLine) + ']: ' + s;

        end

    else

        s:= intro + ': ' + s;

    WriteLnToConsole(s);

    if isWarning then

        AddChatString(#0 + s)

    else

        AddChatString(#5 + s);

    if cTestLua and (not isWarning) then

        halt(HaltTestLuaError);

end;

procedure LuaError(s: shortstring);

begin

    LuaErrorOrWarning(s, false);

end;

procedure LuaWarning(s: shortstring);

begin

    LuaErrorOrWarning(s, true);

end;

procedure LuaCallError(error, call, paramsyntax: shortstring);

begin

    LuaError(call + ': ' + error);

    LuaError('-- SYNTAX: ' + call + ' ( ' + paramsyntax + ' )');

end;

procedure LuaParameterCountError(expected, call, paramsyntax: shortstring; wrongcount: LongInt); 

begin

    // TODO: i18n?

    LuaCallError('Wrong number of parameters! (is: ' + inttostr(wrongcount) + ', should be: '+ expected + ')', call, paramsyntax);

end;

// compare with allowed count

function CheckLuaParamCount(L : Plua_State; count: LongInt; call, paramsyntax: shortstring): boolean; 

var c: LongInt;

begin

    c:= lua_gettop(L);

    if c <> count then

        begin

        LuaParameterCountError('exactly ' + inttostr(count), call, paramsyntax, c);

        exit(false);

        end;

    CheckLuaParamCount:= true;

end;

// check if is either count1 or count2

function CheckAndFetchParamCount(L : Plua_State; count1, count2: LongInt; call, paramsyntax: shortstring; out actual: LongInt): boolean; 

begin

    actual:= lua_gettop(L);

    if (actual <> count1) and (actual <> count2) then

        begin

        LuaParameterCountError('either ' + inttostr(count1) + ' or ' + inttostr(count2), call, paramsyntax, actual);

        exit(false);

        end;

    CheckAndFetchParamCount:= true;

end;

// check if is in range of count1 and count2

function CheckAndFetchParamCountRange(L : Plua_State; count1, count2: LongInt; call, paramsyntax: shortstring; out actual: LongInt): boolean; 

begin

    actual:= lua_gettop(L);

    if (actual < count1) or (actual > count2) then

        begin

        LuaParameterCountError('at least ' + inttostr(count1) + ', but at most ' + inttostr(count2), call, paramsyntax, actual);

        exit(false);

        end;

    CheckAndFetchParamCountRange:= true;

end;

// check if is same or higher as minCount

function CheckAndFetchLuaParamMinCount(L : Plua_State; minCount: LongInt; call, paramsyntax: shortstring; out actual: LongInt): boolean; 

begin

    actual:= lua_gettop(L);

    if (actual < minCount) then

        begin

        LuaParameterCountError(inttostr(minCount) + ' or more', call, paramsyntax, actual);

        exit(false);

        end;

    CheckAndFetchLuaParamMinCount:= true;

end;

function LuaToGearTypeOrd(L : Plua_State; i: LongInt; call, paramsyntax: shortstring): LongInt; 

begin

    if lua_isnoneornil(L, i) then i:= -1

    else i:= Trunc(lua_tonumber(L, i));

    if (i < ord(Low(TGearType))) or (i > ord(High(TGearType))) then

        begin

        LuaCallError('Invalid gearType!', call, paramsyntax);

        LuaToGearTypeOrd:= -1;

        end

    else

        LuaToGearTypeOrd:= i;

end;

function LuaToVisualGearTypeOrd(L : Plua_State; i: LongInt; call, paramsyntax: shortstring): LongInt; 

begin

    if lua_isnoneornil(L, i) then i:= -1

    else i:= Trunc(lua_tonumber(L, i));

    if (i < ord(Low(TVisualGearType))) or (i > ord(High(TVisualGearType))) then

        begin

        LuaCallError('Invalid visualGearType!', call, paramsyntax);

        LuaToVisualGearTypeOrd:= -1;

        end

    else

        LuaToVisualGearTypeOrd:= i;

end;

function LuaToAmmoTypeOrd(L : Plua_State; i: LongInt; call, paramsyntax: shortstring): LongInt; 

begin

    if lua_isnoneornil(L, i) then i:= -1

    else i:= Trunc(lua_tonumber(L, i));

    if (i < ord(Low(TAmmoType))) or (i > ord(High(TAmmoType))) then

        begin

        LuaCallError('Invalid ammoType!', call, paramsyntax);

        LuaToAmmoTypeOrd:= -1;

        end

    else

        LuaToAmmoTypeOrd:= i;

end;

function LuaToStatInfoTypeOrd(L : Plua_State; i: LongInt; call, paramsyntax: shortstring): LongInt; 

begin

    if lua_isnoneornil(L, i) then i:= -1

    else i:= Trunc(lua_tonumber(L, i));

    if (i < ord(Low(TStatInfoType))) or (i > ord(High(TStatInfoType))) then

        begin

        LuaCallError('Invalid statInfoType!', call, paramsyntax);

        LuaToStatInfoTypeOrd:= -1;

        end

    else

        LuaToStatInfoTypeOrd:= i;

end;

function LuaToSoundOrd(L : Plua_State; i: LongInt; call, paramsyntax: shortstring): LongInt; 

begin

    if lua_isnoneornil(L, i) then i:= -1

    else i:= Trunc(lua_tonumber(L, i));

    if (i < ord(Low(TSound))) or (i > ord(High(TSound))) then

        begin

        LuaCallError('Invalid soundId!', call, paramsyntax);

        LuaToSoundOrd:= -1;

        end

    else

        LuaToSoundOrd:= i;

end;

function LuaToHogEffectOrd(L : Plua_State; i: LongInt; call, paramsyntax: shortstring): LongInt; 

begin

    if lua_isnoneornil(L, i) then i:= -1

    else i:= Trunc(lua_tonumber(L, i));

    if (i < ord(Low(THogEffect))) or (i > ord(High(THogEffect))) then

        begin

        LuaCallError('Invalid effect type!', call, paramsyntax);

        LuaToHogEffectOrd:= -1;

        end

    else

        LuaToHogEffectOrd:= i;

end;

function LuaToCapGroupOrd(L : Plua_State; i: LongInt; call, paramsyntax: shortstring): LongInt; 

begin

    if lua_isnoneornil(L, i) then i:= -1

    else i:= Trunc(lua_tonumber(L, i));

    if (i < ord(Low(TCapGroup))) or (i > ord(High(TCapGroup))) then

        begin

        LuaCallError('Invalid capgroup type!', call, paramsyntax);

        LuaToCapGroupOrd:= -1;

        end

    else

        LuaToCapGroupOrd:= i;

end;

function LuaToSpriteOrd(L : Plua_State; i: LongInt; call, paramsyntax: shortstring): LongInt; 

begin

    if lua_isnoneornil(L, i) then i:= -1

    else i:= Trunc(lua_tonumber(L, i));

    if (i < ord(Low(TSprite))) or (i > ord(High(TSprite))) then

        begin

        LuaCallError('Invalid sprite id!', call, paramsyntax);

        LuaToSpriteOrd:= -1;

        end

    else

        LuaToSpriteOrd:= i;

end;

function LuaToMapGenOrd(L : Plua_State; i: LongInt; call, paramsyntax: shortstring): LongInt; 

begin

    if lua_isnoneornil(L, i) then i:= -1

    else i:= Trunc(lua_tonumber(L, i));

    if (i < ord(Low(TMapGen))) or (i > ord(High(TMapGen))) then

        begin

        LuaCallError('Invalid mapgen id!', call, paramsyntax);

        LuaToMapGenOrd:= -1;

        end

    else

        LuaToMapGenOrd:= i;

end;

// wrapped calls

// functions called from Lua:

// function(L : Plua_State) : LongInt; Cdecl;

// where L contains the state, returns the number of return values on the stack

// call CheckLuaParamCount or CheckAndFetchParamCount

// to validate/get the number of passed arguments (see their call definitions)

//

// use as return value the number of variables pushed back to the lua script

function lc_band(L: PLua_State): LongInt; Cdecl;

begin

    if CheckLuaParamCount(L, 2, 'band', 'value1, value2') then

        lua_pushnumber(L, Trunc(lua_tonumber(L, 2)) and Trunc(lua_tonumber(L, 1)))

    else

        lua_pushnil(L);

    lc_band := 1;

end;

function lc_bor(L: PLua_State): LongInt; Cdecl;

begin

    if CheckLuaParamCount(L, 2, 'bor', 'value1, value2') then

        lua_pushnumber(L, Trunc(lua_tonumber(L, 2)) or Trunc(lua_tonumber(L, 1)))

    else

        lua_pushnil(L);

    lc_bor := 1;

end;

function lc_bnot(L: PLua_State): LongInt; Cdecl;

begin

    if CheckLuaParamCount(L, 1, 'bnot', 'value') then

        lua_pushnumber(L, (not Trunc(lua_tonumber(L, 1))))

    else

        lua_pushnil(L);

    lc_bnot := 1;

end;

function lc_div(L: PLua_State): LongInt; Cdecl;

begin

    if CheckLuaParamCount(L, 2, 'div', 'dividend, divisor') then

        lua_pushnumber(L, Trunc(lua_tonumber(L, 1)) div Trunc(lua_tonumber(L, 2)))

    else

        lua_pushnil(L);

    lc_div := 1;

end;

function lc_getinputmask(L : Plua_State) : LongInt; Cdecl;

begin

    if CheckLuaParamCount(L, 0, 'GetInputMask', '') then

        lua_pushnumber(L, InputMask);

    lc_getinputmask:= 1

end;

function lc_setinputmask(L : Plua_State) : LongInt; Cdecl;

begin

    if CheckLuaParamCount(L, 1, 'SetInputMask', 'mask') then

        InputMask:= Trunc(lua_tonumber(L, 1));

    lc_setinputmask:= 0

end;

function lc_writelntoconsole(L : Plua_State) : LongInt; Cdecl;

begin

    if CheckLuaParamCount(L, 1, 'WriteLnToConsole', 'string') then

        WriteLnToConsole('Lua: ' + lua_tostring(L ,1));

    lc_writelntoconsole:= 0;

end;

function lc_writelntochat(L : Plua_State) : LongInt; Cdecl;

begin

    if CheckLuaParamCount(L, 1, 'WriteLnToChat', 'string') then

        AddChatString(#2 + lua_tostring(L, 1));

    lc_writelntochat:= 0;

end;

function lc_parsecommand(L : Plua_State) : LongInt; Cdecl;

var t: PChar;

    i,c: LongWord;

    s: shortstring;

begin

    if CheckLuaParamCount(L, 1, 'ParseCommand', 'string') then

        begin

        t:= lua_tolstring(L, 1, Psize_t(@c));

        for i:= 1 to c do s[i]:= t[i-1];

        s[0]:= char(c);

        ParseCommand(s, true, true);

        end;

    lc_parsecommand:= 0;

end;

// sets weapon to the desired ammo type

function lc_setweapon(L : Plua_State) : LongInt; Cdecl;

var at: LongInt;

const

    call = 'SetWeapon';

    params = 'ammoType';

begin

    // no point to run this without any CurrentHedgehog

    if (CurrentHedgehog <> nil) and (CheckLuaParamCount(L, 1, call, params)) then

        begin

        at:= LuaToAmmoTypeOrd(L, 1, call, params);

        if at >= 0 then

            ParseCommand('setweap ' + char(at), true, true);

        end;

    lc_setweapon:= 0;

end;

// enable/disable cinematic effects

function lc_setcinematicmode(L : Plua_State) : LongInt; Cdecl;

const

    call = 'SetCinematicMode';

    params = 'enable';

begin

    if (CheckLuaParamCount(L, 1, call, params)) then

        begin

        CinematicScript:= lua_toboolean(L, 1);

        end;