From 1f5e9c8f84ee100201c899257dd0521573ed4eed Mon Sep 17 00:00:00 2001
From: Dorian Villet <gnutix@gmail.com>
Date: Sat, 23 Mar 2024 09:49:35 +0100
Subject: [PATCH] YearMonth: deprecate int argument for month, add support for
 Month enum.

---
 src/YearMonth.php       | 37 +++++++++++++++++++++++++++----------
 tests/YearMonthTest.php |  4 ++++
 2 files changed, 31 insertions(+), 10 deletions(-)

diff --git a/src/YearMonth.php b/src/YearMonth.php
index b30ff4a..0c74c02 100644
--- a/src/YearMonth.php
+++ b/src/YearMonth.php
@@ -12,8 +12,11 @@
 use JsonSerializable;
 use Stringable;
 
+use function is_int;
 use function str_pad;
+use function trigger_error;
 
+use const E_USER_DEPRECATED;
 use const STR_PAD_LEFT;
 
 /**
@@ -44,17 +47,25 @@ private function __construct(int $year, int $month)
     /**
      * Obtains an instance of `YearMonth` from a year and month.
      *
-     * @param int $year  The year, from MIN_YEAR to MAX_YEAR.
-     * @param int $month The month-of-year, from 1 (January) to 12 (December).
+     * @param int $year The year, from MIN_YEAR to MAX_YEAR.
      *
      * @throws DateTimeException
      */
-    public static function of(int $year, int $month): YearMonth
+    public static function of(int $year, int|Month $month): YearMonth
     {
         Field\Year::check($year);
-        Field\MonthOfYear::check($month);
 
-        return new YearMonth($year, $month);
+        if (is_int($month)) {
+            // usually we don't use trigger_error() for deprecations, but we can't rely on @deprecated for a parameter type change;
+            // maybe we should revisit using trigger_error() unconditionally for deprecations in the future.
+            trigger_error('Passing an integer to YearMonth::of() second argument is deprecated, pass a Month instance instead.', E_USER_DEPRECATED);
+
+            Field\MonthOfYear::check($month);
+
+            return new YearMonth($year, $month);
+        }
+
+        return new YearMonth($year, $month->value);
     }
 
     /**
@@ -209,17 +220,23 @@ public function withYear(int $year): YearMonth
 
     /**
      * Returns a copy of this YearMonth with the month-of-year altered.
-     *
-     * @throws DateTimeException If the month-of-year is not valid.
      */
-    public function withMonth(int $month): YearMonth
+    public function withMonth(int|Month $month): YearMonth
     {
+        if (is_int($month)) {
+            // usually we don't use trigger_error() for deprecations, but we can't rely on @deprecated for a parameter type change;
+            // maybe we should revisit using trigger_error() unconditionally for deprecations in the future.
+            trigger_error('Passing an integer to YearMonth::withMonth() is deprecated, pass a Month instance instead.', E_USER_DEPRECATED);
+
+            Field\MonthOfYear::check($month);
+        } else {
+            $month = $month->value;
+        }
+
         if ($month === $this->month) {
             return $this;
         }
 
-        Field\MonthOfYear::check($month);
-
         return new YearMonth($this->year, $month);
     }
 
diff --git a/tests/YearMonthTest.php b/tests/YearMonthTest.php
index b0ed78e..eb93a9f 100644
--- a/tests/YearMonthTest.php
+++ b/tests/YearMonthTest.php
@@ -7,6 +7,7 @@
 use Brick\DateTime\Clock\FixedClock;
 use Brick\DateTime\DateTimeException;
 use Brick\DateTime\Instant;
+use Brick\DateTime\Month;
 use Brick\DateTime\TimeZone;
 use Brick\DateTime\YearMonth;
 
@@ -22,6 +23,7 @@ class YearMonthTest extends AbstractTestCase
     public function testOf(): void
     {
         self::assertYearMonthIs(2007, 7, YearMonth::of(2007, 7));
+        self::assertYearMonthIs(2007, 7, YearMonth::of(2007, Month::JULY));
     }
 
     /**
@@ -319,11 +321,13 @@ public function testWithYearWithSameYear(): void
     public function testWithMonth(): void
     {
         self::assertYearMonthIs(2000, 12, YearMonth::of(2000, 1)->withMonth(12));
+        self::assertYearMonthIs(2000, 12, YearMonth::of(2000, 1)->withMonth(Month::DECEMBER));
     }
 
     public function testWithMonthWithSameMonth(): void
     {
         self::assertYearMonthIs(2000, 2, YearMonth::of(2000, 2)->withMonth(2));
+        self::assertYearMonthIs(2000, 2, YearMonth::of(2000, 2)->withMonth(Month::FEBRUARY));
     }
 
     public function testGetFirstDay(): void