Anchor IDL generator

.gitignore

@@ -38,3 +38,5 @@ build*
 .idea
 cmake-build-debug
 cmake-build-release
+
+node_modules/

lib/idl/generator/accounts.js

@@ -0,0 +1,120 @@
+import { Project } from "ts-morph"
+import { CodeBlockWriter } from "ts-morph";
+import { tsTypeFromIdl } from "../generator/helper.js";
+
+export function generateAccounts(idl) {
+    if (idl.accounts === undefined || idl.accounts.length === 0) {
+        return
+    }
+    genAccountFiles(idl)
+}
+
+function accountFieldClass(acc, writer) {
+
+    writer.write(`class ${acc.name}Fields`).inlineBlock(() => {
+        writer.writeLine("public:")
+        for (let i = 0; i < acc.type.fields.length; i++) {
+            writer.writeLine(`${tsTypeFromIdl(acc.type.fields[i].type)} ${acc.type.fields[i].name};`)
+        }
+    }).write(`;`).newLine();
+
+}
+
+function accountClassJson(acc, writer) {
+
+    writer.write(`class ${acc.name}JSON`).inlineBlock(() => {
+        for (let i = 0; i < acc.type.fields.length; i++) {
+            writer.writeLine(`${tsTypeFromIdl(acc.type.fields[i].type)} ${acc.type.fields[i].name};`)
+        }
+    }).write(`;`).newLine();
+
+}
+
+function accountClass(acc, writer) {
+
+    writer.write(`class ${acc.name}`).inlineBlock(() => {
+        for (let i = 0; i < acc.type.fields.length; i++) {
+            writer.writeLine(`${tsTypeFromIdl(acc.type.fields[i].type)} ${acc.type.fields[i].name};`)
+        }
+        writer.blankLine()
+        writer.write(`${acc.name} ( ${acc.name}Fields field)`).inlineBlock(() => {
+            for (let i = 0; i < acc.type.fields.length; i++) {
+                writer.writeLine(`this->${acc.type.fields[i].name} = field.${acc.type.fields[i].name};`)
+            }
+        }
+        ).write(`;`).newLine();
+        writer.blankLine()
+        writer.write(`${acc.name} fetch (Connection c , PublicKey address)`).inlineBlock(() => {
+            writer.writeLine(`int info = c.getAccountInfo(address);`)
+            writer.write(`if (info == NULL)`).inlineBlock(() => {
+                writer.writeLine(`return NULL;`)
+            }
+            ).newLine();
+            writer.writeLine(`return this->decode(info.data);`)
+        }).newLine();
+
+        writer.blankLine()
+        writer.write(`${acc.name} fetchMultiple (Connection c , PublicKey[] addresses)`).inlineBlock(() => {
+            writer.writeLine(`int[] infos =  c.getMultipleAccountsInfo(addresses);`)
+            writer.write(`for (auto info : infos)`).inlineBlock(() => {
+                writer.write(`if (info == NULL)`).inlineBlock(() => {
+                    writer.writeLine(`return NULL;`)
+                }
+                ).newLine();
+                writer.writeLine(`return this->decode(info.data);`)
+            }
+            ).newLine();
+        }).newLine();
+
+        //decode
+
+        writer.blankLine()
+        writer.write(`${acc.name}JSON ToJson`).inlineBlock(() => {
+            writer.writeLine(`json r = {`)
+            for (let i = 0; i < acc.type.fields.length; i++) {
+                writer.writeLine(`${acc.type.fields[i].name} = this.${acc.type.fields[i].name}.toString();`)
+            }
+            writer.writeLine(`} \n return r;`)
+        }).newLine();
+
+        writer.blankLine()
+        writer.write(`${acc.name} FromJson (json ${acc.name}JSON)`).inlineBlock(() => {
+
+            writer.writeLine(`${acc.name}Fields fie;`)
+            writer.writeLine(`${acc.name} obj(fie);`)
+            for (let i = 0; i < acc.type.fields.length; i++) {
+                writer.writeLine(`obj.${acc.type.fields[i].name} = ${acc.name}JSON.${acc.type.fields[i].name};`)
+            }
+            writer.writeLine(`return obj;`)
+
+        }).newLine();
+
+    }).write(`;`).newLine();
+
+}
+
+
+async function genAccountFiles(
+    idl
+) {
+    const project = new Project()
+    idl.accounts?.forEach((acc) => {
+        const src = project.createSourceFile(`accounts/${acc.name}.cpp`, " ", {
+            overwrite: true,
+        })
+        const writers = new CodeBlockWriter({
+            // optional options
+            newLine: "\r\n",         // default: "\n"
+            indentNumberOfSpaces: 2, // default: 4
+            useTabs: false,          // default: false
+            useSingleQuote: true     // default: false
+        });
+
+        writers.writeLine(`#include<string>`)
+        accountFieldClass(acc, writers)
+        accountClassJson(acc, writers)
+        accountClass(acc, writers);
+        src.insertText(0, writer => writer.writeLine(writers.toString()));
+    })
+    await project.save()
+}

lib/idl/generator/helper.js

@@ -0,0 +1,57 @@
+
+export function tsTypeFromIdl(ty) {
+    switch (ty) {
+        case "bool":
+            return "bool"
+        case "u8":
+            return "uint8_t"
+        case "i8":
+            return "int8_t"
+        case "u16":
+            return "uint16_t"
+        case "i16":
+            return "int16_t"
+        case "u32":
+            return "uint32_t"
+        case "i32":
+            return "int32_t"
+        case "f32":
+            return "float"
+        case "u64":
+            return "uint64_t"
+        case "i64":
+            return "int64_t"
+        case "f64":
+            return "double"
+        case "u128":
+            return "uint128_t"
+        case "i128":
+            return "int128_t"
+        case "bytes":
+            return "std::vector<int> Array"
+        case "string":
+            return "std::string"
+        case "publicKey":
+            return "PublicKey"
+        default:
+            if ("vec" in ty) {
+                return `std::vector<int>`
+            }
+            if ("option" in ty) {
+                return `${tsTypeFromIdl(
+                    ty.option,
+                )}`
+            }
+            if ("coption" in ty) {
+                return `${tsTypeFromIdl(
+                    ty.coption,
+                )}`
+            }
+            if ("array" in ty) {
+                return `std::vector<int>`
+            }
+            else {
+                return "std::string"
+            }
+    }
+}

lib/idl/generator/main.js

@@ -0,0 +1,50 @@
+import * as fileSystem from "fs"
+import * as anchor from "@project-serum/anchor"
+import * as solana from "@solana/web3.js"
+
+import { generateAccounts } from "./accounts.js"
+import { generateTypes } from "./types.js"
+import { program } from "commander"
+
+
+async function main() {
+  program
+    .description("generator to interact with anchor IDL")
+    .argument("address", "address of the deployed program")
+    .argument("output", " output directory")
+    .option("-fp, --idlFilePath  [value]", "idl file path")
+    .parse()
+
+  const address = program.args[0]
+  const output = program.args[1]
+  const filepath = program.opts().idlFilePath || null
+
+
+  //Fetch IDL
+  const connection = new solana.Connection(solana.clusterApiUrl('mainnet-beta'))
+  const keypair = anchor.web3.Keypair.generate()
+  const wallet = new anchor.Wallet(keypair)
+  const provider = new anchor.AnchorProvider(connection, wallet)
+  const Program = new solana.PublicKey(address);
+  const idlFromCluster = await anchor.Program.fetchIdl(Program, provider);
+
+  //IDL From File
+  const idlFromFile = filepath ? JSON.parse(fileSystem.readFileSync(filepath, "utf-8")) : null
+
+
+  //generate classes, methods
+
+  //generate accounts
+  console.log("generating accounts")
+  generateAccounts(idlFromCluster)
+  console.log("generating types")
+  generateTypes(idlFromCluster);
+
+
+
+
+  console.log("writing in the  files")
+
+}
+
+main();

lib/idl/generator/types.js

@@ -0,0 +1,123 @@
+import { Project } from "ts-morph"
+import { CodeBlockWriter } from "ts-morph";
+import { tsTypeFromIdl } from "../generator/helper.js";
+
+export function generateTypes(idl) {
+    if (idl.accounts === undefined || idl.accounts.length === 0) {
+        return
+    }
+    genTypesFiles(idl)
+}
+
+function typeDatafieldsClass(acc, writer) {
+
+    writer.write(`class ${acc.name}DataFields`).inlineBlock(() => {
+        writer.writeLine("public:")
+        for (let i = 0; i < acc.type.fields?.length; i++) {
+            writer.writeLine(`${tsTypeFromIdl(acc.type.fields[i]?.type)} ${acc.type.fields[i]?.name};`)
+        }
+
+    }).write(`;`).newLine();
+
+}
+
+function typeDataClassJson(acc, writer) {
+
+    writer.write(`class ${acc.name}DataJSON`).inlineBlock(() => {
+        writer.writeLine("public:")
+        for (let i = 0; i < acc.type.fields?.length; i++) {
+            writer.writeLine(`${tsTypeFromIdl(acc.type.fields[i]?.type)} ${acc.type.fields[i]?.name};`)
+        }
+
+    }).write(`;`).newLine();
+
+}
+
+function typeDataClass(acc, writer) {
+
+    writer.write(`class ${acc.name}Data`).inlineBlock(() => {
+        for (let i = 0; i < acc.type.fields?.length; i++) {
+            writer.writeLine(`${tsTypeFromIdl(acc.type.fields[i]?.type)} ${acc.type.fields[i]?.name};`)
+        }
+        writer.blankLine()
+        writer.write(`${acc.name} ( ${acc.name}Fields field)`).inlineBlock(() => {
+            for (let i = 0; i < acc.type.fields?.length; i++) {
+                writer.writeLine(`this->${acc.type.fields[i]?.name} = field.${acc.type.fields[i]?.name};`)
+            }
+        }
+        ).write(`;`).newLine();
+        writer.blankLine()
+        writer.write(`${acc.name} fetch (Connection c , PublicKey address)`).inlineBlock(() => {
+            writer.writeLine(`int info = c.getAccountInfo(address);`)
+            writer.write(`if (info == NULL)`).inlineBlock(() => {
+                writer.writeLine(`return NULL;`)
+            }
+            ).newLine();
+            writer.writeLine(`return this->decode(info.data);`)
+        }).newLine();
+
+        writer.blankLine()
+        writer.write(`${acc.name} fetchMultiple (Connection c , PublicKey[] addresses)`).inlineBlock(() => {
+            writer.writeLine(`int[] infos =  c.getMultipleAccountsInfo(addresses);`)
+            writer.write(`for (auto info : infos)`).inlineBlock(() => {
+                writer.write(`if (info == NULL)`).inlineBlock(() => {
+                    writer.writeLine(`return NULL;`)
+                }
+                ).newLine();
+                writer.writeLine(`return this->decode(info.data);`)
+            }
+            ).newLine();
+        }).newLine();
+
+        //decode
+
+        writer.blankLine()
+        writer.write(`${acc.name}JSON ToJson`).inlineBlock(() => {
+            writer.writeLine(`json r = {`)
+            for (let i = 0; i < acc.type.fields?.length; i++) {
+                writer.writeLine(`${acc.type.fields[i]?.name} = this.${acc.type.fields[i]?.name}.toString();`)
+            }
+            writer.writeLine(`} \n return r;`)
+        }).newLine();
+
+        writer.blankLine()
+        writer.write(`${acc.name} FromJson (json ${acc.name}JSON)`).inlineBlock(() => {
+
+            writer.writeLine(`${acc.name}Fields fie;`)
+            writer.writeLine(`${acc.name} obj(fie);`)
+            for (let i = 0; i < acc.type.fields?.length; i++) {
+                writer.writeLine(`obj.${acc.type.fields[i]?.name} = ${acc.name}JSON.${acc.type.fields[i]?.name};`)
+            }
+            writer.writeLine(`return obj;`)
+
+        }).newLine();
+
+    }).write(`;`).newLine();
+
+}
+
+async function genTypesFiles(
+    idl
+) {
+
+    const project = new Project()
+    idl.types?.forEach((acc) => {
+        const src = project.createSourceFile(`types/${acc.name}.cpp`, " ", {
+            overwrite: true,
+        })
+        const writers = new CodeBlockWriter({
+            // optional options
+            newLine: "\r\n",         // default: "\n"
+            indentNumberOfSpaces: 2, // default: 4
+            useTabs: false,          // default: false
+            useSingleQuote: true     // default: false
+        });
+
+        writers.writeLine(`#include<string>`)
+        typeDatafieldsClass(acc, writers)
+        typeDataClassJson(acc, writers)
+        typeDataClass(acc, writers);
+        src.insertText(0, writer => writer.writeLine(writers.toString()));
+    })
+    await project.save()
+}