stargaze-studio/contracts/vendingFactory/useContract.ts

import type { logs } from '@cosmjs/stargate'
import { useWallet } from 'contexts/wallet'
import { useCallback, useEffect, useState } from 'react'

import type { VendingFactoryContract, VendingFactoryInstance, VendingFactoryMessages } from './contract'
import { vendingFactory as initContract } from './contract'

/*export interface InstantiateResponse {
  /** The address of the newly instantiated contract *-/
  readonly contractAddress: string
  readonly logs: readonly logs.Log[]
  /** Block height in which the transaction is included *-/
  readonly height: number
  /** Transaction hash (might be used as transaction ID). Guaranteed to be non-empty upper-case hex *-/
  readonly transactionHash: string
  readonly gasWanted: number
  readonly gasUsed: number
}*/

interface InstantiateResponse {
  readonly contractAddress: string
  readonly transactionHash: string
  readonly logs: readonly logs.Log[]
}

export interface UseVendingFactoryContractProps {
  use: (customAddress: string) => VendingFactoryInstance | undefined
  updateContractAddress: (contractAddress: string) => void
  getContractAddress: () => string | undefined
  messages: (contractAddress: string) => VendingFactoryMessages | undefined
}

export function useVendingFactoryContract(): UseVendingFactoryContractProps {
  const wallet = useWallet()

  const [address, setAddress] = useState<string>('')
  const [vendingFactory, setVendingFactory] = useState<VendingFactoryContract>()

  useEffect(() => {
    setAddress(localStorage.getItem('contract_address') || '')
  }, [])

  useEffect(() => {
    const VendingFactoryBaseContract = initContract(wallet.getClient(), wallet.address)
    setVendingFactory(VendingFactoryBaseContract)
  }, [wallet])

  const updateContractAddress = (contractAddress: string) => {
    setAddress(contractAddress)
  }

  const use = useCallback(
    (customAddress = ''): VendingFactoryInstance | undefined => {
      return vendingFactory?.use(address || customAddress)
    },
    [vendingFactory, address],
  )

  const getContractAddress = (): string | undefined => {
    return address
  }

  const messages = useCallback(
    (customAddress = ''): VendingFactoryMessages | undefined => {
      return vendingFactory?.messages(address || customAddress)
    },
    [vendingFactory, address],
  )

  return {
    use,
    updateContractAddress,
    getContractAddress,
    messages,
  }
}