allow configuration of the visibility of types created by codegen

Author: CharlieTap

chasm-gradle-plugin/src/main/kotlin/io/github/charlietap/chasm/gradle/ChasmExtension.kt

@@ -14,6 +14,11 @@ enum class RuntimeDependencyConfiguration {
     IMPLEMENTATION,
 }
 
+enum class TypeVisibility {
+    INTERNAL,
+    PUBLIC,
+}
+
 @ChasmDsl
 open class ChasmExtension
     @Inject

chasm-gradle-plugin/src/main/kotlin/io/github/charlietap/chasm/gradle/ChasmPlugin.kt

@@ -155,6 +155,8 @@ class ChasmPlugin : Plugin<Project> {
             config.set(module.codegenConfig)
             interfaceName.set(module.name)
             packageName.set(module.packageName)
+            interfaceVisibility.set(module.interfaceVisibility)
+            implementationVisibility.set(module.implementationVisibility)
             initializers.set(module.initializers)
             functions.set(module.functions)
             outputDirectory.set(project.layout.buildDirectory.dir("generated/kotlin/$sourceSetName"))

chasm-gradle-plugin/src/main/kotlin/io/github/charlietap/chasm/gradle/ClassImplementationGenerator.kt

@@ -278,9 +278,16 @@ internal class ClassImplementationGenerator(
     operator fun invoke(
         packageName: String,
         interfaceName: String,
+        visibility: TypeVisibility,
         wasmInterface: WasmInterface,
     ): TypeSpec = TypeSpec.classBuilder(interfaceName + "Impl").apply {
 
+        val visibilityModifier = when (visibility) {
+            TypeVisibility.INTERNAL -> KModifier.INTERNAL
+            TypeVisibility.PUBLIC -> KModifier.PUBLIC
+        }
+        addModifiers(visibilityModifier)
+
         addSuperinterface(ClassName(packageName, interfaceName))
 
         addConstructor(constructorGenerator)

chasm-gradle-plugin/src/main/kotlin/io/github/charlietap/chasm/gradle/ClassInterfaceGenerator.kt

@@ -50,9 +50,16 @@ internal class ClassInterfaceGenerator(
     operator fun invoke(
         packageName: String,
         interfaceName: String,
+        visibility: TypeVisibility,
         wasmInterface: WasmInterface,
     ) = TypeSpec.interfaceBuilder(interfaceName).apply {
 
+        val visibilityModifier = when (visibility) {
+            TypeVisibility.INTERNAL -> KModifier.INTERNAL
+            TypeVisibility.PUBLIC -> KModifier.PUBLIC
+        }
+        addModifiers(visibilityModifier)
+
         wasmInterface.functions.forEach { function ->
             addFunction(functionGenerator(packageName, function))
         }

chasm-gradle-plugin/src/main/kotlin/io/github/charlietap/chasm/gradle/CodegenTask.kt

@@ -38,6 +38,12 @@ abstract class CodegenTask
         @get:Input
         abstract val packageName: Property<String>
 
+        @get:Input
+        abstract val interfaceVisibility: Property<TypeVisibility>
+
+        @get:Input
+        abstract val implementationVisibility: Property<TypeVisibility>
+
         @get:Input
         abstract val initializers: SetProperty<String>
 
@@ -73,7 +79,7 @@ abstract class CodegenTask
                 wasmFunctions = functions.get(),
                 logger = logger,
             )
-            val specs = generator(data)
+            val specs = generator(interfaceVisibility.get(), implementationVisibility.get(), data)
 
             val outputDir = outputDirectory.get().asFile
             specs.forEach { spec ->

chasm-gradle-plugin/src/main/kotlin/io/github/charlietap/chasm/gradle/WasmInterfaceGenerator.kt

@@ -38,18 +38,20 @@ internal class WasmInterfaceGenerator(
     private val classImplementationGenerator: ClassImplementationGenerator = ClassImplementationGenerator(),
 ) {
     operator fun invoke(
+        interfaceVisibility: TypeVisibility,
+        implementationVisibility: TypeVisibility,
         wasmInterface: WasmInterface,
     ): List<FileSpec> {
 
         val interfaceFile = FileSpec.builder(wasmInterface.packageName, wasmInterface.interfaceName).apply {
             wasmInterface.types.forEach { type ->
                 addType(dataClassGenerator(wasmInterface.packageName, type))
             }
-            addType(classInterfaceGenerator(wasmInterface.packageName, wasmInterface.interfaceName, wasmInterface))
+            addType(classInterfaceGenerator(wasmInterface.packageName, wasmInterface.interfaceName, interfaceVisibility, wasmInterface))
         }.build()
 
         val implementationFile = FileSpec.builder(wasmInterface.packageName, wasmInterface.interfaceName + "Impl").apply {
-            addType(classImplementationGenerator(wasmInterface.packageName, wasmInterface.interfaceName, wasmInterface))
+            addType(classImplementationGenerator(wasmInterface.packageName, wasmInterface.interfaceName, implementationVisibility, wasmInterface))
         }.build()
 
         return listOf(interfaceFile, implementationFile)

chasm-gradle-plugin/src/main/kotlin/io/github/charlietap/chasm/gradle/WasmModule.kt

@@ -19,11 +19,15 @@ abstract class WasmModule
         abstract val packageName: Property<String>
         abstract val initializers: SetProperty<String>
         abstract val functions: ListProperty<WasmFunction>
+        abstract val interfaceVisibility: Property<TypeVisibility>
+        abstract val implementationVisibility: Property<TypeVisibility>
 
         init {
             binary.convention(project.layout.projectDirectory.file(DEFAULT_MODULE_FILE_PATH))
             codegenConfig.convention(CodegenConfig())
             functions.convention(emptyList())
+            interfaceVisibility.convention(TypeVisibility.PUBLIC)
+            implementationVisibility.convention(TypeVisibility.INTERNAL)
         }
 
         fun function(name: String, configuration: WasmFunctionBuilder.() -> Unit) {