Class: Puma::ControlCLI

# File 'lib/puma/control_cli.rb', line 147

def prepare_configuration
  if @state
    unless File.exist? @state
      raise "State file not found: #{@state}"
    end

    require_relative 'state_file'

    sf = Puma::StateFile.new
    sf.load @state

    @control_url = sf.control_url
    @control_auth_token = sf.control_auth_token
    @pid = sf.pid
  elsif @pidfile
    # get pid from pid_file
    @pid = File.read(@pidfile, mode: 'rb:UTF-8').to_i
  end
end

# File 'lib/puma/control_cli.rb', line 277

def run
  return start if @command == 'start'
  prepare_configuration

  if Puma.windows? || @control_url && !NO_REQ_COMMANDS.include?(@command)
    send_request
  else
    send_signal
  end

rescue => e
  message e.message
  exit 1
end

# File 'lib/puma/control_cli.rb', line 167

def send_request
  uri = URI.parse @control_url

  host = uri.host

  # create server object by scheme
  server =
    case uri.scheme
    when 'ssl'
      require 'openssl'
      host = host[1..-2] if host&.start_with? '['
      OpenSSL::SSL::SSLSocket.new(
        TCPSocket.new(host, uri.port),
        OpenSSL::SSL::SSLContext.new)
        .tap { |ssl| ssl.sync_close = true }  # default is false
        .tap(&:connect)
    when 'tcp'
      host = host[1..-2] if host&.start_with? '['
      TCPSocket.new host, uri.port
    when 'unix'
      # check for abstract UNIXSocket
      UNIXSocket.new(@control_url.start_with?('unix://@') ?
        "\0#{host}#{uri.path}" : "#{host}#{uri.path}")
    else
      raise "Invalid scheme: #{uri.scheme}"
    end

  if @command == 'status'
    message 'Puma is started'
  else
    url = "/#{@command}"

    if @control_auth_token
      url = url + "?token=#{@control_auth_token}"
    end

    server.syswrite "GET #{url} HTTP/1.0\r\n\r\n"

    unless data = server.read
      raise 'Server closed connection before responding'
    end

    response = data.split("\r\n")

    if response.empty?
      raise "Server sent empty response"
    end

    @http, @code, @message = response.first.split(' ',3)

    if @code == '403'
      raise 'Unauthorized access to server (wrong auth token)'
    elsif @code == '404'
      raise "Command error: #{response.last}"
    elsif @code != '200'
      raise "Bad response from server: #{@code}"
    end

    message "Command #{@command} sent success"
    message response.last if PRINTABLE_COMMANDS.include?(@command)
  end
ensure
  if server
    if uri.scheme == 'ssl'
      server.sysclose
    else
      server.close unless server.closed?
    end
  end
end

# File 'lib/puma/control_cli.rb', line 238

def send_signal
  unless @pid
    raise 'Neither pid nor control url available'
  end

  begin
    sig = CMD_PATH_SIG_MAP[@command]

    if sig.nil?
      @stdout.puts "'#{@command}' not available via pid only"
      @stdout.flush unless @stdout.sync
      return
    elsif sig.start_with? 'SIG'
      if Signal.list.key? sig.sub(/\ASIG/, '')
        Process.kill sig, @pid
      else
        raise "Signal '#{sig}' not available'"
      end
    elsif @command == 'status'
      begin
        Process.kill 0, @pid
        @stdout.puts 'Puma is started'
        @stdout.flush unless @stdout.sync
      rescue Errno::ESRCH
        raise 'Puma is not running'
      end
      return
    end
  rescue SystemCallError
    if @command == 'restart'
      start
    else
      raise "No pid '#{@pid}' found"
    end
  end

  message "Command #{@command} sent success"
end

# File 'lib/puma/control_cli.rb', line 293

def start
  require_relative 'cli'

  run_args = []

  run_args += ["-S", @state]  if @state
  run_args += ["-q"] if @quiet
  run_args += ["--pidfile", @pidfile] if @pidfile
  run_args += ["--control-url", @control_url] if @control_url
  run_args += ["--control-token", @control_auth_token] if @control_auth_token
  run_args += ["-C", @config_file] if @config_file
  run_args += ["-e", @environment] if @environment

  log_writer = Puma::LogWriter.new(@stdout, @stderr)

  # replace $0 because puma use it to generate restart command
  puma_cmd = $0.gsub(/pumactl$/, 'puma')
  $0 = puma_cmd if File.exist?(puma_cmd)

  cli = Puma::CLI.new run_args, log_writer
  cli.run
end