Add new tasks to enhance AI workflow. (#837)

Author: kimeta

tests/tck-build-logic/src/main/groovy/org.graalvm.internal.tck-harness.gradle

@@ -30,6 +30,7 @@ import org.graalvm.internal.tck.harness.tasks.FetchExistingLibrariesWithNewerVer
 import org.gradle.util.internal.VersionNumber
 import org.graalvm.internal.tck.harness.tasks.ComputeAndPullAllowedDockerImagesTask
 import org.graalvm.internal.tck.utils.CoordinateUtils
+import org.graalvm.internal.tck.utils.MetadataGenerationUtils
 
 
 import static org.graalvm.internal.tck.Utils.generateTaskName
@@ -334,3 +335,31 @@ tasks.register("fixTestNativeImageRun", FixTestNativeImageRun.class) { task ->
     task.setDescription("Fix test that fails Native Image run for new library version.")
     task.setGroup(METADATA_GROUP)
 }
+
+// gradle addNewEntryTestsIndexJson -Pcoordinates=<maven-coordinates>
+tasks.register("addNewEntryTestsIndexJson", DefaultTask) { task ->
+    task.setDescription("Adds a new entry to tests/src/index.json for a new library version.")
+    task.setGroup(METADATA_GROUP)
+    task.doFirst {
+        if (!project.hasProperty("coordinates")) {
+            throw new GradleException("Missing 'coordinates' property! Rerun Gradle with -Pcoordinates=<group:artifact:version>")
+        }
+        def coordsStr = project.findProperty("coordinates").toString()
+        def coords = CoordinateUtils.fromString(coordsStr)
+        MetadataGenerationUtils.addNewEntryToTestsIndex(project.layout, coords)
+    }
+}
+
+// gradle addLibraryAsLatestMetadataIndexJson -Pcoordinates=<maven-coordinates>
+tasks.register("addLibraryAsLatestMetadataIndexJson", DefaultTask) { task ->
+    task.setDescription("Marks given coordinates as latest in metadata/group/artifact/index.json")
+    task.setGroup(METADATA_GROUP)
+    task.doFirst {
+        if (!project.hasProperty("coordinates")) {
+            throw new GradleException("Missing 'coordinates' property! Rerun Gradle with -Pcoordinates=<group:artifact:version>")
+        }
+        def coordsStr = project.findProperty("coordinates").toString()
+        def coords = CoordinateUtils.fromString(coordsStr)
+        MetadataGenerationUtils.makeVersionLatestInIndexJson(project.layout, coords)
+    }
+}

tests/tck-build-logic/src/main/java/org/graalvm/internal/tck/utils/MetadataGenerationUtils.java

@@ -251,4 +251,65 @@ public static void addNewVersionToTestsIndex(ProjectLayout layout, Coordinates t
         }
         Files.writeString(indexPath, json, StandardCharsets.UTF_8);
     }
+
+    /**
+     * Adds a new entry to tests/src/index.json for the given coordinates.
+     * The entry is inserted by lexicographic order of test-project-path value.
+     */
+    public static void addNewEntryToTestsIndex(ProjectLayout layout, Coordinates newCoords) throws IOException {
+        Path indexPath = GeneralUtils.getPathFromProject(layout, "tests/src/index.json");
+        if (!Files.exists(indexPath)) {
+            return;
+        }
+
+        ObjectMapper mapper = new ObjectMapper().enable(SerializationFeature.INDENT_OUTPUT);
+
+        String content = Files.readString(indexPath, StandardCharsets.UTF_8);
+        if (content == null || content.isBlank()) {
+            throw new GradleException("Cannot find test index file at: " + indexPath);
+        }
+
+        List<TestIndexEntry> entries = mapper.readValue(content, new TypeReference<>() {});
+        String group = newCoords.group();
+        String artifact = newCoords.artifact();
+        String newVersion = newCoords.version();
+        String prefix = group + "/" + artifact + "/";
+        String newTestProjectPath = prefix + newVersion;
+
+        // If the new entry already exists, do nothing
+        for (TestIndexEntry e : entries) {
+            if (newTestProjectPath.equals(e.testProjectPath())) {
+                return;
+            }
+        }
+
+        // Build new entry
+        String libName = group + ":" + artifact;
+        List<String> versions = new ArrayList<>();
+        versions.add(newVersion);
+        LibraryEntry libraryEntry = new LibraryEntry(libName, versions);
+        List<LibraryEntry> libraries = new ArrayList<>();
+        libraries.add(libraryEntry);
+        TestIndexEntry newEntry = new TestIndexEntry(newTestProjectPath, libraries);
+
+        // Insert lexicographic order of test-project-path
+        int insertAt = entries.size();
+        for (int i = 0; i < entries.size(); i++) {
+            String tp = entries.get(i).testProjectPath();
+            if (tp != null && newTestProjectPath.compareTo(tp) < 0) {
+                insertAt = i;
+                break;
+            }
+        }
+
+        entries.add(insertAt, newEntry);
+
+        DefaultPrettyPrinter prettyPrinterNew = new DefaultPrettyPrinter();
+        prettyPrinterNew.indentArraysWith(DefaultIndenter.SYSTEM_LINEFEED_INSTANCE);
+        String jsonNew = mapper.writer(prettyPrinterNew).writeValueAsString(entries);
+        if (!jsonNew.endsWith(System.lineSeparator())) {
+            jsonNew = jsonNew + System.lineSeparator();
+        }
+        Files.writeString(indexPath, jsonNew, StandardCharsets.UTF_8);
+    }
 }