CanCanCommunity/cancancan

require_relative '../sti_detector'

# this class is responsible for detecting sti classes and creating new rules for the
# relevant subclasses, using the inheritance_column as a merger
module CanCan
  module ModelAdapters
    class StiNormalizer
      class << self
        def normalize(rules)
          rules_cache = []
          return unless defined?(ActiveRecord::Base)

          rules.delete_if do |rule|
            subjects = rule.subjects.select do |subject|
              update_rule(subject, rule, rules_cache)
            end
            subjects.length == rule.subjects.length
          end
          rules_cache.each { |rule| rules.push(rule) }
        end

        private

        def update_rule(subject, rule, rules_cache)
          return false unless StiDetector.sti_class?(subject)

          rules_cache.push(build_rule_for_subclass(rule, subject))
          true
        end

        # create a new rule for the subclasses that links on the inheritance_column
        def build_rule_for_subclass(rule, subject)
          sti_conditions = { subject.inheritance_column => subject.sti_name }
          new_rule_conditions =
            if rule.with_scope?
              rule.conditions.where(sti_conditions)
            else
              rule.conditions.merge(sti_conditions)
            end

          CanCan::Rule.new(rule.base_behavior, rule.actions, subject.superclass,
                           new_rule_conditions, rule.block)
        end
      end
    end
  end
end