ic_agent 0.1.1

data/lib/ic_agent/ast/nodes/named_nodes.rb

@@ -0,0 +1,410 @@
+require 'treetop'
+
+module IcAgent
+  module Ast
+    module Nodes
+      class NamedNode < Treetop::Runtime::SyntaxNode
+        def title
+          :named_node
+        end
+
+        def to_array
+          [title] + elements.map(&:to_array)
+        end
+
+        def to_s
+          "#{title.to_s.upcase} #{elements_to_s}"
+        end
+
+        def elements_to_s
+          elements.map(&:to_s).join("\n")
+        end
+      end
+
+      class Instruction < NamedNode
+        def title
+          :instruction
+        end
+
+        def to_s
+          elements_to_s
+        end
+      end
+
+      class Comment < NamedNode
+        def title
+          :comment
+        end
+
+        def to_s
+          "# #{elements[0].to_s}"
+        end
+      end
+
+      class DIDFile < NamedNode
+        def title
+          :did_file
+        end
+
+        def to_s
+          elements_to_s
+        end
+      end
+
+      class TypeDeclaration < NamedNode
+        def title
+          :type_declaration
+        end
+
+        def type_param_name
+          elements[0].text_value
+        end
+
+        def type_param_content
+          elements[1].text_value.gsub("\n", '').gsub(';}', '}')
+        end
+
+        def type_root_opt_code
+          elements[1].opt_code
+        end
+
+        def type_child_items
+          if elements && elements[1] && elements[1].elements && elements[1].elements[0]
+            elements[1].elements[0].elements
+          else
+            []
+          end
+        end
+
+        def type_child_item_keys
+          names = []
+          type_child_items.each do |ele|
+            names << ele.elements[0].text_value.strip
+          end
+          names
+        end
+
+        def type_child_item_values
+          values = []
+          type_child_items.each do |ele|
+
+            # get multi type value
+            replaced_hash = {}
+            modified_str = ele.text_value.gsub("\n", '').gsub(/record\s*{[^{}]*}/) do |match|
+              rad_id = rand(100000..999999)
+              type_name = "record_#{rad_id}"
+              replaced_hash[type_name] = match
+              type_name
+            end
+            modified_str = modified_str.gsub(/variant\s*{[^{}]*}/) do |match|
+              rad_id = rand(100000..999999)
+              type_name = "variant_#{rad_id}"
+              replaced_hash[type_name] = match
+              type_name
+            end
+            replaced_hash.each_key do |key|
+              item_arr = replaced_hash[key].strip.split(';')
+              item_arr.each do |item|
+                multi_item_arr = item.strip.split(':')
+                if multi_item_arr.size > 1
+                  item_value_arr = multi_item_arr[1].strip.split(' ').collect { |v| v.strip.gsub(';', '') }
+                  item_value_arr.delete('{')
+                  item_value_arr.delete('}')
+                  item_value_arr.delete('{}')
+                  values << item_value_arr
+                else
+                  values << []
+                end
+              end
+            end
+
+            # get root type value
+            item_arr = modified_str.strip.split(':')
+            if item_arr.size > 1
+              item_value_arr = item_arr[1].strip.split(' ').collect { |v| v.strip.gsub(';', '') }
+              item_value_arr.delete('{')
+              item_value_arr.delete('}')
+              item_value_arr.delete('{}')
+              item_value_arr -= replaced_hash.keys
+              values << item_value_arr
+            else
+              values << []
+            end
+          end
+          values
+        end
+
+        def type_child_refer_items
+          child_args = type_child_item_values.flatten - IcAgent::Candid::ALL_TYPES
+          child_args.uniq
+        end
+
+        def to_s
+          text_value
+        end
+
+        def to_obj
+          {
+            'type_param_name' => type_param_name,
+            'type_root_opt_code' => type_root_opt_code,
+            'type_child_item_keys' => type_child_item_keys,
+            'type_child_item_values' => type_child_item_values
+          }
+        end
+      end
+
+      class BaseType < NamedNode
+        def title
+          :base_type
+        end
+
+        def to_s
+          elements_to_s
+        end
+      end
+
+      class BaseTypeSingle < NamedNode
+        def title
+          :base_type_single
+        end
+
+        def to_s
+          elements_to_s
+        end
+
+        def opt_code
+          'single'
+        end
+      end
+
+      class BaseTypeRecord < NamedNode
+        def title
+          :base_type_record
+        end
+
+        def to_s
+          elements_to_s
+        end
+
+        def opt_code
+          'record'
+        end
+      end
+
+      class BaseTypeKey < NamedNode
+        def title
+          :base_type_key
+        end
+
+        def to_s
+          elements_to_s
+        end
+      end
+
+      class BaseTypeVariant < NamedNode
+        def title
+          :base_type_variant
+        end
+
+        def to_s
+          elements_to_s
+        end
+
+        def opt_code
+          'variant'
+        end
+      end
+
+      class BaseTypeFunc < NamedNode
+        def title
+          :base_type_func
+        end
+
+        def to_s
+          elements_to_s
+        end
+
+        def opt_code
+          'func'
+        end
+      end
+
+      class BaseTypeOpt < NamedNode
+        def title
+          :base_type_opt
+        end
+
+        def to_s
+          elements_to_s
+        end
+
+        def opt_code
+          'opt'
+        end
+      end
+
+      class BaseTypeVec < NamedNode
+        def title
+          :base_type_vec
+        end
+
+        def to_s
+          elements_to_s
+        end
+
+        def opt_code
+          'vec'
+        end
+      end
+
+      class BaseTypeOther < NamedNode
+        def title
+          :base_type_other
+        end
+
+        def to_s
+          elements_to_s
+        end
+
+        def opt_code
+          text_value
+        end
+      end
+
+      class BaseTypeContent < NamedNode
+        def title
+          :base_type_content
+        end
+
+        def to_s
+          elements_to_s
+        end
+      end
+
+      class BaseTypeChild < NamedNode
+        def title
+          :base_type_child
+        end
+
+        def to_s
+          elements_to_s
+        end
+      end
+
+      class TypeName < NamedNode
+        def title
+          :type_name
+        end
+
+        def to_s
+          elements_to_s
+        end
+      end
+
+      class Service < NamedNode
+        def title
+          :ic_service
+        end
+
+        def to_s
+          elements_to_s
+        end
+      end
+
+      class IcServiceName < NamedNode
+        def title
+          :ic_service_name
+        end
+
+        def to_s
+          "# #{elements[0].to_s}"
+        end
+      end
+
+      class IcServiceMethods < NamedNode
+        def title
+          :ic_service_methods
+        end
+
+        def value
+          elements.map { |update| update.value }
+        end
+      end
+
+      class IcServiceMethodName < NamedNode
+        def title
+          :ic_service_method_name
+        end
+      end
+
+      class IcServiceItem < NamedNode
+        def title
+          :ic_service_item
+        end
+
+        def to_s
+          elements_to_s
+        end
+
+        def to_obj
+          obj = {}
+          elements.each do |element|
+            obj[element.title.to_s] = element.text_value
+          end
+          obj
+        end
+      end
+
+      class IcServiceParam < NamedNode
+        def title
+          :ic_service_param
+        end
+
+        def to_s
+          "# #{elements[0].to_s}"
+        end
+      end
+
+      class IcServiceName < NamedNode
+        def title
+          :ic_service_name
+        end
+
+        def to_s
+          "# #{elements[0].to_s}"
+        end
+      end
+
+      class IcServiceMethodParams < NamedNode
+        def title
+          :ic_service_method_params
+        end
+
+        def to_s
+          elements_to_s
+        end
+      end
+
+      class IcServiceMethodReturn < NamedNode
+        def title
+          :ic_service_method_return
+        end
+
+        def to_s
+          elements_to_s
+        end
+      end
+
+      class IcServiceMethodQuery < NamedNode
+        def title
+          :ic_service_method_query
+        end
+
+        def to_s
+          elements_to_s
+        end
+      end
+    end
+  end
+end
+

data/lib/ic_agent/ast/nodes/string_literal.rb

@@ -0,0 +1,17 @@
+require 'treetop'
+
+module IcAgent
+  module Ast
+    module Nodes
+      class StringLiteral < Treetop::Runtime::SyntaxNode
+        def to_array
+          self.text_value
+        end
+
+        def to_s
+          self.text_value
+        end
+      end
+    end
+  end
+end

data/lib/ic_agent/ast/parser.rb

@@ -0,0 +1,85 @@
+require 'treetop'
+
+module IcAgent
+  module Ast
+    class Parser
+      attr_accessor :parser, :tree
+
+      def initialize
+        Treetop.load(File.expand_path(File.join(File.dirname(__FILE__), 'did_grammar.treetop')))
+        @parser = DIDGrammarParser.new
+      end
+
+      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)
+
+        @tree = tree
+        tree
+      end
+
+      def clean_tree(root_node)
+        return if root_node.elements.nil?
+
+        root_node.elements.delete_if { |node| node.class.name == 'Treetop::Runtime::SyntaxNode' }
+        root_node.elements.each { |node| self.clean_tree(node) }
+      end
+
+      def ic_service
+        tree.elements.each do |ele|
+          return ele if ele.title == :ic_service
+        end
+        nil
+      end
+
+      def ic_types
+        type_arr = []
+        tree.elements.each do |ele|
+          type_arr << ele if ele.title == :type_declaration
+        end
+        type_arr
+      end
+
+      def ic_types_obj
+        obj_arr = []
+        ic_types.each do |ic_type|
+          obj_arr << ic_type.to_obj
+        end
+        obj_arr
+      end
+
+      def ic_service_methods
+        ic_service_tree = ic_service
+        unless ic_service_tree.empty?
+          ic_service_tree.elements.each do |ele|
+            return ele if ele.title == :ic_service_methods
+          end
+        end
+        nil
+      end
+
+      def ic_type_name(ic_type)
+        ic_type.type_param_name
+      end
+
+      def ic_type_names
+        names_arr = []
+        ic_types.each do |ic_type|
+          names_arr << ic_type_name(ic_type)
+        end
+        names_arr
+      end
+
+      def ic_type_by_name(type_name)
+        ic_types.each do |ic_type|
+          return ic_type if type_name == ic_type_name(ic_type)
+        end
+        nil
+      end
+    end
+  end
+end

data/lib/ic_agent/ast/writer.rb

@@ -0,0 +1,19 @@
+module IcAgent
+  module Ast
+    class Writer
+      def initialize(tree)
+        @tree = tree
+      end
+
+      def write(return_type = :string)
+        if return_type == :tree
+          @tree
+        elsif return_type == :array
+          @tree.to_array
+        else
+          @tree.to_s
+        end
+      end
+    end
+  end
+end