pocketpy/src/tools/repl.cpp

#include "pocketpy/tools/repl.hpp"
#include "pocketpy/common/export.h"

namespace pkpy {
REPL::REPL(VM* vm) : vm(vm) {
    vm->stdout_write("pocketpy " PK_VERSION " (" __DATE__ ", " __TIME__ ") ");
    vm->stdout_write(_S("[", sizeof(void*) * 8, " bit] on ", kPlatformStrings[PK_SYS_PLATFORM], "\n"));
    vm->stdout_write("https://github.com/pocketpy/pocketpy" "\n");
    vm->stdout_write("Type \"exit()\" to exit." "\n");
}

bool REPL::input(std::string line) {
    CompileMode mode = REPL_MODE;
    if(need_more_lines) {
        buffer += line;
        buffer += '\n';
        int n = buffer.size();
        if(n >= need_more_lines) {
            for(int i = buffer.size() - need_more_lines; i < buffer.size(); i++) {
                // no enough lines
                if(buffer[i] != '\n') return true;
            }
            need_more_lines = 0;
            line = buffer;
            buffer.clear();
            mode = CELL_MODE;
        } else {
            return true;
        }
    }

    try {
        vm->exec(line, "<stdin>", mode);
    } catch(NeedMoreLines ne) {
        buffer += line;
        buffer += '\n';
        need_more_lines = ne.is_compiling_class ? 3 : 2;
        if(need_more_lines) return true;
    }
    return false;
}

}  // namespace pkpy