class Bullet::Registry::Base

def [](key)
  @registry[key]
end

def add(key, value)
  @registry[key] ||= Set.new
  if value.is_a? Array
    @registry[key] += value
  else
    @registry[key] << value
  end
end

def delete(base)
  @registry.delete(base)
end

def each(&block)
  @registry.each(&block)
end

def include?(key, value)
  !@registry[key].nil? && @registry[key].include?(value)
end

def initialize
  @registry = {}
end

def select(*args, &block)
  @registry.select(*args, &block)
end