diff --git a/lib/letsencrypt-rails-heroku/railtie.rb b/lib/letsencrypt-rails-heroku/railtie.rb index 636da62..11699ae 100644 --- a/lib/letsencrypt-rails-heroku/railtie.rb +++ b/lib/letsencrypt-rails-heroku/railtie.rb @@ -3,6 +3,14 @@ class LetsencryptRailsHerokuRailtie < Rails::Railtie Letsencrypt.configure end + initializer "letsencrypt_rails_heroku.configure_rails_initialization" do |app| + if app.config.force_ssl + app.middleware.insert_before ActionDispatch::SSL, Letsencrypt::Middleware + else + app.middleware.use Letsencrypt::Middleware + end + end + rake_tasks do load 'tasks/letsencrypt.rake' end