lib/cal/ender.rb in cal-0.0.1 vs lib/cal/ender.rb in cal-0.1.0

@@ -1,44 +1,61 @@
 require 'active_support/time'
 require 'active_support/core_ext/array/grouping'
+require 'active_support/core_ext/string/conversions'
 
 module Cal
   class Ender
 
-    def initialize(date)
-      @date = date
+    def initialize(dateable, options = {})
+      options = {:format => :monthly}.merge options
+
+      if options[:format] != :monthly
+        raise ArgumentError, "only supported format currently is :monthly"
+      end
+
+      @date = dateable.to_date
+      @format = options[:format]
     end
 
-    attr_reader :date
+    attr_reader :format, :date
 
     def ==(other)
       other.is_a?(Ender) && other.date == date
     end
 
     def month
-      @mont ||= Month.new self
+      @month ||= Month.new self
     end
 
+    def days
+      @days ||= dates.map { |date| Day.new self, date }
+    end
+
     def weeks
       @weeks ||= days.in_groups_of 7
     end
 
-    def days
-      @days ||= begin
+    def previous
+      self.class.new date.prev_month
+    end
+
+    def next
+      self.class.new date.next_month
+    end
+
+  private
+
+    # TODO: simplify/improve this
+    def dates
+      [].tap do |dates|
         day = date.beginning_of_month.beginning_of_week :sunday
         last_day = date.end_of_month.end_of_week :sunday
 
-        days = []
         while day <= last_day
-          days << Day.new(self, day)
+          dates << day
           day = day.tomorrow
         end
-        days
       end
-    end
-
-    def week_headings
-      @week_headings ||= %w[Sunday Monday Tuesday Wednesday Thursday Friday Saturday]
     end
 
   end
 end
\ No newline at end of file