html beauti

README.md

@@ -49,6 +49,13 @@ plain '<div>Inline html</div>'.html_safe
 render partial: 'shared/navbar'
 ```
 
+### (Optional) Nicely Format the HTML for source inspection
+
+File: `config/environments/development.rb` or `config/environments/test.rb`
+```ruby
+config.middleware.use Ruby2html::HtmlBeautifierMiddleware
+```
+
 #### Or use your current .erb views
 
 ### In your ApplicationController

app/views/home/rb_files.html.rb

@@ -2,7 +2,7 @@
 
 h1 'RbFiles'
 h1 'ok'
-h1 'ok'
+h1 'okddwadwa'
 
 div @value
 

config/environments/development.rb

@@ -80,4 +80,6 @@
 
   # Apply autocorrection by RuboCop to files generated by `bin/rails generate`.
   # config.generators.apply_rubocop_autocorrect_after_generate!
+
+  config.middleware.use Ruby2html::HtmlBeautifierMiddleware
 end

lib/gem/ruby2html/html_beautifier_middleware.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require 'htmlbeautifier'
+
+module Ruby2html
+  class HtmlBeautifierMiddleware
+    def initialize(app)
+      @app = app
+    end
+
+    def call(env)
+      status, headers, body = @app.call(env)
+
+      if headers['Content-Type']&.include?('text/html')
+        new_body = []
+        body.each do |chunk|
+          new_body << HtmlBeautifier.beautify(chunk)
+        end
+        headers['Content-Length'] = new_body.map(&:bytesize).sum.to_s
+        body = new_body
+      end
+
+      [status, headers, body]
+    end
+  end
+end

lib/gem/ruby2html/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Ruby2html
-  VERSION = '1.3.3'
+  VERSION = '1.4.0'
 end

ruby2html.gemspec

@@ -32,4 +32,6 @@ Gem::Specification.new do |spec|
     end
   end
   spec.require_paths = ['lib/gem']
+
+  spec.add_dependency 'htmlbeautifier', '>= 1.4'
 end