module NTimeLine

  # Category of timeline.
  class Category
    attr_reader :display_name, :sub_categories

    def initialize(elt)
      @display_name = elt.elements["dispaly_name"].to_s
      @sub_categories = []
      elt.each_element("sub_categories/category") do |category|
        @sub_categories << Category.new(category)
      end
    end
  end

  # CategoryResponse is a list of categories.
  class CategoryResponse < Array
    def initialize(doc)
      super
      doc.root.each_element("/response/result/categories/category") do |category|
        self << Category.from_xml(category)
      end
    end
  end
end