Add comment handling to Heredocs

rrrene · Mar 30, 2019 · f2a30e1 · f2a30e1
1 parent dd0d568
commit f2a30e1
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/lib/credo/code/heredocs.ex b/lib/credo/code/heredocs.ex
@@ -28,6 +28,10 @@ defmodule Credo.Code.Heredocs do
     parse_code(t, acc <> "\\\"", replacement, empty_line_replacement)
   end
 
+  defp parse_code(<<"#"::utf8, t::binary>>, acc, replacement, empty_line_replacement) do
+    parse_comment(t, acc <> "#", replacement, empty_line_replacement)
+  end
+
   defp parse_code(<<"?\""::utf8, t::binary>>, acc, replacement, empty_line_replacement) do
     parse_code(t, acc <> "?\"", replacement, empty_line_replacement)
   end
@@ -50,6 +54,20 @@ defmodule Credo.Code.Heredocs do
     parse_code(t, acc <> h, replacement, empty_line_replacement)
   end
 
+  defp parse_comment("", acc, _replacement, _empty_line_replacement) do
+    acc
+  end
+
+  defp parse_comment(<<"\n"::utf8, t::binary>>, acc, replacement, empty_line_replacement) do
+    parse_code(t, acc <> "\n", replacement, empty_line_replacement)
+  end
+
+  defp parse_comment(str, acc, replacement, empty_line_replacement) when is_binary(str) do
+    {h, t} = String.next_codepoint(str)
+
+    parse_comment(t, acc <> h, replacement, empty_line_replacement)
+  end
+
   defp parse_heredoc("", acc, _replacement, _empty_line_replacement) do
     acc
   end