mperham--sidekiq/lib/sidekiq/web.rb

require 'sinatra/base'
require 'slim'
require 'sprockets'
require 'multi_json'

module Sidekiq
  class SprocketsMiddleware
    def initialize(app, options={})
      @app = app
      @root = options[:root]
      path   =  options[:path] || 'assets'
      @matcher = /^\/#{path}\/*/
      @environment = ::Sprockets::Environment.new(@root)
      @environment.append_path 'assets/javascripts'
      @environment.append_path 'assets/javascripts/vendor'
      @environment.append_path 'assets/stylesheets'
      @environment.append_path 'assets/stylesheets/vendor'
      @environment.append_path 'assets/images'
    end

    def call(env)
      # Solve the problem of people requesting /sidekiq when they need to request /sidekiq/ so
      # that relative links in templates resolve correctly.
      return [301, { 'Location' => "#{env['SCRIPT_NAME']}/" }, ['redirecting']] if env['SCRIPT_NAME'] == env['REQUEST_PATH']

      return @app.call(env) unless @matcher =~ env["PATH_INFO"]
      env['PATH_INFO'].sub!(@matcher,'')
      @environment.call(env)
    end
  end

  class Web < Sinatra::Base
    dir = File.expand_path(File.dirname(__FILE__) + "/../../web")
    set :views,  "#{dir}/views"
    set :root, "#{dir}/public"
    set :slim, :pretty => true
    use SprocketsMiddleware, :root => dir

    helpers do

      def reset_worker_list
        Sidekiq.redis do |conn|
          workers = conn.smembers('workers')
          workers.each do |name|
            conn.srem('workers', name)
          end
        end
      end

      def workers
        @workers ||= begin
          Sidekiq.redis do |conn|
            conn.smembers('workers').map do |w|
              msg = conn.get("worker:#{w}")
              if msg
                msg = if MultiJson.respond_to?(:adapter)
                  MultiJson.load(msg)
                else
                  MultiJson.decode(msg)
                end
              end
              [w, msg]
            end.sort { |x| x[1] ? -1 : 1 }
          end
        end
      end

      def processed
        Sidekiq.redis { |conn| conn.get('stat:processed') } || 0
      end

      def failed
        Sidekiq.redis { |conn| conn.get('stat:failed') } || 0
      end

      def retry_count
        Sidekiq.redis { |conn| conn.zcard('retry') }
      end

      def retries
        Sidekiq.redis do |conn|
          results = conn.zrange('retry', 0, 25, :withscores => true)
          if MultiJson.respond_to?(:adapter)
            results.each_slice(2).map { |msg, score| [MultiJson.load(msg), Float(score)] }
          else
            results.each_slice(2).map { |msg, score| [MultiJson.decode(msg), Float(score)] }
          end
        end
      end

      def queues
        Sidekiq.redis do |conn|
          conn.smembers('queues').map do |q|
            [q, conn.llen("queue:#{q}") || 0]
          end.sort { |x,y| x[1] <=> y[1] }
        end
      end

      def retries_with_score(score)
        Sidekiq.redis do |conn|
          results = conn.zrangebyscore('retry', score, score)
          if MultiJson.respond_to?(:adapter)
            results.map { |msg| MultiJson.load(msg) }
          else
            results.map { |msg| MultiJson.decode(msg) }
          end
        end
      end

      def location
        Sidekiq.redis { |conn| conn.client.location }
      end

      def root_path
        "#{env['SCRIPT_NAME']}/"
      end

      def current_status
        return 'down' if workers.size == 0
        return 'idle' if workers.size > 0 && workers.map { |x| x[1] }.compact.size == 0
        return 'active'
      end

      def relative_time(time)
        %{<time datetime="#{time.getutc.iso8601}">#{time}</time>}
      end
    end

    get "/" do
      slim :index
    end

    post "/reset" do
      reset_worker_list
      redirect root_path
    end

    get "/queues/:name" do
      halt 404 unless params[:name]
      @name = params[:name]
      if MultiJson.respond_to?(:adapter)
        @messages = Sidekiq.redis {|conn| conn.lrange("queue:#{@name}", 0, 10) }.map { |str| MultiJson.load(str) }
      else
        @messages = Sidekiq.redis {|conn| conn.lrange("queue:#{@name}", 0, 10) }.map { |str| MultiJson.decode(str) }
      end
      slim :queue
    end

    get "/retries/:score" do
      halt 404 unless params[:score]
      @score = params[:score].to_f
      slim :retry
    end

    post "/retries/:score" do
      halt 404 unless params[:score]
      score = params[:score].to_f
      if params['retry']
        Sidekiq.redis do |conn|
          results = conn.zrangebyscore('retry', score, score)
          conn.zremrangebyscore('retry', score, score)
          results.map do |message|
            msg = if MultiJson.respond_to?(:adapter)
              MultiJson.load(message)
            else
              MultiJson.decode(message)
            end
            conn.rpush("queue:#{msg['queue']}", message)
          end
        end
      elsif params['delete']
        Sidekiq.redis do |conn|
          conn.zremrangebyscore('retry', score, score)
        end
      end
      redirect root_path
    end
  end

end