Merge pull request #48 from MaximeRDY/features/add_extension_to_entity

Add Extension for model Entity

Author: kzaitsev

.gitignore

@@ -7,3 +7,5 @@
 /pkg/
 /spec/reports/
 /tmp/
+.ruby-gemset
+.ruby-version

CHANGELOG.md

@@ -3,6 +3,7 @@
 #### Features
 
 * Your contribution here.
+* [#48](https://github.com/ruby-grape/grape-swagger-entity/pull/48): Add support for extensions - [@MaximeRDY](https://github.com/MaximeRDY).
 
 #### Fixes
 

lib/grape-swagger/entity/attribute_parser.rb

@@ -38,6 +38,8 @@ def call(entity_options)
 
           add_attribute_documentation(param, documentation)
 
+          add_extension_documentation(param, documentation)
+
           param
         end
       end
@@ -122,6 +124,10 @@ def add_array_documentation(param, documentation)
         param[:maxItems] = documentation[:max_items] if documentation.key?(:max_items)
         param[:uniqueItems] = documentation[:unique_items] if documentation.key?(:unique_items)
       end
+
+      def add_extension_documentation(param, documentation)
+        GrapeSwagger::DocMethods::Extensions.add_extensions_to_root(documentation, param)
+      end
     end
   end
 end

spec/grape-swagger/entities/response_model_spec.rb

@@ -65,7 +65,7 @@ def app
 
     expect(subject['definitions'].keys).to include 'ThisApi::Entities::Kind'
     expect(subject['definitions']['ThisApi::Entities::Kind']).to eq(
-      'type' => 'object', 'properties' => { 'title' => { 'type' => 'string', 'description' => 'Title of the kind.' } }
+      'type' => 'object', 'properties' => { 'title' => { 'type' => 'string', 'description' => 'Title of the kind.' }, 'content' => { 'description' => 'Content', 'type' => 'string', 'x-some' => 'stuff' } }
     )
 
     expect(subject['definitions'].keys).to include 'ThisApi::Entities::Relation'

spec/grape-swagger/entity/attribute_parser_spec.rb

@@ -60,6 +60,12 @@
 
           it { is_expected.to include(maxLength: 1) }
         end
+
+        context 'when it contains extensions' do
+          let(:entity_options) { { documentation: { type: 'string', desc: 'Colors', x: { some: 'stuff' } } } }
+
+          it { is_expected.to include('x-some' => 'stuff') }
+        end
       end
 
       context 'when it is exposed as an array' do

spec/support/shared_contexts/this_api.rb

@@ -4,6 +4,7 @@ module ThisApi
       module Entities
         class Kind < Grape::Entity
           expose :title, documentation: { type: 'string', desc: 'Title of the kind.' }
+          expose :content, documentation: { type: 'string', desc: 'Content', x: { some: 'stuff' } }
         end
 
         class Relation < Grape::Entity