module Irrc
  module Subquery
    # Public: Generate a child query to resolve IRR / Whois object recursively.
    #
    # object  - IRR / Whois object to extract. (eg: as-set, route-set, aut-num object)
    def fork(object)
      Query.new(object, source: @sources, protocol: @protocols).tap {|q|
        q.parent = self
      }.tap {|c| self.add_child c }
    end

    # Public: Returns the parent (associated) Query object, which is probably as-set.
    def parent
      @parent
    end

    # Public: Set a parent (associated) Query object, which is probably as-set.
    #
    # parent - Parent Query object.
    def parent=(query)
      @parent = query
    end

    # Public: Returns child Query objects
    def children
      @children ||= []
    end

    # Public: Add a child Query object
    def add_child(query)
      children << query
    end

    # Public: Delete a child Query object
    def delete_child(query)
      children.delete(query)
    end

    # Public: Returns the IRR object to query including those from ancestor query objects.
    #
    # Returns: Array of String.
    def ancestor_objects
      @_ancestor_objects ||= Array(parent && parent.ancestor_objects) << object
    end

    # Public: Returns the root IRR object of the nested query
    #
    # Returns: String.
    def root
      @_root ||= if parent
                   parent.root
                 else
                   self
                 end
    end

    # Public: Returns true if the query is root.
    def root?
      root == self
    end

    # Public: Returns true if object is listed in ancestor IRR objects.
    def ancestor_object?(object)
      ancestor_objects.include?(object)
    end
  end
end