require "database_cleaner/strategy"

module DatabaseCleaner
  module Ldap
    class Deletion < DatabaseCleaner::Strategy
      attr_reader :filter

      def initialize(opts={})
        raise ArgumentError,
          "Option :filter must be a Net::LDAP::Filter" unless opts.empty? || opts[:filter].is_a?(Net::LDAP::Filter)

        @filter = opts[:filter]
      end

      def clean
        entries.each { |ldap_entry| delete! ldap_entry.dn }
      end

      def db
        @db ||= Net::LDAP.new
      end

      private

      def entries
        Array db.search(filter: filter)
      end

      def delete!(dn)
        db.delete(dn: dn) or
          raise "Could no delete #{dn}. Reason: #{db.get_operation_result}"
      end
    end
  end
end