chore(sui): view function can only take pure types (#121)

sentioxyz · Nov 12, 2024 · f230826 · f230826
1 parent 4a1c573
commit f230826
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/packages/sui/src/codegen/codegen.ts b/packages/sui/src/codegen/codegen.ts
@@ -121,27 +121,33 @@ export class SuiCodegen extends AbstractCodegen<
     `
   }
 
-  private generateArgs(module: InternalMoveModule, func: InternalMoveFunction) {
+  private generateArgs(module: InternalMoveModule, func: InternalMoveFunction, isView: boolean) {
     const args = []
     const argsLen = func.params.length
     for (const [idx, arg] of func.params.entries()) {
       if (idx === argsLen - 1 && arg.qname === '0x2::tx_context::TxContext') {
         // no op
       } else if (arg.reference) {
         args.push({
-          paramType: `${this.ADDRESS_TYPE} | TransactionObjectArgument | TransactionArgument`,
+          paramType: isView
+            ? this.ADDRESS_TYPE
+            : `${this.ADDRESS_TYPE} | TransactionObjectArgument | TransactionArgument`,
           callValue: `_args.push(transactionArgumentOrObject(args[${idx}], tx))`
         })
       } else if (arg.isVector()) {
         // TODO fix pure vector
         args.push({
-          paramType: `(${this.ADDRESS_TYPE} | TransactionObjectArgument)[] | TransactionArgument`,
+          paramType: isView
+            ? `${this.ADDRESS_TYPE}[]`
+            : `(${this.ADDRESS_TYPE} | TransactionObjectArgument)[] | TransactionArgument`,
           callValue: `_args.push(transactionArgumentOrVec(args[${idx}], tx))`
         })
       } else {
         // Handle pure type
         let pureFunction = ''
-        const paramType = `${this.generateTypeForDescriptor(arg, module.address)} | TransactionArgument`
+        const paramType = isView
+          ? this.generateTypeForDescriptor(arg, module.address)
+          : `${this.generateTypeForDescriptor(arg, module.address)} | TransactionArgument`
 
         switch (arg.qname.toLowerCase()) {
           case 'u8':
@@ -200,7 +206,7 @@ export class SuiCodegen extends AbstractCodegen<
       })
       .join(',')
 
-    const args = this.generateArgs(module, func)
+    const args = this.generateArgs(module, func, true)
     const returnType = `${this.generateFunctionReturnTypeParameters(func, module.address)}`
 
     return `export async function ${camel(normalizeToJSName(func.name))}${genericString}(
@@ -229,7 +235,7 @@ export class SuiCodegen extends AbstractCodegen<
       return ''
     }
 
-    const args = this.generateArgs(module, func)
+    const args = this.generateArgs(module, func, false)
 
     const genericString = this.generateFunctionTypeParameters(func)