Method to add multiple fields

Signed-off-by: Jordan Hollinger <[email protected]>

lib/blueprinter/v2/base.rb

@@ -19,11 +19,11 @@ class << self
         # @api private The fully-qualified name, e.g. "MyBlueprint", or "MyBlueprint.foo.bar"
         attr_accessor :blueprint_name
         # @api private
-        attr_accessor :views, :fields, :excludes, :partials, :used_partials, :eval_mutex
+        attr_accessor :views, :schema, :excludes, :partials, :used_partials, :eval_mutex
       end
 
       self.views = ViewBuilder.new(self)
-      self.fields = {}
+      self.schema = {}
       self.excludes = []
       self.partials = {}
       self.used_partials = []
@@ -35,7 +35,7 @@ class << self
       # Initialize subclass
       def self.inherited(subclass)
         subclass.views = ViewBuilder.new(subclass)
-        subclass.fields = fields.transform_values(&:dup)
+        subclass.schema = schema.transform_values(&:dup)
         subclass.excludes = []
         subclass.partials = partials.dup
         subclass.used_partials = []
@@ -114,7 +114,7 @@ def self.run_eval!
           class_eval(&p)
         end
 
-        excludes.each { |f| fields.delete f }
+        excludes.each { |f| schema.delete f }
         @evaled = true
       end
 

lib/blueprinter/v2/dsl.rb

@@ -47,14 +47,23 @@ def use(*names)
       # @return [Blueprinter::V2::Field]
       #
       def field(name, from: name, **options, &definition)
-        fields[name.to_sym] = Field.new(
+        schema[name.to_sym] = Field.new(
           name: name,
           from: from,
           value_proc: definition,
           options: options.dup
         )
       end
 
+      #
+      # Add multiple fields at once.
+      #
+      def fields(*names)
+        names.each do |name|
+          schema[name.to_sym] = Field.new(name: name, options: {})
+        end
+      end
+
       #
       # Define an association to a single object.
       #
@@ -68,7 +77,7 @@ def field(name, from: name, **options, &definition)
       def object(name, blueprint, from: name, view: nil, **options, &definition)
         raise ArgumentError, 'The :view argument may not be used with V2 Blueprints' if view && blueprint.is_a?(V2)
 
-        fields[name.to_sym] = Association.new(
+        schema[name.to_sym] = Association.new(
           name: name,
           blueprint: blueprint,
           collection: false,
@@ -92,7 +101,7 @@ def object(name, blueprint, from: name, view: nil, **options, &definition)
       def collection(name, blueprint, from: name, view: nil, **options, &definition)
         raise ArgumentError, 'The :view argument may not be used with V2 Blueprints' if view && blueprint.is_a?(V2)
 
-        fields[name.to_sym] = Association.new(
+        schema[name.to_sym] = Association.new(
           name: name,
           blueprint: blueprint,
           collection: true,

lib/blueprinter/v2/reflection.rb

@@ -49,9 +49,9 @@ class View
         # @api private
         def initialize(blueprint, name)
           @name = name
-          @fields = blueprint.fields.select { |_, f| f.is_a? Field }
-          @objects = blueprint.fields.select { |_, f| f.is_a?(Association) && !f.collection }
-          @collections = blueprint.fields.select { |_, f| f.is_a?(Association) && f.collection }
+          @fields = blueprint.schema.select { |_, f| f.is_a? Field }
+          @objects = blueprint.schema.select { |_, f| f.is_a?(Association) && !f.collection }
+          @collections = blueprint.schema.select { |_, f| f.is_a?(Association) && f.collection }
         end
       end
     end

spec/v2/fields_spec.rb

@@ -19,6 +19,25 @@
       expect(ref.fields[:foo].name).to eq :foo
       expect(ref.fields[:foo].value_proc.class.name).to eq "Proc"
     end
+
+    it 'should add multiple fields' do
+      blueprint = Class.new(Blueprinter::V2::Base) do
+        fields :name, :description, :status
+      end
+
+      ref = blueprint.reflections[:default]
+      expect(ref.fields[:name].class.name).to eq "Blueprinter::V2::Field"
+      expect(ref.fields[:name].name).to eq :name
+      expect(ref.fields[:name].options).to eq({})
+
+      expect(ref.fields[:description].class.name).to eq "Blueprinter::V2::Field"
+      expect(ref.fields[:description].name).to eq :description
+      expect(ref.fields[:description].options).to eq({})
+
+      expect(ref.fields[:status].class.name).to eq "Blueprinter::V2::Field"
+      expect(ref.fields[:status].name).to eq :status
+      expect(ref.fields[:status].options).to eq({})
+    end
   end
 
   context "associations" do