Skip to content
This repository
Fetching contributors…

Cannot retrieve contributors at this time

file 56 lines (46 sloc) 1.458 kb
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56
module Puma
  module App
    class Status
      def initialize(server, cli)
        @server = server
        @cli = cli
        @auth_token = nil
      end
      OK_STATUS = '{ "status": "ok" }'.freeze

      attr_accessor :auth_token

      def authenticate(env)
        return true unless @auth_token
        env['QUERY_STRING'].to_s.split(/&;/).include?("token=#{@auth_token}")
      end

      def call(env)
        unless authenticate(env)
          return rack_response(403, 'Invalid auth token', 'text/plain')
        end

        case env['PATH_INFO']
        when "/stop"
          @server.stop
          return rack_response(200, OK_STATUS)

        when "/halt"
          @server.halt
          return rack_response(200, OK_STATUS)

        when "/restart"
          if @cli and @cli.restart_on_stop!
            @server.begin_restart

            return rack_response(200, OK_STATUS)
          else
            return rack_response(200, '{ "status": "not configured" }')
          end

        when "/stats"
          b = @server.backlog
          r = @server.running
            return rack_response(200, %Q!{ "backlog": #{b}, "running": #{r} }!)
        end

        rack_response 404, "Unsupported action", 'text/plain'
      end

      private
      def rack_response(status, body, content_type='application/json')
        [status, { 'Content-Type' => content_type, 'Content-Length' => body.length.to_s }, [body]]
      end
    end
  end
end
Something went wrong with that request. Please try again.