From aea32a9ca103a17f0e2b403b01ec3312f36e9849 Mon Sep 17 00:00:00 2001
From: Evan Phoenix <evan@fallingsnow.net>
Date: Mon, 15 Oct 2012 22:24:22 -0700
Subject: [PATCH] Style cleanup

---
 lib/puma/control_cli.rb | 80 ++++++++++++++++++++++++++++-------------
 1 file changed, 55 insertions(+), 25 deletions(-)

diff --git a/lib/puma/control_cli.rb b/lib/puma/control_cli.rb
index 656fc5ef..f2fc92ce 100644
--- a/lib/puma/control_cli.rb
+++ b/lib/puma/control_cli.rb
@@ -7,7 +7,7 @@ require 'socket'
 module Puma
   class ControlCLI
 
-    COMMANDS = %w{status restart stop halt}
+    COMMANDS = %w{status stats restart stop halt}
 
     def is_windows?
       RUBY_PLATFORM =~ /(win|w)32$/ ? true : false
@@ -18,76 +18,95 @@ module Puma
       @stderr = stderr
       @options = {}
       
-      OptionParser.new do |option|
-        option.banner = "Usage: pumactl (-S status_file | -C url -T token) (#{COMMANDS.join("|")})"
-        option.on "-S", "--state PATH", "Where the state file to use is" do |arg|
+      opts = OptionParser.new do |o|
+        o.banner = "Usage: pumactl (-S status_file | -C url -T token) (#{COMMANDS.join("|")})"
+
+        o.on "-S", "--state PATH", "Where the state file to use is" do |arg|
           @options[:status_path] = arg
         end
-        option.on "-Q", "--quiet", "Not display messages" do |arg|
+
+        o.on "-Q", "--quiet", "Not display messages" do |arg|
           @options[:quiet_flag] = true
         end
-        option.on "-P", "--pidfile PATH", "Pid file" do |arg|
+
+        o.on "-P", "--pidfile PATH", "Pid file" do |arg|
           @options[:pid_file] = arg
         end
-        option.on "-C", "--control-url URL", "The bind url to use for the control server" do |arg|
+
+        o.on "-C", "--control-url URL", "The bind url to use for the control server" do |arg|
           @options[:control_url] = arg
         end
-        option.on "-T", "--control-token TOKEN", "The token to use as authentication for the control server" do |arg|
+
+        o.on "-T", "--control-token TOKEN", "The token to use as authentication for the control server" do |arg|
           @options[:control_auth_token] = arg
         end
-        option.on_tail("-H", "--help", "Show this message") do
+
+        o.on_tail("-H", "--help", "Show this message") do
           @stdout.puts option
           exit
         end
-        option.on_tail("-V", "--version", "Show version") do
+
+        o.on_tail("-V", "--version", "Show version") do
           puts Const::PUMA_VERSION
           exit
         end
-      end.parse!(argv)
+      end
+
+      opts.parse!(argv)
       
       command = argv.shift
       @options[:command] = command if command
-      
+
       # check present of command
       unless @options[:command]
         raise "Available commands: #{COMMANDS.join(", ")}"
-      end      
+      end
+
       unless COMMANDS.include? @options[:command]
         raise "Invalid command: #{@options[:command]}" 
       end
-      
+
     rescue => e
       @stdout.puts e.message
       exit 1
     end
     
-    def message msg
+    def message(msg)
       @stdout.puts msg unless @options[:quiet_flag]
     end
 
     def prepare_configuration
       if @options.has_key? :status_path
-        raise "Status file not found: #{@options[:status_path]} " unless File.exist? @options[:status_path]
+        unless File.exist? @options[:status_path]
+          raise "Status file not found: #{@options[:status_path]}"
+        end
+
         status = YAML.load File.read(@options[:status_path])
+
         if status.has_key? "config"
+
+          conf = status["config"]
+
           # get control_url
-          if status["config"].options.has_key?(:control_url)
-            @options[:control_url] = status["config"].options[:control_url]
+          if url = conf.options[:control_url]
+            @options[:control_url] = url
           end
+
           # get control_auth_token
-          if status["config"].options.has_key?(:control_auth_token)
-            @options[:control_auth_token] = status["config"].options[:control_auth_token]
+          if token = conf.options[:control_auth_token]
+            @options[:control_auth_token] = token
           end
+
           # get pid
           @options[:pid] = status["pid"].to_i
         else
           raise "Invalid status file: #{@options[:status_path]}"
         end
+
       elsif @options.has_key? :pid_file
         # get pid from pid_file
         @options[:pid] = File.open(@options[:pid_file]).gets.to_i
       end
-      
     end
 
     def send_request
@@ -102,23 +121,29 @@ module Puma
       else
         raise "Invalid scheme: #{uri.scheme}"
       end
-      
-      unless @options[:command] == "status"
+
+      if @options[:command] == "status"
+        message "Puma is started"
+      else
         url = "/#{@options[:command]}"
+
         if @options.has_key?(:control_auth_token)
           url = url + "?token=#{@options[:control_auth_token]}"
         end
+
         @server << "GET #{url} HTTP/1.0\r\n\r\n"
+
         response = @server.read.split("\r\n")
+
         (@http,@code,@message) = response.first.split(" ")
+
         if @code == "403"
           raise "Unauthorized access to server (wrong auth token)"
         elsif @code != "200"
           raise "Bad response from server: #{@code}"
         end
+
         message "Command #{@options[:command]} sent success"
-      else
-        message "Puma is started"
       end
       
       @server.close
@@ -126,17 +151,22 @@ module Puma
 
     def send_signal
       Process.getpgid(@options[:pid])
+
       case @options[:command]
       when "restart"
         Process.kill("SIGUSR2", @options[:pid])
+
       when "halt"
         Process.kill("QUIT", @options[:pid])
+
       when "stop"
         Process.kill("SIGTERM", @options[:pid])
+
       else
         message "Puma is started"
         return
       end
+
       message "Command #{@options[:command]} sent success"
     end