import org.example.PlantUMLGenerator;

import org.example.SimpleTransportProtocolParser;

import tools.samt.api.plugin.CodegenFile;

import tools.samt.codegen.Codegen;

import tools.samt.common.DiagnosticController;

import tools.samt.common.FilesKt;

import tools.samt.common.SamtGeneratorConfiguration;

import tools.samt.config.SamtConfigurationParser;

import tools.samt.lexer.Lexer;

import tools.samt.parser.Parser;

import tools.samt.semantic.SemanticModel;

import java.io.StringReader;

import java.nio.file.Path;

public class Main {

public static void main(String[] args) {

// Parse saml.yaml from the test resources

var config = SamtConfigurationParser.parseConfiguration(Path.of("src/test/resources/samt.yaml"));

var sourceDirectory = config.getSource().toUri();

// This will hold all diagnostic messages (errors, warnings, etc.)

var diagnosticController = new DiagnosticController(sourceDirectory);

// Parse all .samt files in the test resources

var sourceFiles = FilesKt.readSamtSource(FilesKt.collectSamtFiles(sourceDirectory), diagnosticController);

// Parse all files and build a semantic model

var fileNodes = sourceFiles.stream().map(sourceFile -> {

var diagnostics = diagnosticController.getOrCreateContext(sourceFile);

var tokens = Lexer.scan(new StringReader(sourceFile.getContent()), diagnostics);

return Parser.parse(sourceFile, tokens, diagnostics);

}).toList();

var semanticModel = SemanticModel.build(fileNodes, diagnosticController);

// Register custom transport protocol parser and generator

Codegen.registerTransportParser(new SimpleTransportProtocolParser());

Codegen.registerGenerator(new PlantUMLGenerator());

if (diagnosticController.hasErrors()) {

// Nicely printing all errors isn't easy here, it's best to use a debugger to analyze the diagnosticController

System.out.println("Has errors!");

return;

}

// Run all configured generators

for (SamtGeneratorConfiguration generatorConfiguration : config.getGenerators()) {

// Print all generated files

for (CodegenFile codegenFile : Codegen.generate(semanticModel, generatorConfiguration, diagnosticController)) {

System.out.println(codegenFile.getFilepath());

System.out.println("-".repeat(codegenFile.getFilepath().length()));

System.out.println(codegenFile.getSource());

}

}

}

}