info.bliki.extensions.scribunto.engine.lua.interfaces.MwMessage 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.engine.lua.interfaces;

import org.luaj.vm2.LuaNumber;
import org.luaj.vm2.LuaString;
import org.luaj.vm2.LuaTable;
import org.luaj.vm2.LuaValue;
import org.luaj.vm2.lib.OneArgFunction;
import org.luaj.vm2.lib.TwoArgFunction;

import java.text.NumberFormat;
import java.util.Locale;

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

public class MwMessage implements MwInterface {
    @Override
    public String name() {
        return "mw.message";
    }

    @Override
    public LuaTable getInterface() {
        LuaTable iface = new LuaTable();
        iface.set("plain", messagePlain());
        iface.set("check", messageCheck());
        return iface;
    }

    private LuaValue messageCheck() {
        return new TwoArgFunction() {
            @Override
            public LuaValue call(LuaValue what, LuaValue data) {
                return FALSE;
            }
        };
    }

    private LuaValue messagePlain() {
        return new OneArgFunction() {
            @Override
            public LuaValue call(LuaValue data) {
                return new Message(data).plain();
            }
        };
    }

    @Override
    public LuaValue getSetupOptions() {
        LuaTable options = new LuaTable();

        options.set("lang", "en");
        return options;
    }

    private static class Message {
        final boolean useDB;
        final String lang;
        final LuaTable params;
        final LuaTable keys;
        final LuaString rawMessage;

        Message(LuaValue data) {
            useDB  = data.get("useDB").optboolean(false);
            lang   = data.get("lang").optjstring("en");
            keys   = data.get("keys").opttable(new LuaTable());
            params = data.get("params").opttable(new LuaTable());
            rawMessage = data.get("rawMessage").optstring(LuaString.valueOf(""));
        }

        public LuaValue plain() {
            String msg = rawMessage.tojstring();
            return toLuaString(replace(msg, params));
        }

        private String replace(String msg, LuaTable params) {
            for (int i=1; i