From 7e86e10be8c0624233a2d27d1729fe6a533a0f5b Mon Sep 17 00:00:00 2001
From: Takeshi KOMIYA <i.tkomiya@gmail.com>
Date: Thu, 12 Sep 2024 02:32:30 +0900
Subject: [PATCH] stdlib: Update signature of CSV.read

* It takes an IO object as `path` argument
* It returns CSV::Table object when headers option given

refs:

* https://docs.ruby-lang.org/en/3.3/CSV.html#method-c-read
* https://docs.ruby-lang.org/en/3.3/CSV.html#class-CSV-label-Option+headers
---
 stdlib/csv/0/csv.rbs    |  3 ++-
 test/stdlib/CSV_test.rb | 28 ++++++++++++++++++++++++++++
 2 files changed, 30 insertions(+), 1 deletion(-)

diff --git a/stdlib/csv/0/csv.rbs b/stdlib/csv/0/csv.rbs
index 63ede413e..59899dd12 100644
--- a/stdlib/csv/0/csv.rbs
+++ b/stdlib/csv/0/csv.rbs
@@ -2031,7 +2031,8 @@ class CSV < Object
   #     File.write(path, string)
   #     CSV.read(path, headers: true) # => #<CSV::Table mode:col_or_row row_count:4>
   #
-  def self.read: (String path, ?::Hash[Symbol, untyped] options) -> ::Array[::Array[String?]]
+  def self.read: (String | IO path, headers: true | :first_row | Array[untyped] | String, **untyped options) -> ::CSV::Table[CSV::Row]
+               | (String | IO path, ?::Hash[Symbol, untyped] options) -> ::Array[::Array[String?]]
 
   # <!--
   #   rdoc-file=lib/csv.rb
diff --git a/test/stdlib/CSV_test.rb b/test/stdlib/CSV_test.rb
index 39246df49..cb428e3dc 100644
--- a/test/stdlib/CSV_test.rb
+++ b/test/stdlib/CSV_test.rb
@@ -35,4 +35,32 @@ def test_foreach
     assert_send_type "(String path, headers: bool, **untyped) -> Enumerator[CSV::Row, void]",
                      CSV, :foreach, path, headers: true, encoding: 'UTF-8'
   end
+
+  def test_read
+    tmpdir = Dir.mktmpdir
+    path = File.join(tmpdir, "example.csv")
+    File.write(path, "a,b,c\n1,2,3\n")
+
+    assert_send_type "(String path, headers: true) -> CSV::Table[CSV::Row]",
+                     CSV, :read, path, headers: true
+    assert_send_type "(IO path, headers: true) -> CSV::Table[CSV::Row]",
+                     CSV, :read, File.open(path), headers: true
+    assert_send_type "(String path, headers: :first_row) -> CSV::Table[CSV::Row]",
+                     CSV, :read, path, headers: :first_row
+    assert_send_type "(IO path, headers: :first_row) -> CSV::Table[CSV::Row]",
+                     CSV, :read, File.open(path), headers: :first_row
+    assert_send_type "(String path, headers: Array[String]) -> CSV::Table[CSV::Row]",
+                     CSV, :read, path, headers: %w[foo bar baz]
+    assert_send_type "(IO path, headers: Array[String]) -> CSV::Table[CSV::Row]",
+                     CSV, :read, File.open(path), headers: %w[foo bar baz]
+    assert_send_type "(String path, headers: String) -> CSV::Table[CSV::Row]",
+                     CSV, :read, path, headers: "foo,bar,baz"
+    assert_send_type "(IO path, headers: String) -> CSV::Table[CSV::Row]",
+                     CSV, :read, File.open(path), headers: "foo,bar,baz"
+
+    assert_send_type "(String path) -> Array[Array[String?]]",
+                     CSV, :read, path
+    assert_send_type "(IO path) -> Array[Array[String?]]",
+                     CSV, :read, File.open(path)
+  end
 end