Merge pull request #13 from samtkit/codegen

Codegen

Author: PascalHonegger

.gitignore

@@ -14,5 +14,10 @@ build
 ehthumbs.db
 Thumbs.db
 
+# SAMT wrapper generated files #
+.samt
+samtw
+samtw.bat
+
 # Random files used for debugging
 specification/examples/debug.samt

build.gradle.kts

@@ -14,6 +14,7 @@ dependencies {
     kover(project(":cli"))
     kover(project(":language-server"))
     kover(project(":samt-config"))
+    kover(project(":codegen"))
 }
 
 koverReport {

cli/build.gradle.kts

@@ -14,6 +14,9 @@ dependencies {
     implementation(project(":lexer"))
     implementation(project(":parser"))
     implementation(project(":semantic"))
+    implementation(project(":samt-config"))
+    implementation(project(":codegen"))
+    implementation(project(":public-api"))
 }
 
 application {

cli/src/main/kotlin/tools/samt/cli/CliArgs.kt

@@ -10,8 +10,8 @@ class CliArgs {
 
 @Parameters(commandDescription = "Compile SAMT files")
 class CompileCommand {
-    @Parameter(description = "Files to compile, defaults to all .samt files in the current directory")
-    var files: List<String> = mutableListOf()
+    @Parameter(description = "SAMT project to compile, defaults to the 'samt.yaml' file in the current directory")
+    var file: String = "./samt.yaml"
 }
 
 @Parameters(commandDescription = "Dump SAMT files in various formats for debugging purposes")
@@ -25,8 +25,8 @@ class DumpCommand {
     @Parameter(names = ["--types"], description = "Dump a visual representation of the resolved types")
     var dumpTypes: Boolean = false
 
-    @Parameter(description = "Files to dump, defaults to all .samt files in the current directory")
-    var files: List<String> = mutableListOf()
+    @Parameter(description = "SAMT project to dump, defaults to the 'samt.yaml' file in the current directory")
+    var file: String = "./samt.yaml"
 }
 
 @Parameters(commandDescription = "Initialize or update the SAMT wrapper")

cli/src/main/kotlin/tools/samt/cli/CliCompiler.kt

@@ -1,13 +1,26 @@
 package tools.samt.cli
 
+import tools.samt.codegen.Codegen
 import tools.samt.common.DiagnosticController
 import tools.samt.common.DiagnosticException
+import tools.samt.common.collectSamtFiles
+import tools.samt.common.readSamtSource
 import tools.samt.lexer.Lexer
 import tools.samt.parser.Parser
 import tools.samt.semantic.SemanticModel
+import java.io.IOException
+import kotlin.io.path.isDirectory
+import kotlin.io.path.notExists
 
 internal fun compile(command: CompileCommand, controller: DiagnosticController) {
-    val sourceFiles = command.files.readSamtSourceFiles(controller)
+    val (configuration ,_) = CliConfigParser.readConfig(command.file, controller) ?: return
+
+    if (configuration.source.notExists() || !configuration.source.isDirectory()) {
+        controller.reportGlobalError("Source path '${configuration.source.toUri()}' does not point to valid directory")
+        return
+    }
+
+    val sourceFiles = collectSamtFiles(configuration.source.toUri()).readSamtSource(controller)
 
     if (controller.hasErrors()) {
         return
@@ -40,7 +53,24 @@ internal fun compile(command: CompileCommand, controller: DiagnosticController)
     }
 
     // build up the semantic model from the AST
-    SemanticModel.build(fileNodes, controller)
+    val model = SemanticModel.build(fileNodes, controller)
 
-    // Code Generators will be called here
+    // if the semantic model failed to build, exit
+    if (controller.hasErrors()) {
+        return
+    }
+
+    if (configuration.generators.isEmpty()) {
+        controller.reportGlobalInfo("No generators configured, did you forget to add a 'generators' section to the 'samt.yaml' configuration?")
+        return
+    }
+
+    for (generator in configuration.generators) {
+        val files = Codegen.generate(model, generator, controller)
+        try {
+            OutputWriter.write(generator.output, files)
+        } catch (e: IOException) {
+            controller.reportGlobalError("Failed to write output for generator '${generator.name}': ${e.message}")
+        }
+    }
 }

cli/src/main/kotlin/tools/samt/cli/CliConfigParser.kt

@@ -0,0 +1,41 @@
+package tools.samt.cli
+
+import tools.samt.common.DiagnosticController
+import tools.samt.common.SamtConfiguration
+import tools.samt.common.SamtLinterConfiguration
+import tools.samt.config.SamtConfigurationParser
+import java.nio.file.InvalidPathException
+import kotlin.io.path.Path
+import kotlin.io.path.notExists
+
+internal object CliConfigParser {
+    fun readConfig(file: String, controller: DiagnosticController): Pair<SamtConfiguration, SamtLinterConfiguration>? {
+        val configFile = try {
+            Path(file)
+        } catch (e: InvalidPathException) {
+            controller.reportGlobalError("Invalid path '${file}': ${e.message}")
+            return null
+        }
+        if (configFile.notExists()) {
+            controller.reportGlobalInfo("Configuration file '${configFile.toUri()}' does not exist, using default configuration")
+        }
+        val configuration = try {
+            SamtConfigurationParser.parseConfiguration(configFile)
+        } catch (e: Exception) {
+            controller.reportGlobalError("Failed to parse configuration file '${configFile.toUri()}': ${e.message}")
+            return null
+        }
+        val samtLintConfigFile = configFile.resolveSibling(".samtrc.yaml")
+        if (samtLintConfigFile.notExists()) {
+            controller.reportGlobalInfo("Lint configuration file '${samtLintConfigFile.toUri()}' does not exist, using default lint configuration")
+        }
+        val linterConfiguration = try {
+            SamtConfigurationParser.parseLinterConfiguration(samtLintConfigFile)
+        } catch (e: Exception) {
+            controller.reportGlobalError("Failed to parse lint configuration file '${samtLintConfigFile.toUri()}': ${e.message}")
+            return null
+        }
+
+        return Pair(configuration, linterConfiguration)
+    }
+}

cli/src/main/kotlin/tools/samt/cli/CliDumper.kt

@@ -3,12 +3,27 @@ package tools.samt.cli
 import com.github.ajalt.mordant.terminal.Terminal
 import tools.samt.common.DiagnosticController
 import tools.samt.common.DiagnosticException
+import tools.samt.common.collectSamtFiles
+import tools.samt.common.readSamtSource
 import tools.samt.lexer.Lexer
 import tools.samt.parser.Parser
 import tools.samt.semantic.SemanticModel
+import kotlin.io.path.isDirectory
+import kotlin.io.path.notExists
 
 internal fun dump(command: DumpCommand, terminal: Terminal, controller: DiagnosticController) {
-    val sourceFiles = command.files.readSamtSourceFiles(controller)
+    val (configuration ,_) = CliConfigParser.readConfig(command.file, controller) ?: return
+
+    if (configuration.source.notExists() || !configuration.source.isDirectory()) {
+        controller.reportGlobalError("Source path '${configuration.source.toUri()}' does not point to valid directory")
+        return
+    }
+
+    val sourceFiles = collectSamtFiles(configuration.source.toUri()).readSamtSource(controller)
+
+    if (controller.hasErrors()) {
+        return
+    }
 
     if (controller.hasErrors()) {
         return

cli/src/main/kotlin/tools/samt/cli/CliFileResolution.kt

@@ -15,9 +15,6 @@ internal class DiagnosticFormatter(
     companion object {
         private const val CONTEXT_ROW_COUNT = 3
 
-        // FIXME: this is a bit of a hack to get the terminal width
-        //        it also means we're assuming this output will only ever be printed in a terminal
-        //        i don't actually know what happens if it doesn't run in a tty setting
         fun format(controller: DiagnosticController, startTimestamp: Long, currentTimestamp: Long, terminalWidth: Int = Terminal().info.width): String {
             val formatter = DiagnosticFormatter(controller, startTimestamp, currentTimestamp, terminalWidth)
             return formatter.format()

cli/src/main/kotlin/tools/samt/cli/DiagnosticFormatter.kt

@@ -0,0 +1,39 @@
+package tools.samt.cli
+
+import tools.samt.api.plugin.CodegenFile
+import java.io.IOException
+import java.nio.file.InvalidPathException
+import java.nio.file.Path
+import kotlin.io.path.*
+
+internal object OutputWriter {
+    @Throws(IOException::class)
+    fun write(outputDirectory: Path, files: List<CodegenFile>) {
+        if (!outputDirectory.exists()) {
+            try {
+                outputDirectory.createDirectories()
+            } catch (e: IOException) {
+                throw IOException("Failed to create output directory '${outputDirectory}'", e)
+            }
+        }
+        if (!outputDirectory.isDirectory()) {
+            throw IOException("Path '${outputDirectory}' does not point to a directory")
+        }
+        for (file in files) {
+            val outputFile = try {
+                outputDirectory.resolve(file.filepath)
+            } catch (e: InvalidPathException) {
+                throw IOException("Invalid path '${file.filepath}'", e)
+            }
+            try {
+                outputFile.parent.createDirectories()
+                if (outputFile.notExists()) {
+                    outputFile.createFile()
+                }
+                outputFile.writeText(file.source)
+            } catch (e: IOException) {
+                throw IOException("Failed to write file '${outputFile.toUri()}'", e)
+            }
+        }
+    }
+}