Some cleanup of code

1 files changed, 23 insertions, 3 deletions

diff --git a/help/app/models/ticket_selection.rb b/help/app/models/ticket_selection.rb
index 77720f7..cb20f5b 100644
--- a/help/app/models/ticket_selection.rb
+++ b/help/app/models/ticket_selection.rb
@@ -2,18 +2,33 @@ class TicketSelection
 
   def initialize(options = {})
     @options = options
+    @options[:open_status] ||= 'open'
+    @options[:sort_order] ||= 'updated_at_desc'
+
+  end
+
+  def tickets
+    #TODO: can this be more succinct?
+    if order
+      Ticket.send(finder_method).startkey(startkey).endkey(endkey).send(order)
+    else
+      Ticket.send(finder_method).startkey(startkey).endkey(endkey)
+    end
   end
 
+  protected
+
+
   def finder_method
     method = 'by_'
-    method += 'includes_post_by_and_' if !@options[:is_admin] or (@options[:admin_status] == 'mine')
+    method += 'includes_post_by_and_' if only_mine?
     method += 'is_open_and_' if @options[:open_status] != 'all'
     method += @options[:sort_order].sub(/_(de|a)sc$/, '')
   end
 
   def startkey
     startkeys = []
-    startkeys << @options[:user_id] if !@options[:is_admin] or (@options[:admin_status] == 'mine')
+    startkeys << @options[:user_id] if only_mine?
     startkeys << (@options[:open_status] == 'open') if @options[:open_status] != 'all'
     startkeys << 0
     startkeys = startkeys.join if startkeys.length == 1 #want string not array if just one thing in array
@@ -21,7 +36,7 @@ class TicketSelection
   end
 
   def endkey
-    endtime = Time.now + 2.days #TODO
+    endtime = Time.now + 2.days #TODO. this obviously isn't ideal
     if self.startkey.is_a?(Array)
       endkeys = self.startkey
       endkeys.pop
@@ -35,4 +50,9 @@ class TicketSelection
     'descending' if @options[:sort_order].end_with? 'desc'
   end
 
+
+  def only_mine?
+    !@options[:is_admin] or (@options[:admin_status] == 'mine')
+  end
+
 end