pilang3/modules/std.cpp

#include <sstream>
#include <memory>
#include "pilang.hpp"
using namespace Pilang3;


class Std {
public:
    static Variable retval(SharedEnvironment, Cmdargs& args) {
        return Variable({
            Variable::id_retval,
            std::make_shared<Variable>(args[0])
        });
    }

    static Variable set(SharedEnvironment env, Cmdargs& args) {
        env->globalVars[std::get<std::string>(args[0].data)] = std::make_shared<Variable>(args[1]);
        return args[1];
    }

    static Variable get(SharedEnvironment env, Cmdargs& args) {
        return *env->globalVars[std::get<std::string>(args[0].data)];
    }

    static Variable concat(SharedEnvironment, Cmdargs& args) {
        std::ostringstream fres;
        for (const auto& arg : args) {
            switch (arg.type) {
                case Variable::id_integer: fres << std::get<integer_type>(arg.data); break;
                case Variable::id_string: fres << std::get<std::string>(arg.data); break;
                case Variable::id_reference: fres << std::get<SharedVariable>(arg.data).get(); break;
                case Variable::id_type: fres << std::get<Variable::Type>(arg.data); break;
                default: fres << "<ID" << arg.type << '>'; break;
            }
        }
        return Variable({
            Variable::id_string,
            fres.str()
        });
    }

    Std() {
        Pilang3::builtinCmds["return"] = {retval, 1, {Variable::id_any}, false};
        Pilang3::builtinCmds["set"] = {set, 2, {Variable::id_string, Variable::id_any}, false};
        Pilang3::builtinCmds["get"] = {get, 1, {Variable::id_string}, false};
        Pilang3::builtinCmds["concat"] = {concat, 0, {}, true};
    }
};

static Std inst;