From 70e96c9f5c1bfcbb056c34fe7afe3d39226d1cc6 Mon Sep 17 00:00:00 2001
From: Jamis Buck <jamis@37signals.com>
Date: Wed, 31 Aug 2005 10:36:35 +0000
Subject: Add Time.days_in_month, and make Time#next_month work when invoked on
 the 31st of a month

git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@2082 5ecf4fe2-1ee6-0310-87b1-e25e094e27de
---
 .../active_support/core_ext/time/calculations.rb   | 33 ++++++++++++++++++----
 1 file changed, 28 insertions(+), 5 deletions(-)

(limited to 'activesupport/lib/active_support/core_ext/time')

diff --git a/activesupport/lib/active_support/core_ext/time/calculations.rb b/activesupport/lib/active_support/core_ext/time/calculations.rb
index 503717f4fe..76d6d4c176 100644
--- a/activesupport/lib/active_support/core_ext/time/calculations.rb
+++ b/activesupport/lib/active_support/core_ext/time/calculations.rb
@@ -3,6 +3,23 @@ module ActiveSupport #:nodoc:
     module Time #:nodoc:
       # Enables the use of time calculations within Time itself
       module Calculations
+        def self.append_features(base) #:nodoc:
+          super
+          base.extend(ClassMethods)
+        end
+
+        module ClassMethods
+          def days_in_month(month, year=nil)
+            if month == 2
+              (year % 4 == 0) && ((year % 100 != 0) || (year % 400 == 0)) ?  29 : 28
+            elsif month <= 7
+              month % 2 == 0 ? 30 : 31
+            else
+              month % 2 == 0 ? 31 : 30
+            end
+          end
+        end
+
         # Seconds since midnight: Time.now.seconds_since_midnight
         def seconds_since_midnight
           self.hour.hours + self.min.minutes + self.sec + (self.usec/1.0e+6)
@@ -47,12 +64,18 @@ module ActiveSupport #:nodoc:
         end
 
         def months_since(months)
-          if months + self.month > 12
-            old_time = self
-            change(:year => self.year + 1, :month => 1).months_since(months + old_time.month - 12 - 1)
-          else
-            change(:year => self.year, :month => self.month + months)
+          year, month, mday = self.year, self.month, self.mday
+
+          month += months
+          while month > 12
+            month -= 12
+            year += 1
           end
+
+          max = ::Time.days_in_month(month, year)
+          mday = max if mday > max
+
+          change(:year => year, :month => month, :mday => mday)
         end
 
         # Returns a new Time representing the time a number of specified years ago
-- 
cgit v1.2.3