lib/wassup/pane_builder.rb in wassup-0.1.2 vs lib/wassup/pane_builder.rb in wassup-0.2.0

@@ -8,12 +8,58 @@
 
     attr_accessor :highlight
 
     attr_accessor :title
 
+    attr_accessor :show_refresh
+
     attr_accessor :interval
     attr_accessor :content_block
     attr_accessor :selection_blocks
+
+    class ContentBuilder
+      attr_accessor :contents
+
+      def initialize(contents)
+        @contents = contents
+        @need_to_clear = true
+        @show_refresh = true
+      end
+
+      def clear=(clear)
+        @need_to_clear = clear
+      end
+
+      def add_row(display, object=nil, page:nil)
+        if @need_to_clear
+          @need_to_clear = false
+          self.contents = []
+        end
+
+        content = nil
+
+        # Create contents if none
+        if page.nil?
+          if self.contents.empty?
+            content = Pane::Content.new
+            self.contents << content
+          else
+            content = self.contents.first
+          end
+        elsif page.is_a?(String)
+          content = self.contents.find do |content|
+            content.title == page
+          end
+
+          if content.nil?
+            content = Pane::Content.new(page)
+            self.contents << content
+          end
+        end
+
+        content.add_row(display, object)
+      end
+    end
 
     def initialize()
       @height = 1
       @weight = 1
       @top = 0