mirror of
				https://gitea.invidious.io/iv-org/shard-kemal.git
				synced 2024-08-15 00:53:36 +00:00 
			
		
		
		
	Use nested module declaration
This commit is contained in:
		
							parent
							
								
									4edebcf8eb
								
							
						
					
					
						commit
						f7484d14d3
					
				
					 9 changed files with 176 additions and 160 deletions
				
			
		|  | @ -1,12 +1,14 @@ | ||||||
| # All loggers must inherit from `Kemal::BaseLogHandler`. | module Kemal | ||||||
| class Kemal::BaseLogHandler < HTTP::Handler |   # All loggers must inherit from `Kemal::BaseLogHandler`. | ||||||
|   def initialize |   class Kemal::BaseLogHandler < HTTP::Handler | ||||||
|   end |     def initialize | ||||||
|  |     end | ||||||
| 
 | 
 | ||||||
|   def call(context) |     def call(context) | ||||||
|     call_next context |       call_next context | ||||||
|   end |     end | ||||||
| 
 | 
 | ||||||
|   def write(message) |     def write(message) | ||||||
|  |     end | ||||||
|   end |   end | ||||||
| end | end | ||||||
|  |  | ||||||
|  | @ -1,33 +1,35 @@ | ||||||
| class Kemal::CommonLogHandler < Kemal::BaseLogHandler | module Kemal | ||||||
|   @handler : IO::FileDescriptor |   class CommonLogHandler < Kemal::BaseLogHandler | ||||||
|   getter handler |     @handler : IO::FileDescriptor | ||||||
|  |     getter handler | ||||||
| 
 | 
 | ||||||
|   def initialize |     def initialize | ||||||
|     @handler = STDOUT |       @handler = STDOUT | ||||||
|   end |     end | ||||||
| 
 | 
 | ||||||
|   def call(context) |     def call(context) | ||||||
|     time = Time.now |       time = Time.now | ||||||
|     call_next(context) |       call_next(context) | ||||||
|     elapsed_text = elapsed_text(Time.now - time) |       elapsed_text = elapsed_text(Time.now - time) | ||||||
|     @handler << time << " " << context.response.status_code << " " << context.request.method << " " << context.request.resource << " " << elapsed_text << "\n" |       @handler << time << " " << context.response.status_code << " " << context.request.method << " " << context.request.resource << " " << elapsed_text << "\n" | ||||||
|     context |       context | ||||||
|   end |     end | ||||||
| 
 | 
 | ||||||
|   def write(message) |     def write(message) | ||||||
|     @handler << message |       @handler << message | ||||||
|   end |     end | ||||||
| 
 | 
 | ||||||
|   private def elapsed_text(elapsed) |     private def elapsed_text(elapsed) | ||||||
|     minutes = elapsed.total_minutes |       minutes = elapsed.total_minutes | ||||||
|     return "#{minutes.round(2)}m" if minutes >= 1 |       return "#{minutes.round(2)}m" if minutes >= 1 | ||||||
| 
 | 
 | ||||||
|     seconds = elapsed.total_seconds |       seconds = elapsed.total_seconds | ||||||
|     return "#{seconds.round(2)}s" if seconds >= 1 |       return "#{seconds.round(2)}s" if seconds >= 1 | ||||||
| 
 | 
 | ||||||
|     millis = elapsed.total_milliseconds |       millis = elapsed.total_milliseconds | ||||||
|     return "#{millis.round(2)}ms" if millis >= 1 |       return "#{millis.round(2)}ms" if millis >= 1 | ||||||
| 
 | 
 | ||||||
|     "#{(millis * 1000).round(2)}µs" |       "#{(millis * 1000).round(2)}µs" | ||||||
|  |     end | ||||||
|   end |   end | ||||||
| end | end | ||||||
|  |  | ||||||
|  | @ -1,4 +1,5 @@ | ||||||
| module Kemal::Middleware | module Kemal::Middleware | ||||||
|  |   # This middleware adds SSL / TLS support. | ||||||
|   class SSL |   class SSL | ||||||
|     getter context |     getter context | ||||||
| 
 | 
 | ||||||
|  |  | ||||||
|  | @ -1,3 +1,5 @@ | ||||||
| # This is here to represent the logger corresponding to Null Object Pattern. | module Kemal | ||||||
| class Kemal::NullLogHandler < Kemal::BaseLogHandler |   # This is here to represent the logger corresponding to Null Object Pattern. | ||||||
|  |   class NullLogHandler < Kemal::BaseLogHandler | ||||||
|  |   end | ||||||
| end | end | ||||||
|  |  | ||||||
|  | @ -1,73 +1,75 @@ | ||||||
| require "json" | require "json" | ||||||
| 
 | 
 | ||||||
| # ParamParser parses the request contents including query_params and body | module Kemal | ||||||
| # and converts them into a params hash which you can within the environment |   # ParamParser parses the request contents including query_params and body | ||||||
| # context. |   # and converts them into a params hash which you can within the environment | ||||||
| alias AllParamTypes = Nil | String | Int64 | Float64 | Bool | Hash(String, JSON::Type) | Array(JSON::Type) |   # context. | ||||||
|  |   alias AllParamTypes = Nil | String | Int64 | Float64 | Bool | Hash(String, JSON::Type) | Array(JSON::Type) | ||||||
| 
 | 
 | ||||||
| class Kemal::ParamParser |   class ParamParser | ||||||
|   URL_ENCODED_FORM = "application/x-www-form-urlencoded" |     URL_ENCODED_FORM = "application/x-www-form-urlencoded" | ||||||
|   APPLICATION_JSON = "application/json" |     APPLICATION_JSON = "application/json" | ||||||
| 
 | 
 | ||||||
|   def initialize(@request : HTTP::Request) |     def initialize(@request : HTTP::Request) | ||||||
|     @url = {} of String => String |       @url = {} of String => String | ||||||
|     @query = HTTP::Params.new({} of String => Array(String)) |       @query = HTTP::Params.new({} of String => Array(String)) | ||||||
|     @body = HTTP::Params.new({} of String => Array(String)) |       @body = HTTP::Params.new({} of String => Array(String)) | ||||||
|     @json = {} of String => AllParamTypes |       @json = {} of String => AllParamTypes | ||||||
|     @url_parsed = false |       @url_parsed = false | ||||||
|     @query_parsed = false |       @query_parsed = false | ||||||
|     @body_parsed = false |       @body_parsed = false | ||||||
|     @json_parsed = false |       @json_parsed = false | ||||||
|   end |     end | ||||||
| 
 | 
 | ||||||
|   {% for method in %w(url query body json) %} |     {% for method in %w(url query body json) %} | ||||||
|   def {{method.id}} |     def {{method.id}} | ||||||
|     # check memoization |       # check memoization | ||||||
|     return @{{method.id}} if @{{method.id}}_parsed |       return @{{method.id}} if @{{method.id}}_parsed | ||||||
| 
 | 
 | ||||||
|     parse_{{method.id}} |       parse_{{method.id}} | ||||||
|     # memoize |       # memoize | ||||||
|     @{{method.id}}_parsed = true |       @{{method.id}}_parsed = true | ||||||
|     @{{method.id}} |       @{{method.id}} | ||||||
|   end |     end | ||||||
|   {% end %} |     {% end %} | ||||||
| 
 | 
 | ||||||
|   def parse_body |     def parse_body | ||||||
|     return if (@request.headers["Content-Type"]? =~ /#{URL_ENCODED_FORM}/).nil? |       return if (@request.headers["Content-Type"]? =~ /#{URL_ENCODED_FORM}/).nil? | ||||||
|     @body = parse_part(@request.body) |       @body = parse_part(@request.body) | ||||||
|   end |     end | ||||||
| 
 | 
 | ||||||
|   def parse_query |     def parse_query | ||||||
|     @query = parse_part(@request.query) |       @query = parse_part(@request.query) | ||||||
|   end |     end | ||||||
| 
 | 
 | ||||||
|   def parse_url |     def parse_url | ||||||
|     if params = @request.url_params |       if params = @request.url_params | ||||||
|       params.each do |key, value| |         params.each do |key, value| | ||||||
|         @url[key.as(String)] = value.as(String) |           @url[key.as(String)] = value.as(String) | ||||||
|  |         end | ||||||
|       end |       end | ||||||
|     end |     end | ||||||
|   end |  | ||||||
| 
 | 
 | ||||||
|   # Parses JSON request body if Content-Type is `application/json`. |     # Parses JSON request body if Content-Type is `application/json`. | ||||||
|   # If request body is a JSON Hash then all the params are parsed and added into `params`. |     # If request body is a JSON Hash then all the params are parsed and added into `params`. | ||||||
|   # If request body is a JSON Array it's added into `params` as `_json` and can be accessed |     # If request body is a JSON Array it's added into `params` as `_json` and can be accessed | ||||||
|   # like params["_json"] |     # like params["_json"] | ||||||
|   def parse_json |     def parse_json | ||||||
|     return unless @request.body && @request.headers["Content-Type"]? == APPLICATION_JSON |       return unless @request.body && @request.headers["Content-Type"]? == APPLICATION_JSON | ||||||
| 
 | 
 | ||||||
|     body = @request.body.as(String) |       body = @request.body.as(String) | ||||||
|     case json = JSON.parse(body).raw |       case json = JSON.parse(body).raw | ||||||
|     when Hash |       when Hash | ||||||
|       json.each do |key, value| |         json.each do |key, value| | ||||||
|         @json[key.as(String)] = value.as(AllParamTypes) |           @json[key.as(String)] = value.as(AllParamTypes) | ||||||
|  |         end | ||||||
|  |       when Array | ||||||
|  |         @json["_json"] = json | ||||||
|       end |       end | ||||||
|     when Array |  | ||||||
|       @json["_json"] = json |  | ||||||
|     end |     end | ||||||
|   end |  | ||||||
| 
 | 
 | ||||||
|   def parse_part(part) |     def parse_part(part) | ||||||
|     HTTP::Params.parse(part || "") |       HTTP::Params.parse(part || "") | ||||||
|  |     end | ||||||
|   end |   end | ||||||
| end | end | ||||||
|  |  | ||||||
|  | @ -1,12 +1,14 @@ | ||||||
| # Route is the main building block of Kemal. | module Kemal | ||||||
| # It takes 3 parameters: Method, path and a block to specify |   # Route is the main building block of Kemal. | ||||||
| # what action to be done if the route is matched. |   # It takes 3 parameters: Method, path and a block to specify | ||||||
| class Kemal::Route |   # what action to be done if the route is matched. | ||||||
|   getter handler |   class Route | ||||||
|   @handler : HTTP::Server::Context -> String |     getter handler | ||||||
|   @method : String |     @handler : HTTP::Server::Context -> String | ||||||
|  |     @method : String | ||||||
| 
 | 
 | ||||||
|   def initialize(@method, @path : String, &handler : HTTP::Server::Context -> _) |     def initialize(@method, @path : String, &handler : HTTP::Server::Context -> _) | ||||||
|     @handler = ->(context : HTTP::Server::Context) { handler.call(context).to_s } |       @handler = ->(context : HTTP::Server::Context) { handler.call(context).to_s } | ||||||
|  |     end | ||||||
|   end |   end | ||||||
| end | end | ||||||
|  |  | ||||||
|  | @ -1,53 +1,54 @@ | ||||||
| require "http/server" |  | ||||||
| require "radix" | require "radix" | ||||||
| 
 | 
 | ||||||
| # Kemal::RouteHandler is the main handler which handles all the HTTP requests. Routing, parsing, rendering e.g | module Kemal | ||||||
| # are done in this handler. |   # Kemal::RouteHandler is the main handler which handles all the HTTP requests. Routing, parsing, rendering e.g | ||||||
| class Kemal::RouteHandler < HTTP::Handler |   # are done in this handler. | ||||||
|   INSTANCE = new |   class RouteHandler < HTTP::Handler | ||||||
|  |     INSTANCE = new | ||||||
| 
 | 
 | ||||||
|   property tree |     property tree | ||||||
| 
 | 
 | ||||||
|   def initialize |     def initialize | ||||||
|     @tree = Radix::Tree(Route).new |       @tree = Radix::Tree(Route).new | ||||||
|   end |  | ||||||
| 
 |  | ||||||
|   def call(context) |  | ||||||
|     context.response.headers.add "X-Powered-By", "Kemal" |  | ||||||
|     context.response.content_type = "text/html" unless context.response.headers.has_key?("Content-Type") |  | ||||||
|     process_request(context) |  | ||||||
|   end |  | ||||||
| 
 |  | ||||||
|   # Adds a given route to routing tree. As an exception each `GET` route additionaly defines |  | ||||||
|   # a corresponding `HEAD` route. |  | ||||||
|   def add_route(method, path, &handler : HTTP::Server::Context -> _) |  | ||||||
|     add_to_radix_tree method, path, Route.new(method, path, &handler) |  | ||||||
|     add_to_radix_tree("HEAD", path, Route.new("HEAD", path, &handler)) if method == "GET" |  | ||||||
|   end |  | ||||||
| 
 |  | ||||||
|   # Check if a route is defined and returns the lookup |  | ||||||
|   def lookup_route(verb, path) |  | ||||||
|     @tree.find radix_path(verb, path) |  | ||||||
|   end |  | ||||||
| 
 |  | ||||||
|   # Processes the route if it's a match. Otherwise renders 404. |  | ||||||
|   def process_request(context) |  | ||||||
|     raise Kemal::Exceptions::RouteNotFound.new(context) unless context.route_defined? |  | ||||||
|     route = context.route_lookup.payload.as(Route) |  | ||||||
|     content = route.handler.call(context) |  | ||||||
|     if Kemal.config.error_handlers.has_key?(context.response.status_code) |  | ||||||
|       raise Kemal::Exceptions::CustomException.new(context) |  | ||||||
|     end |     end | ||||||
|     context.response.print(content) |  | ||||||
|     context |  | ||||||
|   end |  | ||||||
| 
 | 
 | ||||||
|   private def radix_path(method : String, path) |     def call(context) | ||||||
|     "/#{method.downcase}#{path}" |       context.response.headers.add "X-Powered-By", "Kemal" | ||||||
|   end |       context.response.content_type = "text/html" unless context.response.headers.has_key?("Content-Type") | ||||||
|  |       process_request(context) | ||||||
|  |     end | ||||||
| 
 | 
 | ||||||
|   private def add_to_radix_tree(method, path, route) |     # Adds a given route to routing tree. As an exception each `GET` route additionaly defines | ||||||
|     node = radix_path method, path |     # a corresponding `HEAD` route. | ||||||
|     @tree.add node, route |     def add_route(method, path, &handler : HTTP::Server::Context -> _) | ||||||
|  |       add_to_radix_tree method, path, Route.new(method, path, &handler) | ||||||
|  |       add_to_radix_tree("HEAD", path, Route.new("HEAD", path, &handler)) if method == "GET" | ||||||
|  |     end | ||||||
|  | 
 | ||||||
|  |     # Check if a route is defined and returns the lookup | ||||||
|  |     def lookup_route(verb, path) | ||||||
|  |       @tree.find radix_path(verb, path) | ||||||
|  |     end | ||||||
|  | 
 | ||||||
|  |     # Processes the route if it's a match. Otherwise renders 404. | ||||||
|  |     def process_request(context) | ||||||
|  |       raise Kemal::Exceptions::RouteNotFound.new(context) unless context.route_defined? | ||||||
|  |       route = context.route_lookup.payload.as(Route) | ||||||
|  |       content = route.handler.call(context) | ||||||
|  |       if Kemal.config.error_handlers.has_key?(context.response.status_code) | ||||||
|  |         raise Kemal::Exceptions::CustomException.new(context) | ||||||
|  |       end | ||||||
|  |       context.response.print(content) | ||||||
|  |       context | ||||||
|  |     end | ||||||
|  | 
 | ||||||
|  |     private def radix_path(method : String, path) | ||||||
|  |       "/#{method.downcase}#{path}" | ||||||
|  |     end | ||||||
|  | 
 | ||||||
|  |     private def add_to_radix_tree(method, path, route) | ||||||
|  |       node = radix_path method, path | ||||||
|  |       @tree.add node, route | ||||||
|  |     end | ||||||
|   end |   end | ||||||
| end | end | ||||||
|  |  | ||||||
|  | @ -1,11 +1,13 @@ | ||||||
| # Kemal::StaticFileHandler is used to serve static files(.js/.css/.png e.g). | module Kemal | ||||||
| # This handler is on by default and you can disable it like. |   # Kemal::StaticFileHandler is used to serve static files(.js/.css/.png e.g). | ||||||
| # |   # This handler is on by default and you can disable it like. | ||||||
| #   serve_static false |   # | ||||||
| # |   #   serve_static false | ||||||
| class Kemal::StaticFileHandler < HTTP::StaticFileHandler |   # | ||||||
|   def call(context) |   class StaticFileHandler < HTTP::StaticFileHandler | ||||||
|     return call_next(context) if context.request.path.not_nil! == "/" |     def call(context) | ||||||
|     super |       return call_next(context) if context.request.path.not_nil! == "/" | ||||||
|  |       super | ||||||
|  |     end | ||||||
|   end |   end | ||||||
| end | end | ||||||
|  |  | ||||||
|  | @ -1,12 +1,14 @@ | ||||||
| # Kemal::WebSocketHandler is used for building a WebSocket route. | module Kemal | ||||||
| # For each WebSocket route a new handler is created and registered to global handlers. |   # Kemal::WebSocketHandler is used for building a WebSocket route. | ||||||
| class Kemal::WebSocketHandler < HTTP::WebSocketHandler |   # For each WebSocket route a new handler is created and registered to global handlers. | ||||||
|   def initialize(@path : String, &@proc : HTTP::WebSocket, HTTP::Server::Context -> Void) |   class WebSocketHandler < HTTP::WebSocketHandler | ||||||
|     Kemal.config.add_ws_handler self |     def initialize(@path : String, &@proc : HTTP::WebSocket, HTTP::Server::Context -> Void) | ||||||
|   end |       Kemal.config.add_ws_handler self | ||||||
|  |     end | ||||||
| 
 | 
 | ||||||
|   def call(context) |     def call(context) | ||||||
|     return call_next(context) unless context.request.path.not_nil! == @path |       return call_next(context) unless context.request.path.not_nil! == @path | ||||||
|     super |       super | ||||||
|  |     end | ||||||
|   end |   end | ||||||
| end | end | ||||||
|  |  | ||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue