Add JS support to LanguageComposition recipe

openrewrite · Nov 23, 2024 · 885dd3d · 885dd3d
1 parent 7cb42eb
commit 885dd3d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/build.gradle.kts b/build.gradle.kts
@@ -35,6 +35,7 @@ dependencies {
     implementation("org.openrewrite:rewrite-python:$latest")
     implementation("org.openrewrite:rewrite-kotlin:$latest")
     implementation("org.openrewrite:rewrite-csharp:$latest")
+    implementation("org.openrewrite:rewrite-javascript:$latest")
 
     testRuntimeOnly("org.openrewrite:rewrite-java-17")
 }
diff --git a/src/main/java/org/openrewrite/LanguageComposition.java b/src/main/java/org/openrewrite/LanguageComposition.java
@@ -24,6 +24,7 @@
 import org.openrewrite.groovy.tree.G;
 import org.openrewrite.hcl.tree.Hcl;
 import org.openrewrite.java.tree.J;
+import org.openrewrite.javascript.tree.JS;
 import org.openrewrite.json.tree.Json;
 import org.openrewrite.kotlin.tree.K;
 import org.openrewrite.properties.tree.Properties;
@@ -184,6 +185,18 @@ public TreeVisitor<?, ExecutionContext> getScanner(Accumulator acc) {
                                 s.getClass().getName(),
                                 genericLineCount,
                                 hasParseFailure));
+                    } else if (s instanceof JS.CompilationUnit) {
+                        Counts javaScriptCounts = acc.getFolderToLanguageToCounts()
+                                .computeIfAbsent(folderPath, k -> new HashMap<>())
+                                .computeIfAbsent("JavaScript", k -> new Counts());
+                        javaScriptCounts.fileCount++;
+                        javaScriptCounts.lineCount += genericLineCount;
+                        perFileReport.insertRow(ctx, new LanguageCompositionPerFile.Row(
+                                s.getSourcePath().toString(),
+                                "JavaScript",
+                                s.getClass().getName(),
+                                genericLineCount,
+                                hasParseFailure));
                     } else if (s instanceof J) {
                         Counts javaCounts = acc.getFolderToLanguageToCounts()
                                 .computeIfAbsent(folderPath, k -> new HashMap<>())