Files
ivy/frontend/cmd/repl.c

158 lines
3.0 KiB
C

#include "../debug.h"
#include "../line-ed/line-ed.h"
#include "cmd.h"
#include <blue/cmd.h>
#include <blue/term.h>
#include <ivy/lang/lex.h>
#include <ivy/lang/ast.h>
#include <stdbool.h>
enum {
OPT_LEX_ONLY = 0x1000u,
};
static int repl_lex_only()
{
struct line_ed *ed = line_ed_create();
line_ed_set_flags(ed, LINE_ED_REMOVE_CONTINUATIONS);
struct ivy_lexer *lex;
enum ivy_status status = ivy_lexer_create(&lex);
if (status != IVY_OK) {
line_ed_destroy(ed);
return -1;
}
ivy_lexer_set_source(lex, &ed->l_line_source);
while (true) {
struct ivy_token *tok = ivy_lexer_read(lex);
status = ivy_lexer_get_status(lex);
if (status == IVY_ERR_EOF) {
break;
}
if (status != IVY_OK) {
b_err("lex error (%s)",
ivy_status_to_string(ivy_lexer_get_status(lex)));
break;
}
print_lex_token(tok);
}
ivy_lexer_destroy(lex);
line_ed_destroy(ed);
return 0;
}
static void skip_line(struct ivy_lexer *lex)
{
while (ivy_lexer_tokens_available(lex)) {
struct ivy_token *tok = ivy_lexer_read(lex);
bool line_end = (tok->t_type == IVY_TOK_LINEFEED);
ivy_token_destroy(tok);
if (line_end) {
break;
}
}
}
static int repl_full()
{
struct line_ed *ed = line_ed_create();
line_ed_set_flags(ed, LINE_ED_REMOVE_CONTINUATIONS);
struct ivy_lexer *lex;
enum ivy_status status = ivy_lexer_create(&lex);
if (status != IVY_OK) {
line_ed_destroy(ed);
return -1;
}
ivy_lexer_set_source(lex, &ed->l_line_source);
struct ivy_parser *parser;
status = ivy_parser_create(&parser);
if (status != IVY_OK) {
ivy_lexer_destroy(lex);
line_ed_destroy(ed);
return -1;
}
while (true) {
struct ivy_token *tok = ivy_lexer_read(lex);
status = ivy_lexer_get_status(lex);
if (status == IVY_ERR_EOF) {
break;
}
if (status != IVY_OK) {
b_err("lex error (%s)",
ivy_status_to_string(ivy_lexer_get_status(lex)));
continue;
}
status = ivy_parser_push_token(parser, tok);
if (status != IVY_OK) {
b_err("parse error (%s)",
ivy_status_to_string(status));
ivy_token_destroy(tok);
skip_line(lex);
continue;
}
if (ivy_lexer_tokens_available(lex)) {
continue;
}
struct ivy_ast_node *root = ivy_parser_root_node(parser);
ivy_ast_node_print(root);
#if 0
struct ivy_ast_node *node = ivy_parser_dequeue_node(parser);
if (!node) {
continue;
}
ivy_ast_node_print(node);
ivy_ast_node_destroy(node);
#endif
}
ivy_parser_destroy(parser);
ivy_lexer_destroy(lex);
line_ed_destroy(ed);
return 0;
}
int repl(const b_command *cmd, const b_arglist *args, const b_array *_)
{
if (b_arglist_get_count(args, OPT_LEX_ONLY, B_COMMAND_INVALID_ID) > 0) {
return repl_lex_only();
}
return repl_full();
}
B_COMMAND(CMD_REPL, CMD_ROOT)
{
B_COMMAND_NAME("shell");
B_COMMAND_SHORT_NAME('S');
B_COMMAND_DESC("start an interactive Ivy shell.");
B_COMMAND_HELP_OPTION();
B_COMMAND_FUNCTION(repl);
B_COMMAND_OPTION(OPT_LEX_ONLY) {
B_OPTION_LONG_NAME("lex-only");
B_OPTION_SHORT_NAME('L');
B_OPTION_DESC("print the lexical tokens generated by the user input.");
}
}