module Schlep

# File lib/schlep.rb, line 12
def app
  @app ||= ""
end

# File lib/schlep.rb, line 16
def app=(string)
  @app = sanitize string
end

# File lib/schlep.rb, line 20
def configure
  yield self
end

# File lib/schlep.rb, line 24
def envelope(type, message, options = {})
  {
    :timestamp => timestamp,
    :app =>       options[:app]  || app,
    :host =>      options[:host] || host,
    :type =>      type,
    :message =>   serialize_message(message)
  }
end

# File lib/schlep.rb, line 34
def event(type, message, options = {})
  events type, [message], options
end

# File lib/schlep.rb, line 38
def events(type, messages, options = {})
  options.keys.each { |k| options[k] = sanitize(options[k]) }

  messages.map! { |message| envelope(type, message, options).to_json }

  suppress_redis_errors do
    redis.pipelined do
      while messages.any?
        redis.rpush key, messages.shift
      end
    end
  end
end

# File lib/schlep.rb, line 52
def host
  @host ||= sanitize `hostname`
end

# File lib/schlep.rb, line 56
def host=(string)
  @host = sanitize string
end

# File lib/schlep.rb, line 60
def key
  @key ||= :schlep
end

# File lib/schlep.rb, line 64
def redis
  @redis ||= Redis.new redis_options
end

# File lib/schlep.rb, line 68
def redis_options
  return {} unless redis_url

  parsed_url = URI::parse(redis_url)

  {
    :host     => parsed_url.host,
    :port     => parsed_url.port,
    :password => parsed_url.password
  }
end

# File lib/schlep.rb, line 80
def redis_url
  @redis_url ||= ENV["REDIS_URL"] or ENV["REDISTOGO_URL"]
end

# File lib/schlep.rb, line 84
def reset
  %w[app host redis redis_url].each do |ivar|
    instance_variable_set "@#{ivar}", nil
  end
end

# File lib/schlep.rb, line 90
def serialize_message(message)
  return message unless
    message.is_a? String and
    message.match /\{.+\}/

  begin
    JSON.parse message
  rescue JSON::ParserError
    message
  end
end

# File lib/schlep.rb, line 102
def timestamp
  Time.now.to_f
end

# File lib/schlep.rb, line 108
def sanitize(string)
  string.gsub! /\s/, ""
  string.gsub! /^[^\w]+|[^\w]+$/, ""
  string.gsub! /[^\w\.\-]+/, ":"

  string
end

# File lib/schlep.rb, line 116
def suppress_redis_errors
  begin
    yield
  rescue Errno::ECONNREFUSED => e
    puts e.inspect unless ENV['RUBY_ENV'] == 'test'
  end
end