info.bliki.extensions.scribunto.template.Frame Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of bliki-core Show documentation

This is the core project.

The newest version!

package info.bliki.extensions.scribunto.template;

import info.bliki.wiki.filter.ParsedPageName;
import org.luaj.vm2.LuaTable;
import org.luaj.vm2.LuaValue;

import java.util.HashMap;
import java.util.Map;

import static info.bliki.extensions.scribunto.engine.lua.ScribuntoLuaEngine.toLuaString;


public class Frame {
    private final ParsedPageName page;
    private final Map templateParameters;
    private final Frame parent;
    private boolean isSubst;

    public Frame(ParsedPageName page, Map templateParameters, Frame parent, boolean isSubst) {
        this.templateParameters = templateParameters;
        this.page = page;
        this.parent = parent;
        this.isSubst = isSubst;
    }

    public Frame newChild(ParsedPageName pageName, Map templateParameters, boolean isSubst) {
        return new Frame(pageName, templateParameters, this, isSubst);
    }

    public LuaValue getArgument(String name) {
        String value = templateParameters != null ? templateParameters.get(name) : null;
        if (value != null) {
            return toLuaString(value);
        } else {
            return LuaValue.NIL;
        }
    }

    public Map getTemplateParameters() {
        return new HashMap<>(templateParameters);
    }

    public LuaValue getAllArguments() {
        LuaTable table = new LuaTable();
        for (Map.Entry entry: templateParameters.entrySet()) {
            try {
                final int numberedParam = Integer.parseInt(entry.getKey());
                table.set(LuaValue.valueOf(numberedParam), toLuaString(entry.getValue()));
            } catch (NumberFormatException e) {
                table.set(toLuaString(entry.getKey()), toLuaString(entry.getValue()));
            }
        }
        return table;
    }

    public Frame getParent() {
        return parent;
    }

    public String getTitle() {
        return page.pagename;
    }

    public boolean isSubsting() {
        return isSubst;
    }
}