detect enum in FileNameConsistent

xuwei-k · Jan 16, 2024 · 950221d · 950221d
1 parent 91d4234
commit 950221d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/input/src/main/scala-3/fix/FileNameConsistentTest2.scala b/input/src/main/scala-3/fix/FileNameConsistentTest2.scala
@@ -0,0 +1,9 @@
+/*
+rule = FileNameConsistent
+ */
+package fix
+
+enum MyTopLevelEnum { // assert: FileNameConsistent
+  case X1(x: Int)
+  case X2(x: String)
+}
diff --git a/rules/src/main/scala/fix/FileNameConsistent.scala b/rules/src/main/scala/fix/FileNameConsistent.scala
@@ -23,6 +23,8 @@ class FileNameConsistent extends SyntacticRule("FileNameConsistent") {
         TemplateDef(x, x.name.value)
       case x: Defn.Object if x.isTopLevel =>
         TemplateDef(x, x.name.value)
+      case x: Defn.Enum if x.isTopLevel =>
+        TemplateDef(x, x.name.value)
     }
     val packageObjects = doc.tree.collect { case x: Pkg.Object =>
       TemplateDef(x, x.name.value)