refactor: capture generic type info only for boxes, global states, an…

…d local states as type info is not needed for others
algorandfoundation · Jan 2, 2025 · c29c885 · c29c885
1 parent 8a4cf8c
commit c29c885
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/test-transformer/visitors.ts b/src/test-transformer/visitors.ts
@@ -105,6 +105,7 @@ class ExpressionVisitor {
       if (type instanceof ptypes.FunctionPType) type = type.returnType
 
       const isGeneric = isGenericType(type)
+      const needsToCaptureTypeInfo = isStateOrBoxType(type)
       const isArc4Encoded = isArc4EncodedType(type)
       const info = isGeneric || isArc4Encoded ? getGenericTypeInfo(type) : undefined
       let updatedNode = node
@@ -129,7 +130,7 @@ class ExpressionVisitor {
         }
         updatedNode = stubbedFunctionName ? nodeFactory.callStubbedFunction(stubbedFunctionName, updatedNode, infoArg) : updatedNode
       }
-      return isGeneric
+      return needsToCaptureTypeInfo
         ? nodeFactory.captureGenericTypeInfo(ts.visitEachChild(updatedNode, this.visit, this.context), JSON.stringify(info))
         : ts.visitEachChild(updatedNode, this.visit, this.context)
     }
@@ -311,6 +312,9 @@ const isGenericType = (type: ptypes.PType): boolean =>
     ptypes.TuplePType,
   )
 
+const isStateOrBoxType = (type: ptypes.PType): boolean =>
+  instanceOfAny(type, ptypes.BoxMapPType, ptypes.BoxPType, ptypes.GlobalStateType, ptypes.LocalStateType)
+
 const isArc4EncodedType = (type: ptypes.PType): boolean =>
   instanceOfAny(
     type,