mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
907ae13cf6
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@65652 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
711 lines
22 KiB
C
711 lines
22 KiB
C
/* indent-tabs-mode: nil */
|
|
#include "ruby.h"
|
|
#include "ruby/encoding.h"
|
|
#include "ruby/util.h"
|
|
#include "internal.h"
|
|
#include "node.h"
|
|
#include "vm_core.h"
|
|
#include "iseq.h"
|
|
|
|
static VALUE rb_mAST;
|
|
static VALUE rb_cNode;
|
|
|
|
struct ASTNodeData {
|
|
rb_ast_t *ast;
|
|
NODE *node;
|
|
};
|
|
|
|
static void
|
|
node_gc_mark(void *ptr)
|
|
{
|
|
struct ASTNodeData *data = (struct ASTNodeData *)ptr;
|
|
rb_gc_mark((VALUE)data->ast);
|
|
}
|
|
|
|
static const rb_data_type_t rb_node_type = {
|
|
"AST/node",
|
|
{node_gc_mark, RUBY_TYPED_DEFAULT_FREE, 0,},
|
|
0, 0,
|
|
RUBY_TYPED_FREE_IMMEDIATELY,
|
|
};
|
|
|
|
static VALUE rb_ast_node_alloc(VALUE klass);
|
|
|
|
static void
|
|
setup_node(VALUE obj, rb_ast_t *ast, NODE *node)
|
|
{
|
|
struct ASTNodeData *data;
|
|
|
|
TypedData_Get_Struct(obj, struct ASTNodeData, &rb_node_type, data);
|
|
data->ast = ast;
|
|
data->node = node;
|
|
}
|
|
|
|
static VALUE
|
|
ast_new_internal(rb_ast_t *ast, NODE *node)
|
|
{
|
|
VALUE obj;
|
|
|
|
obj = rb_ast_node_alloc(rb_cNode);
|
|
setup_node(obj, ast, node);
|
|
|
|
return obj;
|
|
}
|
|
|
|
VALUE rb_ast_parse_str(VALUE str);
|
|
VALUE rb_ast_parse_file(VALUE path);
|
|
|
|
static VALUE
|
|
ast_parse_new(void)
|
|
{
|
|
return rb_parser_set_context(rb_parser_new(), NULL, 0);
|
|
}
|
|
|
|
static VALUE
|
|
ast_parse_done(rb_ast_t *ast)
|
|
{
|
|
if (!ast->body.root) {
|
|
rb_ast_dispose(ast);
|
|
rb_exc_raise(GET_EC()->errinfo);
|
|
}
|
|
|
|
return ast_new_internal(ast, (NODE *)ast->body.root);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* RubyVM::AbstractSyntaxTree.parse(string) -> RubyVM::AbstractSyntaxTree::Node
|
|
*
|
|
* Parses the given string into an abstract syntax tree,
|
|
* returning the root node of that tree.
|
|
*
|
|
* SyntaxError is raised if the given string is invalid syntax.
|
|
*
|
|
* RubyVM::AbstractSyntaxTree.parse("x = 1 + 2")
|
|
* # => #<RubyVM::AbstractSyntaxTree::Node(NODE_SCOPE(0) 1:0, 1:9): >
|
|
*/
|
|
static VALUE
|
|
rb_ast_s_parse(VALUE module, VALUE str)
|
|
{
|
|
return rb_ast_parse_str(str);
|
|
}
|
|
|
|
VALUE
|
|
rb_ast_parse_str(VALUE str)
|
|
{
|
|
rb_ast_t *ast = 0;
|
|
|
|
str = rb_check_string_type(str);
|
|
ast = rb_parser_compile_string_path(ast_parse_new(), Qnil, str, 1);
|
|
return ast_parse_done(ast);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* RubyVM::AbstractSyntaxTree.parse_file(pathname) -> RubyVM::AbstractSyntaxTree::Node
|
|
*
|
|
* Reads the file from <code>pathname</code>, then parses it like ::parse,
|
|
* returning the root node of the abstract syntax tree.
|
|
*
|
|
* SyntaxError is raised if <code>pathname</code>'s contents are not
|
|
* valid Ruby syntax.
|
|
*
|
|
* RubyVM::AbstractSyntaxTree.parse_file("my-app/app.rb")
|
|
* # => #<RubyVM::AbstractSyntaxTree::Node(NODE_SCOPE(0) 1:0, 31:3): >
|
|
*/
|
|
static VALUE
|
|
rb_ast_s_parse_file(VALUE module, VALUE path)
|
|
{
|
|
return rb_ast_parse_file(path);
|
|
}
|
|
|
|
VALUE
|
|
rb_ast_parse_file(VALUE path)
|
|
{
|
|
VALUE f;
|
|
rb_ast_t *ast = 0;
|
|
rb_encoding *enc = rb_utf8_encoding();
|
|
|
|
FilePathValue(path);
|
|
f = rb_file_open_str(path, "r");
|
|
rb_funcall(f, rb_intern("set_encoding"), 2, rb_enc_from_encoding(enc), rb_str_new_cstr("-"));
|
|
ast = rb_parser_compile_file_path(ast_parse_new(), Qnil, f, 1);
|
|
rb_io_close(f);
|
|
return ast_parse_done(ast);
|
|
}
|
|
|
|
static VALUE node_children(rb_ast_t*, NODE*);
|
|
|
|
static VALUE
|
|
node_find(VALUE self, const int node_id)
|
|
{
|
|
VALUE ary;
|
|
long i;
|
|
struct ASTNodeData *data;
|
|
TypedData_Get_Struct(self, struct ASTNodeData, &rb_node_type, data);
|
|
|
|
if (nd_node_id(data->node) == node_id) return self;
|
|
|
|
ary = node_children(data->ast, data->node);
|
|
|
|
for (i = 0; i < RARRAY_LEN(ary); i++) {
|
|
VALUE child = RARRAY_AREF(ary, i);
|
|
|
|
if (CLASS_OF(child) == rb_cNode) {
|
|
VALUE result = node_find(child, node_id);
|
|
if (RTEST(result)) return result;
|
|
}
|
|
}
|
|
|
|
return Qnil;
|
|
}
|
|
|
|
static VALUE
|
|
script_lines(VALUE path)
|
|
{
|
|
VALUE hash, lines;
|
|
ID script_lines;
|
|
CONST_ID(script_lines, "SCRIPT_LINES__");
|
|
if (!rb_const_defined_at(rb_cObject, script_lines)) return Qnil;
|
|
hash = rb_const_get_at(rb_cObject, script_lines);
|
|
if (!RB_TYPE_P(hash, T_HASH)) return Qnil;
|
|
lines = rb_hash_lookup(hash, path);
|
|
if (!RB_TYPE_P(lines, T_ARRAY)) return Qnil;
|
|
return lines;
|
|
}
|
|
|
|
static VALUE
|
|
rb_ast_s_of(VALUE module, VALUE body)
|
|
{
|
|
VALUE path, node, lines;
|
|
int node_id;
|
|
const rb_iseq_t *iseq = NULL;
|
|
|
|
if (rb_obj_is_proc(body)) {
|
|
iseq = vm_proc_iseq(body);
|
|
|
|
if (!rb_obj_is_iseq((VALUE)iseq)) {
|
|
iseq = NULL;
|
|
}
|
|
}
|
|
else {
|
|
iseq = rb_method_iseq(body);
|
|
}
|
|
|
|
if (!iseq) return Qnil;
|
|
|
|
path = rb_iseq_path(iseq);
|
|
node_id = iseq->body->location.node_id;
|
|
if (!NIL_P(lines = script_lines(path))) {
|
|
node = rb_ast_parse_str(rb_ary_join(lines, Qnil));
|
|
}
|
|
else {
|
|
node = rb_ast_parse_file(path);
|
|
}
|
|
|
|
return node_find(node, node_id);
|
|
}
|
|
|
|
static VALUE
|
|
rb_ast_node_alloc(VALUE klass)
|
|
{
|
|
struct ASTNodeData *data;
|
|
VALUE obj = TypedData_Make_Struct(klass, struct ASTNodeData, &rb_node_type, data);
|
|
|
|
return obj;
|
|
}
|
|
|
|
static const char*
|
|
node_type_to_str(NODE *node)
|
|
{
|
|
return ruby_node_name(nd_type(node));
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* node.type -> string
|
|
*
|
|
* Returns the type of this node as a string.
|
|
*
|
|
* root = RubyVM::AbstractSyntaxTree.parse("x = 1 + 2")
|
|
* root.type # => "NODE_SCOPE"
|
|
* call = root.children[2]
|
|
* call.type # => "NODE_OPCALL"
|
|
*/
|
|
static VALUE
|
|
rb_ast_node_type(VALUE self)
|
|
{
|
|
struct ASTNodeData *data;
|
|
TypedData_Get_Struct(self, struct ASTNodeData, &rb_node_type, data);
|
|
|
|
return rb_fstring_cstr(node_type_to_str(data->node));
|
|
}
|
|
|
|
#define NEW_CHILD(ast, node) node ? ast_new_internal(ast, node) : Qnil
|
|
|
|
static VALUE
|
|
rb_ary_new_from_node_args(rb_ast_t *ast, long n, ...)
|
|
{
|
|
va_list ar;
|
|
VALUE ary;
|
|
long i;
|
|
|
|
ary = rb_ary_new2(n);
|
|
|
|
va_start(ar, n);
|
|
for (i=0; i<n; i++) {
|
|
NODE *node;
|
|
node = va_arg(ar, NODE *);
|
|
rb_ary_push(ary, NEW_CHILD(ast, node));
|
|
}
|
|
va_end(ar);
|
|
return ary;
|
|
}
|
|
|
|
static VALUE
|
|
dump_block(rb_ast_t *ast, NODE *node)
|
|
{
|
|
VALUE ary = rb_ary_new();
|
|
do {
|
|
rb_ary_push(ary, NEW_CHILD(ast, node->nd_head));
|
|
} while (node->nd_next &&
|
|
nd_type(node->nd_next) == NODE_BLOCK &&
|
|
(node = node->nd_next, 1));
|
|
if (node->nd_next) {
|
|
rb_ary_push(ary, NEW_CHILD(ast, node->nd_next));
|
|
}
|
|
|
|
return ary;
|
|
}
|
|
|
|
static VALUE
|
|
dump_array(rb_ast_t *ast, NODE *node)
|
|
{
|
|
VALUE ary = rb_ary_new();
|
|
rb_ary_push(ary, NEW_CHILD(ast, node->nd_head));
|
|
|
|
while (node->nd_next && nd_type(node->nd_next) == NODE_ARRAY) {
|
|
node = node->nd_next;
|
|
rb_ary_push(ary, NEW_CHILD(ast, node->nd_head));
|
|
}
|
|
rb_ary_push(ary, NEW_CHILD(ast, node->nd_next));
|
|
|
|
return ary;
|
|
}
|
|
|
|
static VALUE
|
|
var_name(ID id)
|
|
{
|
|
if (!id) return Qnil;
|
|
if (!rb_id2str(id)) return Qnil;
|
|
return ID2SYM(id);
|
|
}
|
|
|
|
static VALUE
|
|
node_children(rb_ast_t *ast, NODE *node)
|
|
{
|
|
char name[DECIMAL_SIZE_OF_BITS(sizeof(long) * CHAR_BIT) + 2]; /* including '$' */
|
|
|
|
enum node_type type = nd_type(node);
|
|
switch (type) {
|
|
case NODE_BLOCK:
|
|
return dump_block(ast, node);
|
|
case NODE_IF:
|
|
return rb_ary_new_from_node_args(ast, 3, node->nd_cond, node->nd_body, node->nd_else);
|
|
case NODE_UNLESS:
|
|
return rb_ary_new_from_node_args(ast, 3, node->nd_cond, node->nd_body, node->nd_else);
|
|
case NODE_CASE:
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_head, node->nd_body);
|
|
case NODE_CASE2:
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_head, node->nd_body);
|
|
case NODE_WHEN:
|
|
return rb_ary_new_from_node_args(ast, 3, node->nd_head, node->nd_body, node->nd_next);
|
|
case NODE_WHILE:
|
|
goto loop;
|
|
case NODE_UNTIL:
|
|
loop:
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_cond, node->nd_body);
|
|
case NODE_ITER:
|
|
case NODE_FOR:
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_iter, node->nd_body);
|
|
case NODE_FOR_MASGN:
|
|
return rb_ary_new_from_node_args(ast, 1, node->nd_var);
|
|
case NODE_BREAK:
|
|
goto jump;
|
|
case NODE_NEXT:
|
|
goto jump;
|
|
case NODE_RETURN:
|
|
jump:
|
|
return rb_ary_new_from_node_args(ast, 1, node->nd_stts);
|
|
case NODE_REDO:
|
|
return rb_ary_new_from_node_args(ast, 0);
|
|
case NODE_RETRY:
|
|
return rb_ary_new_from_node_args(ast, 0);
|
|
case NODE_BEGIN:
|
|
return rb_ary_new_from_node_args(ast, 1, node->nd_body);
|
|
case NODE_RESCUE:
|
|
return rb_ary_new_from_node_args(ast, 3, node->nd_head, node->nd_resq, node->nd_else);
|
|
case NODE_RESBODY:
|
|
return rb_ary_new_from_node_args(ast, 3, node->nd_args, node->nd_body, node->nd_head);
|
|
case NODE_ENSURE:
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_head, node->nd_ensr);
|
|
case NODE_AND:
|
|
goto andor;
|
|
case NODE_OR:
|
|
andor:
|
|
{
|
|
VALUE ary = rb_ary_new();
|
|
|
|
while (1) {
|
|
rb_ary_push(ary, NEW_CHILD(ast, node->nd_1st));
|
|
if (!node->nd_2nd || nd_type(node->nd_2nd) != (int)type)
|
|
break;
|
|
node = node->nd_2nd;
|
|
}
|
|
rb_ary_push(ary, NEW_CHILD(ast, node->nd_2nd));
|
|
return ary;
|
|
}
|
|
case NODE_MASGN:
|
|
if (NODE_NAMED_REST_P(node->nd_args)) {
|
|
return rb_ary_new_from_node_args(ast, 3, node->nd_value, node->nd_head, node->nd_args);
|
|
}
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_value, node->nd_head);
|
|
case NODE_LASGN:
|
|
goto asgn;
|
|
case NODE_DASGN:
|
|
goto asgn;
|
|
case NODE_DASGN_CURR:
|
|
goto asgn;
|
|
case NODE_IASGN:
|
|
goto asgn;
|
|
case NODE_CVASGN:
|
|
asgn:
|
|
if (NODE_REQUIRED_KEYWORD_P(node)) {
|
|
return rb_ary_new_from_args(1, var_name(node->nd_vid));
|
|
}
|
|
return rb_ary_new_from_args(2, var_name(node->nd_vid), NEW_CHILD(ast, node->nd_value));
|
|
case NODE_GASGN:
|
|
goto asgn;
|
|
case NODE_CDECL:
|
|
if (node->nd_vid) {
|
|
return rb_ary_new_from_args(2, ID2SYM(node->nd_vid), NEW_CHILD(ast, node->nd_value));
|
|
}
|
|
return rb_ary_new_from_args(3, NEW_CHILD(ast, node->nd_else), ID2SYM(node->nd_else->nd_mid), NEW_CHILD(ast, node->nd_value));
|
|
case NODE_OP_ASGN1:
|
|
return rb_ary_new_from_args(4, NEW_CHILD(ast, node->nd_recv),
|
|
ID2SYM(node->nd_mid),
|
|
NEW_CHILD(ast, node->nd_args->nd_head),
|
|
NEW_CHILD(ast, node->nd_args->nd_body));
|
|
case NODE_OP_ASGN2:
|
|
return rb_ary_new_from_args(4, NEW_CHILD(ast, node->nd_recv),
|
|
node->nd_next->nd_aid ? Qtrue : Qfalse,
|
|
ID2SYM(node->nd_next->nd_vid),
|
|
NEW_CHILD(ast, node->nd_value));
|
|
case NODE_OP_ASGN_AND:
|
|
return rb_ary_new_from_args(3, NEW_CHILD(ast, node->nd_head), ID2SYM(idANDOP),
|
|
NEW_CHILD(ast, node->nd_value));
|
|
case NODE_OP_ASGN_OR:
|
|
return rb_ary_new_from_args(3, NEW_CHILD(ast, node->nd_head), ID2SYM(idOROP),
|
|
NEW_CHILD(ast, node->nd_value));
|
|
case NODE_OP_CDECL:
|
|
return rb_ary_new_from_args(3, NEW_CHILD(ast, node->nd_head),
|
|
ID2SYM(node->nd_aid),
|
|
NEW_CHILD(ast, node->nd_value));
|
|
case NODE_CALL:
|
|
case NODE_OPCALL:
|
|
case NODE_QCALL:
|
|
return rb_ary_new_from_args(3, NEW_CHILD(ast, node->nd_recv),
|
|
ID2SYM(node->nd_mid),
|
|
NEW_CHILD(ast, node->nd_args));
|
|
case NODE_FCALL:
|
|
return rb_ary_new_from_args(2, ID2SYM(node->nd_mid),
|
|
NEW_CHILD(ast, node->nd_args));
|
|
case NODE_VCALL:
|
|
return rb_ary_new_from_args(1, ID2SYM(node->nd_mid));
|
|
case NODE_SUPER:
|
|
return rb_ary_new_from_node_args(ast, 1, node->nd_args);
|
|
case NODE_ZSUPER:
|
|
return rb_ary_new_from_node_args(ast, 0);
|
|
case NODE_ARRAY:
|
|
goto ary;
|
|
case NODE_VALUES:
|
|
ary:
|
|
return dump_array(ast, node);
|
|
case NODE_ZARRAY:
|
|
return rb_ary_new_from_node_args(ast, 0);
|
|
case NODE_HASH:
|
|
return rb_ary_new_from_node_args(ast, 1, node->nd_head);
|
|
case NODE_YIELD:
|
|
return rb_ary_new_from_node_args(ast, 1, node->nd_head);
|
|
case NODE_LVAR:
|
|
case NODE_DVAR:
|
|
return rb_ary_new_from_args(1, var_name(node->nd_vid));
|
|
case NODE_IVAR:
|
|
case NODE_CONST:
|
|
case NODE_CVAR:
|
|
case NODE_GVAR:
|
|
return rb_ary_new_from_args(1, ID2SYM(node->nd_vid));
|
|
case NODE_NTH_REF:
|
|
snprintf(name, sizeof(name), "$%ld", node->nd_nth);
|
|
return rb_ary_new_from_args(1, ID2SYM(rb_intern(name)));
|
|
case NODE_BACK_REF:
|
|
name[0] = '$';
|
|
name[1] = (char)node->nd_nth;
|
|
name[2] = '\0';
|
|
return rb_ary_new_from_args(1, ID2SYM(rb_intern(name)));
|
|
case NODE_MATCH:
|
|
goto lit;
|
|
case NODE_MATCH2:
|
|
if (node->nd_args) {
|
|
return rb_ary_new_from_node_args(ast, 3, node->nd_recv, node->nd_value, node->nd_args);
|
|
}
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_recv, node->nd_value);
|
|
case NODE_MATCH3:
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_recv, node->nd_value);
|
|
case NODE_LIT:
|
|
goto lit;
|
|
case NODE_STR:
|
|
goto lit;
|
|
case NODE_XSTR:
|
|
lit:
|
|
return rb_ary_new_from_args(1, node->nd_lit);
|
|
case NODE_ONCE:
|
|
return rb_ary_new_from_node_args(ast, 1, node->nd_body);
|
|
case NODE_DSTR:
|
|
goto dlit;
|
|
case NODE_DXSTR:
|
|
goto dlit;
|
|
case NODE_DREGX:
|
|
goto dlit;
|
|
case NODE_DSYM:
|
|
dlit:
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_next->nd_head, node->nd_next->nd_next);
|
|
case NODE_EVSTR:
|
|
return rb_ary_new_from_node_args(ast, 1, node->nd_body);
|
|
case NODE_ARGSCAT:
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_head, node->nd_body);
|
|
case NODE_ARGSPUSH:
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_head, node->nd_body);
|
|
case NODE_SPLAT:
|
|
return rb_ary_new_from_node_args(ast, 1, node->nd_head);
|
|
case NODE_BLOCK_PASS:
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_head, node->nd_body);
|
|
case NODE_DEFN:
|
|
return rb_ary_new_from_args(2, ID2SYM(node->nd_mid), NEW_CHILD(ast, node->nd_defn));
|
|
case NODE_DEFS:
|
|
return rb_ary_new_from_args(3, NEW_CHILD(ast, node->nd_recv), ID2SYM(node->nd_mid), NEW_CHILD(ast, node->nd_defn));
|
|
case NODE_ALIAS:
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_1st, node->nd_2nd);
|
|
case NODE_VALIAS:
|
|
return rb_ary_new_from_args(2, ID2SYM(node->nd_alias), ID2SYM(node->nd_orig));
|
|
case NODE_UNDEF:
|
|
return rb_ary_new_from_node_args(ast, 1, node->nd_undef);
|
|
case NODE_CLASS:
|
|
return rb_ary_new_from_node_args(ast, 3, node->nd_cpath, node->nd_super, node->nd_body);
|
|
case NODE_MODULE:
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_cpath, node->nd_body);
|
|
case NODE_SCLASS:
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_recv, node->nd_body);
|
|
case NODE_COLON2:
|
|
return rb_ary_new_from_args(2, NEW_CHILD(ast, node->nd_head), ID2SYM(node->nd_mid));
|
|
case NODE_COLON3:
|
|
return rb_ary_new_from_args(1, ID2SYM(node->nd_mid));
|
|
case NODE_DOT2:
|
|
goto dot;
|
|
case NODE_DOT3:
|
|
goto dot;
|
|
case NODE_FLIP2:
|
|
goto dot;
|
|
case NODE_FLIP3:
|
|
dot:
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_beg, node->nd_end);
|
|
case NODE_SELF:
|
|
return rb_ary_new_from_node_args(ast, 0);
|
|
case NODE_NIL:
|
|
return rb_ary_new_from_node_args(ast, 0);
|
|
case NODE_TRUE:
|
|
return rb_ary_new_from_node_args(ast, 0);
|
|
case NODE_FALSE:
|
|
return rb_ary_new_from_node_args(ast, 0);
|
|
case NODE_ERRINFO:
|
|
return rb_ary_new_from_node_args(ast, 0);
|
|
case NODE_DEFINED:
|
|
return rb_ary_new_from_node_args(ast, 1, node->nd_head);
|
|
case NODE_POSTEXE:
|
|
return rb_ary_new_from_node_args(ast, 1, node->nd_body);
|
|
case NODE_ATTRASGN:
|
|
return rb_ary_new_from_args(3, NEW_CHILD(ast, node->nd_recv), ID2SYM(node->nd_mid), NEW_CHILD(ast, node->nd_args));
|
|
case NODE_LAMBDA:
|
|
return rb_ary_new_from_node_args(ast, 1, node->nd_body);
|
|
case NODE_OPT_ARG:
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_body, node->nd_next);
|
|
case NODE_KW_ARG:
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_body, node->nd_next);
|
|
case NODE_POSTARG:
|
|
if (NODE_NAMED_REST_P(node->nd_1st)) {
|
|
return rb_ary_new_from_node_args(ast, 2, node->nd_1st, node->nd_2nd);
|
|
}
|
|
return rb_ary_new_from_node_args(ast, 1, node->nd_2nd);
|
|
case NODE_ARGS:
|
|
{
|
|
struct rb_args_info *ainfo = node->nd_ainfo;
|
|
return rb_ary_new_from_args(10,
|
|
INT2NUM(ainfo->pre_args_num),
|
|
NEW_CHILD(ast, ainfo->pre_init),
|
|
NEW_CHILD(ast, ainfo->opt_args),
|
|
var_name(ainfo->first_post_arg),
|
|
INT2NUM(ainfo->post_args_num),
|
|
NEW_CHILD(ast, ainfo->post_init),
|
|
var_name(ainfo->rest_arg),
|
|
NEW_CHILD(ast, ainfo->kw_args),
|
|
NEW_CHILD(ast, ainfo->kw_rest_arg),
|
|
var_name(ainfo->block_arg));
|
|
}
|
|
case NODE_SCOPE:
|
|
{
|
|
ID *tbl = node->nd_tbl;
|
|
int i, size = tbl ? (int)*tbl++ : 0;
|
|
VALUE locals = rb_ary_new_capa(size);
|
|
for (i = 0; i < size; i++) {
|
|
rb_ary_push(locals, var_name(tbl[i]));
|
|
}
|
|
return rb_ary_new_from_args(3, locals, NEW_CHILD(ast, node->nd_args), NEW_CHILD(ast, node->nd_body));
|
|
}
|
|
case NODE_ARGS_AUX:
|
|
case NODE_LAST:
|
|
break;
|
|
}
|
|
|
|
rb_bug("node_children: unknown node: %s", ruby_node_name(type));
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* node.children -> array
|
|
*
|
|
* Returns AST nodes under this one. Each kind of node
|
|
* has different children, depending on what kind of node it is.
|
|
*
|
|
* The returned array may contain other nodes or <code>nil</code>.
|
|
*/
|
|
static VALUE
|
|
rb_ast_node_children(VALUE self)
|
|
{
|
|
struct ASTNodeData *data;
|
|
TypedData_Get_Struct(self, struct ASTNodeData, &rb_node_type, data);
|
|
|
|
return node_children(data->ast, data->node);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* node.first_lineno -> integer
|
|
*
|
|
* The line number in the source code where this AST's text began.
|
|
*/
|
|
static VALUE
|
|
rb_ast_node_first_lineno(VALUE self)
|
|
{
|
|
struct ASTNodeData *data;
|
|
TypedData_Get_Struct(self, struct ASTNodeData, &rb_node_type, data);
|
|
|
|
return INT2NUM(nd_first_lineno(data->node));
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* node.first_column -> integer
|
|
*
|
|
* The column number in the source code where this AST's text began.
|
|
*/
|
|
static VALUE
|
|
rb_ast_node_first_column(VALUE self)
|
|
{
|
|
struct ASTNodeData *data;
|
|
TypedData_Get_Struct(self, struct ASTNodeData, &rb_node_type, data);
|
|
|
|
return INT2NUM(nd_first_column(data->node));
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* node.last_lineno -> integer
|
|
*
|
|
* The line number in the source code where this AST's text ended.
|
|
*/
|
|
static VALUE
|
|
rb_ast_node_last_lineno(VALUE self)
|
|
{
|
|
struct ASTNodeData *data;
|
|
TypedData_Get_Struct(self, struct ASTNodeData, &rb_node_type, data);
|
|
|
|
return INT2NUM(nd_last_lineno(data->node));
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* node.last_column -> integer
|
|
*
|
|
* The column number in the source code where this AST's text ended.
|
|
*/
|
|
static VALUE
|
|
rb_ast_node_last_column(VALUE self)
|
|
{
|
|
struct ASTNodeData *data;
|
|
TypedData_Get_Struct(self, struct ASTNodeData, &rb_node_type, data);
|
|
|
|
return INT2NUM(nd_last_column(data->node));
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* node.inspect -> string
|
|
*
|
|
* Returns debugging information about this node as a string.
|
|
*/
|
|
static VALUE
|
|
rb_ast_node_inspect(VALUE self)
|
|
{
|
|
VALUE str;
|
|
VALUE cname;
|
|
struct ASTNodeData *data;
|
|
TypedData_Get_Struct(self, struct ASTNodeData, &rb_node_type, data);
|
|
|
|
cname = rb_class_path(rb_obj_class(self));
|
|
str = rb_str_new2("#<");
|
|
|
|
rb_str_append(str, cname);
|
|
rb_str_cat2(str, "(");
|
|
rb_str_catf(str, "%s(%d) %d:%d, %d:%d", node_type_to_str(data->node), nd_type(data->node), nd_first_lineno(data->node), nd_first_column(data->node), nd_last_lineno(data->node), nd_last_column(data->node));
|
|
rb_str_cat2(str, "): >");
|
|
|
|
return str;
|
|
}
|
|
|
|
void
|
|
Init_ast(void)
|
|
{
|
|
/*
|
|
* AbstractSyntaxTree provides methods to parse Ruby code into
|
|
* abstract syntax trees. The nodes in the tree
|
|
* are instances of RubyVM::AbstractSyntaxTree::Node.
|
|
*/
|
|
rb_mAST = rb_define_module_under(rb_cRubyVM, "AbstractSyntaxTree");
|
|
/*
|
|
* RubyVM::AbstractSyntaxTree::Node instances are created by parse methods in
|
|
* RubyVM::AbstractSyntaxTree.
|
|
*/
|
|
rb_cNode = rb_define_class_under(rb_mAST, "Node", rb_cObject);
|
|
|
|
rb_undef_alloc_func(rb_cNode);
|
|
rb_define_singleton_method(rb_mAST, "parse", rb_ast_s_parse, 1);
|
|
rb_define_singleton_method(rb_mAST, "parse_file", rb_ast_s_parse_file, 1);
|
|
rb_define_singleton_method(rb_mAST, "of", rb_ast_s_of, 1);
|
|
rb_define_method(rb_cNode, "type", rb_ast_node_type, 0);
|
|
rb_define_method(rb_cNode, "first_lineno", rb_ast_node_first_lineno, 0);
|
|
rb_define_method(rb_cNode, "first_column", rb_ast_node_first_column, 0);
|
|
rb_define_method(rb_cNode, "last_lineno", rb_ast_node_last_lineno, 0);
|
|
rb_define_method(rb_cNode, "last_column", rb_ast_node_last_column, 0);
|
|
rb_define_method(rb_cNode, "children", rb_ast_node_children, 0);
|
|
rb_define_method(rb_cNode, "inspect", rb_ast_node_inspect, 0);
|
|
}
|