NodeProcessor.kt

NodeProcessor.kt

private fun generateCode(symbol: KSFunctionDeclaration, route: String, args: ArrayList<String>,
                             optionalArgs: ArrayList<String>) {
        val functionName = symbol.simpleName.asString()
        val packageName = symbol.packageName.asString()
        val fileName = "${functionName}GraphNode"

        val typeBuilder = TypeSpec.objectBuilder(fileName)
        
        // add properties and functions to typeBuilder

        val fileSpec = FileSpec.builder(packageName, fileName).addType(typeBuilder.build()).build()

        val dependencies = Dependencies(false, symbol.containingFile!!)
        val file = codeGenerator.createNewFile(dependencies, packageName, fileName)
        file.bufferedWriter().use { fileSpec.writeTo(it) }
    }