iknow_view_models 3.4.1 → 3.5.0

data/lib/view_model/active_record/collection_nested_controller.rb

@@ -3,8 +3,7 @@
 require 'view_model/active_record/nested_controller_base'
 
 # Controller mixin for accessing a root ViewModel which can be accessed in a
-# collection by a parent model. Enabled by calling `nested_in :parent, as:
-# :children` on the viewmodel controller
+# collection by a parent model.
 
 # Contributes the following routes:
 # PUT    /parents/:parent_id/children            #append  -- deserialize (possibly existing) children and append to collection
@@ -32,6 +31,10 @@ module ViewModel::ActiveRecord::CollectionNestedController
     write_association(serialize_context: serialize_context, deserialize_context: deserialize_context, lock_owner: lock_owner, &block)
   end
 
+  def replace_bulk(serialize_context: new_serialize_context, deserialize_context: new_deserialize_context, &block)
+    write_association_bulk(serialize_context: serialize_context, deserialize_context: deserialize_context, &block)
+  end
+
   def disassociate_all(serialize_context: new_serialize_context, deserialize_context: new_deserialize_context, lock_owner: nil)
     destroy_association(true, serialize_context: serialize_context, deserialize_context: deserialize_context, lock_owner: lock_owner)
   end

data/lib/view_model/active_record/controller_base.rb

@@ -102,15 +102,35 @@ module ActionDispatch
         def arvm_resources(resource_name, options = {}, &block)
           except             = options.delete(:except) { [] }
           add_shallow_routes = options.delete(:add_shallow_routes) { true }
+          defaults           = options.delete(:defaults) { {} }
 
           nested = shallow_nesting_depth > 0
 
+          association_name = options.delete(:association_name) { resource_name.to_s }
+          owner_viewmodel  = options.delete(:owner_viewmodel) do
+            if nested
+              parent_resource.name.to_s.singularize.camelize
+            end
+          end
+
+          defaults = {
+            association_name: association_name,
+            owner_viewmodel: owner_viewmodel,
+          }.merge(defaults)
+
           only_routes = []
           only_routes += [:create] unless nested
           only_routes += [:show, :destroy] if add_shallow_routes
           only_routes -= except
 
-          resources resource_name, shallow: true, only: only_routes, **options do
+          # Bulk replace
+          if nested && !except.include?(:replace_bulk)
+            collection do
+              post resource_name, controller: resource_name, action: :replace_bulk, as: nil, defaults: defaults
+            end
+          end
+
+          resources resource_name, shallow: true, only: only_routes, defaults: defaults, **options do
             instance_eval(&block) if block_given?
 
             if nested
@@ -149,16 +169,35 @@ module ActionDispatch
         def arvm_resource(resource_name, options = {}, &block)
           except             = options.delete(:except) { [] }
           add_shallow_routes = options.delete(:add_shallow_routes) { true }
+          defaults           = options.delete(:defaults) { {} }
 
           only_routes = []
-          is_shallow = false
-          resource resource_name, shallow: true, only: only_routes, **options do
-            is_shallow = shallow_nesting_depth > 1
+          nested = shallow_nesting_depth > 0
+
+          association_name = options.delete(:association_name) { resource_name.to_s }
+          owner_viewmodel  = options.delete(:owner_viewmodel) do
+            if nested
+              parent_resource.name.to_s.singularize.camelize
+            end
+          end
+
+          defaults = {
+            association_name: association_name,
+            owner_viewmodel: owner_viewmodel,
+          }.merge(defaults)
+
+          if nested && !except.include?(:create_associated_bulk)
+            collection do
+              post resource_name, controller: resource_name, action: :create_associated_bulk, as: nil, defaults: defaults
+            end
+          end
+
+          resource resource_name, shallow: true, only: only_routes, defaults: defaults, **options do
             instance_eval(&block) if block_given?
 
             name_route = { as: '' } # Only one route may take the name
 
-            if is_shallow
+            if nested
               post('',   action: :create_associated,  **name_route.extract!(:as)) unless except.include?(:create)
               get('',    action: :show_associated,    **name_route.extract!(:as)) unless except.include?(:show)
               delete('', action: :destroy_associated, **name_route.extract!(:as)) unless except.include?(:destroy)
@@ -170,7 +209,7 @@ module ActionDispatch
           end
 
           # singularly nested resources provide collection accessors at the top level
-          if is_shallow && add_shallow_routes
+          if nested && add_shallow_routes
             resources resource_name.to_s.pluralize, shallow: true, only: [:show, :destroy] - except do
               shallow_scope do
                 collection do

data/lib/view_model/active_record/nested_controller_base.rb

@@ -7,15 +7,51 @@ require 'view_model/active_record/controller_base'
 module ViewModel::ActiveRecord::NestedControllerBase
   extend ActiveSupport::Concern
 
+  class ParentProxyModel < ViewModel
+    # Prevent this from appearing in hooks
+    self.synthetic = true
+
+    attr_reader :parent, :association_data, :changed_children
+
+    def initialize(parent, association_data, changed_children)
+      @parent = parent
+      @association_data = association_data
+      @changed_children = changed_children
+    end
+
+    def serialize(json, serialize_context:)
+      ViewModel::Callbacks.wrap_serialize(parent, context: serialize_context) do
+        child_context = parent.context_for_child(association_data.association_name, context: serialize_context)
+
+        json.set!(ViewModel::ID_ATTRIBUTE, parent.id)
+        json.set!(ViewModel::BULK_UPDATE_ATTRIBUTE) do
+          if association_data.referenced? && !association_data.owned?
+            if association_data.collection?
+              json.array!(changed_children) do |child|
+                ViewModel.serialize_as_reference(child, json, serialize_context: child_context)
+              end
+            else
+              ViewModel.serialize_as_reference(changed_children, json, serialize_context: child_context)
+            end
+          else
+            ViewModel.serialize(changed_children, json, serialize_context: child_context)
+          end
+        end
+      end
+    end
+  end
+
   protected
 
   def show_association(scope: nil, serialize_context: new_serialize_context, lock_owner: nil)
+    require_external_referenced_association!
+
     associated_views = nil
     pre_rendered = owner_viewmodel.transaction do
       owner_view = owner_viewmodel.find(owner_viewmodel_id, eager_include: false, lock: lock_owner)
       ViewModel::Callbacks.wrap_serialize(owner_view, context: serialize_context) do
         # Association manipulation methods construct child contexts internally
-        associated_views = owner_view.load_associated(association_name, scope: scope)
+        associated_views = owner_view.load_associated(association_name, scope: scope, serialize_context: serialize_context)
 
         associated_views = yield(associated_views) if block_given?
 
@@ -27,11 +63,27 @@ module ViewModel::ActiveRecord::NestedControllerBase
     associated_views
   end
 
+  # This method always takes direct update hashes, and returns
+  # viewmodels directly.
+  #
+  # There's no multi membership, so when viewing the children of a
+  # single parent each child can only appear once. This means it's
+  # safe to use update hashes directly.
   def write_association(serialize_context: new_serialize_context, deserialize_context: new_deserialize_context, lock_owner: nil)
+    require_external_referenced_association!
+
     association_view = nil
     pre_rendered = owner_viewmodel.transaction do
       update_hash, refs = parse_viewmodel_updates
 
+      update_hash =
+        ViewModel::ActiveRecord.add_reference_indirection(
+          update_hash,
+          association_data: association_data,
+          references:       refs,
+          key:              'write-association',
+        )
+
       owner_view = owner_viewmodel.find(owner_viewmodel_id, eager_include: false, lock: lock_owner)
 
       association_view = owner_view.replace_associated(association_name, update_hash,
@@ -49,7 +101,67 @@ module ViewModel::ActiveRecord::NestedControllerBase
     association_view
   end
 
+  # This method takes direct update hashes for owned associations, and
+  # reference hashes for shared associations. The return value matches
+  # the input structure.
+  #
+  # If an association is referenced and owned, each child may only
+  # appear once so each is guaranteed to have a unique update
+  # hash. This means it's only safe to use update hashes directly in
+  # this case.
+  def write_association_bulk(serialize_context: new_serialize_context, deserialize_context: new_deserialize_context, lock_owner: nil)
+    require_external_referenced_association!
+
+    updated_by_parent_viewmodel = nil
+
+    pre_rendered = owner_viewmodel.transaction do
+      updates_by_parent_id, references = parse_bulk_update
+
+      if association_data.owned?
+        updates_by_parent_id.transform_values!.with_index do |update_hash, index|
+          ViewModel::ActiveRecord.add_reference_indirection(
+            update_hash,
+            association_data: association_data,
+            references:       references,
+            key:              "write-association-bulk-#{index}",
+          )
+        end
+      end
+
+      updated_by_parent_viewmodel =
+        owner_viewmodel.replace_associated_bulk(
+          association_name,
+          updates_by_parent_id,
+          references:          references,
+          deserialize_context: deserialize_context,
+        )
+
+      views = updated_by_parent_viewmodel.flat_map { |_parent_viewmodel, updated_views| Array.wrap(updated_views) }
+
+      ViewModel.preload_for_serialization(views)
+
+      updated_by_parent_viewmodel = yield(updated_by_parent_viewmodel) if block_given?
+
+      return_updates = updated_by_parent_viewmodel.map do |owner_view, updated_views|
+        ParentProxyModel.new(owner_view, association_data, updated_views)
+      end
+
+      return_structure = {
+        ViewModel::TYPE_ATTRIBUTE         => ViewModel::BULK_UPDATE_TYPE,
+        ViewModel::BULK_UPDATES_ATTRIBUTE => return_updates,
+      }
+
+      prerender_viewmodel(return_structure, serialize_context: serialize_context)
+    end
+
+    render_json_string(pre_rendered)
+    updated_by_parent_viewmodel
+  end
+
+
   def destroy_association(collection, serialize_context: new_serialize_context, deserialize_context: new_deserialize_context, lock_owner: nil)
+    require_external_referenced_association!
+
     if lock_owner
       owner_viewmodel.find(owner_viewmodel_id, eager_include: false, lock: lock_owner)
     end
@@ -61,7 +173,7 @@ module ViewModel::ActiveRecord::NestedControllerBase
   end
 
   def association_data
-    owner_viewmodel._association_data(association_name)
+    @association_data ||= owner_viewmodel._association_data(association_name)
   end
 
   def owner_update_hash(update)
@@ -78,22 +190,22 @@ module ViewModel::ActiveRecord::NestedControllerBase
     parse_param(id_param_name, **default)
   end
 
-  included do
-    delegate :owner_viewmodel, :association_name, to: 'self.class'
+  def owner_viewmodel_class_for_name(name)
+    ViewModel::Registry.for_view_name(name)
   end
 
-  class_methods do
-    attr_accessor :owner_viewmodel, :association_name
-
-    def nested_in(owner, as:)
-      unless owner.is_a?(Class) && owner < ViewModel::Record
-        owner = ViewModel::Registry.for_view_name(owner.to_s.camelize)
-      end
+  def owner_viewmodel
+    name = params.fetch(:owner_viewmodel) { raise ArgumentError.new("No owner viewmodel present") }
+    owner_viewmodel_class_for_name(name.to_s.camelize)
+  end
 
-      self.owner_viewmodel = owner
-      raise ArgumentError.new("Could not find owner ViewModel class '#{owner_name}'") if owner_viewmodel.nil?
+  def association_name
+    params.fetch(:association_name) { raise ArgumentError.new('No association name from routes') }
+  end
 
-      self.association_name = as
+  def require_external_referenced_association!
+    unless association_data.referenced? && association_data.external?
+      raise ArgumentError.new("Expected referenced external association: '#{association_name}'")
     end
   end
 end

data/lib/view_model/active_record/singular_nested_controller.rb

@@ -1,8 +1,7 @@
 # frozen_string_literal: true
 
 # Controller mixin for accessing a root ViewModel which can be accessed
-# individually by a parent model. Enabled by calling `nested_in :parent, as:
-# :child` on the viewmodel controller
+# individually by a parent model.
 
 # Contributes the following routes:
 # POST   /parents/:parent_id/child   #create_associated  -- deserialize (possibly existing) child, replacing existing child
@@ -28,6 +27,10 @@ module ViewModel::ActiveRecord::SingularNestedController
     write_association(serialize_context: serialize_context, deserialize_context: deserialize_context, lock_owner: lock_owner, &block)
   end
 
+  def create_associated_bulk(serialize_context: new_serialize_context, deserialize_context: new_deserialize_context, &block)
+    write_association_bulk(serialize_context: serialize_context, deserialize_context: deserialize_context, &block)
+  end
+
   def destroy_associated(serialize_context: new_serialize_context, deserialize_context: new_deserialize_context, lock_owner: nil)
     destroy_association(false, serialize_context: serialize_context, deserialize_context: deserialize_context, lock_owner: lock_owner)
   end

data/lib/view_model/callbacks.rb

@@ -181,7 +181,7 @@ module ViewModel::Callbacks
     # ARVM synthetic views are considered part of their association and as such
     # are not visited by callbacks. Eligibility exclusion is intended to be
     # library-internal: subclasses should not attempt to extend this.
-    view.is_a?(ViewModel::ActiveRecord) && view.class.synthetic
+    view.is_a?(ViewModel) && view.class.synthetic
   end
 
   def self.wrap_serialize(viewmodel, context:)

data/lib/view_model/controller.rb

@@ -75,12 +75,34 @@ module ViewModel::Controller
   protected
 
   def parse_viewmodel_updates
-    update_hash = _extract_update_data(params.fetch(:data))
-    refs        = _extract_param_hash(params.fetch(:references, {}))
+    data_param = params.fetch(:data) do
+      raise ViewModel::Error.new(status: 400, detail: "Missing 'data' parameter")
+    end
+    refs_param = params.fetch(:references, {})
+
+    update_hash = _extract_update_data(data_param)
+    refs        = _extract_param_hash(refs_param)
 
     return update_hash, refs
   end
 
+  def parse_bulk_update
+    data, references = parse_viewmodel_updates
+
+    ViewModel::Schemas.verify_schema!(ViewModel::Schemas::BULK_UPDATE, data)
+
+    updates_by_parent =
+      data.fetch(ViewModel::BULK_UPDATES_ATTRIBUTE).each_with_object({}) do |parent_update, acc|
+        parent_id = parent_update.fetch(ViewModel::ID_ATTRIBUTE)
+        update    = parent_update.fetch(ViewModel::BULK_UPDATE_ATTRIBUTE)
+
+        acc[parent_id] = update
+      end
+
+    return updates_by_parent, references
+  end
+
+
   private
 
   def _extract_update_data(data)

data/lib/view_model/record.rb

@@ -35,7 +35,7 @@ class ViewModel::Record < ViewModel
 
     # Should this class be registered in the viewmodel registry
     def should_register?
-      !abstract_class && !unregistered
+      !abstract_class && !unregistered && !synthetic
     end
 
     # Specifies an attribute from the model to be serialized in this view

data/lib/view_model/schemas.rb

@@ -39,6 +39,50 @@ class ViewModel::Schemas
 
   VIEWMODEL_REFERENCE = JsonSchema.parse!(VIEWMODEL_REFERENCE_SCHEMA)
 
+  BULK_UPDATE_SCHEMA =
+    {
+      'type'                 => 'object',
+      'description'          => 'bulk update collection',
+      'properties'           => {
+        ViewModel::TYPE_ATTRIBUTE => {
+          'type' => 'string',
+          'enum' => [ViewModel::BULK_UPDATE_TYPE],
+        },
+
+        ViewModel::BULK_UPDATES_ATTRIBUTE => {
+          'type'  => 'array',
+          'items' => {
+            'type'                 => 'object',
+            'properties'           => {
+              ViewModel::ID_ATTRIBUTE => ID_SCHEMA,
+
+              # These will be checked by the main deserialize operation. Any operations on the data
+              # before the main serialization must do its own checking of the presented update data.
+
+              ViewModel::BULK_UPDATE_ATTRIBUTE => {
+                'oneOf' => [
+                  { 'type' => 'array' },
+                  { 'type' => 'object' },
+                ]
+              },
+            },
+            'additionalProperties' => false,
+            'required'             => [
+              ViewModel::ID_ATTRIBUTE,
+              ViewModel::BULK_UPDATE_ATTRIBUTE,
+            ],
+          },
+        }
+      },
+      'additionalProperties' => false,
+      'required'             => [
+        ViewModel::TYPE_ATTRIBUTE,
+        ViewModel::BULK_UPDATES_ATTRIBUTE,
+      ],
+    }.freeze
+
+  BULK_UPDATE = JsonSchema.parse!(BULK_UPDATE_SCHEMA)
+
   def self.verify_schema!(schema, value)
     valid, errors = schema.validate(value)
     unless valid

data/test/helpers/arvm_test_utilities.rb

@@ -192,4 +192,69 @@ module ARVMTestUtilities
   def build_fupdate(attrs = {}, &block)
     FupdateBuilder.new.build!(&block).merge(attrs)
   end
+
+  def each_hook_span(trace)
+    return enum_for(:each_hook_span, trace) unless block_given?
+
+    hook_nesting = []
+
+    trace.each_with_index do |t, i|
+      case t.hook
+      when ViewModel::Callbacks::Hook::OnChange,
+        ViewModel::Callbacks::Hook::BeforeValidate
+        # ignore
+      when ViewModel::Callbacks::Hook::BeforeVisit,
+        ViewModel::Callbacks::Hook::BeforeDeserialize
+        hook_nesting.push([t, i])
+
+      when ViewModel::Callbacks::Hook::AfterVisit,
+        ViewModel::Callbacks::Hook::AfterDeserialize
+        (nested_top, nested_index) = hook_nesting.pop
+
+        unless nested_top.hook.name == t.hook.name.sub(/^After/, 'Before')
+          raise "Invalid nesting, processing '#{t.hook.name}', expected matching '#{nested_top.hook.name}'"
+        end
+
+        unless nested_top.view == t.view
+          raise "Invalid nesting, processing '#{t.hook.name}', " \
+                  "expected viewmodel '#{t.view}' to match '#{nested_top.view}'"
+        end
+
+        yield t.view, (nested_index..i), t.hook.name.sub(/^After/, '')
+
+      else
+        raise 'Unexpected hook type'
+      end
+    end
+  end
+
+  def show_span(view, range, hook)
+    "#{view.class.name}(#{view.id}) #{range} #{hook}"
+  end
+
+  def enclosing_hooks(spans, inner_range)
+    spans.select do |_view, range, _hook|
+      inner_range != range && range.cover?(inner_range.min) && range.cover?(inner_range.max)
+    end
+  end
+
+  def assert_all_hooks_nested_inside_parent_hook(trace)
+    spans = each_hook_span(trace).to_a
+
+    spans.reject { |view, _range, _hook| view.class == ParentView }.each do |view, range, hook|
+      enclosing_spans = enclosing_hooks(spans, range)
+
+      enclosing_parent_hook = enclosing_spans.detect do |other_view, _other_range, other_hook|
+        other_hook == hook && other_view.class == ParentView
+      end
+
+      next if enclosing_parent_hook
+
+      self_str      = show_span(view, range, hook)
+      enclosing_str = enclosing_spans.map { |ov, ora, oh| show_span(ov, ora, oh) }.join("\n")
+      assert_not_nil(
+        enclosing_parent_hook,
+        "Invalid nesting of hook: #{self_str}\nEnclosing hooks:\n#{enclosing_str}")
+    end
+  end
 end