From 200b36e2aac1cf0f3f9e7ab511489f6fe585d062 Mon Sep 17 00:00:00 2001
From: ksss <co000ri@gmail.com>
Date: Fri, 28 Mar 2025 17:26:01 +0900
Subject: [PATCH] Support pattern argument for Enumerable#{all,any,none,one}?

---
 core/enumerable.rbs            |  9 +++++++++
 test/stdlib/Enumerable_test.rb | 24 ++++++++++++++++++++++++
 2 files changed, 33 insertions(+)

diff --git a/core/enumerable.rbs b/core/enumerable.rbs
index e0368f5c6..aab7bcf72 100644
--- a/core/enumerable.rbs
+++ b/core/enumerable.rbs
@@ -192,6 +192,11 @@
 #     find exactly one of each Hash entry.
 #
 module Enumerable[unchecked out Elem] : _Each[Elem]
+  %a{private}
+  interface _Pattern
+    def ===: (untyped) -> bool
+  end
+
   # <!--
   #   rdoc-file=enum.c
   #   - all?                  -> true or false
@@ -234,6 +239,7 @@ module Enumerable[unchecked out Elem] : _Each[Elem]
   # Related: #any?, #none? #one?.
   #
   def all?: () -> bool
+          | (_Pattern) -> bool
           | () { (Elem) -> boolish } -> bool
 
   # <!--
@@ -277,6 +283,7 @@ module Enumerable[unchecked out Elem] : _Each[Elem]
   # Related: #all?, #none?, #one?.
   #
   def any?: () -> bool
+          | (_Pattern) -> bool
           | () { (Elem) -> boolish } -> bool
 
   # <!--
@@ -1167,6 +1174,7 @@ module Enumerable[unchecked out Elem] : _Each[Elem]
   # Related: #one?, #all?, #any?.
   #
   def none?: () -> bool
+           | (_Pattern) -> bool
            | () { (Elem) -> boolish } -> bool
 
   # <!--
@@ -1210,6 +1218,7 @@ module Enumerable[unchecked out Elem] : _Each[Elem]
   # Related: #none?, #all?, #any?.
   #
   def one?: () -> bool
+          | (_Pattern) -> bool
           | () { (Elem) -> boolish } -> bool
 
   # <!--
diff --git a/test/stdlib/Enumerable_test.rb b/test/stdlib/Enumerable_test.rb
index 73f258b1d..7c36fff90 100644
--- a/test/stdlib/Enumerable_test.rb
+++ b/test/stdlib/Enumerable_test.rb
@@ -241,4 +241,28 @@ def test_first
     assert_send_type '(ToInt n) -> ::Array[::String]' , TestEnumerable.new, :first, ToInt.new(42)
     assert_send_type '(ToInt n) -> ::Array[::String]' , TestEmptyEnumerable.new, :first, ToInt.new(42)
   end
+
+  def test_all_p
+    assert_send_type '() -> bool', TestEnumerable.new, :all?
+    assert_send_type '(Class) -> bool', TestEnumerable.new, :all?, String
+    assert_send_type '() { (String) -> bool } -> bool', TestEnumerable.new, :all? do |x| x == '0' end
+  end
+
+  def test_any_p
+    assert_send_type '() -> bool', TestEnumerable.new, :any?
+    assert_send_type '(Class) -> bool', TestEnumerable.new, :any?, String
+    assert_send_type '() { (String) -> bool } -> bool', TestEnumerable.new, :any? do |x| x == '0' end
+  end
+
+  def test_none_p
+    assert_send_type '() -> bool', TestEnumerable.new, :none?
+    assert_send_type '(Class) -> bool', TestEnumerable.new, :none?, String
+    assert_send_type '() { (String) -> bool } -> bool', TestEnumerable.new, :none? do |x| x == '0' end
+  end
+
+  def test_one_p
+    assert_send_type '() -> bool', TestEnumerable.new, :one?
+    assert_send_type '(Class) -> bool', TestEnumerable.new, :one?, String
+    assert_send_type '() { (String) -> bool } -> bool', TestEnumerable.new, :one? do |x| x == '0' end
+  end
 end