Generating metas

playground/src/main/java/com/solanamobile/buoy/playground/Main.kt

@@ -3,15 +3,56 @@ package com.solanamobile.buoy.playground
 import com.google.gson.Gson
 import com.solanamobile.buoy.playground.idlspec.IdlRootV1
 import com.solanamobile.web3.core.PublicKey
-import com.squareup.kotlinpoet.FileSpec
-import com.squareup.kotlinpoet.FunSpec
-import com.squareup.kotlinpoet.PropertySpec
-import com.squareup.kotlinpoet.TypeSpec
+import com.solanamobile.web3.core.TransactionInstruction
+import com.squareup.kotlinpoet.*
 import java.io.File
+import java.security.MessageDigest
+import java.util.*
+import kotlin.reflect.KClass
 
 /**
  * ./gradlew :playground:run --args=<Path_to_json>
  */
+
+fun classTypeForArgType( argType: String ): KClass<*> {
+    return when(argType) {
+        "u8" -> UByte::class
+        "u16" -> UShort::class
+        "u32" -> UInt::class
+        "u64" -> ULong::class
+        "i8" -> Byte::class
+        "i16" -> Short::class
+        "i32" -> Int::class
+        "i64" -> Long::class
+        "publicKey" -> PublicKey::class
+        else -> {
+//            throw Exception("This should not happen.")
+            return ClassName("solanapackagename", argType)::class
+        }
+    }
+}
+
+private fun convertCamelToSnakeCase(camelCase : String) : String {
+    val snakeCase = StringBuilder()
+    for(character in camelCase) {
+        if(character.isUpperCase()) {
+            snakeCase.append("_${character.toLowerCase()}")
+        } else {
+            snakeCase.append(character)
+        }
+    }
+    return snakeCase.removePrefix("_").toString()
+}
+
+fun sighash( nameSpace: String = "global", name: String ): Array<Byte> {
+    val preImage = String.format("%s:%s", nameSpace, name)
+    val preImageByteArray = preImage.toByteArray()
+    val md = MessageDigest.getInstance("SHA-256")
+    val digest = md.digest(preImageByteArray)
+    val sighash = digest.copyOfRange(0, 8)
+    return sighash.toTypedArray()
+}
+
 fun main(arguments: Array<String>) {
     println("----------------------------------------")
     println("")
@@ -24,6 +65,7 @@ fun main(arguments: Array<String>) {
 
     //Use the objects to generate the contract file
     val contractFile = FileSpec.builder("com.solanamobile.buoy", idlSource.name)
+
     val mainClass = contractFile.addType(
         TypeSpec.classBuilder(idlSource.name)
             .build()
@@ -32,40 +74,62 @@ fun main(arguments: Array<String>) {
     idlSource.instructions.forEach { instruction ->
         val functionBuilder = FunSpec.builder(instruction.name)
 
+        functionBuilder
+            .returns(TransactionInstruction::class)
+
+        // start the list of function
+        functionBuilder.addStatement("val = listOf(")
+
+        // for each meta descriptor, add a parameter accepting
+        // the pubkey and generate the AccountMeta item in the list
+        instruction.args.forEach { arg ->
+            val type = classTypeForArgType(arg.type)
+            functionBuilder
+                .addParameter(arg.name, type)
+        }
+
         instruction.accounts.forEach { metaDescriptor ->
+
+            val metaItem = String.format("\tAccountMeta(publicKey = %s, isSigner = %s, isWritable = %s)", metaDescriptor.name, metaDescriptor.isSigner.toString(), metaDescriptor.isMut.toString())
+
             functionBuilder
                 .addParameter(metaDescriptor.name, PublicKey::class)
-                .build()
+                .addStatement(metaItem)
+
         }
 
+        // close the listOf function
+        functionBuilder.addStatement(")")
+
+        val sighHashStatement = String.format("val sigHash: Array<Byte> = sighash(\"%s\", \"%s\")", "global", convertCamelToSnakeCase(instruction.name))
+        functionBuilder.addStatement( sighHashStatement )
+
+        val instructionDataStatementBuilder = StringBuilder()
+
+        instructionDataStatementBuilder
+            .append("val instructionData: Array<Byte> = sigHash")
+
+        instruction.args.forEach { arg ->
+            functionBuilder
+                .addStatement(String.format("val %sBytes = %s.toBytes()", arg.name, arg.name))
+            instructionDataStatementBuilder.append( String.format(" + %sBytes", arg.name, arg.name) )
+        }
+
+        val instructionDataStatement = instructionDataStatementBuilder.toString()
+
+        functionBuilder.addStatement(instructionDataStatement)
+
+        functionBuilder.addStatement("return TransactionInstruction(")
+        functionBuilder.addStatement("\tkeys = keys,")
+        functionBuilder.addStatement("\tprogramId = programId,")
+        functionBuilder.addStatement("\tdata = instructionData")
+        functionBuilder.addStatement(")")
+
         mainClass.addFunction(
             functionBuilder.build()
         ).build()
     }
 
     contractFile.build().writeTo(System.out)
 
-    val file = FileSpec.builder("com.solanamobile", "HelloWorld")
-        .addType(
-            TypeSpec.classBuilder("Greeter")
-                .primaryConstructor(
-                    FunSpec.constructorBuilder()
-                        .addParameter("name", String::class)
-                        .build()
-                )
-                .addProperty(
-                    PropertySpec.builder("name", String::class)
-                        .initializer("name")
-                        .build()
-                )
-                .addFunction(
-                    FunSpec.builder("greet")
-                        .addStatement("println(%P)", "Hello, \$name")
-                        .build()
-                )
-                .build()
-        )
-        .build()
-
-    file.writeTo(System.out)
 }

playground/src/main/java/com/solanamobile/buoy/playground/idlspec/IdlSpecV1.kt

@@ -25,6 +25,18 @@ data class Arg(
     val type: String
 )
 
+//data class VecType(
+//    val vec: Any
+//)
+//
+//data class OptionalType(
+//    val option: Any
+//)
+//
+//data class DefinedParam(
+//    val defined: String
+//)
+
 data class AccountData(
     val name: String,
     val type: AccountDataType