Hyperion: Add AI code generation assistance for template, solution, and tests repositories

src/main/java/de/tum/cit/aet/artemis/buildagent/service/BuildJobContainerService.java

@@ -450,13 +450,29 @@ private void addDirectory(String containerId, String directoryName, boolean crea
     }
 
     private void copyToContainer(Path sourcePath, String containerId) {
-        try (final var uploadStream = new ByteArrayInputStream(createTarArchive(sourcePath).toByteArray());
-                final var copyToContainerCommand = buildAgentConfiguration.getDockerClient().copyArchiveToContainerCmd(containerId)
-                        .withRemotePath(LOCAL_CI_DOCKER_CONTAINER_WORKING_DIRECTORY).withTarInputStream(uploadStream)) {
-            copyToContainerCommand.exec();
-        }
-        catch (IOException e) {
-            throw new LocalCIException("Could not copy to container " + containerId, e);
+        final int maxRetries = 3;
+        final long retryDelayMs = 1000;
+
+        for (int attempt = 1; attempt <= maxRetries; attempt++) {
+            try (final var uploadStream = new ByteArrayInputStream(createTarArchive(sourcePath).toByteArray());
+                    final var copyToContainerCommand = buildAgentConfiguration.getDockerClient().copyArchiveToContainerCmd(containerId)
+                            .withRemotePath(LOCAL_CI_DOCKER_CONTAINER_WORKING_DIRECTORY).withTarInputStream(uploadStream)) {
+                copyToContainerCommand.exec();
+                return;
+            }
+            catch (Exception e) {
+                if (attempt == maxRetries) {
+                    throw new LocalCIException("Could not copy to container " + containerId + " after " + maxRetries + " attempts", e);
+                }
+                log.warn("Attempt {} failed to copy to container {}: {}. Retrying in {} ms...", attempt, containerId, e.getMessage(), retryDelayMs);
+                try {
+                    Thread.sleep(retryDelayMs);
+                }
+                catch (InterruptedException ie) {
+                    Thread.currentThread().interrupt();
+                    throw new LocalCIException("Interrupted while retrying copy to container " + containerId, ie);
+                }
+            }
         }
     }
 

src/main/java/de/tum/cit/aet/artemis/hyperion/dto/ArtifactType.java

@@ -1,8 +1,27 @@
 package de.tum.cit.aet.artemis.hyperion.dto;
 
+import com.fasterxml.jackson.annotation.JsonValue;
+
 /**
  * Enum for artifact types in consistency checks.
  */
 public enum ArtifactType {
-    PROBLEM_STATEMENT, TEMPLATE_REPOSITORY, SOLUTION_REPOSITORY, TESTS_REPOSITORY
+
+    PROBLEM_STATEMENT("PROBLEM_STATEMENT"), TEMPLATE_REPOSITORY("TEMPLATE_REPOSITORY"), SOLUTION_REPOSITORY("SOLUTION_REPOSITORY"), TESTS_REPOSITORY("TESTS_REPOSITORY");
+
+    private final String value;
+
+    ArtifactType(String value) {
+        this.value = value;
+    }
+
+    @JsonValue
+    public String getValue() {
+        return value;
+    }
+
+    @Override
+    public String toString() {
+        return value;
+    }
 }

src/main/java/de/tum/cit/aet/artemis/hyperion/dto/CodeGenerationRequestDTO.java

@@ -0,0 +1,15 @@
+package de.tum.cit.aet.artemis.hyperion.dto;
+
+import jakarta.validation.constraints.NotNull;
+
+import com.fasterxml.jackson.annotation.JsonInclude;
+
+import de.tum.cit.aet.artemis.programming.domain.RepositoryType;
+
+/**
+ * DTO for requesting code generation for a programming exercise.
+ * Contains the repository type to determine which generation strategy to use.
+ */
+@JsonInclude(JsonInclude.Include.NON_EMPTY)
+public record CodeGenerationRequestDTO(@NotNull RepositoryType repositoryType) {
+}

src/main/java/de/tum/cit/aet/artemis/hyperion/dto/CodeGenerationResponseDTO.java

@@ -0,0 +1,30 @@
+package de.tum.cit.aet.artemis.hyperion.dto;
+
+import java.util.List;
+
+import com.fasterxml.jackson.annotation.JsonInclude;
+
+/**
+ * DTO for internal code generation response.
+ * Contains the generated content from AI service calls.
+ */
+@JsonInclude(JsonInclude.Include.NON_EMPTY)
+public record CodeGenerationResponseDTO(
+        /**
+         * The solution plan as a string
+         */
+        String solutionPlan,
+
+        /**
+         * List of generated files with path and content
+         */
+        List<GeneratedFile> files) {
+
+    public String getSolutionPlan() {
+        return solutionPlan;
+    }
+
+    public List<GeneratedFile> getFiles() {
+        return files;
+    }
+}

src/main/java/de/tum/cit/aet/artemis/hyperion/dto/CodeGenerationResultDTO.java

@@ -0,0 +1,25 @@
+package de.tum.cit.aet.artemis.hyperion.dto;
+
+import com.fasterxml.jackson.annotation.JsonInclude;
+
+/**
+ * DTO for REST API code generation result.
+ * Contains the result of the code generation and compilation process.
+ */
+@JsonInclude(JsonInclude.Include.NON_EMPTY)
+public record CodeGenerationResultDTO(
+        /**
+         * Whether the code generation and compilation was successful
+         */
+        boolean success,
+
+        /**
+         * Descriptive message about the generation result
+         */
+        String message,
+
+        /**
+         * Number of attempts made during the generation process
+         */
+        int attempts) {
+}

src/main/java/de/tum/cit/aet/artemis/hyperion/dto/GeneratedFile.java

@@ -0,0 +1,4 @@
+package de.tum.cit.aet.artemis.hyperion.dto;
+
+public record GeneratedFile(String path, String content) {
+}

src/main/java/de/tum/cit/aet/artemis/hyperion/service/HyperionProgrammingExerciseContextRendererService.java

@@ -90,6 +90,11 @@ public class HyperionProgrammingExerciseContextRendererService {
 
     private static final Logger log = LoggerFactory.getLogger(HyperionProgrammingExerciseContextRendererService.class);
 
+    private static final Set<String> EXCLUDED_DIRECTORIES = Set.of(".git", ".idea", ".vscode", "target", "build", "out", "bin", "node_modules", ".gradle", ".mvn", "dist", ".next",
+            "coverage");
+
+    private static final Set<String> EXCLUDED_FILES = Set.of(".DS_Store", "Thumbs.db", ".gitkeep");
+
     private final RepositoryService repositoryService;
 
     private final HyperionProgrammingLanguageContextFilterService languageFilter;
@@ -227,4 +232,79 @@ private static String renderFileString(String root, String path, String content)
         });
         return String.join("\n", out);
     }
+
+    /**
+     * Generates a tree-format representation of the repository structure.
+     * Reads the current state fresh each time to capture any changes.
+     *
+     * @param repository the repository to analyze
+     * @return tree-format string representation of the repository structure
+     */
+    public String getRepositoryStructure(Repository repository) {
+        try {
+            File repositoryRoot = repository.getLocalPath().toFile();
+            if (!repositoryRoot.exists() || !repositoryRoot.isDirectory()) {
+                log.warn("Repository path does not exist or is not a directory: {}", repositoryRoot.getAbsolutePath());
+                return "Repository structure could not be determined.";
+            }
+
+            StringBuilder structure = new StringBuilder();
+            structure.append(repositoryRoot.getName()).append("/").append("\n");
+            generateTreeStructure(repositoryRoot, structure, "", true);
+
+            return structure.toString();
+
+        }
+        catch (Exception e) {
+            log.error("Failed to generate repository structure for repository: {}", repository.getLocalPath(), e);
+            return "Repository structure could not be determined due to an error.";
+        }
+    }
+
+    /**
+     * Recursively generates the tree structure representation.
+     *
+     * @param directory the current directory to process
+     * @param structure the StringBuilder to append to
+     * @param prefix    the current line prefix for proper tree formatting
+     * @param isLast    whether this is the last item in its parent directory
+     */
+    private void generateTreeStructure(File directory, StringBuilder structure, String prefix, boolean isLast) {
+        File[] files = directory.listFiles();
+        if (files == null) {
+            return;
+        }
+
+        // Filter out excluded directories and files
+        File[] filteredFiles = Arrays.stream(files).filter(file -> !EXCLUDED_DIRECTORIES.contains(file.getName()) && !EXCLUDED_FILES.contains(file.getName())).sorted((a, b) -> {
+            // Directories first, then files
+            if (a.isDirectory() && !b.isDirectory()) {
+                return -1;
+            }
+            else if (!a.isDirectory() && b.isDirectory()) {
+                return 1;
+            }
+            return a.getName().compareToIgnoreCase(b.getName());
+        }).toArray(File[]::new);
+
+        for (int i = 0; i < filteredFiles.length; i++) {
+            File file = filteredFiles[i];
+            boolean isLastFile = (i == filteredFiles.length - 1);
+
+            structure.append(prefix);
+            structure.append(isLastFile ? "└── " : "├── ");
+            structure.append(file.getName());
+
+            if (file.isDirectory()) {
+                structure.append("/");
+            }
+            structure.append("\n");
+
+            // Recursively process subdirectories
+            if (file.isDirectory()) {
+                String newPrefix = prefix + (isLastFile ? "    " : "│   ");
+                generateTreeStructure(file, structure, newPrefix, false);
+            }
+        }
+    }
 }

src/main/java/de/tum/cit/aet/artemis/hyperion/service/HyperionPromptTemplateService.java

@@ -39,4 +39,27 @@ public String render(String resourcePath, Map<String, String> variables) {
             throw new RuntimeException("Failed to load template: " + resourcePath, e);
         }
     }
+
+    /**
+     * Render the template at the given classpath resource path with the provided variables.
+     * <p>
+     * Supporting placeholders of the form {{var}}
+     *
+     * @param resourcePath classpath to the template resource
+     * @param variables    map of variables used during rendering
+     * @return the rendered string
+     */
+    public String renderObject(String resourcePath, Map<String, Object> variables) {
+        try {
+            var resource = new ClassPathResource(resourcePath);
+            String rendered = StreamUtils.copyToString(resource.getInputStream(), StandardCharsets.UTF_8);
+            for (var entry : variables.entrySet()) {
+                rendered = rendered.replace("{{" + entry.getKey() + "}}", entry.getValue().toString());
+            }
+            return rendered;
+        }
+        catch (IOException e) {
+            throw new RuntimeException("Failed to load template: " + resourcePath, e);
+        }
+    }
 }