# Copyright (c) 2020 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
# frozen_string_literal: true

require 'contrast/extension/module'

module Contrast
  module Agent
    module Assess
      module Policy
        module Propagator
          # Propagation that results in some complex or specific translation
          # of tags from the source to the target. Each node with the CUSTOM
          # action knows the class and method it should call to preform this
          # action.
          module Custom
            class << self
              def propagate propagation_node, preshift, ret, block
                clazz = propagation_node.patch_class
                method = propagation_node.patch_method

                # We cannot flip the String to a Module at patcher creation time -
                # the Module may not exist yet. Instead, we have to defer until the
                # first time the patcher is used.
                if clazz.is_a?(String)
                  clazz = Object.cs__const_get(clazz)
                  propagation_node.patch_class = clazz
                end
                clazz.send(method, propagation_node, preshift, ret, block)
              end
            end
          end
        end
      end
    end
  end
end