class PuppetTwitch::BasicHttpServer

Attributes

logger[RW]

Public Class Methods

new(bind, port, multiple_threads = true) click to toggle source
# File lib/puppet_twitch/basic_http_server.rb, line 11
def initialize(bind, port, multiple_threads = true)
  @bind = bind
  @port = port
  @multiple_threads = multiple_threads
  @logger = PuppetTwitch::Logger.new()
  @logger.level = PuppetTwitch::Logger::DEBUG
  @actions = {}
end

Public Instance Methods

endpoint(endpoint, &block) click to toggle source
# File lib/puppet_twitch/basic_http_server.rb, line 60
def endpoint(endpoint, &block)
  @actions[endpoint] = block
end
form_response(status, body) click to toggle source
# File lib/puppet_twitch/basic_http_server.rb, line 73
    def form_response(status, body)
<<-EOF
HTTP/1.1 #{status}
Content-Type: text/plain
Content-Length: #{body.bytesize}
Connection: close

#{body}
EOF
    end
get_endpoint_and_params(request) click to toggle source
# File lib/puppet_twitch/basic_http_server.rb, line 64
def get_endpoint_and_params(request)
  match = request.match(/^\s*(?<verb>GET|POST|PUT|DELETE) (?<endpoint>[\w\-\/]*)(\?(?<params>\S*)\s*|\s*)HTTP.*/)
  return match['endpoint'], parse_params(match['params'])
end
parse_params(param_string) click to toggle source
# File lib/puppet_twitch/basic_http_server.rb, line 69
def parse_params(param_string)
  PuppetTwitch::ParamParser.parse(param_string.to_s.strip.split('&'))
end
process_request(socket) click to toggle source
# File lib/puppet_twitch/basic_http_server.rb, line 45
def process_request(socket)
  client_ip       = socket.peeraddr[3]
  client_hostname = socket.peeraddr[2]
  request         = socket.gets

  endpoint, params = get_endpoint_and_params(request)
  @logger.info "Connection from: #{client_hostname} (#{client_ip}) | Endpoint: #{endpoint} | Params: #{params}"

  response = @actions.has_key?(endpoint) ? @actions[endpoint].call(params) : [400, "Unrecognised endpoint: #{endpoint}"]
  @logger.debug "#{response[0]}: #{response[1]}"

  socket.puts form_response(response[0], response[1])
  @logger.debug "Closing connection from #{client_ip}"
end
start() click to toggle source
# File lib/puppet_twitch/basic_http_server.rb, line 20
def start()
  @logger.info 'Starting server ...'
  server = TCPServer.new(@bind, @port)
  @logger.info "Listening on #{@bind}:#{@port}"

  server_loop = proc { |socket|
    begin
      process_request socket
    rescue => e
      @logger.error e.to_s
      socket.puts form_response(500, 'Unexpected error')
    ensure
      socket.close
    end
  }

  loop do
    if @multiple_threads
      Thread.fork(server.accept, &server_loop)
    else
      server_loop.call server.accept
    end
  end
end