vcap_relations.rb

module Sequel::Plugins::VcapRelations
  # Depend on the instance_hooks plugin.
  def self.apply(model)
    model.plugin(:instance_hooks)
  end

  module InstanceMethods
    def has_one_to_many?(association)
      association_type(association) == :one_to_many && send(association).any?
    end

    def has_one_to_one?(association)
      association_type(association) == :one_to_one && !!send(association)
    end

    def association_type(association)
      self.class.association_reflection(association)[:type]
    end

    def relationship_dataset(association)
      reflection = self.class.association_reflection(association)
      if (dataset = reflection[:dataset])
        if dataset.arity == 1
          instance_exec(reflection, &dataset)
        else
          instance_exec(&dataset)
        end
      else
        reflection.associated_class.dataset
      end
    end
  end

  module ClassMethods
    # Override many_to_one in order to add <relation>_guid
    # and <relation>_guid= methods.
    #
    # See the default many_to_one implementation for a description of the args
    # and return values.
    def many_to_one(name, opts={})
      define_guid_accessors(name) unless opts.fetch(:without_guid_generation, false)

      opts[:reciprocal] ||= self.name.split('::').last.underscore.pluralize.to_sym
      super
    end

    def one_through_one(name, opts={})
      define_guid_reader(name)
      super
    end

    # Override many_to_many in order to add an override the default Sequel
    # methods for many_to_many relationships.
    #
    # In particular, this enables support of bulk modifying relationships.
    #
    # See the default many_to_many implementation for a description of the args
    # and return values.
    def many_to_many(name, opts={})
      singular_name = name.to_s.singularize
      ids_attr = "#{singular_name}_ids"
      guids_attr = "#{singular_name}_guids"

      define_method("add_#{singular_name}") do |other|
        # sequel is not capable of merging adds to a many_to_many association
        # like it is for a one_to_many and nds up throwing a db exception,
        # so lets squash the add
        db.transaction(savepoint: true) do
          if other.is_a?(Integer)
            super(other) unless send(ids_attr).include? other
          else
            super(other) unless send(name).include? other
          end
        rescue Sequel::UniqueConstraintViolation => e
          # ignore the error and rollback the inner transaction
          raise Sequel::Rollback if opts[:ignored_unique_constraint_violation_errors]&.any? { |pattern| e.message.include?(pattern) }

          raise e
        end
      end

      opts[:reciprocal] ||=
        self.name.split('::').last.underscore.pluralize.to_sym

      define_to_many_methods(name, singular_name, ids_attr, guids_attr)
      super
    end

    # Override one_to_many in order to add an override the default Sequel
    # methods for one_to_many relationships.
    #
    # In particular, this enables support of bulk modifying relationships.
    #
    # See the default one_to_many implementation for a description of the args
    # and return values.
    def one_to_many(name, opts={})
      singular_name = name.to_s.singularize
      ids_attr = "#{singular_name}_ids"
      guids_attr = "#{singular_name}_guids"

      opts[:reciprocal] ||= self.name.split('::').last.underscore.to_sym

      define_to_many_methods(name, singular_name, ids_attr, guids_attr)
      super
    end

    private

    def define_guid_accessors(name)
      define_guid_reader(name)
      define_guid_writer(name)
    end

    def define_guid_writer(name)
      define_method("#{guid_attr(name)}=") do |val|
        other = nil

        if val
          ar    = self.class.association_reflection(name)
          other = ar.associated_class[guid: val]
          raise CloudController::Errors::ApiError.new_from_details('InvalidRelation', "Could not find #{ar.associated_class.name.demodulize} with guid: #{val}") if other.nil?
        end
        send("#{name}=", other)
      end
    end

    def define_guid_reader(name)
      define_method(guid_attr(name)) do
        other = send(name)
        other.guid unless other.nil?
      end
    end

    def guid_attr(name)
      "#{name}_guid"
    end

    # rubocop:todo Metrics/CyclomaticComplexity
    def define_to_many_methods(name, singular_name, ids_attr, guids_attr)
      diff_collections = proc do |a, b|
        cur_set = Set.new(a)
        new_set = Set.new(b)

        intersection = cur_set & new_set
        added = new_set - intersection
        removed = cur_set - intersection

        [added, removed]
      end

      define_method(ids_attr) do
        send(name).collect(&:id)
      end

      # greppable: add_domain_by_guid
      define_method("add_#{singular_name}_by_guid") do |guid|
        ar = self.class.association_reflection(name)
        other = ar.associated_class[guid:]
        raise CloudController::Errors::ApiError.new_from_details('InvalidRelation', "Could not find #{ar.associated_class.name} with guid: #{guid}") if other.nil?

        if pk
          send("add_#{singular_name}", other)
        else
          after_save_hook { send("add_#{singular_name}", other) }
        end
      end

      define_method("#{ids_attr}=") do |ids|
        return unless ids

        ds = send(name)
        ds.each { |r| send("remove_#{singular_name}", r) unless ids.include?(r.id) }
        ids.each { |i| send("add_#{singular_name}", i) }
      end

      define_method(guids_attr) do
        send(name).collect(&:guid)
      end

      define_method("#{guids_attr}=") do |guids|
        return unless guids

        current_guids = send(name).map(&:guid)
        (added, removed) = diff_collections.call(current_guids, guids)
        added.each { |g| send("add_#{singular_name}_by_guid", g) }
        removed.each { |g| send("remove_#{singular_name}_by_guid", g) }
      end

      define_method("remove_#{singular_name}_by_guid") do |guid|
        ar = self.class.association_reflection(name)
        other = ar.associated_class[guid:]
        raise CloudController::Errors::ApiError.new_from_details('InvalidRelation', "Could not find #{ar.associated_class.name} with guid: #{guid}") if other.nil?

        send("remove_#{singular_name}", other)
      end

      define_method("remove_#{singular_name}") do |other|
        if other.is_a?(Integer)
          super(other) if send(ids_attr).include? other
        elsif send(name).map(&:pk).include? other.pk
          super(other)
        end
      end
    end
    # rubocop:enable Metrics/CyclomaticComplexity
  end
end