Use CsvDataTableStore for data table export (#1125)

* Use CsvDataTableStore for data table export

Replace recipeRun.exportDatatablesToCsv() with CsvDataTableStore set on
the ExecutionContext before recipe execution.

Depends on: openrewrite/rewrite#7088

Author: jkschneider

src/main/java/org/openrewrite/maven/AbstractRewriteBaseRunMojo.java

@@ -244,13 +244,20 @@ protected LargeSourceSet loadSourceSet(Path repositoryRoot, Environment env, Exe
 
     protected List<Result> runRecipe(Recipe recipe, LargeSourceSet sourceSet, ExecutionContext ctx) {
         getLog().info("Running recipe(s)...");
-        RecipeRun recipeRun = recipe.run(sourceSet, ctx);
 
+        CsvDataTableStore csvDataTableStore = null;
         if (exportDatatables) {
             String timestamp = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd_HH-mm-ss-SSS"));
             Path datatableDirectoryPath = Paths.get("target", "rewrite", "datatables", timestamp);
             getLog().info(String.format("Printing available datatables to: %s", datatableDirectoryPath));
-            recipeRun.exportDatatablesToCsv(datatableDirectoryPath, ctx);
+            csvDataTableStore = new CsvDataTableStore(datatableDirectoryPath);
+            DataTableExecutionContextView.view(ctx).setDataTableStore(csvDataTableStore);
+        }
+
+        RecipeRun recipeRun = recipe.run(sourceSet, ctx);
+
+        if (csvDataTableStore != null) {
+            csvDataTableStore.close();
         }
 
         return recipeRun.getChangeset().getAllResults().stream().filter(source -> {

src/test/java/org/openrewrite/maven/RewriteRunIT.java

@@ -223,20 +223,30 @@ void datatable_export(MavenExecutionResult result) throws Exception {
                 .findFirst()
                 .orElseThrow(() -> new AssertionError("No timestamped directory found in " + datatablesDir));
 
-            csvFile = timestampedDir.resolve("org.openrewrite.maven.table.DependenciesInUse.csv");
+            try (Stream<Path> csvFiles = Files.list(timestampedDir)) {
+                csvFile = csvFiles
+                    .filter(p -> p.getFileName().toString().startsWith("org.openrewrite.maven.table.DependenciesInUse") &&
+                                 p.getFileName().toString().endsWith(".csv"))
+                    .findFirst()
+                    .orElseThrow(() -> new AssertionError("No DependenciesInUse CSV file found in " + timestampedDir));
+            }
             assertThat(csvFile).exists().isRegularFile();
         }
 
         // Verify CSV contains expected structure and data rows
-        // CSV format: header row, description row, data rows
+        // CSV format: 3 comment lines (@name, @instanceName, @group), header row, data rows
         List<String> lines = Files.readAllLines(csvFile);
-        assertThat(lines)
-          .hasSizeGreaterThanOrEqualTo(4) // header + description + 2 data rows
+        // Filter out comment lines
+        List<String> dataLines = lines.stream()
+                .filter(l -> !l.startsWith("#"))
+                .collect(java.util.stream.Collectors.toList());
+        assertThat(dataLines)
+          .hasSizeGreaterThanOrEqualTo(3) // header + 2 data rows
           .first(as(STRING))
           .contains("Group", "Artifact"); // CSV header
 
-        // Get only the data rows (skip header and description rows)
-        assertThat(lines.subList(2, lines.size()))
+        // Get only the data rows (skip header)
+        assertThat(dataLines.subList(1, dataLines.size()))
           .hasSizeGreaterThanOrEqualTo(2)
           .anySatisfy(line -> assertThat(line).contains("com.google.guava", "guava"))
           .anySatisfy(line -> assertThat(line).contains("org.projectlombok", "lombok"));