Add support for linting ECR files (pt. 2)

README.md

@@ -174,6 +174,7 @@ In this example we define default globs and exclude `src/compiler` folder:
 ``` yaml
 Globs:
   - "**/*.cr"
+  - "**/*.ecr"
   - "!lib"
 
 Excluded:

spec/ameba/source_spec.cr

@@ -70,5 +70,31 @@ module Ameba
           CRYSTAL
       end
     end
+
+    Ameba.ecr_supported? do
+      describe "#ast" do
+        it "parses an ECR file" do
+          source = Source.new <<-ECR, "filename.ecr"
+            hello <%= "world" %>
+            ECR
+
+          source.ast.to_s.should eq(<<-CRYSTAL)
+            __str__ << "hello "
+            ("world").to_s(__str__)
+
+            CRYSTAL
+        end
+
+        it "raises an exception when ECR parsing fails" do
+          source = Source.new <<-ECR, "filename.ecr"
+            hello <%= "world" >
+            ECR
+
+          expect_raises(Crystal::SyntaxException) do
+            source.ast
+          end
+        end
+      end
+    end
   end
 end

src/ameba.cr

@@ -1,11 +1,3 @@
-require "./ameba/*"
-require "./ameba/ast/**"
-require "./ameba/ext/**"
-require "./ameba/rule/**"
-require "./ameba/formatter/*"
-require "./ameba/presenter/*"
-require "./ameba/source/**"
-
 # Ameba's entry module.
 #
 # To run the linter with default parameters:
@@ -40,4 +32,18 @@ module Ameba
   def run(config = Config.load)
     Runner.new(config).run
   end
+
+  macro ecr_supported?(&)
+    {% if compare_versions(Crystal::VERSION, "1.15.0") >= 0 %}
+      {{ yield }}
+    {% end %}
+  end
 end
+
+require "./ameba/*"
+require "./ameba/ast/**"
+require "./ameba/ext/**"
+require "./ameba/rule/**"
+require "./ameba/formatter/*"
+require "./ameba/presenter/*"
+require "./ameba/source/**"

src/ameba/config.cr

@@ -1,5 +1,6 @@
 require "semantic_version"
 require "yaml"
+require "ecr/processor"
 require "./glob_utils"
 
 # A configuration entry for `Ameba::Runner`.
@@ -62,6 +63,10 @@ class Ameba::Config
     !lib
   )
 
+  Ameba.ecr_supported? do
+    DEFAULT_GLOBS << "**/*.ecr"
+  end
+
   getter rules : Array(Rule::Base)
   property severity = Severity::Convention
 
@@ -167,7 +172,7 @@ class Ameba::Config
   # ```
   # config = Ameba::Config.load
   # config.sources # => list of default sources
-  # config.globs = ["**/*.cr"]
+  # config.globs = ["**/*.cr", "**/*.ecr"]
   # config.excluded = ["spec"]
   # config.sources # => list of sources pointing to files found by the wildcards
   # ```

src/ameba/glob_utils.cr

@@ -24,7 +24,14 @@ module Ameba
     def expand(globs)
       globs
         .flat_map do |glob|
-          glob += "/**/*.cr" if File.directory?(glob)
+          if File.directory?(glob)
+            glob += "/**/*.cr"
+
+            Ameba.ecr_supported? do
+              glob += "/**/*.ecr"
+            end
+          end
+
           Dir[glob]
         end
         .uniq!

src/ameba/source.cr

@@ -57,6 +57,24 @@ module Ameba
     # source.ast
     # ```
     getter ast : Crystal::ASTNode do
+      code = @code
+
+      Ameba.ecr_supported? do
+        if @path.ends_with?(".ecr")
+          begin
+            code = ECR.process_string(code, @path)
+          rescue ex : ECR::Lexer::SyntaxException
+            # Need to rescue to add the filename
+            raise Crystal::SyntaxException.new(
+              ex.message,
+              ex.line_number,
+              ex.column_number,
+              @path
+            )
+          end
+        end
+      end
+
       Crystal::Parser.new(code)
         .tap(&.wants_doc = true)
         .tap(&.filename = path)