Stirling-PDF/src/main/java/stirling/software/SPDF/config/ConfigInitializer.java

package stirling.software.SPDF.config;

import java.io.BufferedReader;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.URISyntaxException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;

import org.springframework.context.ApplicationContextInitializer;
import org.springframework.context.ConfigurableApplicationContext;

public class ConfigInitializer
        implements ApplicationContextInitializer<ConfigurableApplicationContext> {

    @Override
    public void initialize(ConfigurableApplicationContext applicationContext) {
        try {
            ensureConfigExists();
        } catch (Exception e) {
            throw new RuntimeException("Failed to initialize application configuration", e);
        }
    }

    public void ensureConfigExists() throws IOException, URISyntaxException {
        // Define the path to the external config directory
        Path destPath = Paths.get("configs", "settings.yml");

        // Check if the file already exists
        if (Files.notExists(destPath)) {
            // Ensure the destination directory exists
            Files.createDirectories(destPath.getParent());

            // Copy the resource from classpath to the external directory
            try (InputStream in =
                    getClass().getClassLoader().getResourceAsStream("settings.yml.template")) {
                if (in != null) {
                    Files.copy(in, destPath);
                } else {
                    throw new FileNotFoundException(
                            "Resource file not found: settings.yml.template");
                }
            }
        } else {
            // Load the template content from classpath
            List<String> templateLines;
            try (InputStream in =
                    getClass().getClassLoader().getResourceAsStream("settings.yml.template")) {
                if (in == null) {
                    throw new FileNotFoundException(
                            "Resource file not found: settings.yml.template");
                }
                templateLines = new ArrayList<>();
                try (var reader = new InputStreamReader(in)) {
                    try (var bufferedReader = new BufferedReader(reader)) {
                        String line;
                        while ((line = bufferedReader.readLine()) != null) {
                            templateLines.add(line);
                        }
                    }
                }
            }

            // Read the user settings file if it exists
            Path userPath = Paths.get("configs", "settings.yml");
            List<String> userLines =
                    Files.exists(userPath) ? Files.readAllLines(userPath) : new ArrayList<>();

            List<String> resultLines = new ArrayList<>();
            int position = 0;
            for (String templateLine : templateLines) {
                // Check if the line is a comment
                if (templateLine.trim().startsWith("#")) {
                    String entry = templateLine.trim().substring(1).trim();
                    if (!entry.isEmpty()) {
                        // Check if this comment has been uncommented in userLines
                        String key = entry.split(":")[0].trim();
                        addLine(resultLines, userLines, templateLine, key, position);
                    } else {
                        resultLines.add(templateLine);
                    }
                }
                // Check if the line is a key-value pair
                else if (templateLine.contains(":")) {
                    String key = templateLine.split(":")[0].trim();
                    addLine(resultLines, userLines, templateLine, key, position);
                }
                // Handle empty lines
                else if (templateLine.trim().length() == 0) {
                    resultLines.add("");
                }
                position++;
            }

            // Write the result to the user settings file
            Files.write(userPath, resultLines);
        }

        // Ensure the custom settings file exists
        Path customSettingsPath = Paths.get("configs", "custom_settings.yml");
        if (!Files.exists(customSettingsPath)) {
            Files.createFile(customSettingsPath);
        }
    }

    // TODO check parent value instead of just indent lines for duplicate keys (like enabled etc)
    private static void addLine(
            List<String> resultLines,
            List<String> userLines,
            String templateLine,
            String key,
            int position) {
        boolean added = false;
        int templateIndentationLevel = getIndentationLevel(templateLine);
        int pos = 0;
        for (String settingsLine : userLines) {
            if (settingsLine.trim().startsWith(key + ":") && position == pos) {
                int settingsIndentationLevel = getIndentationLevel(settingsLine);
                // Check if it is correct settingsLine and has the same parent as templateLine
                if (settingsIndentationLevel == templateIndentationLevel) {
                    resultLines.add(settingsLine);
                    added = true;
                    break;
                }
            }
            pos++;
        }
        if (!added) {
            resultLines.add(templateLine);
        }
    }

    private static int getIndentationLevel(String line) {
        int indentationLevel = 0;
        String trimmedLine = line.trim();
        if (trimmedLine.startsWith("#")) {
            line = trimmedLine.substring(1);
        }
        for (char c : line.toCharArray()) {
            if (c == ' ') {
                indentationLevel++;
            } else {
                break;
            }
        }
        return indentationLevel;
    }
}