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

@@ -7,16 +7,18 @@
     attr_accessor :left
 
     attr_accessor :highlight
 
     attr_accessor :title
+    attr_accessor :description
 
     attr_accessor :show_refresh
 
     attr_accessor :interval
     attr_accessor :content_block
     attr_accessor :selection_blocks
+    attr_accessor :selection_blocks_description
 
     class ContentBuilder
       attr_accessor :contents
 
       def initialize(contents)
@@ -68,15 +70,26 @@
 
       @highlight = false
       @interval = 60 * 5 
 
       @selection_blocks = {}
+      @selection_blocks_description = {}
     end
 
     def content(&block)
       self.content_block = block
     end
-    def selection(input=10, &block)
+    def selection(input=10, description=nil, &block)
+      if input.to_s.downcase == "enter"
+        input = 10
+      end
+
+      description_input = input
+      if input.to_s == "10"
+        input = "<Enter>"
+      end
+
       self.selection_blocks[input] = block
+      self.selection_blocks_description[input] = description
     end
   end
 end