# frozen_string_literal: true

module Rake
  module Funnel
    module Support
      class Patch
        def initialize(context = nil)
          @context = context

          yield self if block_given?
        end

        def setup(&block)
          @setup = block
        end

        def reset(&block)
          @reset = block
        end

        def apply!
          return self if @memo

          @memo = (@setup || noop).call(@context)

          self
        end

        def revert!
          return self unless @memo

          (@reset || noop).call(@memo)
          @memo = nil

          self
        end

        private

        def noop
          proc {}
        end
      end
    end
  end
end