WIP: Add support for primitive data types in responses

lib/grape-swagger/endpoint.rb

@@ -213,6 +213,9 @@ def response_object(route, options)
           next
         end
 
+        # Build primitive type response
+        build_primitive_response(memo, route, value, options) if value[:type]
+
         # Explicitly request no model with { model: '' }
         next if value[:model] == ''
 
@@ -284,6 +287,18 @@ def default_code_from_route(route)
       [default_code]
     end
 
+    def build_primitive_response(memo, route, value, options)
+      type = GrapeSwagger::DocMethods::DataType.call(value[:type])
+
+      if memo[value[:code]].include?(:schema) && value.include?(:as)
+        memo[value[:code]][:schema][:properties].merge!(value[:as] => { type: type })
+      elsif value.include?(:as)
+        memo[value[:code]][:schema] = { type: :object, properties: { value[:as] => { type: type } } }  
+      else
+        memo[value[:code]][:schema] = { type: type }
+      end
+    end
+
     def build_memo_schema(memo, route, value, response_model, options)
       if memo[value[:code]][:schema] && value[:as]
         memo[value[:code]][:schema][:properties].merge!(build_reference(route, value, response_model, options))
@@ -464,6 +479,7 @@ def success_code_from_entity(route, entity)
         default_code[:as] = entity[:as] if entity[:as]
         default_code[:is_array] = entity[:is_array] if entity[:is_array]
         default_code[:required] = entity[:required] if entity[:required]
+        default_code[:type] = entity[:type] if entity[:type]
       else
         default_code = GrapeSwagger::DocMethods::StatusCodes.get[route.request_method.downcase.to_sym]
         default_code[:model] = entity if entity

spec/swagger_v2/api_swagger_v2_response_with_primitive_types_spec.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe 'response' do
+  include_context "#{MODEL_PARSER} swagger example"
+
+  before :all do
+    module TheApi
+      class ResponseApiModels < Grape::API
+        format :json
+
+        desc 'This returns something',
+             success: [
+              { model: Entities::UseResponse, as: :user_response },
+              { type: 'Integer', as: :integer_response },
+            ],
+             failure: [
+               { code: 400, message: 'NotFound', model: '' },
+               { code: 404, message: 'BadRequest', model: Entities::ApiError }
+             ],
+             default_response: { message: 'Error', model: Entities::ApiError }
+        get '/use-response' do
+          { 'declared_params' => declared(params) }
+        end
+
+        add_swagger_documentation
+      end
+    end
+  end
+
+  def app
+    TheApi::ResponseApiModels
+  end
+
+  describe 'uses entity as response object implicitly with route name' do
+    subject do
+      get '/swagger_doc/use-response'
+      JSON.parse(last_response.body)
+    end
+
+    specify do
+      expect(subject['paths']['/use-response']['get']).to eql(
+        'description' => 'This returns something',
+        'produces' => ['application/json'],
+        'responses' => {
+          '200' => { 'description' => 'This returns something', 'schema' => { 'type' => 'object', 'properties' => { 'user_response' => { '$ref' => '#/definitions/UseResponse' }, 'integer_response' => {'type' => 'integer'} }}} },
+          '400' => { 'description' => 'NotFound' },
+          '404' => { 'description' => 'BadRequest', 'schema' => { '$ref' => '#/definitions/ApiError' } },
+          'default' => { 'description' => 'Error', 'schema' => { '$ref' => '#/definitions/ApiError' } }
+        },
+        'tags' => ['use-response'],
+        'operationId' => 'getUseResponse'
+      )
+      expect(subject['definitions']).to eql(swagger_entity_as_response_object)
+    end
+  end
+end