mirror of
https://gitea.invidious.io/iv-org/shard-kemal.git
synced 2024-08-15 00:53:36 +00:00
Ability to optionally disable powered-by header (#449)
This commit is contained in:
parent
73045bb138
commit
f3f7e319ae
4 changed files with 16 additions and 1 deletions
|
@ -17,6 +17,10 @@ describe "Config" do
|
|||
config.env.should eq "production"
|
||||
end
|
||||
|
||||
it "sets default powered_by_header to true" do
|
||||
Kemal.config.powered_by_header.should be_true
|
||||
end
|
||||
|
||||
it "sets host binding" do
|
||||
config = Kemal.config
|
||||
config.host_binding = "127.0.0.1"
|
||||
|
|
|
@ -19,4 +19,14 @@ describe "Kemal::InitHandler" do
|
|||
Kemal::InitHandler::INSTANCE.call(context)
|
||||
context.response.headers["X-Powered-By"].should eq "Kemal"
|
||||
end
|
||||
|
||||
it "does not initialize context with X-Powered-By: Kemal if disabled" do
|
||||
Kemal.config.powered_by_header = false
|
||||
request = HTTP::Request.new("GET", "/")
|
||||
io = IO::Memory.new
|
||||
response = HTTP::Server::Response.new(io)
|
||||
context = HTTP::Server::Context.new(request, response)
|
||||
Kemal::InitHandler::INSTANCE.call(context)
|
||||
context.response.headers["X-Powered-By"]?.should be_nil
|
||||
end
|
||||
end
|
||||
|
|
|
@ -22,6 +22,7 @@ module Kemal
|
|||
property always_rescue, server : HTTP::Server?, extra_options, shutdown_message
|
||||
property serve_static : (Bool | Hash(String, Bool))
|
||||
property static_headers : (HTTP::Server::Response, String, File::Stat -> Void)?
|
||||
property powered_by_header : Bool = true
|
||||
|
||||
def initialize
|
||||
@host_binding = "0.0.0.0"
|
||||
|
|
|
@ -7,7 +7,7 @@ module Kemal
|
|||
INSTANCE = new
|
||||
|
||||
def call(context : HTTP::Server::Context)
|
||||
context.response.headers.add "X-Powered-By", "Kemal"
|
||||
context.response.headers.add "X-Powered-By", "Kemal" if Kemal.config.powered_by_header
|
||||
context.response.content_type = "text/html" unless context.response.headers.has_key?("Content-Type")
|
||||
call_next context
|
||||
end
|
||||
|
|
Loading…
Reference in a new issue