palkan/clowne

# frozen_string_literal: true

module Clowne
  module Utils
    class Plan # :nodoc: all
      class TwoPhaseSet
        def initialize
          @added = {}
          @removed = []
        end

        def []=(k, v)
          return if @removed.include?(k)

          @added[k] = v
        end

        def delete(k)
          return if @removed.include?(k)

          @removed << k
          @added.delete(k)
        end

        def values
          @added.values
        end
      end

      def initialize(registry)
        @registry = registry
        @data = {}
      end

      def add(type, declaration)
        data[type] = [] unless data.key?(type)
        data[type] << declaration
      end

      def add_to(type, id, declaration)
        data[type] = TwoPhaseSet.new unless data.key?(type)
        data[type][id] = declaration
      end

      def set(type, declaration)
        data[type] = declaration
      end

      def get(type)
        data[type]
      end

      def remove(type)
        data.delete(type)
      end

      def remove_from(type, id)
        return unless data[type]

        data[type].delete(id)
      end

      def declarations(reload = false)
        return @declarations if !reload && instance_variable_defined?(:@declarations)

        @declarations =
          registry.actions.flat_map do |type|
            value = data[type]
            next if value.nil?

            value = value.values if value.is_a?(TwoPhaseSet)
            value = Array(value)
            value.map { |v| [type, v] }
          end.compact
      end

      def dup
        self.class.new(registry).tap do |duped|
          data.each do |k, v|
            duped.set(k, v.dup)
          end
        end
      end

      private

      attr_reader :data, :registry
    end
  end
end