mirror of
				https://gitea.invidious.io/iv-org/shard-kemal.git
				synced 2024-08-15 00:53:36 +00:00 
			
		
		
		
	Add HTTPBasicAuth middleware
This commit is contained in:
		
							parent
							
								
									8b6700695d
								
							
						
					
					
						commit
						743fd3682d
					
				
					 4 changed files with 65 additions and 0 deletions
				
			
		
							
								
								
									
										25
									
								
								spec/middleware/http_basic_auth_spec.cr
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										25
									
								
								spec/middleware/http_basic_auth_spec.cr
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,25 @@ | ||||||
|  | require "../spec_helper" | ||||||
|  | 
 | ||||||
|  | describe "Kemal::Middleware::HTTPBasicAuth" do | ||||||
|  |   it "goes to next handler with correct credentials" do | ||||||
|  |     auth_handler = Kemal::Middleware::HTTPBasicAuth.new("serdar", "123") | ||||||
|  |     request = HTTP::Request.new( | ||||||
|  |       "GET", | ||||||
|  |       "/", | ||||||
|  |       headers: HTTP::Headers{"Authorization": "Basic c2VyZGFyOjEyMw=="}, | ||||||
|  |     ) | ||||||
|  |     response = auth_handler.call(request) | ||||||
|  |     response.status_code.should eq 404 | ||||||
|  |   end | ||||||
|  | 
 | ||||||
|  |   it "returns 401 with incorrect credentials" do | ||||||
|  |     auth_handler = Kemal::Middleware::HTTPBasicAuth.new("serdar", "123") | ||||||
|  |     request = HTTP::Request.new( | ||||||
|  |       "GET", | ||||||
|  |       "/", | ||||||
|  |       headers: HTTP::Headers{"Authorization": "NotBasic"}, | ||||||
|  |     ) | ||||||
|  |     response = auth_handler.call(request) | ||||||
|  |     response.status_code.should eq 401 | ||||||
|  |   end | ||||||
|  | end | ||||||
|  | @ -1,5 +1,6 @@ | ||||||
| require "spec" | require "spec" | ||||||
| require "../src/kemal/*" | require "../src/kemal/*" | ||||||
|  | require "../src/kemal/middleware/*" | ||||||
| 
 | 
 | ||||||
| include Kemal | include Kemal | ||||||
| 
 | 
 | ||||||
|  |  | ||||||
|  | @ -20,3 +20,9 @@ macro redirect(url) | ||||||
|   env.response.headers.add "Location", {{url}} |   env.response.headers.add "Location", {{url}} | ||||||
|   env.response.status_code = 301 |   env.response.status_code = 301 | ||||||
| end | end | ||||||
|  | 
 | ||||||
|  | # Uses Kemal::Middleware::HTTPBasicAuth to easily add HTTP Basic Auth support. | ||||||
|  | macro basic_auth(username, password) | ||||||
|  |   auth_handler = Kemal::Middleware::HTTPBasicAuth.new("serdar", "123") | ||||||
|  |   Kemal.config.add_handler auth_handler | ||||||
|  | end | ||||||
|  |  | ||||||
							
								
								
									
										33
									
								
								src/kemal/middleware/http_basic_auth.cr
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										33
									
								
								src/kemal/middleware/http_basic_auth.cr
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,33 @@ | ||||||
|  | require "base64" | ||||||
|  | 
 | ||||||
|  | module Kemal::Middleware | ||||||
|  |   # This middleware adds HTTP Basic Auth support to your application. | ||||||
|  |   # Returns 401 "Unauthorized" with wrong credentials. | ||||||
|  |   # | ||||||
|  |   # auth_handler = Kemal::Middleware::HTTPBasicAuth.new("username", "password") | ||||||
|  |   # Kemal.config.add_handler auth_handler | ||||||
|  |   # | ||||||
|  |   class HTTPBasicAuth < HTTP::Handler | ||||||
|  |     BASIC = "Basic" | ||||||
|  |     AUTH  = "Authorization" | ||||||
|  | 
 | ||||||
|  |     def initialize(@username, @password) | ||||||
|  |     end | ||||||
|  | 
 | ||||||
|  |     def call(request) | ||||||
|  |       if request.headers[AUTH]? | ||||||
|  |         if value = request.headers[AUTH] | ||||||
|  |           if value.size > 0 && value.starts_with?(BASIC) | ||||||
|  |             return call_next(request) if authorized?(value) | ||||||
|  |           end | ||||||
|  |         end | ||||||
|  |       end | ||||||
|  |       HTTP::Response.new(401, "Unauthorized") | ||||||
|  |     end | ||||||
|  | 
 | ||||||
|  |     def authorized?(value) | ||||||
|  |       username, password = Base64.decode_string(value[BASIC.size + 1..-1]).split(":") | ||||||
|  |       @username == username && @password == password | ||||||
|  |     end | ||||||
|  |   end | ||||||
|  | end | ||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue