/**

** @file mruby/compile.h - mruby parser

**

** See Copyright Notice in mruby.h

*/

#ifndef MRUBY_COMPILE_H

#define MRUBY_COMPILE_H

#include "common.h"

#include "mruby/mempool.h"

/**

* mruby Compiler

*/

MRB_BEGIN_DECL

#include <mruby.h>

struct mrb_parser_state;

/* load context */

typedef struct mrb_ccontext {

mrb_sym *syms;

int slen;

char *filename;

uint16_t lineno;

int (*partial_hook)(struct mrb_parser_state*);

void *partial_data;

struct RClass *target_class;

mrb_bool capture_errors:1;

mrb_bool dump_result:1;

mrb_bool no_exec:1;

mrb_bool keep_lv:1;

mrb_bool no_optimize:1;

mrb_bool no_ext_ops:1;

mrb_bool no_return_value:1;

const struct RProc *upper;

size_t parser_nerr;

} mrb_ccontext; /* compiler context */

MRB_API mrb_ccontext* mrb_ccontext_new(mrb_state *mrb);

MRB_API void mrb_ccontext_free(mrb_state *mrb, mrb_ccontext *cxt);

MRB_API const char *mrb_ccontext_filename(mrb_state *mrb, mrb_ccontext *c, const char *s);

MRB_API void mrb_ccontext_partial_hook(mrb_ccontext *c, int (*partial_hook)(struct mrb_parser_state*), void*data);

MRB_API void mrb_ccontext_cleanup_local_variables(mrb_ccontext *c);

/* compatibility macros */

#define mrbc_context mrb_ccontext

#define mrbc_context_new mrb_ccontext_new

#define mrbc_context_free mrb_ccontext_free

#define mrbc_filename mrb_ccontext_filename

#define mrbc_partial_hook mrb_ccontext_partial_hook

#define mrbc_cleanup_local_variables mrb_ccontext_cleanup_local_variables

/* AST node structure */

typedef struct mrb_ast_node mrb_ast_node;

/* lexer states */

enum mrb_lex_state_enum {

EXPR_BEG, /* ignore newline, +/- is a sign. */

EXPR_END, /* newline significant, +/- is an operator. */

EXPR_ENDARG, /* ditto, and unbound braces. */

EXPR_ENDFN, /* ditto, and unbound braces. */

EXPR_ARG, /* newline significant, +/- is an operator. */

EXPR_CMDARG, /* newline significant, +/- is an operator. */

EXPR_MID, /* newline significant, +/- is a sign. */

EXPR_FNAME, /* ignore newline, no reserved words. */

EXPR_DOT, /* right after '.' or '::', no reserved words. */

EXPR_CLASS, /* immediate after 'class', no here document. */

EXPR_VALUE, /* alike EXPR_BEG but label is disallowed. */

EXPR_MAX_STATE

};

/* saved error message */

struct mrb_parser_message {

uint16_t lineno;

int column;

char* message;

};

#define MRB_PARSER_TOKBUF_MAX (UINT16_MAX-1)

#define MRB_PARSER_TOKBUF_SIZE 256

/* parser structure */

struct mrb_parser_state {

mrb_state *mrb;

mempool *pool;

mrb_ast_node *cells;

const char *s, *send;

#ifndef MRB_NO_STDIO

/* If both f and s are non-null, it will be taken preferentially from s until s < send. */

FILE *f;

#endif

mrb_ccontext *cxt;

mrb_sym filename_sym;

uint16_t lineno;

int column;

enum mrb_lex_state_enum lstate;

struct parser_lex_strterm *lex_strterm;

unsigned int cond_stack;

unsigned int cmdarg_stack;

int paren_nest;

int lpar_beg;

int in_def, in_single, in_kwarg;

mrb_bool cmd_start:1;

mrb_ast_node *locals;

mrb_ast_node *pb;

char *tokbuf;

char buf[MRB_PARSER_TOKBUF_SIZE];

int tidx;

int tsiz;

mrb_ast_node *heredocs_from_nextline;

mrb_ast_node *parsing_heredoc;

void *ylval;

size_t nerr;

size_t nwarn;

mrb_ast_node *tree;

mrb_bool no_optimize:1;

mrb_bool capture_errors:1;

mrb_bool no_ext_ops:1;

mrb_bool no_return_value:1;

const struct RProc *upper;

struct mrb_parser_message error_buffer[10];

struct mrb_parser_message warn_buffer[10];

mrb_sym* filename_table;

uint16_t filename_table_length;

uint16_t current_filename_index;

/* Variable-sized node management */

mrb_ast_node *nvars;

};

MRB_API struct mrb_parser_state* mrb_parser_new(mrb_state*);

MRB_API void mrb_parser_free(struct mrb_parser_state*);

MRB_API void mrb_parser_parse(struct mrb_parser_state*,mrb_ccontext*);

MRB_API void mrb_parser_set_filename(struct mrb_parser_state*, char const*);

MRB_API mrb_sym mrb_parser_get_filename(struct mrb_parser_state*, uint16_t idx);

/* utility functions */

#ifndef MRB_NO_STDIO

MRB_API struct mrb_parser_state* mrb_parse_file(mrb_state*,FILE*,mrb_ccontext*);

#endif

MRB_API struct mrb_parser_state* mrb_parse_string(mrb_state*,const char*,mrb_ccontext*);

MRB_API struct mrb_parser_state* mrb_parse_nstring(mrb_state*,const char*,size_t,mrb_ccontext*);

MRB_API struct RProc* mrb_generate_code(mrb_state*, struct mrb_parser_state*);

MRB_API mrb_value mrb_load_exec(mrb_state *mrb, struct mrb_parser_state *p, mrb_ccontext *c);

/**

* program load functions

*

* Please note! Currently due to interactions with the GC calling these functions will

* leak one RProc object per function call.

* To prevent this save the current memory arena before calling and restore the arena

* right after, like so

*

* int ai = mrb_gc_arena_save(mrb);

* mrb_value status = mrb_load_string(mrb, buffer);

* mrb_gc_arena_restore(mrb, ai);

*

* Also, when called from a C function defined as a method, the current stack is destroyed.

* If processing continues after this function, the objects obtained from the arguments

* must be protected as needed before this function.

*/

#ifndef MRB_NO_STDIO

MRB_API mrb_value mrb_load_file(mrb_state*,FILE*);

MRB_API mrb_value mrb_load_file_cxt(mrb_state*,FILE*, mrb_ccontext *cxt);

MRB_API mrb_value mrb_load_detect_file_cxt(mrb_state *mrb, FILE *fp, mrb_ccontext *c);

#endif

MRB_API mrb_value mrb_load_string(mrb_state *mrb, const char *s);

MRB_API mrb_value mrb_load_nstring(mrb_state *mrb, const char *s, size_t len);

MRB_API mrb_value mrb_load_string_cxt(mrb_state *mrb, const char *s, mrb_ccontext *cxt);

MRB_API mrb_value mrb_load_nstring_cxt(mrb_state *mrb, const char *s, size_t len, mrb_ccontext *cxt);

/** @} */

MRB_END_DECL

#endif /* MRUBY_COMPILE_H */