class IcAgent::Ast::StatementParser

The StatementParser class provides methods to parse data and generate an Abstract Syntax Tree (AST) for nested types based on the specified grammar.

Constants

REFER_TYPE_CLASS
REFER_TYPE_KEYS
TREE_TYPE_CLASS

Attributes

parser[RW]

source_tree[RW]

tree[RW]

Public Class Methods

new() click to toggle source

# File lib/ic_agent/ast/statement_parser.rb, line 25
def initialize
  # Loads the Treetop grammar from the specified file.
  Treetop.load(File.expand_path(File.join(File.dirname(__FILE__), 'nested_type_grammar.treetop')))
  @parser = TypeGrammarParser.new
end

Public Instance Methods

clean_tree(root_node) click to toggle source

Cleans up the AST tree by removing unnecessary nodes.

# File lib/ic_agent/ast/statement_parser.rb, line 53
def clean_tree(root_node)
  return if root_node.elements.nil?

  root_node.elements.delete_if { |node| node.class.name == 'Treetop::Runtime::SyntaxNode' and node.parent.class.name != 'IcAgent::Ast::Nodes::IcBaseTypeValue' and node.parent.class.name != "IcAgent::Ast::Nodes::IcTypeDef" }
  root_node.elements.each { |node| self.clean_tree(node) }
end

gen_source_tree(root_node, tree_root_node = nil, tree_current_node = nil) click to toggle source

Generates the source tree from the AST tree. @param [Object] root_node @param [nil] tree_root_node @param [nil] tree_current_node

# File lib/ic_agent/ast/statement_parser.rb, line 64
def gen_source_tree(root_node, tree_root_node = nil, tree_current_node = nil)
  return if root_node.elements.nil?

  tree_root_node = tree_root_node.nil? ? Tree::TreeNode.new('root', { 'total_child': 0, 'ic_type': nil, 'refer_type': [], 'prototype': root_node.source_content, 'content': root_node.source_content }) : tree_root_node
  tree_current_node = tree_current_node.nil? ? tree_root_node : tree_current_node

  root_node.elements.each do |node|
    if TREE_TYPE_CLASS.include?(node.class.name) && node.source_content != tree_root_node.content[:prototype]

      id = tree_root_node.content[:total_child] + 1
      new_tree_node = Tree::TreeNode.new("node_#{id}", { 'total_child': 0, 'ic_type': nil, 'prototype': root_node.source_content, 'content': root_node.source_content })
      tree_current_node << new_tree_node
      tree_root_node.content[:total_child] = id

      # set refer_type
      unless Regexp.union(REFER_TYPE_KEYS) === root_node.source_content
        # func type content
        if root_node.source_content.index('->')
          param_arr = []
          temp_param_arr = root_node.source_content.strip.split(' ').collect { |v| v.strip.gsub(';', '') }
          temp_param_arr.delete_if {|v| !v.index('(') && !v.index(')') }
          temp_param_arr.each {|v| param_arr = param_arr + v.sub('(', '').sub(')', '').split(',')}
        else
          param_arr = root_node.source_content.strip.split(' ').collect { |v| v.strip.gsub(';', '') }
          param_arr = param_arr - IcAgent::Candid::ALL_TYPES
        end
        tree_root_node.content[:refer_type] = (tree_root_node.content[:refer_type] + param_arr).uniq
      end

      self.source_tree = tree_root_node
      self.gen_source_tree(node, tree_root_node, new_tree_node)
    else
      self.gen_source_tree(node, tree_root_node, tree_current_node)
    end
  end

  self.source_tree = tree_root_node
end

ic_statement_childs() click to toggle source

Returns the child nodes of the AST statement.

# File lib/ic_agent/ast/statement_parser.rb, line 109
def ic_statement_childs
  if tree.elements[0] && tree.elements[0].elements[0].elements[0]
    tree.elements[0].elements[0].elements[0].elements
  end
end

ic_statement_root() click to toggle source

Returns the root node of the AST statement.

# File lib/ic_agent/ast/statement_parser.rb, line 104
def ic_statement_root
  tree.elements[0]
end

parse(data, return_type = :string) click to toggle source

Parses the input data and generates the Abstract Syntax Tree (AST) for nested types based on the grammar.

Parameters:

data: The input data to be parsed.
return_type: The desired format for the parsed AST (:string by default).

Returns:

The generated AST in the specified format.

# File lib/ic_agent/ast/statement_parser.rb, line 39
def parse(data, return_type = :string)
  tree = @parser.parse(data)
  raise Exception, "Parse error at offset: #{@parser.index} #{@parser.failure_reason}" if tree.nil?

  # this edits the tree in place
  clean_tree(tree)

  # generate soure tree
  gen_source_tree(tree)
  @tree = tree
  tree
end