io.cpp

Go to the documentation of this file.

#include "StdLib.hpp"
 
static std::string toHex(int value)
{
    std::stringstream ss;
    ss << std::showbase << std::hex << value;
    return ss.str();
}
 
namespace Phasor
{
 
void StdLib::registerIOFunctions(VM *vm)
{
    vm->registerNativeFunction("c_fmt", StdLib::io_c_format);
    vm->registerNativeFunction("prints", StdLib::io_prints);
    vm->registerNativeFunction("printf", StdLib::io_printf);
    vm->registerNativeFunction("puts", StdLib::io_puts);
    vm->registerNativeFunction("putf", StdLib::io_putf);
    vm->registerNativeFunction("puts_error", StdLib::io_puts_error);
    vm->registerNativeFunction("putf_error", StdLib::io_putf_error);
#ifndef SANDBOXED
    vm->registerNativeFunction("clear", StdLib::io_clear);
    vm->registerNativeFunction("gets", StdLib::io_gets);
#endif
}
 
#ifndef SANDBOXED
Value StdLib::io_clear(const std::vector<Value> &args, VM *vm)
{
    checkArgCount(args, 0, "clear");
    vm->regRun(OpCode::PRINT_R, "\033[2J\033[H");
    return Value();
}
#endif
 
Value StdLib::io_c_format(const std::vector<Value> &args, VM *)
{
    if (args.empty())
    {
        return Value(""); // Return empty string if no arguments
    }
 
    const std::string &fmt = args[0].asString();
    std::string        out;
    size_t             argIndex = 1; // Start from 1 because args[0] is the format string
 
    for (size_t i = 0; i < fmt.size(); ++i)
    {
        if (fmt[i] == '%' && i + 1 < fmt.size())
        {
            char spec = fmt[++i]; // Move to the format specifier
            if (argIndex < args.size())
            {
                const Value &v = args[argIndex++];
                switch (spec)
                {
                case 's':
                case 'c':
                    out += v.asString();
                    break;
                case 'd':
                    out += std::to_string(v.asInt());
                    break;
                case 'f':
                    out += std::to_string(v.asFloat());
                    break;
                case '%':
                    out += '%';
                    break;
                case 'x':
                    out += "0x" + toHex((int)v.asInt());
                    break;
                default:
                    // Unknown specifier, include it literally
                    out += '%';
                    out += spec;
                    break;
                }
                continue;
            }
        }
        out += fmt[i];
    }
    return Value(out);
}
 
Value StdLib::io_prints(const std::vector<Value> &args, VM *vm)
{
    checkArgCount(args, 1, "prints");
    vm->regRun(OpCode::PRINT_R, args[0]);
    return Value("");
}
 
Value StdLib::io_printf(const std::vector<Value> &args, VM *vm)
{
    checkArgCount(args, 1, "printf", true);
    std::vector<Value> formatArgs(args.begin(), args.end());
    vm->regRun(OpCode::PRINT_R, io_c_format(formatArgs, vm));
    return Value("");
}
 
Value StdLib::io_puts(const std::vector<Value> &args, VM *vm)
{
    checkArgCount(args, 1, "puts", true);
    std::string input = args[0].toString();
    vm->regRun(OpCode::PRINT_R, input + "\n");
    return Value("");
}
 
Value StdLib::io_putf(const std::vector<Value> &args, VM *vm)
{
    checkArgCount(args, 1, "putf", true);
    std::vector<Value> formatArgs(args.begin(), args.end());
    std::string        input = io_c_format(formatArgs, vm).toString();
    vm->regRun(OpCode::PRINT_R, input + "\n");
    return Value("");
}
 
#ifndef SANDBOXED
Value StdLib::io_gets(const std::vector<Value> &args, VM *vm)
{
    checkArgCount(args, 0, "gets");
    return vm->regRun(OpCode::READLINE_R, REGISTER1);
}
#endif
 
Value StdLib::io_puts_error(const std::vector<Value> &args, VM *vm)
{
    checkArgCount(args, 1, "puts_error", true);
    std::string input = args[0].toString();
    vm->regRun(OpCode::PRINTERROR_R, input + "\n");
    return Value("");
}
 
Value StdLib::io_putf_error(const std::vector<Value> &args, VM *vm)
{
    checkArgCount(args, 1, "putf_error", true);
    std::vector<Value> formatArgs(args.begin(), args.end());
    std::string        input = io_c_format(formatArgs, vm).toString();
    vm->regRun(OpCode::PRINTERROR_R, input + "\n");
    return Value("");
}
} // namespace Phasor