lib/dassets-erubi.rb in dassets-erubi-0.0.1 vs lib/dassets-erubi.rb in dassets-erubi-0.1.0

@@ -2,8 +2,21 @@
 
 require "erubi"
 require "dassets/engine"
 require "dassets-erubi/version"
 
-module Dassets; end
-module Dassets::Erubi
+module Dassets::Erubi; end
+class Dassets::Erubi::Engine < Dassets::Engine
+  def self.ERB_EXTENSIONS
+    ["erb", "erubis", "erubi"]
+  end
+
+  def ext(input_ext)
+    return "" if self.class.ERB_EXTENSIONS.include?(input_ext)
+
+    input_ext
+  end
+
+  def compile(input_content)
+    eval(::Erubi::Engine.new(input_content, freeze: true).src)
+  end
 end