@sentio/sdk 1.20.1 → 1.21.0

package/lib/aptos/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/aptos/utils.ts"],"names":[],"mappings":";;;AAEa,QAAA,QAAQ,GAAG,IAAI,CAAA;AAEf,QAAA,UAAU,GAAG,QAAQ,CAAA;AAElC,SAAgB,kBAAkB,CAAC,OAAe;IAChD,OAAO,OAAO,KAAK,KAAK,IAAI,OAAO,KAAK,KAAK,IAAI,OAAO,KAAK,KAAK,CAAA;AACpE,CAAC;AAFD,gDAEC;AAED,SAAgB,WAAW,CAAC,MAAkB;IAC5C,OAAO,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,GAAG,gBAAQ,GAAG,MAAM,CAAC,IAAI,CAAA;AAC9D,CAAC;AAFD,kCAEC;AAED,SAAgB,kBAAkB,CAAC,IAAY;IAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,gBAAQ,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;IAC9C,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAA;AAC7B,CAAC;AAHD,gDAGC","sourcesContent":["import { MoveModule } from 'aptos/src/generated'\n\nexport const SPLITTER = '::'\n\nexport const VECTOR_STR = 'vector'\n\nexport function isFrameworkAccount(account: string) {\n  return account === '0x1' || account === '0x2' || account === '0x3'\n}\n\nexport function moduleQname(module: MoveModule): string {\n  return module.address.toLowerCase() + SPLITTER + module.name\n}\n\nexport function moduleQnameForType(type: string): [string, string] {\n  const parts = type.split(SPLITTER).slice(0, 2)\n  return [parts[0], parts[1]]\n}\n"]}

package/lib/aptos-codegen/codegen.d.ts

@@ -1,12 +1,15 @@
 import { MoveModuleBytecode } from 'aptos/src/generated';
+import { AccountRegister } from './typegen';
+export declare function generate(srcDir: string, outputDir: string): void;
 interface Config {
     srcFile: string;
     outputDir: string;
 }
-export declare class AptosCodegen {
+export declare class AccountCodegen {
     modules: MoveModuleBytecode[];
     config: Config;
-    constructor(config: Config);
+    loader: AccountRegister;
+    constructor(loader: AccountRegister, config: Config);
     generate(): void;
 }
 export {};

package/lib/aptos-codegen/codegen.js

@@ -3,22 +3,67 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.AptosCodegen = void 0;
+exports.AccountCodegen = exports.generate = void 0;
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const prettier_1 = __importDefault(require("prettier"));
-class AptosCodegen {
+const typegen_1 = require("./typegen");
+const utils_1 = require("../aptos/utils");
+function generate(srcDir, outputDir) {
+    const files = fs_1.default.readdirSync(srcDir);
+    outputDir = path_1.default.resolve(outputDir);
+    fs_1.default.mkdirSync(outputDir, { recursive: true });
+    const loader = new typegen_1.AccountRegister();
+    // The first path, identify import relation and module name (filename) of those imports
+    for (const file of files) {
+        if (!file.endsWith('.json')) {
+            continue;
+        }
+        // Reading file is duplicated, but since they are small files probably file
+        // TODO add file manager class
+        const json = fs_1.default.readFileSync(path_1.default.join(srcDir, file), 'utf-8');
+        const modules = JSON.parse(json);
+        for (const module of modules) {
+            if (module.abi) {
+                loader.register(module.abi, path_1.default.basename(file, '.json'));
+            }
+        }
+    }
+    for (const file of files) {
+        if (!file.endsWith('.json')) {
+            continue;
+        }
+        const codeGen = new AccountCodegen(loader, {
+            srcFile: path_1.default.join(srcDir, file),
+            outputDir: outputDir,
+        });
+        codeGen.generate();
+    }
+    // when generating user code, don't need to generate framework account
+    loader.pendingAccounts.delete('0x1');
+    loader.pendingAccounts.delete('0x2');
+    loader.pendingAccounts.delete('0x3');
+    if (loader.pendingAccounts.size > 0) {
+        // TODO automatic download dependencies
+        console.error('Missing ABIs from the following accounts', loader.pendingAccounts);
+    }
+}
+exports.generate = generate;
+class AccountCodegen {
     modules;
     config;
-    constructor(config) {
+    loader;
+    constructor(loader, config) {
         const json = fs_1.default.readFileSync(config.srcFile, 'utf-8');
         this.modules = JSON.parse(json);
         this.config = config;
+        this.loader = loader;
     }
     generate() {
         if (!this.modules) {
             return;
         }
+        const baseName = path_1.default.basename(this.config.srcFile, '.json');
         let address;
         for (const module of this.modules) {
             if (module.abi && module.abi.address) {
@@ -28,7 +73,32 @@ class AptosCodegen {
         if (!address) {
             return;
         }
-        const imports = `import { aptos } from "@sentio/sdk"`;
+        const imports = `
+    import { aptos } from "@sentio/sdk"
+    import { Address, MoveModule } from "aptos/src/generated"
+    `;
+        const dependedAccounts = [];
+        const moduleImports = [];
+        const info = this.loader.accountImports.get(address);
+        if (info) {
+            for (const [account, moduleImported] of info.imports.entries()) {
+                // Remap to user's filename if possible, TODO codepath not well tested
+                let tsAccountModule = './' + (this.loader.accountImports.get(account)?.moduleName || account);
+                if ((0, utils_1.isFrameworkAccount)(account) && !(0, utils_1.isFrameworkAccount)(address)) {
+                    // Decide where to find runtime library
+                    let srcRoot = 'lib';
+                    if (__dirname.includes('sdk/src/aptos-codegen')) {
+                        srcRoot = 'src';
+                    }
+                    tsAccountModule = `@sentio/sdk/${srcRoot}/builtin/aptos/${account}`;
+                }
+                const items = Array.from(moduleImported);
+                moduleImports.push(`import { ${items.join(',')} } from "${tsAccountModule}"`);
+                // Ideally we should use per module's load types, but it doesn't matter since we are loading the entire
+                // account modules anyway
+                items.forEach((m) => dependedAccounts.push(m));
+            }
+        }
         let source = `
     /* Autogenerated file. Do not edit manually. */
     /* tslint:disable */
@@ -38,43 +108,147 @@ class AptosCodegen {
   
     ${imports}
     
-    ${this.modules.map(generateModule).join('\n')}
+    ${moduleImports.join('\n')}
+    
+    ${this.modules.map((m) => generateModule(m, dependedAccounts)).join('\n')}
+    
+    function loadAllTypes(registry: aptos.TypeRegistry) {
+      ${dependedAccounts.map((m) => `${m}.loadTypes(registry)`).join('\n')}
+
+      ${this.modules
+            .map((m) => {
+            return `registry.load(${m.abi?.name}.ABI)`;
+        })
+            .join('\n')}
+    }
     `; // source
-        const baseName = path_1.default.basename(this.config.srcFile, '.json');
         source = prettier_1.default.format(source, { parser: 'typescript' });
         fs_1.default.writeFileSync(path_1.default.join(this.config.outputDir, baseName + '.ts'), source);
     }
 }
-exports.AptosCodegen = AptosCodegen;
-function generateModule(moduleByteCode) {
+exports.AccountCodegen = AccountCodegen;
+function generateModule(moduleByteCode, dependedModules) {
     if (!moduleByteCode.abi) {
         return '';
     }
     const module = moduleByteCode.abi;
     const functions = module.exposed_functions.map((f) => generateOnEntryFunctions(module, f)).filter((s) => s !== '');
     const events = module.structs.map((e) => generateOnEvents(module, e)).filter((s) => s !== '');
-    return `
-  export class ${module.name} extends aptos.AptosBaseProcessor {
-    static DEFAULT_OPTIONS: aptos.NamedAptosBindOptions = {
+    const structs = module.structs.map((s) => generateStructs(module, s));
+    const callArgs = module.exposed_functions.map((f) => generateCallArgsStructs(module, f));
+    let processor = '';
+    if (functions.length > 0 || events.length > 0) {
+        processor = `export class ${module.name} extends aptos.AptosBaseProcessor {
+
+    constructor(options: aptos.AptosBindOptions) {
+      super("${module.name}", options)
+    }
+    static DEFAULT_OPTIONS: aptos.AptosBindOptions = {
       address: "${module.address}",
       network: aptos.AptosNetwork.TEST_NET       
     }
-    static bind(options = ${module.name}.DEFAULT_OPTIONS): ${module.name} {
-      return new ${module.name}({ ...options, name: "${module.name}" })
+
+    static bind(options: Partial<aptos.AptosBindOptions> = {}): ${module.name} {
+      return new ${module.name}({ ...${module.name}.DEFAULT_OPTIONS, ...options })
     }
     
     ${functions.join('\n')}
     
     ${events.join('\n')}
-  }`;
+    
+    loadTypesInternal(registry: aptos.TypeRegistry) {
+      loadAllTypes(registry)
+    }
+  }
+  `;
+    }
+    return `
+  ${processor}
+
+  export namespace ${module.name} {
+    ${structs.join('\n')}
+    
+    ${callArgs.join('\n')}
+       
+    export function loadTypes(registry: aptos.TypeRegistry) {
+      loadAllTypes(registry)
+    }
+    export const ABI: MoveModule = JSON.parse('${JSON.stringify(module)}')
+ }
+  `;
+}
+function generateStructs(module, struct) {
+    const genericString = generateStructTypeParameters(struct);
+    const fields = struct.fields.map((field) => {
+        return `${field.name}: ${(0, typegen_1.generateType)(field.type)}`;
+    });
+    let eventPayload = '';
+    if (isEvent(struct)) {
+        eventPayload = `
+    export interface ${struct.name}Instance${genericString} extends 
+        aptos.TypedEventInstance<${struct.name}${genericString}> {
+      data_typed: ${struct.name}${genericString}
+    }
+    `;
+    }
+    return `
+  export class ${struct.name}${genericString} {
+    ${fields.join('\n')} 
+  }
+  
+  ${eventPayload}
+  `;
+}
+function generateFunctionTypeParameters(func) {
+    let genericString = '';
+    if (func.generic_type_params && func.generic_type_params.length > 0) {
+        const params = func.generic_type_params
+            .map((v, idx) => {
+            return 'T' + idx;
+        })
+            .join(',');
+        genericString = `<${params}>`;
+    }
+    return genericString;
+}
+function generateStructTypeParameters(struct) {
+    let genericString = '';
+    if (struct.generic_type_params && struct.generic_type_params.length > 0) {
+        const params = struct.generic_type_params
+            .map((v, idx) => {
+            return 'T' + idx;
+        })
+            .join(',');
+        genericString = `<${params}>`;
+    }
+    return genericString;
+}
+function generateCallArgsStructs(module, func) {
+    if (!func.is_entry) {
+        return;
+    }
+    // the first param is always signer so ignore
+    // TODO check if there is any edge case
+    const fields = func.params.slice(1).map((param) => {
+        return `${(0, typegen_1.generateType)(param)}`;
+    });
+    const camelFuncName = capitalizeFirstChar(camelize(func.name));
+    const genericString = generateFunctionTypeParameters(func);
+    return `
+  export interface ${camelFuncName}Payload${genericString} 
+      extends aptos.TypedEntryFunctionPayload<[${fields.join(',')}]> {
+    arguments_typed: [${fields.join(',')}]
+  }
+  `;
 }
 function generateOnEntryFunctions(module, func) {
     if (!func.is_entry) {
         return '';
     }
+    const genericString = generateFunctionTypeParameters(func);
     const camelFuncName = capitalizeFirstChar(camelize(func.name));
     const source = `
-  onEntry${camelFuncName}(func: (call: aptos.TransactionPayload_EntryFunctionPayload, ctx: aptos.AptosContext) => void): ${module.name} {
+  onEntry${camelFuncName}${genericString}(func: (call: ${module.name}.${camelFuncName}Payload${genericString}, ctx: aptos.AptosContext) => void): ${module.name} {
     this.onEntryFunctionCall(func, {
       function: '${module.name}::${func.name}'
     })
@@ -82,15 +256,18 @@ function generateOnEntryFunctions(module, func) {
   }`;
     return source;
 }
+function isEvent(struct) {
+    return struct.abilities.includes('drop') && struct.abilities.includes('store') && struct.name.endsWith('Event');
+}
 function generateOnEvents(module, struct) {
     // for struct that has drop + store
-    if (!struct.abilities.includes('drop') || !struct.abilities.includes('store') || !struct.name.endsWith('Event')) {
+    if (!isEvent(struct)) {
         return '';
     }
     const source = `
-  onEvent${struct.name}(func: (event: aptos.Event, ctx: aptos.AptosContext) => void): ${module.name} {
+  onEvent${struct.name}(func: (event: ${module.name}.${struct.name}Instance, ctx: aptos.AptosContext) => void): ${module.name} {
     this.onEvent(func, {
-      type: '${module.address}::${module.name}::${struct.name}'
+      type: '${module.name}::${struct.name}'
     })
     return this
   }

package/lib/aptos-codegen/codegen.js.map

@@ -1 +1 @@
-{"version":3,"file":"codegen.js","sourceRoot":"","sources":["../../src/aptos-codegen/codegen.ts"],"names":[],"mappings":";;;;;;AACA,4CAAmB;AACnB,gDAAuB;AAGvB,wDAA+B;AAO/B,MAAa,YAAY;IACvB,OAAO,CAAsB;IAC7B,MAAM,CAAQ;IAEd,YAAY,MAAc;QACxB,MAAM,IAAI,GAAG,YAAE,CAAC,YAAY,CAAC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;QACrD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QAC/B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;IACtB,CAAC;IAED,QAAQ;QACN,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE;YACjB,OAAM;SACP;QACD,IAAI,OAA2B,CAAA;QAC/B,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE;YACjC,IAAI,MAAM,CAAC,GAAG,IAAI,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE;gBACpC,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,CAAA;aAC7B;SACF;QACD,IAAI,CAAC,OAAO,EAAE;YACZ,OAAM;SACP;QAED,MAAM,OAAO,GAAG,qCAAqC,CAAA;QAErD,IAAI,MAAM,GAAG;;;;;uCAKsB,OAAO;;MAExC,OAAO;;MAEP,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;KAC5C,CAAA,CAAC,SAAS;QAEX,MAAM,QAAQ,GAAG,cAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;QAC5D,MAAM,GAAG,kBAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAA;QAC1D,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,QAAQ,GAAG,KAAK,CAAC,EAAE,MAAM,CAAC,CAAA;IAC9E,CAAC;CACF;AA1CD,oCA0CC;AAED,SAAS,cAAc,CAAC,cAAkC;IACxD,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE;QACvB,OAAO,EAAE,CAAA;KACV;IACD,MAAM,MAAM,GAAG,cAAc,CAAC,GAAG,CAAA;IAEjC,MAAM,SAAS,GAAG,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,wBAAwB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAA;IAClH,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAA;IAE7F,OAAO;iBACQ,MAAM,CAAC,IAAI;;kBAEV,MAAM,CAAC,OAAO;;;4BAGJ,MAAM,CAAC,IAAI,sBAAsB,MAAM,CAAC,IAAI;mBACrD,MAAM,CAAC,IAAI,yBAAyB,MAAM,CAAC,IAAI;;;MAG5D,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;;MAEpB,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;IACnB,CAAA;AACJ,CAAC;AAED,SAAS,wBAAwB,CAAC,MAAkB,EAAE,IAAkB;IACtE,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;QAClB,OAAO,EAAE,CAAA;KACV;IAED,MAAM,aAAa,GAAG,mBAAmB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;IAC9D,MAAM,MAAM,GAAG;WACN,aAAa,mGAAmG,MAAM,CAAC,IAAI;;mBAEnH,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI;;;IAGxC,CAAA;IAEF,OAAO,MAAM,CAAA;AACf,CAAC;AAED,SAAS,gBAAgB,CAAC,MAAkB,EAAE,MAAkB;IAC9D,mCAAmC;IACnC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE;QAC/G,OAAO,EAAE,CAAA;KACV;IACD,MAAM,MAAM,GAAG;WACN,MAAM,CAAC,IAAI,kEAAkE,MAAM,CAAC,IAAI;;eAEpF,MAAM,CAAC,OAAO,KAAK,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI;;;;GAI1D,CAAA;IACD,OAAO,MAAM,CAAA;AACf,CAAC;AAED,SAAS,QAAQ,CAAC,KAAa;IAC7B,OAAO,KAAK;SACT,KAAK,CAAC,GAAG,CAAC;SACV,MAAM,CACL,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,CACf,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,EAAE,EACvG,EAAE,CACH,CAAA;AACL,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAa;IACxC,IAAI,CAAC,KAAK,EAAE;QACV,OAAO,KAAK,CAAA;KACb;IACD,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAA;AAC9E,CAAC","sourcesContent":["import { MoveModuleBytecode } from 'aptos/src/generated'\nimport fs from 'fs'\nimport path from 'path'\nimport { MoveFunction, MoveModule } from 'aptos/src/generated'\nimport { MoveStruct } from 'aptos/src/generated/models/MoveStruct'\nimport prettier from 'prettier'\n\ninterface Config {\n  srcFile: string\n  outputDir: string\n}\n\nexport class AptosCodegen {\n  modules: MoveModuleBytecode[]\n  config: Config\n\n  constructor(config: Config) {\n    const json = fs.readFileSync(config.srcFile, 'utf-8')\n    this.modules = JSON.parse(json)\n    this.config = config\n  }\n\n  generate() {\n    if (!this.modules) {\n      return\n    }\n    let address: string | undefined\n    for (const module of this.modules) {\n      if (module.abi && module.abi.address) {\n        address = module.abi.address\n      }\n    }\n    if (!address) {\n      return\n    }\n\n    const imports = `import { aptos } from \"@sentio/sdk\"`\n\n    let source = `\n    /* Autogenerated file. Do not edit manually. */\n    /* tslint:disable */\n    /* eslint-disable */\n  \n    /* Generated modules for account ${address} */\n  \n    ${imports}\n    \n    ${this.modules.map(generateModule).join('\\n')}\n    ` // source\n\n    const baseName = path.basename(this.config.srcFile, '.json')\n    source = prettier.format(source, { parser: 'typescript' })\n    fs.writeFileSync(path.join(this.config.outputDir, baseName + '.ts'), source)\n  }\n}\n\nfunction generateModule(moduleByteCode: MoveModuleBytecode) {\n  if (!moduleByteCode.abi) {\n    return ''\n  }\n  const module = moduleByteCode.abi\n\n  const functions = module.exposed_functions.map((f) => generateOnEntryFunctions(module, f)).filter((s) => s !== '')\n  const events = module.structs.map((e) => generateOnEvents(module, e)).filter((s) => s !== '')\n\n  return `\n  export class ${module.name} extends aptos.AptosBaseProcessor {\n    static DEFAULT_OPTIONS: aptos.NamedAptosBindOptions = {\n      address: \"${module.address}\",\n      network: aptos.AptosNetwork.TEST_NET       \n    }\n    static bind(options = ${module.name}.DEFAULT_OPTIONS): ${module.name} {\n      return new ${module.name}({ ...options, name: \"${module.name}\" })\n    }\n    \n    ${functions.join('\\n')}\n    \n    ${events.join('\\n')}\n  }`\n}\n\nfunction generateOnEntryFunctions(module: MoveModule, func: MoveFunction) {\n  if (!func.is_entry) {\n    return ''\n  }\n\n  const camelFuncName = capitalizeFirstChar(camelize(func.name))\n  const source = `\n  onEntry${camelFuncName}(func: (call: aptos.TransactionPayload_EntryFunctionPayload, ctx: aptos.AptosContext) => void): ${module.name} {\n    this.onEntryFunctionCall(func, {\n      function: '${module.name}::${func.name}'\n    })\n    return this\n  }`\n\n  return source\n}\n\nfunction generateOnEvents(module: MoveModule, struct: MoveStruct): string {\n  // for struct that has drop + store\n  if (!struct.abilities.includes('drop') || !struct.abilities.includes('store') || !struct.name.endsWith('Event')) {\n    return ''\n  }\n  const source = `\n  onEvent${struct.name}(func: (event: aptos.Event, ctx: aptos.AptosContext) => void): ${module.name} {\n    this.onEvent(func, {\n      type: '${module.address}::${module.name}::${struct.name}'\n    })\n    return this\n  }\n  `\n  return source\n}\n\nfunction camelize(input: string): string {\n  return input\n    .split('_')\n    .reduce(\n      (res, word, i) =>\n        i === 0 ? word.toLowerCase() : `${res}${word.charAt(0).toUpperCase()}${word.substr(1).toLowerCase()}`,\n      ''\n    )\n}\n\nfunction capitalizeFirstChar(input: string): string {\n  if (!input) {\n    return input\n  }\n  return input[0].toUpperCase() + (input.length > 1 ? input.substring(1) : '')\n}\n"]}
+{"version":3,"file":"codegen.js","sourceRoot":"","sources":["../../src/aptos-codegen/codegen.ts"],"names":[],"mappings":";;;;;;AAAA,4CAAmB;AACnB,gDAAuB;AACvB,wDAA+B;AAE/B,uCAAyD;AACzD,0CAAmD;AAEnD,SAAgB,QAAQ,CAAC,MAAc,EAAE,SAAiB;IACxD,MAAM,KAAK,GAAG,YAAE,CAAC,WAAW,CAAC,MAAM,CAAC,CAAA;IACpC,SAAS,GAAG,cAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;IAEnC,YAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;IAE5C,MAAM,MAAM,GAAG,IAAI,yBAAe,EAAE,CAAA;IAEpC,uFAAuF;IACvF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;QACxB,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE;YAC3B,SAAQ;SACT;QACD,2EAA2E;QAC3E,8BAA8B;QAC9B,MAAM,IAAI,GAAG,YAAE,CAAC,YAAY,CAAC,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,CAAC,CAAA;QAC9D,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QAEhC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE;YAC5B,IAAI,MAAM,CAAC,GAAG,EAAE;gBACd,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,EAAE,cAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAA;aAC1D;SACF;KACF;IAED,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;QACxB,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE;YAC3B,SAAQ;SACT;QACD,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC,MAAM,EAAE;YACzC,OAAO,EAAE,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC;YAChC,SAAS,EAAE,SAAS;SACrB,CAAC,CAAA;QAEF,OAAO,CAAC,QAAQ,EAAE,CAAA;KACnB;IAED,sEAAsE;IACtE,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IACpC,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IACpC,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IAEpC,IAAI,MAAM,CAAC,eAAe,CAAC,IAAI,GAAG,CAAC,EAAE;QACnC,uCAAuC;QACvC,OAAO,CAAC,KAAK,CAAC,0CAA0C,EAAE,MAAM,CAAC,eAAe,CAAC,CAAA;KAClF;AACH,CAAC;AA9CD,4BA8CC;AAOD,MAAa,cAAc;IACzB,OAAO,CAAsB;IAC7B,MAAM,CAAQ;IACd,MAAM,CAAiB;IAEvB,YAAY,MAAuB,EAAE,MAAc;QACjD,MAAM,IAAI,GAAG,YAAE,CAAC,YAAY,CAAC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;QACrD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QAC/B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;QACpB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;IACtB,CAAC;IAED,QAAQ;QACN,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE;YACjB,OAAM;SACP;QACD,MAAM,QAAQ,GAAG,cAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;QAE5D,IAAI,OAA2B,CAAA;QAC/B,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE;YACjC,IAAI,MAAM,CAAC,GAAG,IAAI,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE;gBACpC,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,CAAA;aAC7B;SACF;QACD,IAAI,CAAC,OAAO,EAAE;YACZ,OAAM;SACP;QAED,MAAM,OAAO,GAAG;;;KAGf,CAAA;QAED,MAAM,gBAAgB,GAAa,EAAE,CAAA;QAErC,MAAM,aAAa,GAAa,EAAE,CAAA;QAElC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;QAEpD,IAAI,IAAI,EAAE;YACR,KAAK,MAAM,CAAC,OAAO,EAAE,cAAc,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE;gBAC9D,sEAAsE;gBACtE,IAAI,eAAe,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,UAAU,IAAI,OAAO,CAAC,CAAA;gBAC7F,IAAI,IAAA,0BAAkB,EAAC,OAAO,CAAC,IAAI,CAAC,IAAA,0BAAkB,EAAC,OAAO,CAAC,EAAE;oBAC/D,uCAAuC;oBACvC,IAAI,OAAO,GAAG,KAAK,CAAA;oBACnB,IAAI,SAAS,CAAC,QAAQ,CAAC,uBAAuB,CAAC,EAAE;wBAC/C,OAAO,GAAG,KAAK,CAAA;qBAChB;oBACD,eAAe,GAAG,eAAe,OAAO,kBAAkB,OAAO,EAAE,CAAA;iBACpE;gBACD,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAA;gBACxC,aAAa,CAAC,IAAI,CAAC,YAAY,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,YAAY,eAAe,GAAG,CAAC,CAAA;gBAE7E,uGAAuG;gBACvG,yBAAyB;gBACzB,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAA;aAC/C;SACF;QAED,IAAI,MAAM,GAAG;;;;;uCAKsB,OAAO;;MAExC,OAAO;;MAEP,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC;;MAExB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;;QAGrE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;QAElE,IAAI,CAAC,OAAO;aACX,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACT,OAAO,iBAAiB,CAAC,CAAC,GAAG,EAAE,IAAI,OAAO,CAAA;QAC5C,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC;;KAEd,CAAA,CAAC,SAAS;QAEX,MAAM,GAAG,kBAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAA;QAC1D,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,QAAQ,GAAG,KAAK,CAAC,EAAE,MAAM,CAAC,CAAA;IAC9E,CAAC;CACF;AAvFD,wCAuFC;AAED,SAAS,cAAc,CAAC,cAAkC,EAAE,eAAyB;IACnF,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE;QACvB,OAAO,EAAE,CAAA;KACV;IACD,MAAM,MAAM,GAAG,cAAc,CAAC,GAAG,CAAA;IAEjC,MAAM,SAAS,GAAG,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,wBAAwB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAA;IAClH,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAA;IAC7F,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAA;IACrE,MAAM,QAAQ,GAAG,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,uBAAuB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAA;IAExF,IAAI,SAAS,GAAG,EAAE,CAAA;IAClB,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE;QAC7C,SAAS,GAAG,gBAAgB,MAAM,CAAC,IAAI;;;eAG5B,MAAM,CAAC,IAAI;;;kBAGR,MAAM,CAAC,OAAO;;;;kEAIkC,MAAM,CAAC,IAAI;mBAC1D,MAAM,CAAC,IAAI,SAAS,MAAM,CAAC,IAAI;;;MAG5C,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;;MAEpB,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;;;;;;GAMpB,CAAA;KACA;IAED,OAAO;IACL,SAAS;;qBAEQ,MAAM,CAAC,IAAI;MAC1B,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;;MAElB,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC;;;;;iDAKwB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;;GAEpE,CAAA;AACH,CAAC;AAED,SAAS,eAAe,CAAC,MAAkB,EAAE,MAAkB;IAC7D,MAAM,aAAa,GAAG,4BAA4B,CAAC,MAAM,CAAC,CAAA;IAE1D,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;QACzC,OAAO,GAAG,KAAK,CAAC,IAAI,KAAK,IAAA,sBAAY,EAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAA;IACrD,CAAC,CAAC,CAAA;IAEF,IAAI,YAAY,GAAG,EAAE,CAAA;IACrB,IAAI,OAAO,CAAC,MAAM,CAAC,EAAE;QACnB,YAAY,GAAG;uBACI,MAAM,CAAC,IAAI,WAAW,aAAa;mCACvB,MAAM,CAAC,IAAI,GAAG,aAAa;oBAC1C,MAAM,CAAC,IAAI,GAAG,aAAa;;KAE1C,CAAA;KACF;IAED,OAAO;iBACQ,MAAM,CAAC,IAAI,GAAG,aAAa;MACtC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;;;IAGnB,YAAY;GACb,CAAA;AACH,CAAC;AAED,SAAS,8BAA8B,CAAC,IAAkB;IACxD,IAAI,aAAa,GAAG,EAAE,CAAA;IACtB,IAAI,IAAI,CAAC,mBAAmB,IAAI,IAAI,CAAC,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE;QACnE,MAAM,MAAM,GAAG,IAAI,CAAC,mBAAmB;aACpC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE;YACd,OAAO,GAAG,GAAG,GAAG,CAAA;QAClB,CAAC,CAAC;aACD,IAAI,CAAC,GAAG,CAAC,CAAA;QACZ,aAAa,GAAG,IAAI,MAAM,GAAG,CAAA;KAC9B;IACD,OAAO,aAAa,CAAA;AACtB,CAAC;AAED,SAAS,4BAA4B,CAAC,MAAkB;IACtD,IAAI,aAAa,GAAG,EAAE,CAAA;IAEtB,IAAI,MAAM,CAAC,mBAAmB,IAAI,MAAM,CAAC,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE;QACvE,MAAM,MAAM,GAAG,MAAM,CAAC,mBAAmB;aACtC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE;YACd,OAAO,GAAG,GAAG,GAAG,CAAA;QAClB,CAAC,CAAC;aACD,IAAI,CAAC,GAAG,CAAC,CAAA;QACZ,aAAa,GAAG,IAAI,MAAM,GAAG,CAAA;KAC9B;IACD,OAAO,aAAa,CAAA;AACtB,CAAC;AAED,SAAS,uBAAuB,CAAC,MAAkB,EAAE,IAAkB;IACrE,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;QAClB,OAAM;KACP;IAED,6CAA6C;IAC7C,uCAAuC;IACvC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;QAChD,OAAO,GAAG,IAAA,sBAAY,EAAC,KAAK,CAAC,EAAE,CAAA;IACjC,CAAC,CAAC,CAAA;IAEF,MAAM,aAAa,GAAG,mBAAmB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;IAE9D,MAAM,aAAa,GAAG,8BAA8B,CAAC,IAAI,CAAC,CAAA;IAC1D,OAAO;qBACY,aAAa,UAAU,aAAa;iDACR,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC;wBACzC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC;;GAErC,CAAA;AACH,CAAC;AAED,SAAS,wBAAwB,CAAC,MAAkB,EAAE,IAAkB;IACtE,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;QAClB,OAAO,EAAE,CAAA;KACV;IAED,MAAM,aAAa,GAAG,8BAA8B,CAAC,IAAI,CAAC,CAAA;IAE1D,MAAM,aAAa,GAAG,mBAAmB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;IAC9D,MAAM,MAAM,GAAG;WACN,aAAa,GAAG,aAAa,iBAAiB,MAAM,CAAC,IAAI,IAAI,aAAa,UAAU,aAAa,wCAAwC,MAAM,CAAC,IAAI;;mBAE5I,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI;;;IAGxC,CAAA;IAEF,OAAO,MAAM,CAAA;AACf,CAAC;AAED,SAAS,OAAO,CAAC,MAAkB;IACjC,OAAO,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAA;AACjH,CAAC;AAED,SAAS,gBAAgB,CAAC,MAAkB,EAAE,MAAkB;IAC9D,mCAAmC;IACnC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE;QACpB,OAAO,EAAE,CAAA;KACV;IACD,MAAM,MAAM,GAAG;WACN,MAAM,CAAC,IAAI,kBAAkB,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,gDAAgD,MAAM,CAAC,IAAI;;eAE9G,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI;;;;GAIvC,CAAA;IACD,OAAO,MAAM,CAAA;AACf,CAAC;AAED,SAAS,QAAQ,CAAC,KAAa;IAC7B,OAAO,KAAK;SACT,KAAK,CAAC,GAAG,CAAC;SACV,MAAM,CACL,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,CACf,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,EAAE,EACvG,EAAE,CACH,CAAA;AACL,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAa;IACxC,IAAI,CAAC,KAAK,EAAE;QACV,OAAO,KAAK,CAAA;KACb;IACD,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAA;AAC9E,CAAC","sourcesContent":["import fs from 'fs'\nimport path from 'path'\nimport prettier from 'prettier'\nimport { MoveFunction, MoveModule, MoveStruct, MoveModuleBytecode } from 'aptos/src/generated'\nimport { generateType, AccountRegister } from './typegen'\nimport { isFrameworkAccount } from '../aptos/utils'\n\nexport function generate(srcDir: string, outputDir: string) {\n  const files = fs.readdirSync(srcDir)\n  outputDir = path.resolve(outputDir)\n\n  fs.mkdirSync(outputDir, { recursive: true })\n\n  const loader = new AccountRegister()\n\n  // The first path, identify import relation and module name (filename) of those imports\n  for (const file of files) {\n    if (!file.endsWith('.json')) {\n      continue\n    }\n    // Reading file is duplicated, but since they are small files probably file\n    // TODO add file manager class\n    const json = fs.readFileSync(path.join(srcDir, file), 'utf-8')\n    const modules = JSON.parse(json)\n\n    for (const module of modules) {\n      if (module.abi) {\n        loader.register(module.abi, path.basename(file, '.json'))\n      }\n    }\n  }\n\n  for (const file of files) {\n    if (!file.endsWith('.json')) {\n      continue\n    }\n    const codeGen = new AccountCodegen(loader, {\n      srcFile: path.join(srcDir, file),\n      outputDir: outputDir,\n    })\n\n    codeGen.generate()\n  }\n\n  // when generating user code, don't need to generate framework account\n  loader.pendingAccounts.delete('0x1')\n  loader.pendingAccounts.delete('0x2')\n  loader.pendingAccounts.delete('0x3')\n\n  if (loader.pendingAccounts.size > 0) {\n    // TODO automatic download dependencies\n    console.error('Missing ABIs from the following accounts', loader.pendingAccounts)\n  }\n}\n\ninterface Config {\n  srcFile: string\n  outputDir: string\n}\n\nexport class AccountCodegen {\n  modules: MoveModuleBytecode[]\n  config: Config\n  loader: AccountRegister\n\n  constructor(loader: AccountRegister, config: Config) {\n    const json = fs.readFileSync(config.srcFile, 'utf-8')\n    this.modules = JSON.parse(json)\n    this.config = config\n    this.loader = loader\n  }\n\n  generate() {\n    if (!this.modules) {\n      return\n    }\n    const baseName = path.basename(this.config.srcFile, '.json')\n\n    let address: string | undefined\n    for (const module of this.modules) {\n      if (module.abi && module.abi.address) {\n        address = module.abi.address\n      }\n    }\n    if (!address) {\n      return\n    }\n\n    const imports = `\n    import { aptos } from \"@sentio/sdk\"\n    import { Address, MoveModule } from \"aptos/src/generated\"\n    `\n\n    const dependedAccounts: string[] = []\n\n    const moduleImports: string[] = []\n\n    const info = this.loader.accountImports.get(address)\n\n    if (info) {\n      for (const [account, moduleImported] of info.imports.entries()) {\n        // Remap to user's filename if possible, TODO codepath not well tested\n        let tsAccountModule = './' + (this.loader.accountImports.get(account)?.moduleName || account)\n        if (isFrameworkAccount(account) && !isFrameworkAccount(address)) {\n          // Decide where to find runtime library\n          let srcRoot = 'lib'\n          if (__dirname.includes('sdk/src/aptos-codegen')) {\n            srcRoot = 'src'\n          }\n          tsAccountModule = `@sentio/sdk/${srcRoot}/builtin/aptos/${account}`\n        }\n        const items = Array.from(moduleImported)\n        moduleImports.push(`import { ${items.join(',')} } from \"${tsAccountModule}\"`)\n\n        // Ideally we should use per module's load types, but it doesn't matter since we are loading the entire\n        // account modules anyway\n        items.forEach((m) => dependedAccounts.push(m))\n      }\n    }\n\n    let source = `\n    /* Autogenerated file. Do not edit manually. */\n    /* tslint:disable */\n    /* eslint-disable */\n  \n    /* Generated modules for account ${address} */\n  \n    ${imports}\n    \n    ${moduleImports.join('\\n')}\n    \n    ${this.modules.map((m) => generateModule(m, dependedAccounts)).join('\\n')}\n    \n    function loadAllTypes(registry: aptos.TypeRegistry) {\n      ${dependedAccounts.map((m) => `${m}.loadTypes(registry)`).join('\\n')}\n\n      ${this.modules\n        .map((m) => {\n          return `registry.load(${m.abi?.name}.ABI)`\n        })\n        .join('\\n')}\n    }\n    ` // source\n\n    source = prettier.format(source, { parser: 'typescript' })\n    fs.writeFileSync(path.join(this.config.outputDir, baseName + '.ts'), source)\n  }\n}\n\nfunction generateModule(moduleByteCode: MoveModuleBytecode, dependedModules: string[]) {\n  if (!moduleByteCode.abi) {\n    return ''\n  }\n  const module = moduleByteCode.abi\n\n  const functions = module.exposed_functions.map((f) => generateOnEntryFunctions(module, f)).filter((s) => s !== '')\n  const events = module.structs.map((e) => generateOnEvents(module, e)).filter((s) => s !== '')\n  const structs = module.structs.map((s) => generateStructs(module, s))\n  const callArgs = module.exposed_functions.map((f) => generateCallArgsStructs(module, f))\n\n  let processor = ''\n  if (functions.length > 0 || events.length > 0) {\n    processor = `export class ${module.name} extends aptos.AptosBaseProcessor {\n\n    constructor(options: aptos.AptosBindOptions) {\n      super(\"${module.name}\", options)\n    }\n    static DEFAULT_OPTIONS: aptos.AptosBindOptions = {\n      address: \"${module.address}\",\n      network: aptos.AptosNetwork.TEST_NET       \n    }\n\n    static bind(options: Partial<aptos.AptosBindOptions> = {}): ${module.name} {\n      return new ${module.name}({ ...${module.name}.DEFAULT_OPTIONS, ...options })\n    }\n    \n    ${functions.join('\\n')}\n    \n    ${events.join('\\n')}\n    \n    loadTypesInternal(registry: aptos.TypeRegistry) {\n      loadAllTypes(registry)\n    }\n  }\n  `\n  }\n\n  return `\n  ${processor}\n\n  export namespace ${module.name} {\n    ${structs.join('\\n')}\n    \n    ${callArgs.join('\\n')}\n       \n    export function loadTypes(registry: aptos.TypeRegistry) {\n      loadAllTypes(registry)\n    }\n    export const ABI: MoveModule = JSON.parse('${JSON.stringify(module)}')\n }\n  `\n}\n\nfunction generateStructs(module: MoveModule, struct: MoveStruct) {\n  const genericString = generateStructTypeParameters(struct)\n\n  const fields = struct.fields.map((field) => {\n    return `${field.name}: ${generateType(field.type)}`\n  })\n\n  let eventPayload = ''\n  if (isEvent(struct)) {\n    eventPayload = `\n    export interface ${struct.name}Instance${genericString} extends \n        aptos.TypedEventInstance<${struct.name}${genericString}> {\n      data_typed: ${struct.name}${genericString}\n    }\n    `\n  }\n\n  return `\n  export class ${struct.name}${genericString} {\n    ${fields.join('\\n')} \n  }\n  \n  ${eventPayload}\n  `\n}\n\nfunction generateFunctionTypeParameters(func: MoveFunction) {\n  let genericString = ''\n  if (func.generic_type_params && func.generic_type_params.length > 0) {\n    const params = func.generic_type_params\n      .map((v, idx) => {\n        return 'T' + idx\n      })\n      .join(',')\n    genericString = `<${params}>`\n  }\n  return genericString\n}\n\nfunction generateStructTypeParameters(struct: MoveStruct) {\n  let genericString = ''\n\n  if (struct.generic_type_params && struct.generic_type_params.length > 0) {\n    const params = struct.generic_type_params\n      .map((v, idx) => {\n        return 'T' + idx\n      })\n      .join(',')\n    genericString = `<${params}>`\n  }\n  return genericString\n}\n\nfunction generateCallArgsStructs(module: MoveModule, func: MoveFunction) {\n  if (!func.is_entry) {\n    return\n  }\n\n  // the first param is always signer so ignore\n  // TODO check if there is any edge case\n  const fields = func.params.slice(1).map((param) => {\n    return `${generateType(param)}`\n  })\n\n  const camelFuncName = capitalizeFirstChar(camelize(func.name))\n\n  const genericString = generateFunctionTypeParameters(func)\n  return `\n  export interface ${camelFuncName}Payload${genericString} \n      extends aptos.TypedEntryFunctionPayload<[${fields.join(',')}]> {\n    arguments_typed: [${fields.join(',')}]\n  }\n  `\n}\n\nfunction generateOnEntryFunctions(module: MoveModule, func: MoveFunction) {\n  if (!func.is_entry) {\n    return ''\n  }\n\n  const genericString = generateFunctionTypeParameters(func)\n\n  const camelFuncName = capitalizeFirstChar(camelize(func.name))\n  const source = `\n  onEntry${camelFuncName}${genericString}(func: (call: ${module.name}.${camelFuncName}Payload${genericString}, ctx: aptos.AptosContext) => void): ${module.name} {\n    this.onEntryFunctionCall(func, {\n      function: '${module.name}::${func.name}'\n    })\n    return this\n  }`\n\n  return source\n}\n\nfunction isEvent(struct: MoveStruct) {\n  return struct.abilities.includes('drop') && struct.abilities.includes('store') && struct.name.endsWith('Event')\n}\n\nfunction generateOnEvents(module: MoveModule, struct: MoveStruct): string {\n  // for struct that has drop + store\n  if (!isEvent(struct)) {\n    return ''\n  }\n  const source = `\n  onEvent${struct.name}(func: (event: ${module.name}.${struct.name}Instance, ctx: aptos.AptosContext) => void): ${module.name} {\n    this.onEvent(func, {\n      type: '${module.name}::${struct.name}'\n    })\n    return this\n  }\n  `\n  return source\n}\n\nfunction camelize(input: string): string {\n  return input\n    .split('_')\n    .reduce(\n      (res, word, i) =>\n        i === 0 ? word.toLowerCase() : `${res}${word.charAt(0).toUpperCase()}${word.substr(1).toLowerCase()}`,\n      ''\n    )\n}\n\nfunction capitalizeFirstChar(input: string): string {\n  if (!input) {\n    return input\n  }\n  return input[0].toUpperCase() + (input.length > 1 ? input.substring(1) : '')\n}\n"]}

package/lib/aptos-codegen/typegen.d.ts

@@ -0,0 +1,18 @@
+import { MoveModule } from 'aptos/src/generated';
+import { TypeDescriptor } from '../aptos/types';
+export declare function parseMoveType(type: string): TypeDescriptor;
+export declare function generateType(type: string, ctx?: any): string;
+export declare class AccountModulesImportInfo {
+    imports: Map<string, Set<string>>;
+    account: string;
+    moduleName: string;
+    constructor(account: string, tsModuleName: string);
+    addImport(account: string, module: string): void;
+}
+export declare class AccountRegister {
+    accountImports: Map<string, AccountModulesImportInfo>;
+    pendingAccounts: Set<string>;
+    typeRegistry: Map<string, TypeDescriptor>;
+    private loadTypeDescriptor;
+    register(module: MoveModule, tsModuleName: string): AccountModulesImportInfo;
+}

package/lib/aptos-codegen/typegen.js

@@ -0,0 +1,196 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AccountRegister = exports.AccountModulesImportInfo = exports.generateType = exports.parseMoveType = void 0;
+const types_1 = require("../aptos/types");
+const utils_1 = require("../aptos/utils");
+function generateTypeForDescriptor(type) {
+    // TODO &signer is defintely an address, but what if &OTHER_TYPE?
+    if (type.qname.startsWith('&')) {
+        return 'Address';
+    }
+    switch (type.qname) {
+        case 'signer': // TODO check this
+        case 'address':
+            return 'Address';
+        case '0x1::string::String':
+            return 'string';
+        case 'bool':
+            return 'Boolean';
+        case 'u8':
+        case 'u16':
+        case 'u32':
+            return 'number';
+        case 'u64':
+        case 'u128':
+            return 'bigint';
+    }
+    if (type.qname === utils_1.VECTOR_STR) {
+        // vector<u8> as hex string
+        const elementTypeQname = type.typeArgs[0].qname;
+        if (elementTypeQname === 'u8') {
+            return 'string';
+        }
+        if (elementTypeQname.startsWith('T') && !elementTypeQname.includes(utils_1.SPLITTER)) {
+            return `${elementTypeQname}[] | string`;
+        }
+        return generateTypeForDescriptor(type.typeArgs[0]) + '[]';
+    }
+    const simpleName = generateSimpleType(type.qname);
+    if (simpleName.length === 0) {
+        console.error('unexpected error');
+    }
+    if (simpleName.includes('vector')) {
+        console.error('unexpected error');
+    }
+    if (type.typeArgs.length > 0) {
+        // return simpleName
+        return simpleName + '<' + type.typeArgs.map((t) => generateTypeForDescriptor(t)).join(',') + '>';
+    }
+    return simpleName;
+}
+function generateSimpleType(type) {
+    const parts = type.split(utils_1.SPLITTER);
+    if (parts.length < 2) {
+        return parts[0];
+    }
+    return parts.slice(1).join('.');
+}
+function parseMoveType(type) {
+    // type = type.replace('&', '')
+    type = type.replace('&mut ', '&');
+    // TODO replace ' ' is not exactly safe, need to double check this
+    type = type.replace(' ', '');
+    const stack = [new types_1.TypeDescriptor('')];
+    let buffer = [];
+    // xxx:asdf<g1<a,<c,d>>, b, g2<a,b>, e>
+    for (let i = 0; i < type.length; i++) {
+        const ch = type[i];
+        if (ch === '<') {
+            // const symbol = type.slice(symbolStart, i)
+            // symbolStart =
+            const symbol = buffer.join('');
+            buffer = [];
+            stack[stack.length - 1].qname = symbol;
+            stack.push(new types_1.TypeDescriptor(''));
+            continue;
+        }
+        if (ch === '>') {
+            const typeParam = stack.pop();
+            if (!typeParam) {
+                throw Error('Uxpectecd stack size');
+            }
+            if (buffer.length > 0) {
+                typeParam.qname = buffer.join('');
+                buffer = [];
+            }
+            stack[stack.length - 1].typeArgs.push(typeParam);
+            continue;
+        }
+        if (ch === ',') {
+            const typeParam = stack.pop();
+            if (!typeParam) {
+                throw Error('Uxpectecd stack size');
+            }
+            if (buffer.length > 0) {
+                typeParam.qname = buffer.join('');
+                buffer = [];
+            }
+            stack[stack.length - 1].typeArgs.push(typeParam);
+            // continue parse next param
+            stack.push(new types_1.TypeDescriptor(''));
+            continue;
+        }
+        buffer.push(ch);
+    }
+    if (buffer.length > 0) {
+        stack[stack.length - 1].qname = buffer.join('');
+    }
+    const res = stack.pop();
+    if (!res || stack.length > 0) {
+        throw Error('Uxpectecd stack size');
+    }
+    return res;
+}
+exports.parseMoveType = parseMoveType;
+// TODO ctx need to have type parameters
+function generateType(type, ctx) {
+    return generateTypeForDescriptor(parseMoveType(type));
+}
+exports.generateType = generateType;
+class AccountModulesImportInfo {
+    // account to module
+    imports;
+    account;
+    moduleName;
+    constructor(account, tsModuleName) {
+        this.account = account;
+        this.moduleName = tsModuleName;
+        this.imports = new Map();
+    }
+    addImport(account, module) {
+        if (account === this.account) {
+            return;
+        }
+        let accountModules = this.imports.get(account);
+        if (!accountModules) {
+            accountModules = new Set();
+            this.imports.set(account, accountModules);
+        }
+        accountModules.add(module);
+    }
+}
+exports.AccountModulesImportInfo = AccountModulesImportInfo;
+class AccountRegister {
+    accountImports = new Map();
+    pendingAccounts = new Set();
+    // loadedAccount = new Set<string>()
+    typeRegistry = new Map();
+    loadTypeDescriptor(type) {
+        let descriptor = this.typeRegistry.get(type);
+        // const descriptparseMoveType(type)
+        if (!descriptor) {
+            descriptor = parseMoveType(type);
+            this.typeRegistry.set(type, descriptor);
+        }
+        return descriptor;
+    }
+    register(module, tsModuleName) {
+        const currentModuleFqn = (0, utils_1.moduleQname)(module);
+        let accountModuleImports = this.accountImports.get(module.address);
+        if (!accountModuleImports) {
+            accountModuleImports = new AccountModulesImportInfo(module.address, tsModuleName);
+            this.accountImports.set(module.address, accountModuleImports);
+            // the account has already be processed, delete pending task
+            this.pendingAccounts.delete(module.address);
+        }
+        for (const struct of module.structs) {
+            for (const field of struct.fields) {
+                for (const type of this.loadTypeDescriptor(field.type).dependedTypes()) {
+                    const [account, module] = (0, utils_1.moduleQnameForType)(type);
+                    accountModuleImports.addImport(account, module);
+                    if (!this.accountImports.has(account)) {
+                        this.pendingAccounts.add(account);
+                    }
+                }
+            }
+        }
+        for (const func of module.exposed_functions) {
+            if (!func.is_entry) {
+                continue;
+            }
+            for (const param of func.params) {
+                for (const type of this.loadTypeDescriptor(param).dependedTypes()) {
+                    const [account, module] = (0, utils_1.moduleQnameForType)(type);
+                    accountModuleImports.addImport(account, module);
+                    if (!this.accountImports.has(account)) {
+                        this.pendingAccounts.add(account);
+                    }
+                }
+            }
+        }
+        this.accountImports.set(currentModuleFqn, accountModuleImports);
+        return accountModuleImports;
+    }
+}
+exports.AccountRegister = AccountRegister;
+//# sourceMappingURL=typegen.js.map

package/lib/aptos-codegen/typegen.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"typegen.js","sourceRoot":"","sources":["../../src/aptos-codegen/typegen.ts"],"names":[],"mappings":";;;AACA,0CAA+C;AAC/C,0CAAsF;AAEtF,SAAS,yBAAyB,CAAC,IAAoB;IACrD,iEAAiE;IACjE,IAAI,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE;QAC9B,OAAO,SAAS,CAAA;KACjB;IAED,QAAQ,IAAI,CAAC,KAAK,EAAE;QAClB,KAAK,QAAQ,CAAC,CAAC,kBAAkB;QACjC,KAAK,SAAS;YACZ,OAAO,SAAS,CAAA;QAClB,KAAK,qBAAqB;YACxB,OAAO,QAAQ,CAAA;QACjB,KAAK,MAAM;YACT,OAAO,SAAS,CAAA;QAClB,KAAK,IAAI,CAAC;QACV,KAAK,KAAK,CAAC;QACX,KAAK,KAAK;YACR,OAAO,QAAQ,CAAA;QACjB,KAAK,KAAK,CAAC;QACX,KAAK,MAAM;YACT,OAAO,QAAQ,CAAA;KAClB;IAED,IAAI,IAAI,CAAC,KAAK,KAAK,kBAAU,EAAE;QAC7B,2BAA2B;QAC3B,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAA;QAC/C,IAAI,gBAAgB,KAAK,IAAI,EAAE;YAC7B,OAAO,QAAQ,CAAA;SAChB;QACD,IAAI,gBAAgB,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,gBAAQ,CAAC,EAAE;YAC5E,OAAO,GAAG,gBAAgB,aAAa,CAAA;SACxC;QACD,OAAO,yBAAyB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAA;KAC1D;IAED,MAAM,UAAU,GAAG,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IACjD,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE;QAC3B,OAAO,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAA;KAClC;IACD,IAAI,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE;QACjC,OAAO,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAA;KAClC;IACD,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;QAC5B,oBAAoB;QACpB,OAAO,UAAU,GAAG,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,CAAA;KACjG;IACD,OAAO,UAAU,CAAA;AACnB,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAY;IACtC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,gBAAQ,CAAC,CAAA;IAClC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE;QACpB,OAAO,KAAK,CAAC,CAAC,CAAC,CAAA;KAChB;IAED,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;AACjC,CAAC;AAED,SAAgB,aAAa,CAAC,IAAY;IACxC,+BAA+B;IAE/B,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,CAAA;IAEjC,kEAAkE;IAClE,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,CAAA;IAE5B,MAAM,KAAK,GAAqB,CAAC,IAAI,sBAAc,CAAC,EAAE,CAAC,CAAC,CAAA;IACxD,IAAI,MAAM,GAAG,EAAE,CAAA;IAEf,uCAAuC;IACvC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;QACpC,MAAM,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,CAAA;QAClB,IAAI,EAAE,KAAK,GAAG,EAAE;YACd,4CAA4C;YAC5C,gBAAgB;YAChB,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;YAC9B,MAAM,GAAG,EAAE,CAAA;YACX,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,KAAK,GAAG,MAAM,CAAA;YACtC,KAAK,CAAC,IAAI,CAAC,IAAI,sBAAc,CAAC,EAAE,CAAC,CAAC,CAAA;YAClC,SAAQ;SACT;QACD,IAAI,EAAE,KAAK,GAAG,EAAE;YACd,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,EAAE,CAAA;YAC7B,IAAI,CAAC,SAAS,EAAE;gBACd,MAAM,KAAK,CAAC,sBAAsB,CAAC,CAAA;aACpC;YACD,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE;gBACrB,SAAS,CAAC,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;gBACjC,MAAM,GAAG,EAAE,CAAA;aACZ;YACD,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;YAChD,SAAQ;SACT;QACD,IAAI,EAAE,KAAK,GAAG,EAAE;YACd,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,EAAE,CAAA;YAC7B,IAAI,CAAC,SAAS,EAAE;gBACd,MAAM,KAAK,CAAC,sBAAsB,CAAC,CAAA;aACpC;YACD,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE;gBACrB,SAAS,CAAC,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;gBACjC,MAAM,GAAG,EAAE,CAAA;aACZ;YAED,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;YAChD,4BAA4B;YAC5B,KAAK,CAAC,IAAI,CAAC,IAAI,sBAAc,CAAC,EAAE,CAAC,CAAC,CAAA;YAClC,SAAQ;SACT;QAED,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;KAChB;IAED,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE;QACrB,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;KAChD;IAED,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,EAAE,CAAA;IACvB,IAAI,CAAC,GAAG,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE;QAC5B,MAAM,KAAK,CAAC,sBAAsB,CAAC,CAAA;KACpC;IACD,OAAO,GAAG,CAAA;AACZ,CAAC;AA/DD,sCA+DC;AAED,wCAAwC;AACxC,SAAgB,YAAY,CAAC,IAAY,EAAE,GAAS;IAClD,OAAO,yBAAyB,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAA;AACvD,CAAC;AAFD,oCAEC;AAED,MAAa,wBAAwB;IACnC,oBAAoB;IACpB,OAAO,CAA0B;IACjC,OAAO,CAAQ;IACf,UAAU,CAAQ;IAElB,YAAY,OAAe,EAAE,YAAoB;QAC/C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAA;QACtB,IAAI,CAAC,UAAU,GAAG,YAAY,CAAA;QAC9B,IAAI,CAAC,OAAO,GAAG,IAAI,GAAG,EAAuB,CAAA;IAC/C,CAAC;IAED,SAAS,CAAC,OAAe,EAAE,MAAc;QACvC,IAAI,OAAO,KAAK,IAAI,CAAC,OAAO,EAAE;YAC5B,OAAM;SACP;QACD,IAAI,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;QAC9C,IAAI,CAAC,cAAc,EAAE;YACnB,cAAc,GAAG,IAAI,GAAG,EAAU,CAAA;YAClC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,cAAc,CAAC,CAAA;SAC1C;QACD,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;IAC5B,CAAC;CACF;AAvBD,4DAuBC;AAED,MAAa,eAAe;IAC1B,cAAc,GAAG,IAAI,GAAG,EAAoC,CAAA;IAC5D,eAAe,GAAG,IAAI,GAAG,EAAU,CAAA;IAEnC,oCAAoC;IACpC,YAAY,GAAG,IAAI,GAAG,EAA0B,CAAA;IAExC,kBAAkB,CAAC,IAAY;QACrC,IAAI,UAAU,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;QAE5C,oCAAoC;QACpC,IAAI,CAAC,UAAU,EAAE;YACf,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,CAAA;YAChC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,EAAE,UAAU,CAAC,CAAA;SACxC;QACD,OAAO,UAAU,CAAA;IACnB,CAAC;IAED,QAAQ,CAAC,MAAkB,EAAE,YAAoB;QAC/C,MAAM,gBAAgB,GAAG,IAAA,mBAAW,EAAC,MAAM,CAAC,CAAA;QAE5C,IAAI,oBAAoB,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;QAClE,IAAI,CAAC,oBAAoB,EAAE;YACzB,oBAAoB,GAAG,IAAI,wBAAwB,CAAC,MAAM,CAAC,OAAO,EAAE,YAAY,CAAC,CAAA;YACjF,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,EAAE,oBAAoB,CAAC,CAAA;YAC7D,4DAA4D;YAC5D,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;SAC5C;QAED,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,OAAO,EAAE;YACnC,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE;gBACjC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,EAAE;oBACtE,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,GAAG,IAAA,0BAAkB,EAAC,IAAI,CAAC,CAAA;oBAClD,oBAAoB,CAAC,SAAS,CAAC,OAAO,EAAE,MAAM,CAAC,CAAA;oBAC/C,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;wBACrC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;qBAClC;iBACF;aACF;SACF;QAED,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,iBAAiB,EAAE;YAC3C,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;gBAClB,SAAQ;aACT;YACD,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE;gBAC/B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,EAAE;oBACjE,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,GAAG,IAAA,0BAAkB,EAAC,IAAI,CAAC,CAAA;oBAClD,oBAAoB,CAAC,SAAS,CAAC,OAAO,EAAE,MAAM,CAAC,CAAA;oBAC/C,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;wBACrC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;qBAClC;iBACF;aACF;SACF;QACD,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,gBAAgB,EAAE,oBAAoB,CAAC,CAAA;QAC/D,OAAO,oBAAoB,CAAA;IAC7B,CAAC;CACF;AA1DD,0CA0DC","sourcesContent":["import { MoveModule } from 'aptos/src/generated'\nimport { TypeDescriptor } from '../aptos/types'\nimport { moduleQname, moduleQnameForType, SPLITTER, VECTOR_STR } from '../aptos/utils'\n\nfunction generateTypeForDescriptor(type: TypeDescriptor): string {\n  // TODO &signer is defintely an address, but what if &OTHER_TYPE?\n  if (type.qname.startsWith('&')) {\n    return 'Address'\n  }\n\n  switch (type.qname) {\n    case 'signer': // TODO check this\n    case 'address':\n      return 'Address'\n    case '0x1::string::String':\n      return 'string'\n    case 'bool':\n      return 'Boolean'\n    case 'u8':\n    case 'u16':\n    case 'u32':\n      return 'number'\n    case 'u64':\n    case 'u128':\n      return 'bigint'\n  }\n\n  if (type.qname === VECTOR_STR) {\n    // vector<u8> as hex string\n    const elementTypeQname = type.typeArgs[0].qname\n    if (elementTypeQname === 'u8') {\n      return 'string'\n    }\n    if (elementTypeQname.startsWith('T') && !elementTypeQname.includes(SPLITTER)) {\n      return `${elementTypeQname}[] | string`\n    }\n    return generateTypeForDescriptor(type.typeArgs[0]) + '[]'\n  }\n\n  const simpleName = generateSimpleType(type.qname)\n  if (simpleName.length === 0) {\n    console.error('unexpected error')\n  }\n  if (simpleName.includes('vector')) {\n    console.error('unexpected error')\n  }\n  if (type.typeArgs.length > 0) {\n    // return simpleName\n    return simpleName + '<' + type.typeArgs.map((t) => generateTypeForDescriptor(t)).join(',') + '>'\n  }\n  return simpleName\n}\n\nfunction generateSimpleType(type: string): string {\n  const parts = type.split(SPLITTER)\n  if (parts.length < 2) {\n    return parts[0]\n  }\n\n  return parts.slice(1).join('.')\n}\n\nexport function parseMoveType(type: string) {\n  // type = type.replace('&', '')\n\n  type = type.replace('&mut ', '&')\n\n  // TODO replace ' ' is not exactly safe, need to double check this\n  type = type.replace(' ', '')\n\n  const stack: TypeDescriptor[] = [new TypeDescriptor('')]\n  let buffer = []\n\n  // xxx:asdf<g1<a,<c,d>>, b, g2<a,b>, e>\n  for (let i = 0; i < type.length; i++) {\n    const ch = type[i]\n    if (ch === '<') {\n      // const symbol = type.slice(symbolStart, i)\n      // symbolStart =\n      const symbol = buffer.join('')\n      buffer = []\n      stack[stack.length - 1].qname = symbol\n      stack.push(new TypeDescriptor(''))\n      continue\n    }\n    if (ch === '>') {\n      const typeParam = stack.pop()\n      if (!typeParam) {\n        throw Error('Uxpectecd stack size')\n      }\n      if (buffer.length > 0) {\n        typeParam.qname = buffer.join('')\n        buffer = []\n      }\n      stack[stack.length - 1].typeArgs.push(typeParam)\n      continue\n    }\n    if (ch === ',') {\n      const typeParam = stack.pop()\n      if (!typeParam) {\n        throw Error('Uxpectecd stack size')\n      }\n      if (buffer.length > 0) {\n        typeParam.qname = buffer.join('')\n        buffer = []\n      }\n\n      stack[stack.length - 1].typeArgs.push(typeParam)\n      // continue parse next param\n      stack.push(new TypeDescriptor(''))\n      continue\n    }\n\n    buffer.push(ch)\n  }\n\n  if (buffer.length > 0) {\n    stack[stack.length - 1].qname = buffer.join('')\n  }\n\n  const res = stack.pop()\n  if (!res || stack.length > 0) {\n    throw Error('Uxpectecd stack size')\n  }\n  return res\n}\n\n// TODO ctx need to have type parameters\nexport function generateType(type: string, ctx?: any): string {\n  return generateTypeForDescriptor(parseMoveType(type))\n}\n\nexport class AccountModulesImportInfo {\n  // account to module\n  imports: Map<string, Set<string>>\n  account: string\n  moduleName: string\n\n  constructor(account: string, tsModuleName: string) {\n    this.account = account\n    this.moduleName = tsModuleName\n    this.imports = new Map<string, Set<string>>()\n  }\n\n  addImport(account: string, module: string) {\n    if (account === this.account) {\n      return\n    }\n    let accountModules = this.imports.get(account)\n    if (!accountModules) {\n      accountModules = new Set<string>()\n      this.imports.set(account, accountModules)\n    }\n    accountModules.add(module)\n  }\n}\n\nexport class AccountRegister {\n  accountImports = new Map<string, AccountModulesImportInfo>()\n  pendingAccounts = new Set<string>()\n\n  // loadedAccount = new Set<string>()\n  typeRegistry = new Map<string, TypeDescriptor>()\n\n  private loadTypeDescriptor(type: string) {\n    let descriptor = this.typeRegistry.get(type)\n\n    // const descriptparseMoveType(type)\n    if (!descriptor) {\n      descriptor = parseMoveType(type)\n      this.typeRegistry.set(type, descriptor)\n    }\n    return descriptor\n  }\n\n  register(module: MoveModule, tsModuleName: string): AccountModulesImportInfo {\n    const currentModuleFqn = moduleQname(module)\n\n    let accountModuleImports = this.accountImports.get(module.address)\n    if (!accountModuleImports) {\n      accountModuleImports = new AccountModulesImportInfo(module.address, tsModuleName)\n      this.accountImports.set(module.address, accountModuleImports)\n      // the account has already be processed, delete pending task\n      this.pendingAccounts.delete(module.address)\n    }\n\n    for (const struct of module.structs) {\n      for (const field of struct.fields) {\n        for (const type of this.loadTypeDescriptor(field.type).dependedTypes()) {\n          const [account, module] = moduleQnameForType(type)\n          accountModuleImports.addImport(account, module)\n          if (!this.accountImports.has(account)) {\n            this.pendingAccounts.add(account)\n          }\n        }\n      }\n    }\n\n    for (const func of module.exposed_functions) {\n      if (!func.is_entry) {\n        continue\n      }\n      for (const param of func.params) {\n        for (const type of this.loadTypeDescriptor(param).dependedTypes()) {\n          const [account, module] = moduleQnameForType(type)\n          accountModuleImports.addImport(account, module)\n          if (!this.accountImports.has(account)) {\n            this.pendingAccounts.add(account)\n          }\n        }\n      }\n    }\n    this.accountImports.set(currentModuleFqn, accountModuleImports)\n    return accountModuleImports\n  }\n}\n"]}

package/lib/binds.d.ts

@@ -2,8 +2,7 @@ import { BindOptions } from './core/bind-options';
 import { BaseProcessor, ContractView } from './core';
 import { Networkish } from '@ethersproject/networks';
 import { BaseContract } from 'ethers';
-export declare function getContractName(abiName: string, contractName: string | undefined, address: string, network?: Networkish): string;
-export declare function getProcessor(abiName: string, opts: BindOptions): BaseProcessor<any, any> | undefined;
-export declare function addProcessor(abiName: string, opts: BindOptions, processor: BaseProcessor<any, any>): void;
+export declare function getProcessor(opts: BindOptions): BaseProcessor<any, any> | undefined;
+export declare function addProcessor(opts: BindOptions, processor: BaseProcessor<any, any>): void;
 export declare function getContractByABI(abiName: string, address: string, network: Networkish): ContractView<BaseContract> | undefined;
 export declare function addContractByABI(abiName: string, address: string, network: Networkish, contract: ContractView<BaseContract>): Map<string, ContractView<BaseContract>>;