class Faker::UniqueGenerator

def self.clear
  marked_unique.each(&:clear)
  marked_unique.clear
end

def self.marked_unique
  Thread.current[:faker_unique_generator_marked_unique] ||= Set.new
end

def clear
  previous_results.clear
end

def exclude(name, arguments, values)
  values ||= []
  values.each do |value|
    previous_results[[name, arguments]] << value
  end
end

def initialize(generator, max_retries)
  @generator = generator
  @max_retries = max_retries
end

def method_missing(name, *arguments)
  self.class.marked_unique.add(self)
  @max_retries.times do
    result = @generator.public_send(name, *arguments)
    next if previous_results[[name, arguments]].include?(result)
    previous_results[[name, arguments]] << result
    return result
  end
  raise RetryLimitExceeded, "Retry limit exceeded for #{name}"
end

def previous_results
  Thread.current[:faker_unique_generator_previous_results] ||= {}
  Thread.current[:faker_unique_generator_previous_results][@generator] ||= Hash.new { |hash, key| hash[key] = Set.new }
end

def respond_to_missing?(method_name, include_private = false)
  method_name.to_s.start_with?('faker_') || super
end