@sentio/sdk 1.0.0-development → 1.0.0-rc.3

package/lib/aptos/utils.js

@@ -0,0 +1,27 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getMeaningfulFunctionParams = exports.moduleQnameForType = exports.moduleQname = exports.isFrameworkAccount = exports.VECTOR_STR = exports.SPLITTER = void 0;
+exports.SPLITTER = '::';
+exports.VECTOR_STR = 'vector';
+function isFrameworkAccount(account) {
+    return account === '0x1' || account === '0x2' || account === '0x3';
+}
+exports.isFrameworkAccount = isFrameworkAccount;
+function moduleQname(module) {
+    return module.address.toLowerCase() + exports.SPLITTER + module.name;
+}
+exports.moduleQname = moduleQname;
+function moduleQnameForType(type) {
+    const parts = type.split(exports.SPLITTER).slice(0, 2);
+    return [parts[0], parts[1]];
+}
+exports.moduleQnameForType = moduleQnameForType;
+function getMeaningfulFunctionParams(func) {
+    let params = func.params;
+    if (func.params[0] === '&signer') {
+        params = params.slice(1);
+    }
+    return params;
+}
+exports.getMeaningfulFunctionParams = getMeaningfulFunctionParams;
+//# sourceMappingURL=utils.js.map

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;AAED,SAAgB,2BAA2B,CAAC,IAAkB;IAC5D,IAAI,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;IACxB,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,SAAS,EAAE;QAChC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;KACzB;IACD,OAAO,MAAM,CAAA;AACf,CAAC;AAND,kEAMC","sourcesContent":["import { MoveFunction, MoveModule } from 'aptos-sdk/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\nexport function getMeaningfulFunctionParams(func: MoveFunction): string[] {\n  let params = func.params\n  if (func.params[0] === '&signer') {\n    params = params.slice(1)\n  }\n  return params\n}\n"]}

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

@@ -0,0 +1,23 @@
+import { MoveModuleBytecode } from 'aptos-sdk/src/generated';
+import { AccountRegister } from './typegen';
+import { AptosNetwork } from '../aptos/network';
+export declare function codeGenAptosProcessor(abisDir: string, outDir?: string): void;
+export declare function generate(srcDir: string, outputDir: string): Promise<void>;
+export declare function generateForNetwork(srcDir: string, outputDir: string, network: AptosNetwork): Promise<void>;
+interface OutputFile {
+    fileName: string;
+    fileContent: string;
+}
+interface Config {
+    fileName: string;
+    outputDir: string;
+    network: AptosNetwork;
+}
+export declare class AccountCodegen {
+    modules: MoveModuleBytecode[];
+    config: Config;
+    loader: AccountRegister;
+    constructor(loader: AccountRegister, modules: MoveModuleBytecode[], config: Config);
+    generate(): OutputFile[];
+}
+export {};

package/lib/aptos-codegen/codegen.js

@@ -0,0 +1,364 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AccountCodegen = exports.generateForNetwork = exports.generate = exports.codeGenAptosProcessor = void 0;
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const prettier_1 = __importDefault(require("prettier"));
+const typegen_1 = require("./typegen");
+const utils_1 = require("../aptos/utils");
+const chalk_1 = __importDefault(require("chalk"));
+const network_1 = require("../aptos/network");
+const types_1 = require("../aptos/types");
+function codeGenAptosProcessor(abisDir, outDir = 'src/types/aptos') {
+    if (!fs_1.default.existsSync(abisDir)) {
+        return;
+    }
+    console.log(chalk_1.default.green('Generated Types for Aptos'));
+    generate(abisDir, outDir);
+}
+exports.codeGenAptosProcessor = codeGenAptosProcessor;
+async function generate(srcDir, outputDir) {
+    await generateForNetwork(srcDir, outputDir, network_1.AptosNetwork.MAIN_NET);
+    await generateForNetwork(path_1.default.join(srcDir, 'testnet'), path_1.default.join(outputDir, 'testnet'), network_1.AptosNetwork.TEST_NET);
+}
+exports.generate = generate;
+async function generateForNetwork(srcDir, outputDir, network) {
+    if (!fs_1.default.existsSync(srcDir)) {
+        return;
+    }
+    if (network === network_1.AptosNetwork.TEST_NET) {
+        console.log('Found testnet directory, generate code for testnet modules');
+    }
+    const files = fs_1.default.readdirSync(srcDir);
+    outputDir = path_1.default.resolve(outputDir);
+    const outputs = [];
+    fs_1.default.mkdirSync(outputDir, { recursive: true });
+    const loader = new typegen_1.AccountRegister();
+    // when generating user code, don't need to generate framework account
+    loader.accountImports.set('0x1', new typegen_1.AccountModulesImportInfo('0x1', '0x1'));
+    loader.accountImports.set('0x2', new typegen_1.AccountModulesImportInfo('0x2', '0x2'));
+    loader.accountImports.set('0x3', new typegen_1.AccountModulesImportInfo('0x3', '0x3'));
+    const client = (0, network_1.getRpcClient)(network);
+    for (const file of files) {
+        if (!file.endsWith('.json')) {
+            continue;
+        }
+        const fullPath = path_1.default.resolve(srcDir, file);
+        const modules = JSON.parse(fs_1.default.readFileSync(fullPath, 'utf-8'));
+        for (const module of modules) {
+            if (module.abi) {
+                loader.register(module.abi, path_1.default.basename(file, '.json'));
+            }
+        }
+        const codeGen = new AccountCodegen(loader, modules, {
+            fileName: path_1.default.basename(file, '.json'),
+            outputDir: outputDir,
+            network,
+        });
+        outputs.push(...codeGen.generate());
+    }
+    while (loader.pendingAccounts.size > 0) {
+        for (const account of loader.pendingAccounts) {
+            console.log(`download dependent module for account ${account} at ${(0, network_1.getAptosChainName)(network)}`);
+            try {
+                const modules = await client.getAccountModules(account);
+                fs_1.default.writeFileSync(path_1.default.resolve(srcDir, account + '.json'), JSON.stringify(modules, null, '\t'));
+                for (const module of modules) {
+                    if (module.abi) {
+                        loader.register(module.abi, account);
+                    }
+                }
+                const codeGen = new AccountCodegen(loader, modules, {
+                    fileName: account,
+                    outputDir: outputDir,
+                    network,
+                });
+                outputs.push(...codeGen.generate());
+            }
+            catch (e) {
+                console.error(chalk_1.default.red('Error downloading account module, check if you choose the right network，or download account modules manually into your director'));
+                console.error(e);
+                process.exit(1);
+            }
+        }
+    }
+    for (const output of outputs) {
+        const content = prettier_1.default.format(output.fileContent, { parser: 'typescript' });
+        fs_1.default.writeFileSync(path_1.default.join(outputDir, output.fileName), content);
+    }
+}
+exports.generateForNetwork = generateForNetwork;
+class AccountCodegen {
+    modules;
+    config;
+    loader;
+    constructor(loader, modules, config) {
+        // const json = fs.readFileSync(config.srcFile, 'utf-8')
+        this.modules = modules;
+        this.config = config;
+        this.loader = loader;
+    }
+    generate() {
+        if (!this.modules) {
+            return [];
+        }
+        // const baseName = path.basename(this.config.fileName, '.json')
+        let address;
+        for (const module of this.modules) {
+            if (module.abi && module.abi.address) {
+                address = module.abi.address;
+            }
+        }
+        if (!address) {
+            return [];
+        }
+        const imports = `
+    import { aptos } from "@sentio/sdk"
+    import { Address, MoveModule } from "aptos-sdk/src/generated"
+    `;
+        const dependedAccounts = [];
+        const moduleImports = [];
+        const info = this.loader.accountImports.get(address);
+        if (info) {
+            for (const [account] 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}`;
+                }
+                moduleImports.push(`import * as _${account} from "${tsAccountModule}"`);
+                dependedAccounts.push(account);
+            }
+        }
+        const source = `
+    /* Autogenerated file. Do not edit manually. */
+    /* tslint:disable */
+    /* eslint-disable */
+  
+    /* Generated modules for account ${address} */
+  
+    ${imports}
+    
+    ${moduleImports.join('\n')}
+    
+    ${this.modules.map((m) => generateModule(m, this.config.network)).join('\n')}
+    
+    export function loadAllTypes(_r: aptos.TypeRegistry) {
+      ${dependedAccounts.map((a) => `_${a}.loadAllTypes(_r)`).join('\n')}
+
+      ${this.modules
+            .map((m) => {
+            return `_r.load(${m.abi?.name}.ABI)`;
+        })
+            .join('\n')}
+    }
+    `; // source
+        return [
+            {
+                fileName: this.config.fileName + '.ts',
+                fileContent: source,
+            },
+        ];
+    }
+}
+exports.AccountCodegen = AccountCodegen;
+function generateNetworkOption(network) {
+    switch (network) {
+        case network_1.AptosNetwork.TEST_NET:
+            return 'TEST_NET';
+    }
+    return 'MAIN_NET';
+}
+function generateModule(moduleByteCode, network) {
+    if (!moduleByteCode.abi) {
+        return '';
+    }
+    const module = moduleByteCode.abi;
+    const functions = module.exposed_functions.map((f) => generateOnEntryFunctions(module, f)).filter((s) => s !== '');
+    const eventStructs = getEventStructs(module);
+    const eventTypes = new Set(eventStructs.keys());
+    const events = Array.from(eventStructs.values())
+        .map((e) => generateOnEvents(module, e))
+        .filter((s) => s !== '');
+    const structs = module.structs.map((s) => generateStructs(module, s, eventTypes));
+    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.${generateNetworkOption(network)}       
+    }
+
+    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(_r: aptos.TypeRegistry) {
+      loadAllTypes(_r)
+    }
+    export const ABI: MoveModule = JSON.parse('${JSON.stringify(module)}')
+ }
+  `;
+}
+function generateStructs(module, struct, events) {
+    const genericString = generateStructTypeParameters(struct);
+    const genericStringAny = generateStructTypeParameters(struct, true);
+    const fields = struct.fields.map((field) => {
+        return `${field.name}: ${(0, typegen_1.generateType)(field.type, module.address)}`;
+    });
+    let eventPayload = '';
+    if (events.has((0, utils_1.moduleQname)(module) + utils_1.SPLITTER + struct.name)) {
+        eventPayload = `
+    export interface ${struct.name}Instance extends 
+        aptos.TypedEventInstance<${struct.name}${genericStringAny}> {
+      data_typed: ${struct.name}${genericStringAny}
+      type_arguments: [${struct.generic_type_params.map((_) => 'string').join(', ')}]
+    }
+    `;
+    }
+    return `
+  export class ${struct.name}${genericString} {
+    static TYPE_QNAME = '${module.address}::${module.name}::${struct.name}'
+    ${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}=any`;
+        })
+            .join(',');
+        genericString = `<${params}>`;
+    }
+    return genericString;
+}
+function generateStructTypeParameters(struct, useAny = false) {
+    let genericString = '';
+    if (struct.generic_type_params && struct.generic_type_params.length > 0) {
+        const params = struct.generic_type_params
+            .map((v, idx) => {
+            return useAny ? 'any' : 'T' + idx;
+        })
+            .join(',');
+        genericString = `<${params}>`;
+    }
+    return genericString;
+}
+function generateCallArgsStructs(module, func) {
+    if (!func.is_entry) {
+        return;
+    }
+    const fields = (0, utils_1.getMeaningfulFunctionParams)(func).map((param) => {
+        return `${(0, typegen_1.generateType)(param, module.address)}`;
+    });
+    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(',')}],
+    type_arguments: [${func.generic_type_params.map((_) => 'string').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: ${module.name}.${camelFuncName}Payload, ctx: aptos.AptosContext) => void, filter?: aptos.CallFilter): ${module.name} {
+    this.onEntryFunctionCall(func, {
+      ...filter,
+      function: '${module.name}::${func.name}'
+    })
+    return this
+  }`;
+    return source;
+}
+function getEventStructs(module) {
+    const qname = (0, utils_1.moduleQname)(module);
+    const structMap = new Map();
+    const eventMap = new Map();
+    for (const struct of module.structs) {
+        structMap.set(qname + utils_1.SPLITTER + struct.name, struct);
+    }
+    for (const struct of module.structs) {
+        for (const field of struct.fields) {
+            const t = (0, types_1.parseMoveType)(field.type);
+            if (t.qname === '0x1::event::EventHandle') {
+                const event = t.typeArgs[0].qname;
+                const eventStruct = structMap.get(event);
+                if (eventStruct) {
+                    eventMap.set(event, eventStruct);
+                }
+            }
+        }
+    }
+    return eventMap;
+}
+function generateOnEvents(module, struct) {
+    // for struct that has drop + store
+    // if (!isEvent(struct, module)) {
+    //   return ''
+    // }
+    // const genericString = generateStructTypeParameters(struct)
+    const source = `
+  onEvent${struct.name}(func: (event: ${module.name}.${struct.name}Instance, ctx: aptos.AptosContext) => void): ${module.name} {
+    this.onEvent(func, {
+      type: '${module.name}::${struct.name}'
+    })
+    return this
+  }
+  `;
+    return source;
+}
+function camelize(input) {
+    return input
+        .split('_')
+        .reduce((res, word, i) => i === 0 ? word.toLowerCase() : `${res}${word.charAt(0).toUpperCase()}${word.substr(1).toLowerCase()}`, '');
+}
+function capitalizeFirstChar(input) {
+    if (!input) {
+        return input;
+    }
+    return input[0].toUpperCase() + (input.length > 1 ? input.substring(1) : '');
+}
+//# sourceMappingURL=codegen.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"codegen.js","sourceRoot":"","sources":["../../src/aptos-codegen/codegen.ts"],"names":[],"mappings":";;;;;;AAAA,4CAAmB;AACnB,gDAAuB;AACvB,wDAA+B;AAE/B,uCAAmF;AACnF,0CAAuG;AACvG,kDAAyB;AACzB,8CAAgF;AAChF,0CAA8C;AAE9C,SAAgB,qBAAqB,CAAC,OAAe,EAAE,MAAM,GAAG,iBAAiB;IAC/E,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE;QAC3B,OAAM;KACP;IACD,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC,CAAA;IACrD,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC,CAAA;AAC3B,CAAC;AAND,sDAMC;AAEM,KAAK,UAAU,QAAQ,CAAC,MAAc,EAAE,SAAiB;IAC9D,MAAM,kBAAkB,CAAC,MAAM,EAAE,SAAS,EAAE,sBAAY,CAAC,QAAQ,CAAC,CAAA;IAClE,MAAM,kBAAkB,CAAC,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,CAAC,EAAE,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,sBAAY,CAAC,QAAQ,CAAC,CAAA;AAChH,CAAC;AAHD,4BAGC;AAEM,KAAK,UAAU,kBAAkB,CAAC,MAAc,EAAE,SAAiB,EAAE,OAAqB;IAC/F,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE;QAC1B,OAAM;KACP;IACD,IAAI,OAAO,KAAK,sBAAY,CAAC,QAAQ,EAAE;QACrC,OAAO,CAAC,GAAG,CAAC,4DAA4D,CAAC,CAAA;KAC1E;IACD,MAAM,KAAK,GAAG,YAAE,CAAC,WAAW,CAAC,MAAM,CAAC,CAAA;IACpC,SAAS,GAAG,cAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;IACnC,MAAM,OAAO,GAAiB,EAAE,CAAA;IAEhC,YAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;IAE5C,MAAM,MAAM,GAAG,IAAI,yBAAe,EAAE,CAAA;IAEpC,sEAAsE;IACtE,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,kCAAwB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAA;IAC5E,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,kCAAwB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAA;IAC5E,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,kCAAwB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAA;IAC5E,MAAM,MAAM,GAAG,IAAA,sBAAY,EAAC,OAAO,CAAC,CAAA;IAEpC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;QACxB,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE;YAC3B,SAAQ;SACT;QACD,MAAM,QAAQ,GAAG,cAAI,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAA;QAC3C,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,YAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAA;QAE9D,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;QACD,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC,MAAM,EAAE,OAAO,EAAE;YAClD,QAAQ,EAAE,cAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC;YACtC,SAAS,EAAE,SAAS;YACpB,OAAO;SACR,CAAC,CAAA;QAEF,OAAO,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAA;KACpC;IAED,OAAO,MAAM,CAAC,eAAe,CAAC,IAAI,GAAG,CAAC,EAAE;QACtC,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,eAAe,EAAE;YAC5C,OAAO,CAAC,GAAG,CAAC,yCAAyC,OAAO,OAAO,IAAA,2BAAiB,EAAC,OAAO,CAAC,EAAE,CAAC,CAAA;YAEhG,IAAI;gBACF,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAA;gBACvD,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAA;gBAE9F,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE;oBAC5B,IAAI,MAAM,CAAC,GAAG,EAAE;wBACd,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,EAAE,OAAO,CAAC,CAAA;qBACrC;iBACF;gBACD,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC,MAAM,EAAE,OAAO,EAAE;oBAClD,QAAQ,EAAE,OAAO;oBACjB,SAAS,EAAE,SAAS;oBACpB,OAAO;iBACR,CAAC,CAAA;gBAEF,OAAO,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAA;aACpC;YAAC,OAAO,CAAC,EAAE;gBACV,OAAO,CAAC,KAAK,CACX,eAAK,CAAC,GAAG,CACP,iIAAiI,CAClI,CACF,CAAA;gBACD,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;gBAChB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;aAChB;SACF;KACF;IAED,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE;QAC5B,MAAM,OAAO,GAAG,kBAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAA;QAC7E,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,EAAE,OAAO,CAAC,CAAA;KACjE;AACH,CAAC;AA9ED,gDA8EC;AAaD,MAAa,cAAc;IACzB,OAAO,CAAsB;IAC7B,MAAM,CAAQ;IACd,MAAM,CAAiB;IAEvB,YAAY,MAAuB,EAAE,OAA6B,EAAE,MAAc;QAChF,wDAAwD;QACxD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAA;QACtB,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,OAAO,EAAE,CAAA;SACV;QACD,gEAAgE;QAEhE,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,OAAO,EAAE,CAAA;SACV;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,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE;gBAC9C,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,aAAa,CAAC,IAAI,CAAC,gBAAgB,OAAO,UAAU,eAAe,GAAG,CAAC,CAAA;gBAEvE,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;aAC/B;SACF;QAED,MAAM,MAAM,GAAG;;;;;uCAKoB,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,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;;QAGxE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;QAEhE,IAAI,CAAC,OAAO;aACX,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACT,OAAO,WAAW,CAAC,CAAC,GAAG,EAAE,IAAI,OAAO,CAAA;QACtC,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC;;KAEd,CAAA,CAAC,SAAS;QAEX,OAAO;YACL;gBACE,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,GAAG,KAAK;gBACtC,WAAW,EAAE,MAAM;aACpB;SACF,CAAA;IACH,CAAC;CACF;AAxFD,wCAwFC;AAED,SAAS,qBAAqB,CAAC,OAAqB;IAClD,QAAQ,OAAO,EAAE;QACf,KAAK,sBAAY,CAAC,QAAQ;YACxB,OAAO,UAAU,CAAA;KACpB;IACD,OAAO,UAAU,CAAA;AACnB,CAAC;AAED,SAAS,cAAc,CAAC,cAAkC,EAAE,OAAqB;IAC/E,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;IAElH,MAAM,YAAY,GAAG,eAAe,CAAC,MAAM,CAAC,CAAA;IAC5C,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAA;IAC/C,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;SAC7C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;SACvC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAA;IAC1B,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,EAAE,UAAU,CAAC,CAAC,CAAA;IACjF,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;oCACI,qBAAqB,CAAC,OAAO,CAAC;;;kEAGA,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,EAAE,MAAmB;IAClF,MAAM,aAAa,GAAG,4BAA4B,CAAC,MAAM,CAAC,CAAA;IAC1D,MAAM,gBAAgB,GAAG,4BAA4B,CAAC,MAAM,EAAE,IAAI,CAAC,CAAA;IAEnE,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,EAAE,MAAM,CAAC,OAAO,CAAC,EAAE,CAAA;IACrE,CAAC,CAAC,CAAA;IAEF,IAAI,YAAY,GAAG,EAAE,CAAA;IACrB,IAAI,MAAM,CAAC,GAAG,CAAC,IAAA,mBAAW,EAAC,MAAM,CAAC,GAAG,gBAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,EAAE;QAC5D,YAAY,GAAG;uBACI,MAAM,CAAC,IAAI;mCACC,MAAM,CAAC,IAAI,GAAG,gBAAgB;oBAC7C,MAAM,CAAC,IAAI,GAAG,gBAAgB;yBACzB,MAAM,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;KAE9E,CAAA;KACF;IAED,OAAO;iBACQ,MAAM,CAAC,IAAI,GAAG,aAAa;2BACjB,MAAM,CAAC,OAAO,KAAK,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI;MACnE,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,IAAI,GAAG,MAAM,CAAA;QACtB,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,EAAE,MAAM,GAAG,KAAK;IACtE,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,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,CAAA;QACnC,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,MAAM,MAAM,GAAG,IAAA,mCAA2B,EAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;QAC7D,OAAO,GAAG,IAAA,sBAAY,EAAC,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC,EAAE,CAAA;IACjD,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;uBACjB,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;GAE5E,CAAA;AACH,CAAC;AAED,SAAS,wBAAwB,CAAC,MAAkB,EAAE,IAAkB;IACtE,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;QAClB,OAAO,EAAE,CAAA;KACV;IAED,6DAA6D;IAE7D,MAAM,aAAa,GAAG,mBAAmB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;IAC9D,MAAM,MAAM,GAAG;WACN,aAAa,iBAAiB,MAAM,CAAC,IAAI,IAAI,aAAa,0EAA0E,MAAM,CAAC,IAAI;;;mBAGvI,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI;;;IAGxC,CAAA;IAEF,OAAO,MAAM,CAAA;AACf,CAAC;AAED,SAAS,eAAe,CAAC,MAAkB;IACzC,MAAM,KAAK,GAAG,IAAA,mBAAW,EAAC,MAAM,CAAC,CAAA;IACjC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAsB,CAAA;IAC/C,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAsB,CAAA;IAE9C,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,OAAO,EAAE;QACnC,SAAS,CAAC,GAAG,CAAC,KAAK,GAAG,gBAAQ,GAAG,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAA;KACtD;IAED,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,OAAO,EAAE;QACnC,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE;YACjC,MAAM,CAAC,GAAG,IAAA,qBAAa,EAAC,KAAK,CAAC,IAAI,CAAC,CAAA;YACnC,IAAI,CAAC,CAAC,KAAK,KAAK,yBAAyB,EAAE;gBACzC,MAAM,KAAK,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAA;gBACjC,MAAM,WAAW,GAAG,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;gBACxC,IAAI,WAAW,EAAE;oBACf,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,WAAW,CAAC,CAAA;iBACjC;aACF;SACF;KACF;IAED,OAAO,QAAQ,CAAA;AACjB,CAAC;AAED,SAAS,gBAAgB,CAAC,MAAkB,EAAE,MAAkB;IAC9D,mCAAmC;IACnC,kCAAkC;IAClC,cAAc;IACd,IAAI;IAEJ,6DAA6D;IAE7D,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, MoveModuleBytecode, MoveStruct } from 'aptos-sdk/src/generated'\nimport { AccountModulesImportInfo, AccountRegister, generateType } from './typegen'\nimport { getMeaningfulFunctionParams, isFrameworkAccount, moduleQname, SPLITTER } from '../aptos/utils'\nimport chalk from 'chalk'\nimport { AptosNetwork, getAptosChainName, getRpcClient } from '../aptos/network'\nimport { parseMoveType } from '../aptos/types'\n\nexport function codeGenAptosProcessor(abisDir: string, outDir = 'src/types/aptos') {\n  if (!fs.existsSync(abisDir)) {\n    return\n  }\n  console.log(chalk.green('Generated Types for Aptos'))\n  generate(abisDir, outDir)\n}\n\nexport async function generate(srcDir: string, outputDir: string) {\n  await generateForNetwork(srcDir, outputDir, AptosNetwork.MAIN_NET)\n  await generateForNetwork(path.join(srcDir, 'testnet'), path.join(outputDir, 'testnet'), AptosNetwork.TEST_NET)\n}\n\nexport async function generateForNetwork(srcDir: string, outputDir: string, network: AptosNetwork) {\n  if (!fs.existsSync(srcDir)) {\n    return\n  }\n  if (network === AptosNetwork.TEST_NET) {\n    console.log('Found testnet directory, generate code for testnet modules')\n  }\n  const files = fs.readdirSync(srcDir)\n  outputDir = path.resolve(outputDir)\n  const outputs: OutputFile[] = []\n\n  fs.mkdirSync(outputDir, { recursive: true })\n\n  const loader = new AccountRegister()\n\n  // when generating user code, don't need to generate framework account\n  loader.accountImports.set('0x1', new AccountModulesImportInfo('0x1', '0x1'))\n  loader.accountImports.set('0x2', new AccountModulesImportInfo('0x2', '0x2'))\n  loader.accountImports.set('0x3', new AccountModulesImportInfo('0x3', '0x3'))\n  const client = getRpcClient(network)\n\n  for (const file of files) {\n    if (!file.endsWith('.json')) {\n      continue\n    }\n    const fullPath = path.resolve(srcDir, file)\n    const modules = JSON.parse(fs.readFileSync(fullPath, 'utf-8'))\n\n    for (const module of modules) {\n      if (module.abi) {\n        loader.register(module.abi, path.basename(file, '.json'))\n      }\n    }\n    const codeGen = new AccountCodegen(loader, modules, {\n      fileName: path.basename(file, '.json'),\n      outputDir: outputDir,\n      network,\n    })\n\n    outputs.push(...codeGen.generate())\n  }\n\n  while (loader.pendingAccounts.size > 0) {\n    for (const account of loader.pendingAccounts) {\n      console.log(`download dependent module for account ${account} at ${getAptosChainName(network)}`)\n\n      try {\n        const modules = await client.getAccountModules(account)\n        fs.writeFileSync(path.resolve(srcDir, account + '.json'), JSON.stringify(modules, null, '\\t'))\n\n        for (const module of modules) {\n          if (module.abi) {\n            loader.register(module.abi, account)\n          }\n        }\n        const codeGen = new AccountCodegen(loader, modules, {\n          fileName: account,\n          outputDir: outputDir,\n          network,\n        })\n\n        outputs.push(...codeGen.generate())\n      } catch (e) {\n        console.error(\n          chalk.red(\n            'Error downloading account module, check if you choose the right network，or download account modules manually into your director'\n          )\n        )\n        console.error(e)\n        process.exit(1)\n      }\n    }\n  }\n\n  for (const output of outputs) {\n    const content = prettier.format(output.fileContent, { parser: 'typescript' })\n    fs.writeFileSync(path.join(outputDir, output.fileName), content)\n  }\n}\n\ninterface OutputFile {\n  fileName: string\n  fileContent: string\n}\n\ninterface Config {\n  fileName: string\n  outputDir: string\n  network: AptosNetwork\n}\n\nexport class AccountCodegen {\n  modules: MoveModuleBytecode[]\n  config: Config\n  loader: AccountRegister\n\n  constructor(loader: AccountRegister, modules: MoveModuleBytecode[], config: Config) {\n    // const json = fs.readFileSync(config.srcFile, 'utf-8')\n    this.modules = modules\n    this.config = config\n    this.loader = loader\n  }\n\n  generate(): OutputFile[] {\n    if (!this.modules) {\n      return []\n    }\n    // const baseName = path.basename(this.config.fileName, '.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-sdk/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] 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        moduleImports.push(`import * as _${account} from \"${tsAccountModule}\"`)\n\n        dependedAccounts.push(account)\n      }\n    }\n\n    const 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, this.config.network)).join('\\n')}\n    \n    export function loadAllTypes(_r: aptos.TypeRegistry) {\n      ${dependedAccounts.map((a) => `_${a}.loadAllTypes(_r)`).join('\\n')}\n\n      ${this.modules\n        .map((m) => {\n          return `_r.load(${m.abi?.name}.ABI)`\n        })\n        .join('\\n')}\n    }\n    ` // source\n\n    return [\n      {\n        fileName: this.config.fileName + '.ts',\n        fileContent: source,\n      },\n    ]\n  }\n}\n\nfunction generateNetworkOption(network: AptosNetwork) {\n  switch (network) {\n    case AptosNetwork.TEST_NET:\n      return 'TEST_NET'\n  }\n  return 'MAIN_NET'\n}\n\nfunction generateModule(moduleByteCode: MoveModuleBytecode, network: AptosNetwork) {\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\n  const eventStructs = getEventStructs(module)\n  const eventTypes = new Set(eventStructs.keys())\n  const events = Array.from(eventStructs.values())\n    .map((e) => generateOnEvents(module, e))\n    .filter((s) => s !== '')\n  const structs = module.structs.map((s) => generateStructs(module, s, eventTypes))\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.${generateNetworkOption(network)}       \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(_r: aptos.TypeRegistry) {\n      loadAllTypes(_r)\n    }\n    export const ABI: MoveModule = JSON.parse('${JSON.stringify(module)}')\n }\n  `\n}\n\nfunction generateStructs(module: MoveModule, struct: MoveStruct, events: Set<string>) {\n  const genericString = generateStructTypeParameters(struct)\n  const genericStringAny = generateStructTypeParameters(struct, true)\n\n  const fields = struct.fields.map((field) => {\n    return `${field.name}: ${generateType(field.type, module.address)}`\n  })\n\n  let eventPayload = ''\n  if (events.has(moduleQname(module) + SPLITTER + struct.name)) {\n    eventPayload = `\n    export interface ${struct.name}Instance extends \n        aptos.TypedEventInstance<${struct.name}${genericStringAny}> {\n      data_typed: ${struct.name}${genericStringAny}\n      type_arguments: [${struct.generic_type_params.map((_) => 'string').join(', ')}]\n    }\n    `\n  }\n\n  return `\n  export class ${struct.name}${genericString} {\n    static TYPE_QNAME = '${module.address}::${module.name}::${struct.name}'\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}=any`\n      })\n      .join(',')\n    genericString = `<${params}>`\n  }\n  return genericString\n}\n\nfunction generateStructTypeParameters(struct: MoveStruct, useAny = false) {\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 useAny ? 'any' : '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  const fields = getMeaningfulFunctionParams(func).map((param) => {\n    return `${generateType(param, module.address)}`\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    type_arguments: [${func.generic_type_params.map((_) => 'string').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}(func: (call: ${module.name}.${camelFuncName}Payload, ctx: aptos.AptosContext) => void, filter?: aptos.CallFilter): ${module.name} {\n    this.onEntryFunctionCall(func, {\n      ...filter,\n      function: '${module.name}::${func.name}'\n    })\n    return this\n  }`\n\n  return source\n}\n\nfunction getEventStructs(module: MoveModule) {\n  const qname = moduleQname(module)\n  const structMap = new Map<string, MoveStruct>()\n  const eventMap = new Map<string, MoveStruct>()\n\n  for (const struct of module.structs) {\n    structMap.set(qname + SPLITTER + struct.name, struct)\n  }\n\n  for (const struct of module.structs) {\n    for (const field of struct.fields) {\n      const t = parseMoveType(field.type)\n      if (t.qname === '0x1::event::EventHandle') {\n        const event = t.typeArgs[0].qname\n        const eventStruct = structMap.get(event)\n        if (eventStruct) {\n          eventMap.set(event, eventStruct)\n        }\n      }\n    }\n  }\n\n  return eventMap\n}\n\nfunction generateOnEvents(module: MoveModule, struct: MoveStruct): string {\n  // for struct that has drop + store\n  // if (!isEvent(struct, module)) {\n  //   return ''\n  // }\n\n  // const genericString = generateStructTypeParameters(struct)\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,17 @@
+import { MoveModule } from 'aptos-sdk/src/generated';
+import { TypeDescriptor } from '../aptos/types';
+export declare function generateType(type: string, currentAddress: string): 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,142 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AccountRegister = exports.AccountModulesImportInfo = exports.generateType = void 0;
+const types_1 = require("../aptos/types");
+const utils_1 = require("../aptos/utils");
+function generateTypeForDescriptor(type, currentAddress) {
+    // 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], currentAddress) + '[]';
+    }
+    const simpleName = generateSimpleType(type.qname, currentAddress);
+    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, currentAddress)).join(',') + '>';
+    }
+    return simpleName;
+}
+function generateSimpleType(type, currentAddress) {
+    const parts = type.split(utils_1.SPLITTER);
+    if (parts.length < 2) {
+        return parts[0];
+    }
+    if (parts[0] === currentAddress) {
+        return parts.slice(1).join('.');
+    }
+    return '_' + parts.join('.');
+}
+// TODO ctx need to have type parameters
+function generateType(type, currentAddress) {
+    return generateTypeForDescriptor((0, types_1.parseMoveType)(type), currentAddress);
+}
+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 = (0, types_1.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,0CAA8D;AAC9D,0CAAsF;AAEtF,SAAS,yBAAyB,CAAC,IAAoB,EAAE,cAAsB;IAC7E,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,EAAE,cAAc,CAAC,GAAG,IAAI,CAAA;KAC1E;IAED,MAAM,UAAU,GAAG,kBAAkB,CAAC,IAAI,CAAC,KAAK,EAAE,cAAc,CAAC,CAAA;IACjE,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,EAAE,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,CAAA;KACjH;IACD,OAAO,UAAU,CAAA;AACnB,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAY,EAAE,cAAsB;IAC9D,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;IACD,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,cAAc,EAAE;QAC/B,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;KAChC;IACD,OAAO,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;AAC9B,CAAC;AAED,wCAAwC;AACxC,SAAgB,YAAY,CAAC,IAAY,EAAE,cAAsB;IAC/D,OAAO,yBAAyB,CAAC,IAAA,qBAAa,EAAC,IAAI,CAAC,EAAE,cAAc,CAAC,CAAA;AACvE,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,IAAA,qBAAa,EAAC,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-sdk/src/generated'\nimport { parseMoveType, TypeDescriptor } from '../aptos/types'\nimport { moduleQname, moduleQnameForType, SPLITTER, VECTOR_STR } from '../aptos/utils'\n\nfunction generateTypeForDescriptor(type: TypeDescriptor, currentAddress: string): 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], currentAddress) + '[]'\n  }\n\n  const simpleName = generateSimpleType(type.qname, currentAddress)\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, currentAddress)).join(',') + '>'\n  }\n  return simpleName\n}\n\nfunction generateSimpleType(type: string, currentAddress: string): string {\n  const parts = type.split(SPLITTER)\n  if (parts.length < 2) {\n    return parts[0]\n  }\n  if (parts[0] === currentAddress) {\n    return parts.slice(1).join('.')\n  }\n  return '_' + parts.join('.')\n}\n\n// TODO ctx need to have type parameters\nexport function generateType(type: string, currentAddress: string): string {\n  return generateTypeForDescriptor(parseMoveType(type), currentAddress)\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"]}