RubyGems - mlserver - Versions diffs - 1.0.0 - Mend

mlserver 1.0.0

Files changed (14) hide show

checksums.yaml +7 -0
data/bin/mlserver +3 -0
data/lib/MLserver/client_handler.rb +11 -0
data/lib/MLserver/error_response.rb +60 -0
data/lib/MLserver/html/error_page.template.html +10 -0
data/lib/MLserver/logger.rb +51 -0
data/lib/MLserver/redirect_response.rb +13 -0
data/lib/MLserver/request.rb +31 -0
data/lib/MLserver/request_parser.rb +42 -0
data/lib/MLserver/response.rb +41 -0
data/lib/MLserver/server.rb +61 -0
data/lib/MLserver/settings.rb +23 -0
data/lib/MLserver.rb +18 -0
metadata +82 -0

checksums.yaml ADDED Viewed

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 6b326cb18fd091f0272824b58d166980a1e3add7e87d508dd4895388eae223c8
+  data.tar.gz: 26e225d538c7791fe256aa31f1cd14e40b505ffcdb3217296ec602ad1318fe8d
+SHA512:
+  metadata.gz: 2fc0d8d718f919271737cddc7fc361e47b5cdba17091c8b53684612dabbca8b5ed42aaf80c0f6f06fa5e4163ee41f8957a4f4ef3a27dbc3905a8f2c34512c286
+  data.tar.gz: 58c17b39a1fc34cc1aaf72242683d3fb22d3a177a362b235964d94ca961da435400c2f684239dfe7af67db6f410a245a882d8ef305675a982a7e2c81c6b5b97f

data/bin/mlserver ADDED Viewed

@@ -0,0 +1,3 @@
+#!/usr/bin/env ruby
+require "MLserver"

data/lib/MLserver/client_handler.rb ADDED Viewed

@@ -0,0 +1,11 @@
+module MLserver
+  class ClientHandler
+    def initialize(&block)
+      @block = block
+    end
+    def run(request, client)
+      @block.call(request, client)
+    end
+  end
+end

data/lib/MLserver/error_response.rb ADDED Viewed

@@ -0,0 +1,60 @@
+module MLserver
+  class ErrorResponse
+    @@error_messages_by_code = {
+      400 => "bad request",
+      401 => "unauthorized",
+      403 => "forbidden",
+      404 => "page not found",
+      405 => "method not allowed",
+      406 => "not acceptable",
+      407 => "proxy authentication required",
+      408 => "request timeout",
+      409 => "conflict",
+      410 => "resource gone",
+      411 => "length required",
+      412 => "precondition failed",
+      413 => "payload too large",
+      414 => "URI too long",
+      415 => "unsupported media type",
+      416 => "range not satisfiable",
+      417 => "expectation failed",
+      418 => "I'm a teapot",
+      421 => "misdirected request",
+      422 => "unprocessable entity",
+      423 => "resource locked",
+      424 => "failed dependency",
+      425 => "too early",
+      426 => "upgrade required",
+      428 => "precondition required",
+      429 => "too many requests",
+      431 => "request header fields too large",
+      451 => "unavailable for legal reasons",
+      500 => "internal server error",
+      501 => "not implemented",
+      502 => "bad gateway",
+      503 => "service unavailable",
+      504 => "gateway timeout",
+      505 => "HTTP version not supported",
+      506 => "variant also negotiates",
+      507 => "insufficient storage",
+      508 => "loop detected",
+      510 => "not extended",
+      511 => "network authentication required"
+    }
+    def initialize(code, message: nil, httpver: "HTTP/1.0")
+      code = code.to_i
+      @code = code
+      @message = message ? message : "Error: #{code} (#{@@error_messages_by_code[code]})"
+      @httpver = httpver
+    end
+    def response
+      return Response.new(status: @code, data: html_page, content_type: "text/html", httpver: @httpver, headers: {Connection: "close"})
+    end
+    def html_page
+      return File.read(File.dirname(__FILE__) + "/html/error_page.template.html").gsub("[ecode]", @code.to_s).gsub("[emsg]", @message)
+    end
+  end
+end

data/lib/MLserver/html/error_page.template.html ADDED Viewed

@@ -0,0 +1,10 @@
+<!DOCTYPE html>
+<html>
+  <head>
+    <title>Error [ecode]</title>
+  </head>
+  <body>
+    <h1>[emsg]</h1>
+  </body>
+</html>

data/lib/MLserver/logger.rb ADDED Viewed

@@ -0,0 +1,51 @@
+module MLserver
+  class Logger
+    @@log_levels = [:info, :warn, :error]
+    def initialize(out:, err:, log_colors: {}, outputs: {error: :err})
+      @out = out
+      @err = err
+      @log_colors = log_colors
+      @outputs = outputs
+    end
+    def log(message, level = :info)
+      out = @out
+      out = @err if @outputs[level] == :err
+      if @log_colors[level]
+        message = message.color @log_colors[level]
+      end
+      out.puts message
+    end
+    def format_ip_address(address)
+      ipv4_re = /^((25[0-5]|(2[0-4]|1\d|[1-9]|)\d)\.?\b){4}$/
+      ipv4_anywhere_re = /((25[0-5]|(2[0-4]|1\d|[1-9]|)\d)\.?\b){4}/
+      ipv4_in_ipv6_re = /^(\:\:ffff\:)((25[0-5]|(2[0-4]|1\d|[1-9]|)\d)\.?\b){4}$/
+      ipv6_re = /(([0-9a-fA-F]{1,4}:){7,7}[0-9a-fA-F]{1,4}|([0-9a-fA-F]{1,4}:){1,7}:|([0-9a-fA-F]{1,4}:){1,6}:[0-9a-fA-F]{1,4}|([0-9a-fA-F]{1,4}:){1,5}(:[0-9a-fA-F]{1,4}){1,2}|([0-9a-fA-F]{1,4}:){1,4}(:[0-9a-fA-F]{1,4}){1,3}|([0-9a-fA-F]{1,4}:){1,3}(:[0-9a-fA-F]{1,4}){1,4}|([0-9a-fA-F]{1,4}:){1,2}(:[0-9a-fA-F]{1,4}){1,5}|[0-9a-fA-F]{1,4}:((:[0-9a-fA-F]{1,4}){1,6})|:((:[0-9a-fA-F]{1,4}){1,7}|:)|fe80:(:[0-9a-fA-F]{0,4}){0,4}%[0-9a-zA-Z]{1,}|::(ffff(:0{1,4}){0,1}:){0,1}((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])|([0-9a-fA-F]{1,4}:){1,4}:((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9]))/
+      return address if address.match?(ipv4_re)
+      if address.match?(ipv4_in_ipv6_re)
+        ipv4_address = address.match(ipv4_anywhere_re).to_s
+        return ipv4_address
+      end
+      if address.match?(ipv6_re)
+        return "[#{address}]"
+      end
+    end
+    def log_traffic(ip, direction, data)
+      symbol = (direction == :incoming ? "=>" : "<=")
+      log("#{format_ip_address ip} #{symbol} #{data}")
+    end
+  end
+  dlcolors = {
+    warn: :yellow,
+    error: :red
+  }
+  DefaultLogger = Logger.new(out: STDOUT, err: STDERR, log_colors: dlcolors)
+end

data/lib/MLserver/redirect_response.rb ADDED Viewed

@@ -0,0 +1,13 @@
+module MLserver
+  class RedirectResponse
+    def initialize(url, type: 302, httpver: "HTTP/1.1")
+      @type = type.to_i
+      @url = url.to_s
+      @httpver = httpver
+    end
+    def response
+      return Response.new(status: @type, headers: {Location: @url, Connection: "close"}, data: "Redirecting... If you don't get redirected, go to the following URL: #{@url}", httpver: @httpver)
+    end
+  end
+end

data/lib/MLserver/request.rb ADDED Viewed

@@ -0,0 +1,31 @@
+module MLserver
+  class Request
+    def initialize(headers: {}, request:, data: "", client:)
+      @headers = headers
+      @request = request
+      @data = data
+      @client = client
+      @request_split = @request.split(" ")
+    end
+    def method
+      @request_split[0].upcase
+    end
+    def path
+      @request_split[1]
+    end
+    def httpver
+      @request_split[2]
+    end
+    def respond(response)
+      MLserver.settings.logger.log_traffic @client.peeraddr[2], :outgoing, "#{response.httpver} #{response.status}"
+      @client.puts response.to_s
+    end
+    attr_reader :headers, :request, :data
+  end
+end

data/lib/MLserver/request_parser.rb ADDED Viewed

@@ -0,0 +1,42 @@
+module MLserver
+  module RequestParser
+    def self.parse_request(client)
+      keepReading = true
+      headers = {}
+      data = ""
+      req = client.gets.to_s
+      while req.gsub("\r\n", "") == "" do
+        req = client.gets.to_s
+      end
+      #Close if bad request
+      if req.to_s.length < 14
+        client.puts ErrorResponse.new(400).response.to_s
+        client.close
+        Thread.exit
+      end
+      #Get all headers
+      while keepReading do
+        x = client.gets.to_s
+        if x.chomp.length == 0
+          keepReading = false
+        else
+          if x.split(": ").length < 2
+            client.puts ErrorResponse.new(400).response.to_s
+            client.close
+            Thread.exit
+          end
+          headers[x.split(": ")[0].to_sym] = x.split(": ")[1].chomp
+        end
+      end
+      #Get payload data
+      data = client.read(headers[:"Content-Length"].to_i)
+      return Request.new(headers: headers, request: req, data: data, client: client)
+    end
+  end
+end

data/lib/MLserver/response.rb ADDED Viewed

@@ -0,0 +1,41 @@
+module MLserver
+  class Response
+    @@default_headers = {
+      Server: "MLserver #{MLserver.version}",
+      "Content-Type": "text/plain"
+    }
+    def initialize(status:200, headers:{}, data:"", content_type: nil, httpver: "HTTP/1.1")
+      @status = status
+      @data = data
+      @httpver = httpver
+      @headers = @@default_headers.merge(response_specific_headers).merge(headers)
+      @headers[:"Content-Type"] = content_type if content_type
+    end
+    def to_s(array:false)
+      status_line = "#{httpver} #{status}"
+      headers = @headers.map { |header|
+        k=header[0].to_s
+        v=header[1].to_s
+        header = "#{k}: #{v}"
+      }
+      headers = headers.join("\r\n") if !array
+      data = @data
+      return "#{status_line}\r\n#{headers}\r\n\r\n#{data}\r\n" if !array
+      return [status_line, headers, "", data, ""] if array
+    end
+    attr_accessor :status, :headers, :data, :httpver
+    private
+    def response_specific_headers
+      {
+        Date: Time.now.strftime("%a, %d %b %Y %H:%M:%S %Z"),
+        "Content-Length": @data.length,
+      }
+    end
+  end
+end

data/lib/MLserver/server.rb ADDED Viewed

@@ -0,0 +1,61 @@
+module MLserver
+  module Server
+    @@valid_http_versions = ["HTTP/1.0", "HTTP/1.1"]
+    def self.start
+      settings = MLserver.settings
+      host = settings.host
+      port = settings.port
+      handler = settings.handler
+      logger = settings.logger
+      logger.log "MLserver #{MLserver.version}"
+      server = TCPServer.new(host, port)
+      logger.log "Listening on #{logger.format_ip_address host}:#{port}"
+      loop do
+        Thread.start(server.accept) do |client|
+          loop do
+            r=RequestParser.parse_request(client)
+            logger.log_traffic client.peeraddr[2], :incoming, "#{r.method} #{r.path} #{r.httpver}"
+            if !@@valid_http_versions.include?(r.httpver)
+              client_ip = client.peeraddr[2]
+              resp = MLserver::ErrorResponse.new(505)
+              r.respond resp.response
+              client.close
+              Thread.exit
+            end
+            if r.httpver == "HTTP/1.1"
+              if !r.headers[:Host]
+                r.respond ErrorResponse.new(400).response
+                client.close
+                Thread.exit
+              elsif settings.force_host
+                if !settings.force_host.include?(r.headers[:Host])
+                  r.respond ErrorResponse.new(400).response
+                  client.close
+                  Thread.exit
+                end
+              end
+            end
+            handler.run(r, client)
+            if r.httpver == "HTTP/1.0" || r.headers[:Connection] == "close"
+              client.close
+              Thread.exit
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/MLserver/settings.rb ADDED Viewed

@@ -0,0 +1,23 @@
+module MLserver
+  @@settings = nil
+  class Settings
+    def initialize(host: "0.0.0.0", port: "5555", handler:, logger:, force_host: false)
+      @host = host
+      @port = port.to_i
+      @handler = handler
+      @logger = logger
+      @force_host = force_host
+    end
+    attr_accessor :host, :port, :handler, :logger, :force_host
+  end
+  def self.settings
+    @@settings
+  end
+  def self.settings=(x)
+    @@settings=x
+  end
+end

data/lib/MLserver.rb ADDED Viewed

@@ -0,0 +1,18 @@
+module MLserver
+  def self.version
+    "1.0.0"
+  end
+end
+require "socket"
+require "rbtext"
+require "rbtext/string_methods"
+require_relative "MLserver/request.rb"
+require_relative "MLserver/request_parser.rb"
+require_relative "MLserver/response.rb"
+require_relative "MLserver/server.rb"
+require_relative "MLserver/client_handler.rb"
+require_relative "MLserver/logger.rb"
+require_relative "MLserver/error_response.rb"
+require_relative "MLserver/redirect_response.rb"
+require_relative "MLserver/settings.rb"

metadata ADDED Viewed

@@ -0,0 +1,82 @@
+--- !ruby/object:Gem::Specification
+name: mlserver
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Matthias Lee
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2023-04-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: argparse
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.3
+- !ruby/object:Gem::Dependency
+  name: rbtext
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.3.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.3.3
+description: A simple web server
+email: matthias@matthiasclee.com
+executables:
+- mlserver
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/mlserver
+- lib/MLserver.rb
+- lib/MLserver/client_handler.rb
+- lib/MLserver/error_response.rb
+- lib/MLserver/html/error_page.template.html
+- lib/MLserver/logger.rb
+- lib/MLserver/redirect_response.rb
+- lib/MLserver/request.rb
+- lib/MLserver/request_parser.rb
+- lib/MLserver/response.rb
+- lib/MLserver/server.rb
+- lib/MLserver/settings.rb
+homepage: https://github.com/Matthiasclee/MLServer
+licenses: []
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.1.6
+signing_key:
+specification_version: 4
+summary: A simple web server
+test_files: []