From c6296c023231fe3489c10de585219db312c13605 Mon Sep 17 00:00:00 2001
From: ksss <co000ri@gmail.com>
Date: Tue, 25 Jun 2024 14:20:16 +0900
Subject: [PATCH 1/2] Fix double-splat

---
 Gemfile.lock                        | 2 +-
 Rakefile                            | 2 +-
 core/complex.rbs                    | 5 ++++-
 core/float.rbs                      | 6 ++++--
 core/integer.rbs                    | 6 +++---
 core/rational.rbs                   | 6 ++++--
 stdlib/bigdecimal/0/big_decimal.rbs | 5 ++++-
 test/raap.txt                       | 5 +++++
 8 files changed, 26 insertions(+), 11 deletions(-)

diff --git a/Gemfile.lock b/Gemfile.lock
index 031ca8ba0..d9a4ce0d6 100644
--- a/Gemfile.lock
+++ b/Gemfile.lock
@@ -71,7 +71,7 @@ GEM
     psych (4.0.6)
       stringio
     public_suffix (6.0.0)
-    raap (0.8.0)
+    raap (0.10.0)
       rbs (~> 3.0)
       timeout (~> 0.4)
     racc (1.8.0)
diff --git a/Rakefile b/Rakefile
index f4561e93e..a6f6c31f3 100644
--- a/Rakefile
+++ b/Rakefile
@@ -106,7 +106,7 @@ task :typecheck_test => :compile do
 end
 
 task :raap => :compile do
-  sh %q[cat test/raap.txt | egrep -v '^#|^$' | xargs bundle exec raap]
+  sh %q[cat test/raap.txt | egrep -v '^#|^$' | xargs bundle exec raap --require bigdecimal/util --library bigdecimal]
 end
 
 task :rubocop do
diff --git a/core/complex.rbs b/core/complex.rbs
index b251a051b..dd3885465 100644
--- a/core/complex.rbs
+++ b/core/complex.rbs
@@ -134,7 +134,10 @@ class Complex < Numeric
   #     Complex('i') ** 2             # => (-1+0i)
   #     Complex(-8) ** Rational(1, 3) # => (1.0000000000000002+1.7320508075688772i)
   #
-  def **: (Numeric) -> Complex
+  def **: (Integer) -> Complex
+        | (Float) -> Complex
+        | (Rational) -> Complex
+        | (Complex) -> Complex
 
   # <!--
   #   rdoc-file=complex.c
diff --git a/core/float.rbs b/core/float.rbs
index 005ff31aa..617205986 100644
--- a/core/float.rbs
+++ b/core/float.rbs
@@ -142,8 +142,10 @@ class Float < Numeric
   #     f ** Rational(2, 1) # => 9.8596
   #     f ** Complex(2, 0)  # => (9.8596+0i)
   #
-  def **: (Complex) -> Complex
-        | (Numeric) -> Float
+  def **: (Integer) -> Float
+        | (Float) -> (Float | Complex)
+        | (Rational) -> (Float | Complex)
+        | (Complex) -> Complex
 
   # <!--
   #   rdoc-file=numeric.c
diff --git a/core/integer.rbs b/core/integer.rbs
index 5922c979a..9c64d1999 100644
--- a/core/integer.rbs
+++ b/core/integer.rbs
@@ -230,9 +230,9 @@ class Integer < Numeric
   #     2 ** Rational(3, 1) # => (8/1)
   #     2 ** Complex(3, 0)  # => (8+0i)
   #
-  def **: (Integer) -> Numeric
-        | (Float) -> Numeric
-        | (Rational) -> Numeric
+  def **: (Integer) -> (Integer | Rational)
+        | (Float) -> (Float | Complex)
+        | (Rational) -> (Rational | Float | Complex)
         | (Complex) -> Complex
 
   # <!--
diff --git a/core/rational.rbs b/core/rational.rbs
index 3953bd2c2..b7992816c 100644
--- a/core/rational.rbs
+++ b/core/rational.rbs
@@ -84,8 +84,10 @@ class Rational < Numeric
   #     Rational(1, 2) ** 0               #=> (1/1)
   #     Rational(1, 2) ** 0.0             #=> 1.0
   #
-  def **: (Complex) -> Complex
-        | (Numeric) -> Numeric
+  def **: (Integer) -> Rational
+        | (Float) -> (Float | Complex)
+        | (Rational) -> (Float | Rational | Complex)
+        | (Complex) -> (Rational | Complex)
 
   # <!--
   #   rdoc-file=rational.c
diff --git a/stdlib/bigdecimal/0/big_decimal.rbs b/stdlib/bigdecimal/0/big_decimal.rbs
index 5c8c5d58a..b03c83f6b 100644
--- a/stdlib/bigdecimal/0/big_decimal.rbs
+++ b/stdlib/bigdecimal/0/big_decimal.rbs
@@ -441,7 +441,10 @@ class BigDecimal < Numeric
   #
   # Related: BigDecimal#power.
   #
-  def **: (Numeric) -> BigDecimal
+  def **: (Integer) -> BigDecimal
+        | (Float) -> (Integer | BigDecimal)
+        | (Rational) -> (Integer | BigDecimal)
+        | (BigDecimal) -> BigDecimal
 
   # <!--
   #   rdoc-file=ext/bigdecimal/bigdecimal.c
diff --git a/test/raap.txt b/test/raap.txt
index 450803fa2..c43b0ad03 100644
--- a/test/raap.txt
+++ b/test/raap.txt
@@ -4,3 +4,8 @@
 
 Set[Integer]
 Enumerable[Integer]#to_set
+Integer#**
+Float#**
+Rational#**
+Complex#**
+BigDecimal#**

From 7bd6754416f43efa455ddce181f604e26392e871 Mon Sep 17 00:00:00 2001
From: ksss <co000ri@gmail.com>
Date: Tue, 25 Jun 2024 16:29:46 +0900
Subject: [PATCH 2/2] Add overload with BigDecimal

---
 stdlib/bigdecimal/0/big_decimal.rbs | 61 +++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)

diff --git a/stdlib/bigdecimal/0/big_decimal.rbs b/stdlib/bigdecimal/0/big_decimal.rbs
index b03c83f6b..3d5b0a278 100644
--- a/stdlib/bigdecimal/0/big_decimal.rbs
+++ b/stdlib/bigdecimal/0/big_decimal.rbs
@@ -1500,6 +1500,23 @@ class Integer
   #
   def -: (BigDecimal) -> BigDecimal
     | ...
+
+  # <!--
+  #   rdoc-file=numeric.c
+  #   - self ** numeric -> numeric_result
+  # -->
+  # Raises `self` to the power of `numeric`:
+  #
+  #     2 ** 3              # => 8
+  #     2 ** -3             # => (1/8)
+  #     -2 ** 3             # => -8
+  #     -2 ** -3            # => (-1/8)
+  #     2 ** 3.3            # => 9.849155306759329
+  #     2 ** Rational(3, 1) # => (8/1)
+  #     2 ** Complex(3, 0)  # => (8+0i)
+  #
+  def **: (BigDecimal) -> BigDecimal
+        | ...
 end
 
 %a{annotate:rdoc:skip}
@@ -1584,6 +1601,22 @@ class Float
   #
   def -: (BigDecimal) -> BigDecimal
     | ...
+
+  # <!--
+  #   rdoc-file=numeric.c
+  #   - self ** other -> numeric
+  # -->
+  # Raises `self` to the power of `other`:
+  #
+  #     f = 3.14
+  #     f ** 2              # => 9.8596
+  #     f ** -2             # => 0.1014239928597509
+  #     f ** 2.1            # => 11.054834900588839
+  #     f ** Rational(2, 1) # => 9.8596
+  #     f ** Complex(2, 0)  # => (9.8596+0i)
+  #
+  def **: (BigDecimal) -> BigDecimal
+        | ...
 end
 
 %a{annotate:rdoc:skip}
@@ -1687,6 +1720,22 @@ class Rational
   #
   def -: (BigDecimal) -> BigDecimal
     | ...
+
+  # <!--
+  #   rdoc-file=rational.c
+  #   - rat ** numeric  ->  numeric
+  # -->
+  # Performs exponentiation.
+  #
+  #     Rational(2)    ** Rational(3)     #=> (8/1)
+  #     Rational(10)   ** -2              #=> (1/100)
+  #     Rational(10)   ** -2.0            #=> 0.01
+  #     Rational(-4)   ** Rational(1, 2)  #=> (0.0+2.0i)
+  #     Rational(1, 2) ** 0               #=> (1/1)
+  #     Rational(1, 2) ** 0.0             #=> 1.0
+  #
+  def **: (BigDecimal) -> (Rational | BigDecimal)
+        | ...
 end
 
 %a{annotate:rdoc:skip}
@@ -1771,6 +1820,18 @@ class Complex
   #
   def -: (BigDecimal) -> Complex
     | ...
+
+  # <!--
+  #   rdoc-file=complex.c
+  #   - complex ** numeric -> new_complex
+  # -->
+  # Returns `self` raised to power `numeric`:
+  #
+  #     Complex('i') ** 2             # => (-1+0i)
+  #     Complex(-8) ** Rational(1, 3) # => (1.0000000000000002+1.7320508075688772i)
+  #
+  def **: (BigDecimal) -> Complex
+        | ...
 end
 
 %a{annotate:rdoc:skip}