handle arrays of primitive data types in response

Author: khiav223577

lib/grape-swagger/endpoint.rb

@@ -323,13 +323,15 @@ def build_response_for_type_parameter(memo, _route, value, _options)
     end
 
     def build_response_schema(value)
-      type, format = prepare_type_and_format(value)
+      return { value[:as] => build_response_schema(value.except(:as)) } if value.include?(:as)
 
-      if value.include?(:as)
-        return { value[:as] => { type: type, format: format }.compact }
-      else
-        return { type: type }
+      if value[:type].is_a?(Array)
+        items = build_response_schema({ **value, type: value[:type].first })
+        return { type: 'array', items: items }
       end
+
+      type, format = prepare_type_and_format(value)
+      { type: type, format: format }.compact
     end
 
     def prepare_type_and_format(value)

spec/swagger_v2/api_swagger_v2_response_with_models_and_primitive_types_spec.rb

@@ -16,7 +16,12 @@ class ResponseApiModelsAndPrimitiveTypes < Grape::API
                { model: Entities::UseResponse, as: :user_response },
                { type: 'String', as: :string_response },
                { type: 'Float', as: :float_response },
-               { type: 'Hash', as: :hash_response }
+               { type: 'Hash', as: :hash_response },
+               { type: Array[Integer], as: :array_of_integer_response },
+               { type: Array[String], as: :array_of_string_response },
+               { type: Array[Float], as: :array_of_float_response },
+               { type: Array[Hash], as: :array_of_hash_response },
+               { type: Array[Array[Float]], as: :array_of_array_of_float_response }
              ],
              failure: [
                { code: 400, message: 'NotFound', model: '' },
@@ -56,7 +61,43 @@ def app
                 'integer_response' => { 'type' => 'integer', 'format' => 'int32' },
                 'string_response' => { 'type' => 'string' },
                 'float_response' => { 'type' => 'number', 'format' => 'float' },
-                'hash_response' => { 'type' => 'object' }
+                'hash_response' => { 'type' => 'object' },
+                'array_of_integer_response' => {
+                  'type' => 'array',
+                  'items' => {
+                    'type' => 'integer',
+                    'format' => 'int32'
+                  }
+                },
+                'array_of_string_response' => {
+                  'type' => 'array',
+                  'items' => {
+                    'type' => 'string'
+                  }
+                },
+                'array_of_float_response' => {
+                  'type' => 'array',
+                  'items' => {
+                    'type' => 'number',
+                    'format' => 'float'
+                  }
+                },
+                'array_of_hash_response' => {
+                  'type' => 'array',
+                  'items' => {
+                    'type' => 'object'
+                  }
+                },
+                'array_of_array_of_float_response' => {
+                  'type' => 'array',
+                  'items' => {
+                    'type' => 'array',
+                    'items' => {
+                      'type' => 'number',
+                      'format' => 'float'
+                    }
+                  }
+                }
               }
             }
           },