Merge pull request #496 from soutaro/rack-response

Fix rack `response` type
ruby · Jan 9, 2024 · ec140af · ec140af
2 parents 13ae8cd + 1c71abb
commit ec140af
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/gems/rack/2.2/_test/string_body.rb b/gems/rack/2.2/_test/string_body.rb
@@ -0,0 +1,10 @@
+# @type var app: Rack::_Application
+app = _ = nil
+
+middleware = -> (env) do
+  # @type var env: Rack::env
+  # @type block: Rack::response
+  [404, { "FOO" => "BAR" }, "Not found"]
+end
+
+middleware #: Rack::_Application
diff --git a/gems/rack/2.2/rack.rbs b/gems/rack/2.2/rack.rbs
@@ -64,7 +64,16 @@ module Rack
   def self.release: -> String
 
   type env = Hash[String, untyped]
-  type response = [_ToI, _Each[[String, String]], _Each[String]]
+
+  type string_body = _ToStr
+
+  type enumerable_body = _Each[String]
+
+  interface _StreamingBody
+    def call: (untyped stream) -> void
+  end
+
+  type response = [_ToI, _Each[[String, String]], string_body | enumerable_body | _StreamingBody]
 
   interface _Application
     def call: (env) -> response