From 7d66cef6bc7447189a0339daf62ce46222bf38a5 Mon Sep 17 00:00:00 2001
From: Alex Kotov <kotovalexarian@gmail.com>
Date: Wed, 10 May 2023 00:43:15 +0400
Subject: [PATCH] Use middleware Rack::Head, Rack::Protection

---
 diversipub.gemspec         | 7 ++++---
 lib/diversipub/rack_app.rb | 2 ++
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/diversipub.gemspec b/diversipub.gemspec
index 14872a4..3612823 100644
--- a/diversipub.gemspec
+++ b/diversipub.gemspec
@@ -43,9 +43,10 @@ Gem::Specification.new do |spec|
 
   spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename f }
 
-  spec.add_runtime_dependency 'puma',    '~> 6.2', '>= 6.2.2'
-  spec.add_runtime_dependency 'rack',    '~> 2.2', '>= 2.2.7'
-  spec.add_runtime_dependency 'sinatra', '~> 3.0', '>= 3.0.6'
+  spec.add_runtime_dependency 'puma',            '~> 6.2', '>= 6.2.2'
+  spec.add_runtime_dependency 'rack',            '~> 2.2', '>= 2.2.7'
+  spec.add_runtime_dependency 'rack-protection', '~> 3.0', '>= 3.0.6'
+  spec.add_runtime_dependency 'sinatra',         '~> 3.0', '>= 3.0.6'
 
   spec.add_development_dependency 'bundler',             '~> 2.4'
   spec.add_development_dependency 'pry',                 '~> 0.14'
diff --git a/lib/diversipub/rack_app.rb b/lib/diversipub/rack_app.rb
index 53ff43c..4d63f86 100644
--- a/lib/diversipub/rack_app.rb
+++ b/lib/diversipub/rack_app.rb
@@ -21,7 +21,9 @@ module Diversipub
 
     def builder
       @builder ||= Rack::Builder.new.tap do |builder|
+        builder.use Rack::Head
         builder.use Rack::Static, **rack_static_kwargs
+        builder.use Rack::Protection
         builder.run @sinatra_app
       end
     end