fix(sui): don't include txcontext arg in builder functions (#81)

sentioxyz · Feb 7, 2024 · 2c2b65c · 2c2b65c
1 parent 45ea76b
commit 2c2b65c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/packages/move/src/types.ts b/packages/move/src/types.ts
@@ -155,6 +155,11 @@ export class TypeDescriptor<T = any> {
     const parts = this.qname.split(SPLITTER)
     return parts[parts.length - 1]
   }
+
+  module(): string {
+    const parts = this.qname.split(SPLITTER)
+    return parts[parts.length - 2]
+  }
 }
 
 export function parseMoveType(type: string): TypeDescriptor {

diff --git a/packages/sui/src/codegen/codegen.ts b/packages/sui/src/codegen/codegen.ts
@@ -121,8 +121,11 @@ export class SuiCodegen extends AbstractCodegen<
 
   private generateArgs(module: InternalMoveModule, func: InternalMoveFunction) {
     const args = []
+    const argsLen = func.params.length
     for (const [idx, arg] of func.params.entries()) {
-      if (arg.reference) {
+      if (idx === argsLen - 1 && arg.name() == 'TxContext' && arg.module() == 'tx_context') {
+        continue
+      } else if (arg.reference) {
         args.push({
           paramType: `${this.ADDRESS_TYPE} | ObjectCallArg | TransactionArgument`,
           callValue: `_args.push(transactionArgumentOrObject(args[${idx}], tx))`
@@ -197,7 +200,7 @@ export class SuiCodegen extends AbstractCodegen<
     return `export function ${camel(normalizeToJSName(func.name))}${genericString}(tx: TransactionBlock, 
       args: [${args.map((a) => a.paramType).join(',')}],
       ${typeParamArg.length > 0 ? `typeArguments: [${typeParamArg}]` : ``} ):
-       TransactionArgument & [ ${'TransactionArgument,'.repeat(func.params.length)} ] {
+       TransactionArgument & [ ${'TransactionArgument,'.repeat(args.length)} ] {
       const _args: any[] = []
       ${args.map((a) => a.callValue).join('\n')}