Improve data structure

2 files changed, 21 insertions, 21 deletions

diff --git a/lib/carlosgoce/calendar.rb b/lib/carlosgoce/calendar.rb
index 6a5d8f6..1379e48 100644
--- a/lib/carlosgoce/calendar.rb
+++ b/lib/carlosgoce/calendar.rb
@@ -12,29 +12,30 @@ module CarlosGoce
 
     def to_h
       Hash.new.tap {|h|
-        (1..12).each do |month|
-          h[month] = Hash.new.tap {|m|
-            m[:days] = (1..Time.days_in_month(month, @year)).to_a
-            m[:name] = I18n.t('date.month_names')[month].downcase
+        h[:months] = Hash.new.tap do |h|
+          (1..12).each do |month|
+            h[month] = Hash.new.tap {|m|
+              m[:days] = (1..Time.days_in_month(month, @year)).to_a
+              m[:name] = I18n.t('date.month_names')[month].downcase
 
-            # todo: Need improvement. Not too performant...
-            # todo: Maby it should be moved to it's own class too to keep things simple
-            m[:days_names] = Array.new.tap {|a|
-              m[:days].each do |d|
-                t = Time.new @year, month, d
-                a << I18n.t('date.day_names')[t.wday].downcase
-              end
+              # todo: Need improvement. Not too performant...
+              # todo: Maby it should be moved to it's own class too to keep things simple
+              m[:days_names] = Array.new.tap {|a|
+                m[:days].each do |d|
+                  t = Time.new @year, month, d
+                  a << I18n.t('date.day_names')[t.wday].downcase
+                end
+              }
             }
-          }
-        end
 
-        h[:formatted_days] = Array.new.tap {|a|
-          # todo Why I need to use 1?
-          days_before_week_start = Date.new(@year, 1, h[1][:days].first).wday
-          empty_days = [''] * (days_before_week_start - 1)
+            h[month][:formatted_days] = Array.new.tap {|a|
+              days_before_week_start = Date.new(@year, 1, h[1][:days].first).wday
+              empty_days = [''] * (days_before_week_start - 1)
 
-          a << (empty_days + h[1][:days])
-        }
+              a << (empty_days + h[1][:days])
+            }
+          end
+        end
       }
     end
 
diff --git a/lib/carlosgoce/layout/layouts.rb b/lib/carlosgoce/layout/layouts.rb
index 2382573..65da961 100644
--- a/lib/carlosgoce/layout/layouts.rb
+++ b/lib/carlosgoce/layout/layouts.rb
@@ -7,12 +7,11 @@ module CarlosGoce
     class Simple
       def create (file, data, year)
         tables = []
-
         Prawn::Document.generate(file) do
           font_size = 16
 
           data.each do |k, month|
-            days = data[k][:days]
+            days = data[k][:formatted_days]
             cells = (days).to_a.each_slice(7).to_a
             cells.unshift [data[k][:name]]