#include "magic-v2.hpp"
// magic-v2.cpp - second generation magic parser
//
// Copyright © 2014 Ben Longbons <b.r.longbons@gmail.com>
//
// This file is part of The Mana World (Athena server)
//
// 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, either version 3 of the License, or
// (at your option) any later version.
//
// 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, see <http://www.gnu.org/licenses/>.
#include <set>
#include "../sexpr/parser.hpp"
#include "../mmo/dumb_ptr.hpp"
#include "itemdb.hpp"
#include "magic-expr.hpp"
#include "../poison.hpp"
namespace magic_v2
{
static
std::map<RString, proc_t> procs;
static
std::map<RString, val_t> const_defm;
static
size_t intern_id(ZString id_name)
{
// TODO use InternPool
size_t i;
for (i = 0; i < magic_conf.varv.size(); i++)
if (id_name == magic_conf.varv[i].name)
return i;
// i = magic_conf.varv.size();
/* Must add new */
magic_conf_t::mcvar new_var {};
new_var.name = id_name;
new_var.val.ty = TYPE::UNDEF;
magic_conf.varv.push_back(new_var);
return i;
}
inline
bool INTERN_ASSERT(ZString name, int id)
{
int zid = intern_id(name);
if (zid != id)
{
FPRINTF(stderr,
"[magic-conf] INTERNAL ERROR: Builtin special var %s interned to %d, not %d as it should be!\n",
name, zid, id);
}
return zid == id;
}
static
bool init0()
{
bool ok = true;
ok &= INTERN_ASSERT("min_casttime", VAR_MIN_CASTTIME);
ok &= INTERN_ASSERT("obscure_chance", VAR_OBSCURE_CHANCE);
ok &= INTERN_ASSERT("caster", VAR_CASTER);
ok &= INTERN_ASSERT("spellpower", VAR_SPELLPOWER);
ok &= INTERN_ASSERT("self_spell", VAR_SPELL);
ok &= INTERN_ASSERT("self_invocation", VAR_INVOCATION);
ok &= INTERN_ASSERT("target", VAR_TARGET);
ok &= INTERN_ASSERT("script_target", VAR_SCRIPTTARGET);
ok &= INTERN_ASSERT("location", VAR_LOCATION);
return ok;
}
static
bool bind_constant(io::LineSpan span, RString name, val_t *val)
{
if (!const_defm.insert({name, *val}).second)
{
span.error(STRPRINTF("Redefinition of constant '%s'", name));
return false;
}
return true;
}
static
val_t *find_constant(RString name)
{
auto it = const_defm.find(name);
if (it != const_defm.end())
return &it->second;
return NULL;
}
static
dumb_ptr<effect_t> new_effect(EFFECT ty)
{
auto effect = dumb_ptr<effect_t>::make();
effect->ty = ty;
return effect;
}
static
dumb_ptr<effect_t> set_effect_continuation(dumb_ptr<effect_t> src, dumb_ptr<effect_t> continuation)
{
dumb_ptr<effect_t> retval = src;
/* This function is completely analogous to `spellguard_implication' above; read the control flow implications above first before pondering it. */
if (src == continuation)
return retval;
/* For FOR and FOREACH, we use special stack handlers and thus don't have to set
* the continuation. It's only IF that we need to handle in this fashion. */
if (src->ty == EFFECT::IF)
{
set_effect_continuation(src->e.e_if.true_branch, continuation);
set_effect_continuation(src->e.e_if.false_branch, continuation);
}
if (src->next)
set_effect_continuation(src->next, continuation);
else
src->next = continuation;
return retval;
}
static
dumb_ptr<spellguard_t> new_spellguard(SPELLGUARD ty)
{
dumb_ptr<spellguard_t> retval = dumb_ptr<spellguard_t>::make();
retval->ty = ty;
return retval;
}
static
dumb_ptr<spellguard_t> spellguard_implication(dumb_ptr<spellguard_t> a, dumb_ptr<spellguard_t> b)
{
dumb_ptr<spellguard_t> retval = a;
if (a == b)
{
/* This can happen due to reference sharing:
* e.g.,
* (R0 -> (R1 | R2)) => (R3)
* yields
* (R0 -> (R1 -> R3 | R2 -> R3))
*
* So if we now add => R4 to that, we want
* (R0 -> (R1 -> R3 -> R4 | R2 -> R3 -> R4))
*
* but we only need to add it once, because the R3 reference is shared.
*/
return retval;
}
/* If the premise is a disjunction, b is the continuation of _all_ branches */
if (a->ty == SPELLGUARD::CHOICE)
spellguard_implication(a->s.s_alt, b);
if (a->next)
spellguard_implication(a->next, b);
else
// this is the important bit
a->next = b;
return retval;
}
static
bool add_spell(io::LineSpan span, dumb_ptr<spell_t> spell)
{
auto pair1 = magic_conf.spells_by_name.insert({spell->name, spell});
if (!pair1.second)
{
span.error(STRPRINTF("Attempt to redefine spell '%s'", spell->name));
return false;
}
auto pair2 = magic_conf.spells_by_invocation.insert({spell->invocation, spell});
if (!pair2.second)
{
span.error(STRPRINTF("Attempt to redefine spell invocation '%s'", spell->invocation));
magic_conf.spells_by_name.erase(pair1.first);
return false;
}
return true;
}
static
bool add_teleport_anchor(io::LineSpan span, dumb_ptr<teleport_anchor_t> anchor)
{
auto pair1 = magic_conf.anchors_by_name.insert({anchor->name, anchor});
if (!pair1.second)
{
span.error(STRPRINTF("Attempt to redefine teleport anchor '%s'", anchor->name));
return false;
}
auto pair2 = magic_conf.anchors_by_invocation.insert({anchor->name, anchor});
if (!pair2.second)
{
span.error(STRPRINTF("Attempt to redefine anchor invocation '%s'", anchor->invocation));
magic_conf.anchors_by_name.erase(pair1.first);
return false;
}
return true;
}
static
bool install_proc(io::LineSpan span, dumb_ptr<proc_t> proc)
{
RString name = proc->name;
if (!procs.insert({name, std::move(*proc)}).second)
{
span.error("procedure already exists");
return false;
}
return true;
}
static
bool call_proc(io::LineSpan span, ZString name, dumb_ptr<std::vector<dumb_ptr<expr_t>>> argvp, dumb_ptr<effect_t>& retval)
{
auto pi = procs.find(name);
if (pi == procs.end())
{
span.error(STRPRINTF("Unknown procedure '%s'", name));
return false;
}
proc_t *p = &pi->second;
if (p->argv.size() != argvp->size())
{
span.error(STRPRINTF("Procedure %s/%zu invoked with %zu parameters",
name, p->argv.size(), argvp->size()));
return false;
}
retval = new_effect(EFFECT::CALL);
retval->e.e_call.body = p->body;
retval->e.e_call.formalv = &p->argv;
retval->e.e_call.actualvp = argvp;
return true;
}
static
bool op_effect(io::LineSpan span, ZString name, Slice<dumb_ptr<expr_t>> argv, dumb_ptr<effect_t>& effect)
{
op_t *op = magic_get_op(name);
if (!op)
{
span.error(STRPRINTF("Unknown operation '%s'", name));
return false;
}
if (op->signature.size() != argv.size())
{
span.error(STRPRINTF("Incorrect number of arguments to operation '%s': Expected %zu, found %zu",
name, op->signature.size(), argv.size()));
return false;
}
effect = new_effect(EFFECT::OP);
effect->e.e_op.line_nr = span.begin.line;
effect->e.e_op.column = span.begin.column;
effect->e.e_op.opp = op;
assert (argv.size() <= MAX_ARGS);
effect->e.e_op.args_nr = argv.size();
std::copy(argv.begin(), argv.end(), effect->e.e_op.args);
return true;
}
static
dumb_ptr<expr_t> dot_expr(dumb_ptr<expr_t> expr, int id)
{
dumb_ptr<expr_t> retval = magic_new_expr(EXPR::SPELLFIELD);
retval->e.e_field.id = id;
retval->e.e_field.expr = expr;
return retval;
}
static
bool fun_expr(io::LineSpan span, ZString name, Slice<dumb_ptr<expr_t>> argv, dumb_ptr<expr_t>& expr)
{
fun_t *fun = magic_get_fun(name);
if (!fun)
{
span.error(STRPRINTF("Unknown function '%s'", name));
return false;
}
if (fun->signature.size() != argv.size())
{
span.error(STRPRINTF("Incorrect number of arguments to function '%s': Expected %zu, found %zu",
name, fun->signature.size(), argv.size()));
return false;
}
expr = magic_new_expr(EXPR::FUNAPP);
expr->e.e_funapp.line_nr = span.begin.line;
expr->e.e_funapp.column = span.begin.column;
expr->e.e_funapp.funp = fun;
assert (argv.size() <= MAX_ARGS);
expr->e.e_funapp.args_nr = argv.size();
std::copy(argv.begin(), argv.end(), expr->e.e_funapp.args);
return true;
}
static
dumb_ptr<expr_t> BIN_EXPR(io::LineSpan span, ZString name, dumb_ptr<expr_t> left, dumb_ptr<expr_t> right)
{
dumb_ptr<expr_t> e[2];
e[0] = left;
e[1] = right;
dumb_ptr<expr_t> rv;
if (!fun_expr(span, name, e, rv))
abort();
return rv;
}
static
bool fail(const sexpr::SExpr& s, ZString msg)
{
s._span.error(msg);
return false;
}
}
namespace magic_v2
{
using sexpr::SExpr;
static
bool parse_expression(const SExpr& x, dumb_ptr<expr_t>& out);
static
bool parse_effect(const SExpr& s, dumb_ptr<effect_t>& out);
static
bool parse_spellguard(const SExpr& s, dumb_ptr<spellguard_t>& out);
static
bool parse_spellbody(const SExpr& s, dumb_ptr<spellguard_t>& out);
// Note: anything with dumb_ptr leaks memory on failure
// once this is all done, we can convert it to unique_ptr
// (may require bimaps somewhere)
static
bool is_comment(const SExpr& s)
{
if (s._type == sexpr::STRING)
return true;
if (s._type != sexpr::LIST)
return false;
if (s._list.empty())
return false;
if (s._list[0]._type != sexpr::TOKEN)
return false;
return s._list[0]._str == "DISABLED";
}
static
bool parse_loc(const SExpr& s, e_location_t& loc)
{
if (s._type != sexpr::LIST)
return fail(s, "loc not list");
if (s._list.size() != 4)
return fail(s, "loc not 3 args");
if (s._list[0]._type != sexpr::TOKEN)
return fail(s._list[0], "loc cmd not tok");
if (s._list[0]._str != "@")
return fail(s._list[0], "loc cmd not cmd");
return parse_expression(s._list[1], loc.m)
&& parse_expression(s._list[2], loc.x)
&& parse_expression(s._list[3], loc.y);
}
static
bool parse_expression(const SExpr& x, dumb_ptr<expr_t>& out)
{
switch (x._type)
{
case sexpr::INT:
{
val_t val;
val.ty = TYPE::INT;
val.v.v_int = x._int;
if (val.v.v_int != x._int)
return fail(x, "integer too large");
out = magic_new_expr(EXPR::VAL);
out->e.e_val = val;
return true;
}
case sexpr::STRING:
{
val_t val;
val.ty = TYPE::STRING;
val.v.v_string = dumb_string::copys(x._str);
out = magic_new_expr(EXPR::VAL);
out->e.e_val = val;
return true;
}
case sexpr::TOKEN:
{
ZString dirs[8] = {
ZString("S"), ZString("SW"), ZString("W"), ZString("NW"), ZString("N"), ZString("NE"), ZString("E"), ZString("SE"),
};
auto begin = std::begin(dirs);
auto end = std::end(dirs);
auto it = std::find(begin, end, x._str);
if (it != end)
{
val_t val;
val.ty = TYPE::DIR;
val.v.v_dir = static_cast<DIR>(it - begin);
out = magic_new_expr(EXPR::VAL);
out->e.e_val = val;
return true;
}
}
{
if (val_t *val = find_constant(x._str))
{
out = magic_new_expr(EXPR::VAL);
out->e.e_val = *val;
return true;
}
else
{
out = magic_new_expr(EXPR::ID);
out->e.e_id = intern_id(x._str);
return true;
}
}
break;
case sexpr::LIST:
if (x._list.empty())
return fail(x, "empty list");
{
if (x._list[0]._type != sexpr::TOKEN)
return fail(x._list[0], "op not token");
ZString op = x._list[0]._str;
// area
if (op == "@")
{
e_location_t loc;
if (!parse_loc(x, loc))
return false;
out = magic_new_expr(EXPR::AREA);
out->e.e_area.ty = AREA::LOCATION;
out->e.e_area.a.a_loc = loc;
return true;
}
if (op == "@+")
{
e_location_t loc;
dumb_ptr<expr_t> width;
dumb_ptr<expr_t> height;
if (!parse_loc(x._list[1], loc))
return false;
if (!parse_expression(x._list[2], width))
return false;
if (!parse_expression(x._list[3], height))
return false;
out = magic_new_expr(EXPR::AREA);
out->e.e_area.ty = AREA::RECT;
out->e.e_area.a.a_rect.loc = loc;
out->e.e_area.a.a_rect.width = width;
out->e.e_area.a.a_rect.height = height;
return true;
}
if (op == "TOWARDS")
{
e_location_t loc;
dumb_ptr<expr_t> dir;
dumb_ptr<expr_t> width;
dumb_ptr<expr_t> depth;
if (!parse_loc(x._list[1], loc))
return false;
if (!parse_expression(x._list[2], dir))
return false;
if (!parse_expression(x._list[3], width))
return false;
if (!parse_expression(x._list[4], depth))
return false;
out = magic_new_expr(EXPR::AREA);
out->e.e_area.ty = AREA::BAR;
out->e.e_area.a.a_bar.loc = loc;
out->e.e_area.a.a_bar.dir = dir;
out->e.e_area.a.a_bar.width = width;
out->e.e_area.a.a_bar.depth = depth;
return true;
}
if (op == ".")
{
if (x._list.size() != 3)
return fail(x, ". not 2");
dumb_ptr<expr_t> expr;
if (!parse_expression(x._list[1], expr))
return false;
if (x._list[2]._type != sexpr::TOKEN)
return fail(x._list[2], ".elem not name");
ZString elem = x._list[2]._str;
out = dot_expr(expr, intern_id(elem));
return true;
}
static
std::set<ZString> ops =
{
"<", ">", "<=", ">=", "==", "!=",
"+", "-", "*", "%", "/",
"&", "^", "|", "<<", ">>",
"&&", "||",
};
// TODO implement unary operators
if (ops.count(op))
{
// operators are n-ary and left-associative
if (x._list.size() < 3)
return fail(x, "operator not at least 2 args");
auto begin = x._list.begin() + 1;
auto end = x._list.end();
if (!parse_expression(*begin, out))
return false;
++begin;
for (; begin != end; ++begin)
{
dumb_ptr<expr_t> tmp;
if (!parse_expression(*begin, tmp))
return false;
out = BIN_EXPR(x._span, op, out, tmp);
}
return true;
}
std::vector<dumb_ptr<expr_t>> argv;
for (auto it = x._list.begin() + 1, end = x._list.end(); it != end; ++it)
{
dumb_ptr<expr_t> expr;
if (!parse_expression(*it, expr))
return false;
argv.push_back(expr);
}
return fun_expr(x._span, op, argv, out);
}
break;
}
abort();
}
static
bool parse_item(const SExpr& s, int& id, int& count)
{
if (s._type == sexpr::STRING)
{
count = 1;
item_data *item = itemdb_searchname(s._str);
if (!item)
return fail(s, "no such item");
id = item->nameid;
return true;
}
if (s._type != sexpr::LIST)
return fail(s, "item not string or list");
if (s._list.size() != 2)
return fail(s, "item list is not pair");
if (s._list[0]._type != sexpr::INT)
return fail(s._list[0], "item pair first not int");
count = s._list[0]._int;
if (s._list[1]._type != sexpr::STRING)
return fail(s._list[1], "item pair second not name");
item_data *item = itemdb_searchname(s._list[1]._str);
if (!item)
return fail(s, "no such item");
id = item->nameid;
return true;
}
static
bool parse_spellguard(const SExpr& s, dumb_ptr<spellguard_t>& out)
{
if (s._type != sexpr::LIST)
return fail(s, "not list");
if (s._list.empty())
return fail(s, "empty list");
if (s._list[0]._type != sexpr::TOKEN)
return fail(s._list[0], "not token");
ZString cmd = s._list[0]._str;
if (cmd == "OR")
{
auto begin = s._list.begin() + 1;
auto end = s._list.end();
if (begin == end)
return fail(s, "missing arguments");
if (!parse_spellguard(*begin, out))
return false;
++begin;
for (; begin != end; ++begin)
{
dumb_ptr<spellguard_t> alt;
if (!parse_spellguard(*begin, alt))
return false;
dumb_ptr<spellguard_t> choice = new_spellguard(SPELLGUARD::CHOICE);
choice->next = out;
choice->s.s_alt = alt;
out = choice;
}
return true;
}
if (cmd == "GUARD")
{
auto begin = s._list.begin() + 1;
auto end = s._list.end();
while (is_comment(end[-1]))
--end;
if (begin == end)
return fail(s, "missing arguments");
if (!parse_spellguard(end[-1], out))
return false;
--end;
for (; begin != end; --end)
{
if (is_comment(end[-1]))
continue;
dumb_ptr<spellguard_t> implier;
if (!parse_spellguard(end[-1], implier))
return false;
out = spellguard_implication(implier, out);
}
return true;
}
if (cmd == "REQUIRE")
{
if (s._list.size() != 2)
return fail(s, "not one argument");
dumb_ptr<expr_t> condition;
if (!parse_expression(s._list[1], condition))
return false;
out = new_spellguard(SPELLGUARD::CONDITION);
out->s.s_condition = condition;
return true;
}
if (cmd == "MANA")
{
if (s._list.size() != 2)
return fail(s, "not one argument");
dumb_ptr<expr_t> mana;
if (!parse_expression(s._list[1], mana))
return false;
out = new_spellguard(SPELLGUARD::MANA);
out->s.s_mana = mana;
return true;
}
if (cmd == "CASTTIME")
{
if (s._list.size() != 2)
return fail(s, "not one argument");
dumb_ptr<expr_t> casttime;
if (!parse_expression(s._list[1], casttime))
return false;
out = new_spellguard(SPELLGUARD::CASTTIME);
out->s.s_casttime = casttime;
return true;
}
if (cmd == "CATALYSTS")
{
dumb_ptr<component_t> items = nullptr;
for (auto it = s._list.begin() + 1, end = s._list.end(); it != end; ++it)
{
int id, count;
if (!parse_item(*it, id, count))
return false;
magic_add_component(&items, id, count);
}
out = new_spellguard(SPELLGUARD::CATALYSTS);
out->s.s_catalysts = items;
return true;
}
if (cmd == "COMPONENTS")
{
dumb_ptr<component_t> items = nullptr;
for (auto it = s._list.begin() + 1, end = s._list.end(); it != end; ++it)
{
int id, count;
if (!parse_item(*it, id, count))
return false;
magic_add_component(&items, id, count);
}
out = new_spellguard(SPELLGUARD::COMPONENTS);
out->s.s_components = items;
return true;
}
return fail(s._list[0], "unknown guard");
}
static
bool build_effect_list(std::vector<SExpr>::const_iterator begin,
std::vector<SExpr>::const_iterator end, dumb_ptr<effect_t>& out)
{
// these backward lists could be forward by keeping the reference
// I know this is true because Linus said so
out = new_effect(EFFECT::SKIP);
while (end != begin)
{
const SExpr& s = *--end;
if (is_comment(s))
continue;
dumb_ptr<effect_t> chain;
if (!parse_effect(s, chain))
return false;
out = set_effect_continuation(chain, out);
}
return true;
}
static
bool parse_effect(const SExpr& s, dumb_ptr<effect_t>& out)
{
if (s._type != sexpr::LIST)
return fail(s, "not list");
if (s._list.empty())
return fail(s, "empty list");
if (s._list[0]._type != sexpr::TOKEN)
return fail(s._list[0], "not token");
ZString cmd = s._list[0]._str;
if (cmd == "BLOCK")
{
return build_effect_list(s._list.begin() + 1, s._list.end(), out);
}
if (cmd == "SET")
{
if (s._list.size() != 3)
return fail(s, "not 2 args");
if (s._list[1]._type != sexpr::TOKEN)
return fail(s._list[1], "not token");
ZString name = s._list[1]._str;
if (find_constant(name))
return fail(s._list[1], "assigning to constant");
dumb_ptr<expr_t> expr;
if (!parse_expression(s._list[2], expr))
return false;
out = new_effect(EFFECT::ASSIGN);
out->e.e_assign.id = intern_id(name);
out->e.e_assign.expr = expr;
return true;
}
if (cmd == "SCRIPT")
{
if (s._list.size() != 2)
return fail(s, "not 1 arg");
if (s._list[1]._type != sexpr::STRING)
return fail(s._list[1], "not string");
ZString body = s._list[1]._str;
std::unique_ptr<const ScriptBuffer> script = parse_script(body, s._list[1]._span.begin.line, true);
if (!script)
return fail(s._list[1], "script does not compile");
out = new_effect(EFFECT::SCRIPT);
out->e.e_script = dumb_ptr<const ScriptBuffer>(script.release());
return true;
}
if (cmd == "SKIP")
{
if (s._list.size() != 1)
return fail(s, "not 0 arg");
out = new_effect(EFFECT::SKIP);
return true;
}
if (cmd == "ABORT")
{
if (s._list.size() != 1)
return fail(s, "not 0 arg");
out = new_effect(EFFECT::ABORT);
return true;
}
if (cmd == "END")
{
if (s._list.size() != 1)
return fail(s, "not 0 arg");
out = new_effect(EFFECT::END);
return true;
}
if (cmd == "BREAK")
{
if (s._list.size() != 1)
return fail(s, "not 0 arg");
out = new_effect(EFFECT::BREAK);
return true;
}
if (cmd == "FOREACH")
{
if (s._list.size() != 5)
return fail(s, "not 4 arg");
if (s._list[1]._type != sexpr::TOKEN)
return fail(s._list[1], "foreach type not token");
ZString type = s._list[1]._str;
FOREACH_FILTER filter;
if (type == "PC")
filter = FOREACH_FILTER::PC;
else if (type == "MOB")
filter = FOREACH_FILTER::MOB;
else if (type == "ENTITY")
filter = FOREACH_FILTER::ENTITY;
else if (type == "SPELL")
filter = FOREACH_FILTER::SPELL;
else if (type == "TARGET")
filter = FOREACH_FILTER::TARGET;
else if (type == "NPC")
filter = FOREACH_FILTER::NPC;
else
return fail(s._list[1], "unknown foreach filter");
if (s._list[2]._type != sexpr::TOKEN)
return fail(s._list[2], "foreach var not token");
ZString var = s._list[2]._str;
dumb_ptr<expr_t> area;
dumb_ptr<effect_t> effect;
if (!parse_expression(s._list[3], area))
return false;
if (!parse_effect(s._list[4], effect))
return false;
out = new_effect(EFFECT::FOREACH);
out->e.e_foreach.id = intern_id(var);
out->e.e_foreach.area = area;
out->e.e_foreach.body = effect;
out->e.e_foreach.filter = filter;
return true;
}
if (cmd == "FOR")
{
if (s._list.size() != 5)
return fail(s, "not 4 arg");
if (s._list[1]._type != sexpr::TOKEN)
return fail(s._list[1], "for var not token");
ZString var = s._list[1]._str;
dumb_ptr<expr_t> low;
dumb_ptr<expr_t> high;
dumb_ptr<effect_t> effect;
if (!parse_expression(s._list[2], low))
return false;
if (!parse_expression(s._list[3], high))
return false;
if (!parse_effect(s._list[4], effect))
return false;
out = new_effect(EFFECT::FOR);
out->e.e_for.id = intern_id(var);
out->e.e_for.start = low;
out->e.e_for.stop = high;
out->e.e_for.body = effect;
return true;
}
if (cmd == "IF")
{
if (s._list.size() != 3 && s._list.size() != 4)
return fail(s, "not 2 or 3 args");
dumb_ptr<expr_t> cond;
dumb_ptr<effect_t> if_true;
dumb_ptr<effect_t> if_false;
if (!parse_expression(s._list[1], cond))
return false;
if (!parse_effect(s._list[2], if_true))
return false;
if (s._list.size() == 4)
{
if (!parse_effect(s._list[3], if_false))
return false;
}
else
if_false = new_effect(EFFECT::SKIP);
out = new_effect(EFFECT::IF);
out->e.e_if.cond = cond;
out->e.e_if.true_branch = if_true;
out->e.e_if.false_branch = if_false;
return true;
}
if (cmd == "WAIT")
{
if (s._list.size() != 2)
return fail(s, "not 1 arg");
dumb_ptr<expr_t> expr;
if (!parse_expression(s._list[1], expr))
return false;
out = new_effect(EFFECT::SLEEP);
out->e.e_sleep = expr;
return true;
}
if (cmd == "CALL")
{
if (s._list.size() < 2)
return fail(s, "call what?");
if (s._list[1]._type != sexpr::TOKEN)
return fail(s._list[1], "call token please");
ZString func = s._list[1]._str;
auto argvp = dumb_ptr<std::vector<dumb_ptr<expr_t>>>::make();
for (auto it = s._list.begin() + 2, end = s._list.end(); it != end; ++it)
{
dumb_ptr<expr_t> expr;
if (!parse_expression(*it, expr))
return false;
argvp->push_back(expr);
}
return call_proc(s._span, func, argvp, out);
}
auto argv = std::vector<dumb_ptr<expr_t>>();
for (auto it = s._list.begin() + 1, end = s._list.end(); it != end; ++it)
{
dumb_ptr<expr_t> expr;
if (!parse_expression(*it, expr))
return false;
argv.push_back(expr);
}
return op_effect(s._span, cmd, argv, out);
}
static
bool parse_spellbody(const SExpr& s, dumb_ptr<spellguard_t>& out)
{
if (s._type != sexpr::LIST)
return fail(s, "not list");
if (s._list.empty())
return fail(s, "empty list");
if (s._list[0]._type != sexpr::TOKEN)
return fail(s._list[0], "not token");
ZString cmd = s._list[0]._str;
if (cmd == "=>")
{
if (s._list.size() != 3)
return fail(s, "list does not have exactly 2 arguments");
dumb_ptr<spellguard_t> guard;
if (!parse_spellguard(s._list[1], guard))
return false;
dumb_ptr<spellguard_t> body;
if (!parse_spellbody(s._list[2], body))
return false;
out = spellguard_implication(guard, body);
return true;
}
if (cmd == "|")
{
if (s._list.size() == 1)
return fail(s, "spellbody choice empty");
auto begin = s._list.begin() + 1;
auto end = s._list.end();
if (!parse_spellbody(*begin, out))
return false;
++begin;
for (; begin != end; ++begin)
{
dumb_ptr<spellguard_t> alt;
if (!parse_spellbody(*begin, alt))
return false;
auto tmp = out;
out = new_spellguard(SPELLGUARD::CHOICE);
out->next = tmp;
out->s.s_alt = alt;
}
return true;
}
if (cmd == "EFFECT")
{
auto begin = s._list.begin() + 1;
auto end = s._list.end();
dumb_ptr<effect_t> effect, attrig, atend;
// decreasing end can never pass begin, since we know that
// begin[-1] is token EFFECT
while (is_comment(end[-1]))
--end;
if (end[-1]._type == sexpr::LIST && !end[-1]._list.empty()
&& end[-1]._list[0]._type == sexpr::TOKEN
&& end[-1]._list[0]._str == "ATEND")
{
auto atb = end[-1]._list.begin() + 1;
auto ate = end[-1]._list.end();
if (!build_effect_list(atb, ate, atend))
return false;
--end;
while (is_comment(end[-1]))
--end;
}
else
{
atend = nullptr;
}
if (end[-1]._type == sexpr::LIST && !end[-1]._list.empty()
&& end[-1]._list[0]._type == sexpr::TOKEN
&& end[-1]._list[0]._str == "ATTRIGGER")
{
auto atb = end[-1]._list.begin() + 1;
auto ate = end[-1]._list.end();
if (!build_effect_list(atb, ate, attrig))
return false;
--end;
}
else
{
attrig = nullptr;
}
if (!build_effect_list(begin, end, effect))
return false;
out = new_spellguard(SPELLGUARD::EFFECT);
out->s.s_effect.effect = effect;
out->s.s_effect.at_trigger = attrig;
out->s.s_effect.at_end = atend;
return true;
}
return fail(s._list[0], "unknown spellbody");
}
static
bool parse_top_set(const std::vector<SExpr>& in)
{
if (in.size() != 3)
return fail(in[0], "not 2 arguments");
ZString name = in[1]._str;
dumb_ptr<expr_t> expr;
if (!parse_expression(in[2], expr))
return false;
if (find_constant(name))
return fail(in[1], "assign constant");
size_t var_id = intern_id(name);
magic_eval(dumb_ptr<env_t>(&magic_default_env), &magic_conf.varv[var_id].val, expr);
return true;
}
static
bool parse_const(io::LineSpan span, const std::vector<SExpr>& in)
{
if (in.size() != 3)
return fail(in[0], "not 2 arguments");
if (in[1]._type != sexpr::TOKEN)
return fail(in[1], "not token");
ZString name = in[1]._str;
dumb_ptr<expr_t> expr;
if (!parse_expression(in[2], expr))
return false;
val_t tmp;
magic_eval(dumb_ptr<env_t>(&magic_default_env), &tmp, expr);
return bind_constant(span, name, &tmp);
}
static
bool parse_anchor(io::LineSpan span, const std::vector<SExpr>& in)
{
if (in.size() != 4)
return fail(in[0], "not 3 arguments");
auto anchor = dumb_ptr<teleport_anchor_t>::make();
if (in[1]._type != sexpr::TOKEN)
return fail(in[1], "not token");
anchor->name = in[1]._str;
if (in[2]._type != sexpr::STRING)
return fail(in[2], "not string");
anchor->invocation = in[2]._str;
dumb_ptr<expr_t> expr;
if (!parse_expression(in[3], expr))
return false;
anchor->location = expr;
return add_teleport_anchor(span, anchor);
}
static
bool parse_proc(io::LineSpan span, const std::vector<SExpr>& in)
{
if (in.size() < 4)
return fail(in[0], "not at least 3 arguments");
auto proc = dumb_ptr<proc_t>::make();
if (in[1]._type != sexpr::TOKEN)
return fail(in[1], "name not token");
proc->name = in[1]._str;
if (in[2]._type != sexpr::LIST)
return fail(in[2], "args not list");
for (const SExpr& arg : in[2]._list)
{
if (arg._type != sexpr::TOKEN)
return fail(arg, "arg not token");
proc->argv.push_back(intern_id(arg._str));
}
if (!build_effect_list(in.begin() + 3, in.end(), proc->body))
return false;
return install_proc(span, proc);
}
static
bool parse_spell(io::LineSpan span, const std::vector<SExpr>& in)
{
if (in.size() < 6)
return fail(in[0], "not at least 5 arguments");
if (in[1]._type != sexpr::LIST)
return fail(in[1], "flags not list");
auto spell = dumb_ptr<spell_t>::make();
for (const SExpr& s : in[1]._list)
{
if (s._type != sexpr::TOKEN)
return fail(s, "flag not token");
SPELL_FLAG flag = SPELL_FLAG::ZERO;
if (s._str == "LOCAL")
flag = SPELL_FLAG::LOCAL;
else if (s._str == "NONMAGIC")
flag = SPELL_FLAG::NONMAGIC;
else if (s._str == "SILENT")
flag = SPELL_FLAG::SILENT;
else
return fail(s, "unknown flag");
if (bool(spell->flags & flag))
return fail(s, "duplicate flag");
spell->flags |= flag;
}
if (in[2]._type != sexpr::TOKEN)
return fail(in[2], "name not token");
spell->name = in[2]._str;
if (in[3]._type != sexpr::STRING)
return fail(in[3], "invoc not string");
spell->invocation = in[3]._str;
if (in[4]._type != sexpr::LIST)
return fail(in[4], "spellarg not list");
if (in[4]._list.size() == 0)
{
spell->spellarg_ty = SPELLARG::NONE;
}
else
{
if (in[4]._list.size() != 2)
return fail(in[4], "spellarg not empty list or pair");
if (in[4]._list[0]._type != sexpr::TOKEN)
return fail(in[4]._list[0], "spellarg type not token");
if (in[4]._list[1]._type != sexpr::TOKEN)
return fail(in[4]._list[1], "spellarg name not token");
ZString ty = in[4]._list[0]._str;
if (ty == "PC")
spell->spellarg_ty = SPELLARG::PC;
else if (ty == "STRING")
spell->spellarg_ty = SPELLARG::STRING;
else
return fail(in[4]._list[0], "unknown spellarg type");
ZString an = in[4]._list[1]._str;
spell->arg = intern_id(an);
}
std::vector<SExpr>::const_iterator it = in.begin() + 5;
for (;; ++it)
{
if (it == in.end())
return fail(it[-1], "end of list scanning LET defs");
if (is_comment(*it))
continue;
if (it->_type != sexpr::LIST || it->_list.empty())
break;
if (it->_list[0]._type != sexpr::TOKEN || it->_list[0]._str != "LET")
break;
if (it->_list[1]._type != sexpr::TOKEN)
return fail(it->_list[1], "let name not token");
ZString name = it->_list[1]._str;
if (find_constant(name))
return fail(it->_list[1], "constant exists");
dumb_ptr<expr_t> expr;
if (!parse_expression(it->_list[2], expr))
return false;
letdef_t let;
let.id = intern_id(name);
let.expr = expr;
spell->letdefv.push_back(let);
}
if (it + 1 != in.end())
return fail(*it, "expected only one body entry besides LET");
// formally, 'guard' only refers to the first argument of '=>'
// but internally, spellbodies use the same thing
dumb_ptr<spellguard_t> guard;
if (!parse_spellbody(*it, guard))
return false;
spell->spellguard = guard;
return add_spell(span, spell);
}
static
bool parse_top(io::LineSpan span, const std::vector<SExpr>& vs)
{
if (vs.empty())
{
span.error("Empty list at top");
return false;
}
if (vs[0]._type != sexpr::TOKEN)
return fail(vs[0], "top not token");
ZString cmd = vs[0]._str;
if (cmd == "CONST")
return parse_const(span, vs);
if (cmd == "PROCEDURE")
return parse_proc(span, vs);
if (cmd == "SET")
return parse_top_set(vs);
if (cmd == "SPELL")
return parse_spell(span, vs);
if (cmd == "TELEPORT-ANCHOR")
return parse_anchor(span, vs);
return fail(vs[0], "Unknown top-level command");
}
static
bool loop(sexpr::Lexer& in)
{
SExpr s;
while (sexpr::parse(in, s))
{
if (is_comment(s))
continue;
if (s._type != sexpr::LIST)
return fail(s, "top-level entity not a list or comment");
if (!parse_top(s._span, s._list))
return false;
}
// handle low-level errors
if (in.peek() != sexpr::TOK_EOF)
{
in.span().error("parser gave up before end of file");
return false;
}
return true;
}
} // namespace magic_v2
bool magic_init0()
{
return magic_v2::init0();
}
bool load_magic_file_v2(ZString filename)
{
sexpr::Lexer in(filename);
bool rv = magic_v2::loop(in);
if (!rv)
{
in.span().error(STRPRINTF("next token: %s '%s'", sexpr::token_name(in.peek()), in.val_string()));
}
return rv;
}