RubyGems - mlserver - Versions diffs - 1.0.0 - Mend

mlserver 1.0.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

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: []