require 'sidekiq/api'

module Sidekiq
  class SortedEntry
    module UniqueExtension
      def self.included(base)
        base.class_eval do
          include SidekiqUniqueJobs::Unlockable
          alias_method :delete_orig, :delete
          alias_method :delete, :delete_ext
          alias_method :remove_job_orig, :remove_job
          alias_method :remove_job, :remove_job_ext
        end
      end

      def delete_ext
        SidekiqUniqueJobs::Unlockable.unlock(item) if delete_orig
      end

      private

      def remove_job_ext
        remove_job_orig do |message|
          SidekiqUniqueJobs::Unlockable.unlock(Sidekiq.load_json(message))
          yield message
        end
      end
    end

    include UniqueExtension
  end

  class ScheduledSet
    module UniqueExtension
      def self.included(base)
        base.class_eval do
          include SidekiqUniqueJobs::Unlockable
          alias_method :delete_orig, :delete
          alias_method :delete, :delete_ext
        end
      end

      def delete_ext
        SidekiqUniqueJobs::Unlockable.unlock(item) if delete_orig
      end

      def remove_job_ext
        remove_job_orig do |message|
          SidekiqUniqueJobs::Unlockable.unlock(Sidekiq.load_json(message))
          yield message
        end
      end
    end
    include UniqueExtension
  end

  class Job
    module UniqueExtension
      def self.included(base)
        base.class_eval do
          include SidekiqUniqueJobs::Unlockable
          alias_method :delete_orig, :delete
          alias_method :delete, :delete_ext
        end
      end

      def delete_ext
        SidekiqUniqueJobs::Unlockable.unlock(item)
        delete_orig
      end
    end

    include UniqueExtension
  end

  class Queue
    module UniqueExtension
      def self.included(base)
        base.class_eval do
          include SidekiqUniqueJobs::Unlockable
          alias_method :clear_orig, :clear
          alias_method :clear, :clear_ext
        end
      end

      def clear_ext
        each(&:delete)
        clear_orig
      end
    end

    include UniqueExtension
  end

  class JobSet
    module UniqueExtension
      def self.included(base)
        base.class_eval do
          include SidekiqUniqueJobs::Unlockable
          if base.method_defined?(:clear)
            alias_method :clear_orig, :clear
            alias_method :clear, :clear_ext
          end

          if base.method_defined?(:delete_by_value)
            alias_method :delete_by_value_orig, :delete_by_value
            alias_method :delete_by_value, :delete_by_value_ext
          end
        end
      end

      def clear_ext
        each(&:delete)
        clear_orig
      end

      def delete_by_value_ext(name, value)
        SidekiqUniqueJobs::Unlockable.unlock(JSON.parse(value)) if delete_by_value_orig(name, value)
      end
    end

    include UniqueExtension
  end
end