Support Scala 3 dialect based on source path

Author: kaplan-shaked

resources/scala-3/Scala3Class.scala_test

@@ -0,0 +1 @@
+case class Scala3Class(a: Int)(using b: String)

src/FileParser.scala

@@ -1,4 +1,5 @@
 import scala.meta._
+import scala.meta.Dialect
 
 case class ScalaFile(
     imports: List[String],
@@ -44,9 +45,8 @@ object FileParser {
     }
   }
 
-  def parse(content: String): ScalaFile = {
+  def parse(content: String, dialect: Dialect = dialects.Scala213Source3): ScalaFile = {
     val input = Input.String(content)
-    val dialect = dialects.Scala213Source3
     val exampleTree: Source = dialect(input).parse[Source].get
 
     val tree =
@@ -78,6 +78,10 @@ object FileParser {
     val path = java.nio.file.Paths.get(filePath)
     val bytes = java.nio.file.Files.readAllBytes(path)
     val text = new String(bytes, "UTF-8")
-    parse(text)
+    val dialect =
+      if (filePath.split(java.io.File.separator).contains("scala-3"))
+        dialects.Scala3
+      else dialects.Scala213Source3
+    parse(text, dialect)
   }
 }

test/FileParser.test.scala

@@ -63,4 +63,13 @@ class FileParserTest extends munit.FunSuite {
     val parsedFile = FileParser.fromPathToClassDef(file)
     println(parsedFile)
   }
+  test("uses Scala3 dialect for sources in scala-3 directory") {
+    val file = Thread
+      .currentThread()
+      .getContextClassLoader
+      .getResource("scala-3/Scala3Class.scala_test")
+      .getPath
+    val parsedFile = FileParser.fromPathToClassDef(file)
+    assert(parsedFile.classes.head.name == "Scala3Class")
+  }
 }