class Redis::SubscribedClient

def call(*args)
  @client.process(args)
end

def initialize(client)
  @client = client
end

def psubscribe(*channels, &block)
  subscription("psubscribe", "punsubscribe", channels, block)
end

def punsubscribe(*channels)
  call(:punsubscribe, *channels)
end

def subscribe(*channels, &block)
  subscription("subscribe", "unsubscribe", channels, block)
end

def subscription(start, stop, channels, block)
  sub = Subscription.new(&block)
  begin
    @client.call_loop(start, *channels) do |line|
      type, *rest = line
      sub.callbacks[type].call(*rest)
      break if type == stop && rest.last == 0
    end
  ensure
    send(stop)
  end
end

def unsubscribe(*channels)
  call(:unsubscribe, *channels)
end