Class: AbstractSyntaxTreeKit

Inherits:

Object

Object
AbstractSyntaxTreeKit

show all

Defined in:: lib/astkit.rb,
lib/astkit/node.rb,
lib/astkit/node/if.rb,
lib/astkit/node/or.rb,
lib/astkit/version.rb,
lib/astkit/node/and.rb,
lib/astkit/node/for.rb,
lib/astkit/node/lit.rb,
lib/astkit/node/nil.rb,
lib/astkit/node/str.rb,
lib/astkit/node/args.rb,
lib/astkit/node/call.rb,
lib/astkit/node/case.rb,
lib/astkit/node/cvar.rb,
lib/astkit/node/defn.rb,
lib/astkit/node/defs.rb,
lib/astkit/node/dot2.rb,
lib/astkit/node/dot3.rb,
lib/astkit/node/dstr.rb,
lib/astkit/node/dsym.rb,
lib/astkit/node/dvar.rb,
lib/astkit/node/gvar.rb,
lib/astkit/node/hash.rb,
lib/astkit/node/iter.rb,
lib/astkit/node/ivar.rb,
lib/astkit/node/lvar.rb,
lib/astkit/node/next.rb,
lib/astkit/node/once.rb,
lib/astkit/node/redo.rb,
lib/astkit/node/self.rb,
lib/astkit/node/true.rb,
lib/astkit/node/when.rb,
lib/astkit/node/xstr.rb,
lib/astkit/node/alias.rb,
lib/astkit/node/array.rb,
lib/astkit/node/block.rb,
lib/astkit/node/break.rb,
lib/astkit/node/case2.rb,
lib/astkit/node/cdecl.rb,
lib/astkit/node/class.rb,
lib/astkit/node/const.rb,
lib/astkit/node/dasgn.rb,
lib/astkit/node/dregx.rb,
lib/astkit/node/dxstr.rb,
lib/astkit/node/evstr.rb,
lib/astkit/node/false.rb,
lib/astkit/node/fcall.rb,
lib/astkit/node/flip2.rb,
lib/astkit/node/flip3.rb,
lib/astkit/node/gasgn.rb,
lib/astkit/node/iasgn.rb,
lib/astkit/node/lasgn.rb,
lib/astkit/node/masgn.rb,
lib/astkit/node/match.rb,
lib/astkit/node/qcall.rb,
lib/astkit/node/retry.rb,
lib/astkit/node/scope.rb,
lib/astkit/node/splat.rb,
lib/astkit/node/super.rb,
lib/astkit/node/undef.rb,
lib/astkit/node/until.rb,
lib/astkit/node/vcall.rb,
lib/astkit/node/while.rb,
lib/astkit/node/yield.rb,
lib/astkit/node/colon2.rb,
lib/astkit/node/colon3.rb,
lib/astkit/node/cvasgn.rb,
lib/astkit/node/ensure.rb,
lib/astkit/node/kw_arg.rb,
lib/astkit/node/lambda.rb,
lib/astkit/node/match2.rb,
lib/astkit/node/match3.rb,
lib/astkit/node/module.rb,
lib/astkit/node/nbegin.rb,
lib/astkit/node/opcall.rb,
lib/astkit/node/rescue.rb,
lib/astkit/node/return.rb,
lib/astkit/node/sclass.rb,
lib/astkit/node/unless.rb,
lib/astkit/node/valias.rb,
lib/astkit/node/values.rb,
lib/astkit/node/zarray.rb,
lib/astkit/node/zsuper.rb,
lib/astkit/node/defined.rb,
lib/astkit/node/errinfo.rb,
lib/astkit/node/nth_ref.rb,
lib/astkit/node/resbody.rb,
lib/astkit/node/argspush.rb,
lib/astkit/node/attrasgn.rb,
lib/astkit/node/back_ref.rb,
lib/astkit/node/op_asgn1.rb,
lib/astkit/node/op_asgn2.rb,
lib/astkit/node/op_cdecl.rb,
lib/astkit/node/dasgn_curr.rb,
lib/astkit/node/op_asgn_or.rb,
lib/astkit/node/op_asgn_and.rb

Defined Under Namespace

Classes: HashElement, Node

Constant Summary collapse

VERSION =

'0.1.1'

Class Method Summary collapse

Instance Method Summary collapse

#traverse(node) ⇒ Object

Class Method Details

.parse(src) ⇒ `Object`

# File 'lib/astkit.rb', line 100

def self.parse(src)
  root = RubyVM::AbstractSyntaxTree.parse(src)
  new.traverse(root)
end

.walk(node) {|node| ... } ⇒ `Object`

Yields:

(node)

# File 'lib/astkit.rb', line 105

def self.walk(node, &block)
  yield node

  case node.type
  when :SCOPE
    if node.body.instance_of?(Array)
      node.body.each do |child|
        walk(child, &block)
      end
    else
      walk(node.body, &block)
    end
  when :FCALL
    walk(node.arguments, &block)
  when :ARRAY
  when :ZARRAY
  when :LIT
  when :STR
  when :ARGS
  else
    raise "Unexpected node type[#{node.type}]"
  end
end

Instance Method Details

#traverse(node) ⇒ `Object`

# File 'lib/astkit.rb', line 129

def traverse(node)
  return node unless node.instance_of?(RubyVM::AbstractSyntaxTree::Node)

  case node.type
  when :SCOPE
    Node::SCOPE.new(
      node: node,
      local_table: node.children[0],
      arguments: traverse(node.children[1]),
      body: body(node.children[2])
    )
  when :ARRAY
    Node::ARRAY.new(
      node: node,
      elements: elements(node)
    )
  when :ZARRAY
    Node::ZARRAY.new(node: node)
  when :STR
    Node::STR.new(
      node: node,
      value: node.children[0]
    )
  when :FCALL
    Node::FCALL.new(
      node: node,
      method_id: node.children[0],
      arguments: traverse(node.children[1])
    )
  when :LIT
    Node::LIT.new(
      node: node,
      value: node.children[0]
    )
  when :DEFN
    Node::DEFN.new(
      node: node,
      method_id: node.children[0],
      method_definition: traverse(node.children[1])
    )
  when :KW_ARG
    Node::KW_ARG.new(
      node: node,
      body: traverse(node.children[0]),
      next_keyword_argument: traverse(node.children[1])
    )
  when :LASGN
    Node::LASGN.new(
      node: node,
      name: node.children[0],
      value: traverse(node.children[1])
    )
  when :ARGS
    Node::ARGS.new(
      node: node,
      parameter_count: node.children[0],
      rest_argument: node.children[6],
      keyword_arguments: traverse(node.children[7]),
      block_argument: node.children[9]
    )
  when :CDECL
    if node.children[1].instance_of?(Symbol)
      Node::CDECL.new(
        node: node,
        extension: traverse(node.children[0]),
        name: node.children[1],
        value: traverse(node.children[2])
      )
    else
      Node::CDECL.new(
        node: node,
        name: node.children[0],
        value: traverse(node.children[1])
      )
    end
  when :COLON2
    Node::COLON2.new(
      node: node,
      receiver: traverse(node.children[0]),
      name: node.children[1]
    )
  when :COLON3
    Node::COLON3.new(
      node: node,
      name: node.children[0]
    )
  when :CONST
    Node::CONST.new(
      node: node,
      name: node.children[0]
    )
  when :CALL
    Node::CALL.new(
      node: node,
      receiver: traverse(node.children[0]),
      method_id: node.children[1],
      arguments: traverse(node.children[2])
    )
  when :OPCALL
    opcall(node)
  when :HASH
    hash(node)
  when :NIL
    Node::NIL.new(node: node)
  when :TRUE
    Node::TRUE.new(node: node)
  when :FALSE
    Node::FALSE.new(node: node)
  when :OR
    Node::OR.new(
      node: node,
      left: traverse(node.children[0]),
      right: traverse(node.children[1])
    )
  when :AND
    Node::AND.new(
      node: node,
      left: traverse(node.children[0]),
      right: traverse(node.children[1])
    )
  when :LVAR
    Node::LVAR.new(
      node: node,
      name: node.children[0]
    )
  when :IF
    Node::IF.new(
      node: node,
      condition: traverse(node.children[0]),
      body: traverse(node.children[1]),
      els: traverse(node.children[2])
    )
  when :UNLESS
    Node::UNLESS.new(
      node: node,
      condition: traverse(node.children[0]),
      body: traverse(node.children[1]),
      els: traverse(node.children[2])
    )
  when :CASE
    Node::CASE.new(
      node: node,
      expression: traverse(node.children[0]),
      case_clauses: case_clauses(node.children[1])
    )
  when :CASE2
    Node::CASE2.new(
      node: node,
      case_clauses: case_clauses(node.children[1])
    )
  when :WHEN
    if node.children[2]&.type == :WHEN
      Node::WHEN.new(
        node: node,
        expressions: traverse(node.children[0]),
        body: traverse(node.children[1])
      )
    else
      Node::WHEN.new(
        node: node,
        expressions: traverse(node.children[0]),
        body: traverse(node.children[1]),
        els: traverse(node.children[2])
      )
    end
  when :FOR
    Node::FOR.new(
      node: node,
      receiver: traverse(node.children[0]),
      body: traverse(node.children[1])
    )
  when :VCALL
    vcall(node)
  when :BEGIN
    nbegin(node)
  when :ITER
    Node::ITER.new(
      node: node,
      receiver: traverse(node.children[0]),
      body: traverse(node.children[1])
    )
  when :DVAR
    dvar(node)
  when :BLOCK
    block(node)
  when :DOT2
    Node::DOT2.new(
      node: node,
      nd_begin: traverse(node.children[0]),
      nd_end: traverse(node.children[1])
    )
  when :DOT3
    Node::DOT3.new(
      node: node,
      nd_begin: traverse(node.children[0]),
      nd_end: traverse(node.children[1])
    )
  when :MASGN
    Node::MASGN.new(
      node: node,
      rhs: traverse(node.children[0]),
      lhs: traverse(node.children[1]),
      splat: traverse(node.children[2])
    )
  when :WHILE
    Node::WHILE.new(
      node: node,
      condition: traverse(node.children[0]),
      body: traverse(node.children[1])
    )
  when :UNTIL
    Node::UNTIL.new(
      node: node,
      condition: traverse(node.children[0]),
      body: traverse(node.children[1])
    )
  when :BREAK
    Node::BREAK.new(node: node)
  when :NEXT
    Node::NEXT.new(node: node)
  when :REDO
    Node::REDO.new(node: node)
  when :RETRY
    Node::RETRY.new(node: node)
  when :RESCUE
    Node::RESCUE.new(
      node: node,
      body: traverse(node.children[0]),
      rescue_body: traverse(node.children[1]),
      else_body: traverse(node.children[2])
    )
  when :RESBODY
    Node::RESBODY.new(
      node: node,
      exceptions: traverse(node.children[0]),
      clause: traverse(node.children[1]),
      next_rescue: traverse(node.children[2])
    )
  when :ERRINFO
    Node::ERRINFO.new(node: node)
  when :ENSURE
    Node::ENSURE.new(
      node: node,
      body: traverse(node.children[0]),
      clause: traverse(node.children[1])
    )
  when :GASGN
    Node::GASGN.new(
      node: node,
      name: node.children[0],
      value: traverse(node.children[1])
    )
  when :MODULE
    Node::MODULE.new(
      node: node,
      path: traverse(node.children[0]),
      body: traverse(node.children[1])
    )
  when :CLASS
    Node::CLASS.new(
      node: node,
      path: traverse(node.children[0]),
      superclass: traverse(node.children[1]),
      body: traverse(node.children[2])
    )
  when :IASGN
    Node::IASGN.new(
      node: node,
      name: node.children[0],
      value: traverse(node.children[1])
    )
  when :CVASGN
    Node::CVASGN.new(
      node: node,
      name: node.children[0],
      value: traverse(node.children[1])
    )
  when :FLIP2
    Node::FLIP2.new(
      node: node,
      nd_begin: traverse(node.children[0]),
      nd_end: traverse(node.children[1])
    )
  when :FLIP3
    Node::FLIP3.new(
      node: node,
      nd_begin: traverse(node.children[0]),
      nd_end: traverse(node.children[1])
    )
  when :SELF
    Node::SELF.new(node: node)
  when :GVAR
    Node::GVAR.new(
      node: node,
      name: node.children[0]
    )
  when :IVAR
    Node::IVAR.new(
      node: node,
      name: node.children[0]
    )
  when :CVAR
    Node::CVAR.new(
      node: node,
      name: node.children[0]
    )
  when :NTH_REF
    Node::NTH_REF.new(
      node: node,
      name: node.children[0]
    )
  when :BACK_REF
    Node::BACK_REF.new(
      node: node,
      name: node.children[0]
    )
  when :QCALL
    Node::QCALL.new(
      node: node,
      receiver: traverse(node.children[0]),
      method_id: node.children[1],
      arguments: traverse(node.children[2])
    )
  when :ZSUPER
    Node::ZSUPER.new(node: node)
  when :SUPER
    Node::SUPER.new(
      node: node,
      arguments: traverse(node.children[0])
    )
  when :LAMBDA
    Node::LAMBDA.new(
      node: node,
      body: traverse(node.children[0])
    )
  when :ATTRASGN
    Node::ATTRASGN.new(
      node: node,
      receiver: traverse(node.children[0]),
      method_id: node.children[1],
      arguments: traverse(node.children[2])
    )
  when :DSYM
    Node::DSYM.new(
      node: node,
      interpolation: traverse(node.children[0]),
      tail_str: traverse(node.children[1])
    )
  when :DSTR
    Node::DSTR.new(
      node: node,
      pre_str: node.children[0],
      interpolation: traverse(node.children[1]),
      tail_str: traverse(node.children[2])
    )
  when :EVSTR
    Node::EVSTR.new(
      node: node,
      body: traverse(node.children[0])
    )
  when :DEFINED
    Node::DEFINED.new(
      node: node,
      expression: traverse(node.children[0])
    )
  when :SCLASS
    Node::SCLASS.new(
      node: node,
      receiver: traverse(node.children[0]),
      body: traverse(node.children[1])
    )
  when :MATCH
    Node::MATCH.new(
      node: node,
      value: traverse(node.children[0])
    )
  when :MATCH2
    Node::MATCH2.new(
      node: node,
      regexp: traverse(node.children[0]),
      string: traverse(node.children[1])
    )
  when :MATCH3
    Node::MATCH3.new(
      node: node,
      regexp: traverse(node.children[0]),
      string: traverse(node.children[1])
    )
  when :XSTR
    Node::XSTR.new(
      node: node,
      value: node.children[0]
    )
  when :RETURN
    Node::RETURN.new(node: node)
  when :VALUES
    Node::VALUES.new(
      node: node,
      elements: elements(node)
    )
  when :YIELD
    Node::YIELD.new(node: node)
  when :DASGN
    Node::DASGN.new(
      node: node,
      name: node.children[0],
      value: traverse(node.children[1])
    )
  when :DASGN_CURR
    Node::DASGN_CURR.new(
      node: node,
      name: node.children[0],
      value: traverse(node.children[1])
    )
  when :OP_ASGN1
    Node::OP_ASGN1.new(
      node: node,
      receiver: traverse(node.children[0]),
      operator: node.children[1],
      index: traverse(node.children[2]),
      value: traverse(node.children[3])
    )
  when :OP_ASGN2
    # p node.children
  when :OP_ASGN_AND
    Node::OP_ASGN_AND.new(
      node: node,
      variable: traverse(node.children[0]),
      value: traverse(node.children[2])
    )
  when :OP_ASGN_OR
    Node::OP_ASGN_OR.new(
      node: node,
      variable: traverse(node.children[0]),
      value: traverse(node.children[2])
    )
  when :OP_CDECL
    Node::OP_CDECL.new(
      node: node,
      namespace: traverse(node.children[0]),
      operator: node.children[1],
      value: traverse(node.children[2])
    )
  when :DXSTR
    Node::DXSTR.new(
      node: node,
      pre_str: node.children[0],
      interpolation: traverse(node.children[1]),
      tail_str: traverse(node.children[2])
    )
  when :ONCE
    Node::ONCE.new(
      node: node,
      body: traverse(node.children[0])
    )
  when :DREGX
    Node::DREGX.new(
      node: node,
      interpolation: traverse(node.children[0]),
      tail_str: traverse(node.children[1])
    )
  when :UNDEF
    Node::UNDEF.new(
      node: node,
      old_name: traverse(node.children[0])
    )
  when :VALIAS
    Node::VALIAS.new(
      node: node,
      new_name: node.children[0],
      old_name: node.children[1]
    )
  when :ALIAS
    Node::VALIAS.new(
      node: node,
      new_name: traverse(node.children[0]),
      old_name: traverse(node.children[1])
    )
  when :DEFS
    Node::DEFS.new(
      node: node,
      receiver: traverse(node.children[0]),
      name: node.children[1],
      body: traverse(node.children[2])
    )
  when :SPLAT
    Node::SPLAT.new(
      node: node,
      element: traverse(node.children[0])
    )
  when :ARGSPUSH
    Node::ARGSPUSH.new(
      node: node,
      array: traverse(node.children[0]),
      element: traverse(node.children[1])
    )
  else
    raise "Unexpected node type[#{node.type}]"
  end
end

Class: AbstractSyntaxTreeKit

Defined Under Namespace

Constant Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.parse(src) ⇒ Object

.walk(node) {|node| ... } ⇒ Object

Instance Method Details

#traverse(node) ⇒ Object

.parse(src) ⇒ `Object`

.walk(node) {|node| ... } ⇒ `Object`

#traverse(node) ⇒ `Object`