Contents

module Rubrowser
  class Tree
    attr_reader :name, :children, :occurences, :parent

    def self.from_parsers(parsers)
      return Tree.new if parsers.empty?

      definitions = parsers.map(&:definitions).reduce(:+).uniq
      occurences = parsers.map(&:occurences).reduce(:+).uniq
      Tree.new.tap do |tree|
        definitions.each { |definition| tree.add_child(definition) }
        occurences.each { |occurence| tree.add_occurence(*occurence.first) }
      end
    end

    def initialize(name = 'Kernel', parent = nil)
      @name = name
      @parent = parent
      @children = Set.new
      @occurences = Set.new
    end

    def root?
      @parent.nil?
    end

    def id
      return name if root? || parent.root?
      "#{parent.id}::#{name}".to_sym
    end

    def add_child(child_name_path = [])
      return if child_name_path.empty?
      child = get_or_create_child(child_name_path[0])
      children.add(child)
      child.add_child(child_name_path[1..-1])
    end

    def add_occurence(user, constants)
      user_node = find_node(user)
      occured_node = constants.map { |constant| find_node(constant) }.compact.first
      return unless user_node && occured_node
      user_node.occurences << occured_node
    end

    def find_node(path)
      return self if path.empty?
      child = children.find { |c| c.name == path.first }
      return unless child
      child.find_node(path[1..-1])
    end

    def get_or_create_child(child_name)
      children.find { |child| child_name == child.name } || Tree.new(child_name, self)
    end

    def eq?(other)
      other.name == name
    end

    def to_h
      {
        id: id,
        name: name
      }.tap do |hash|
        hash[:children] = children.map(&:to_h) unless children.empty?
        hash[:occurences] = occurences.map(&:id) unless occurences.empty?
      end
    end

    private

    attr_writer :occurences
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
rubrowser-0.1.5	lib/tree.rb
rubrowser-0.1.4	lib/tree.rb
rubrowser-0.1.3	lib/tree.rb
rubrowser-0.1.2	lib/tree.rb
rubrowser-0.1.1	lib/tree.rb
rubrowser-0.1.0	lib/tree.rb