lib/hamlit/cli.rb in hamlit-2.8.1 vs lib/hamlit/cli.rb in hamlit-2.8.2

@@ -16,12 +16,18 @@
       puts eval(code)
     end
 
     desc 'compile HAML', 'Show compile result'
     option :actionview, type: :boolean, default: false, aliases: %w[-a]
+    option :color, type: :boolean, default: false, aliases: %w[-c]
     def compile(file)
-      print_code generate_code(file)
+      code = generate_code(file)
+      if options[:color]
+        colored_puts generate_code(file)
+      else
+        puts code
+      end
     end
 
     desc 'temple HAML', 'Show temple intermediate expression'
     def temple(file)
       pp generate_temple(file)
@@ -93,14 +99,12 @@
     def method_missing(*args)
       return super(*args) if args.length > 1
       render(args.first.to_s)
     end
 
-    def print_code(code)
+    def colored_puts(code)
       require 'pry'
       puts Pry.Code(code).highlighted
-    rescue LoadError
-      puts code
     end
 
     # Enable colored pretty printing only for development environment.
     def pp(arg)
       require 'pry'