Use Railtie instead of AcitveSupport.on_load

lib/hamlit.rb

@@ -4,7 +4,6 @@ require 'hamlit/template'
 require 'hamlit/version'
 
 begin
-  require 'rails'
-  require 'hamlit/rails_template'
+  require 'hamlit/railtie'
 rescue LoadError
 end

lib/hamlit/rails_template.rb

@@ -1,18 +0,0 @@
-require 'temple'
-
-module Hamlit
-  RailsTemplate = Temple::Templates::Rails.create(
-    Hamlit::Engine,
-    generator:   Temple::Generators::RailsOutputBuffer,
-    register_as: :haml,
-    escape_html: true,
-    streaming:   true,
-    ugly:        true,
-  )
-
-  ActiveSupport.on_load(:action_view) do
-    # Haml extends Haml::Helpers in action view each time.
-    # It costs much, so Hamlit includes a compatible module at first.
-    ActionView::Base.send :include, Hamlit::Helpers
-  end
-end

lib/hamlit/railtie.rb

@@ -0,0 +1,21 @@
+require 'rails'
+require 'temple'
+
+module Hamlit
+  class Railtie < ::Rails::Railtie
+    initializer :hamlit do |app|
+      Hamlit::RailsTemplate = Temple::Templates::Rails.create(
+        Hamlit::Engine,
+        generator:   Temple::Generators::RailsOutputBuffer,
+        register_as: :haml,
+        escape_html: true,
+        streaming:   true,
+        ugly:        true,
+      )
+
+      # Haml extends Haml::Helpers in ActionView each time.
+      # It costs much, so Hamlit includes a compatible module at first.
+      ActionView::Base.send :include, Hamlit::Helpers
+    end
+  end
+end