fix(contracts): fix useWrappedContract returns

paltalabs · Jul 30, 2024 · f05418a · f05418a
1 parent 05f4768
commit f05418a
Showing 1 changed file with 8 additions and 9 deletions.
diff --git a/packages/contracts/src/useRegisteredContract.tsx b/packages/contracts/src/useRegisteredContract.tsx
@@ -60,33 +60,34 @@ export type WrappedContract = {
  * @param deploymentInfo - The deployment information for the contract.
  * @returns The `WrappedContract` object.
  */
-export const useWrappedContract = (deploymentInfo: ContractDeploymentInfo) => {
+export const useWrappedContract = (
+  deploymentInfo: ContractDeploymentInfo | undefined
+) => {
   const sorobanContext = useSorobanReact()
   const [wrappedContract, setwWrappedContract] = useState<
     WrappedContract | undefined
-  >()
+  >(undefined)
 
   const wrappedInvokeFunction = useCallback(
     async (args: WrappedContractInvokeArgs) => {
       const contractInvokeArgs: InvokeArgs = {
         ...args,
         sorobanContext,
-        contractAddress: deploymentInfo.contractAddress,
+        contractAddress: deploymentInfo?.contractAddress!,
       }
       return contractInvoke(contractInvokeArgs)
     },
     [sorobanContext, deploymentInfo]
   )
 
-  const createWrappedContract = () => {
+  useEffect(() => {
+    if (!deploymentInfo) return
+
     const newWrappedContract: WrappedContract = {
       deploymentInfo,
       invoke: wrappedInvokeFunction,
     }
     setwWrappedContract(newWrappedContract)
-  }
-  useEffect(() => {
-    createWrappedContract()
   }, [deploymentInfo, wrappedInvokeFunction])
 
   return wrappedContract
@@ -109,7 +110,5 @@ export const useRegisteredContract = (contractId: string) => {
     networkPassphrase
   )
 
-  if (!deploymentInfo) return undefined
-
   return useWrappedContract(deploymentInfo)
 }