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