A simple proxy server in Ruby

ruby_proxy.rb

require 'net/http'
require 'net/https'
require 'uri'
require 'pry'

class Proxy
  def server(port)
    puts "Starting Proxy server on port #{port}"
    @socket = TCPServer.new('localhost', port)
    loop do
      s = @socket.accept
      Thread.new(s, &method(:forward))
    end
  end

  def forward(from)
    request = from.recv(1024)
    url = request[/^\w+\s+(\S+)/, 1]
    url = URI.parse(url)
    http = Net::HTTP.new(url.host, url.port)
    request_uri = url.request_uri

    http.open_timeout = http.read_timeout = 60
    request = Net::HTTP::Get.new(request_uri)
    response = http.request(request)

    puts "#{response.body.length} - #{url}"

    from.write(response.body)
    from.close
  end
end

if ARGV.empty?
  port = 2345
elsif ARGV.size == 1
  port = ARGV[0].to_i
else
  puts 'Usage: proxy.rb [port]'
  exit 1
end


Proxy.new.server port