115 lines
3.6 KiB
TypeScript
115 lines
3.6 KiB
TypeScript
import type { Coin } from '@cosmjs/proto-signing'
|
|
import type { logs } from '@cosmjs/stargate'
|
|
import { useWallet } from 'contexts/wallet'
|
|
import { useCallback, useEffect, useState } from 'react'
|
|
|
|
import type { BadgeHubContract, BadgeHubInstance, BadgeHubMessages, MigrateResponse } from './contract'
|
|
import { badgeHub 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 UseBadgeHubContractProps {
|
|
instantiate: (
|
|
codeId: number,
|
|
initMsg: Record<string, unknown>,
|
|
label: string,
|
|
admin?: string,
|
|
funds?: Coin[],
|
|
) => Promise<InstantiateResponse>
|
|
migrate: (contractAddress: string, codeId: number, migrateMsg: Record<string, unknown>) => Promise<MigrateResponse>
|
|
use: (customAddress: string) => BadgeHubInstance | undefined
|
|
updateContractAddress: (contractAddress: string) => void
|
|
getContractAddress: () => string | undefined
|
|
messages: (contractAddress: string) => BadgeHubMessages | undefined
|
|
}
|
|
|
|
export function useBadgeHubContract(): UseBadgeHubContractProps {
|
|
const wallet = useWallet()
|
|
|
|
const [address, setAddress] = useState<string>('')
|
|
const [badgeHub, setBadgeHub] = useState<BadgeHubContract>()
|
|
|
|
useEffect(() => {
|
|
setAddress(localStorage.getItem('contract_address') || '')
|
|
}, [])
|
|
|
|
useEffect(() => {
|
|
const BadgeHubBaseContract = initContract(wallet.getClient(), wallet.address)
|
|
setBadgeHub(BadgeHubBaseContract)
|
|
}, [wallet])
|
|
|
|
const updateContractAddress = (contractAddress: string) => {
|
|
setAddress(contractAddress)
|
|
}
|
|
|
|
const instantiate = useCallback(
|
|
(codeId: number, initMsg: Record<string, unknown>, label: string, admin?: string): Promise<InstantiateResponse> => {
|
|
return new Promise((resolve, reject) => {
|
|
if (!badgeHub) {
|
|
reject(new Error('Contract is not initialized.'))
|
|
return
|
|
}
|
|
badgeHub.instantiate(wallet.address, codeId, initMsg, label, admin).then(resolve).catch(reject)
|
|
})
|
|
},
|
|
[badgeHub, wallet],
|
|
)
|
|
|
|
const migrate = useCallback(
|
|
(contractAddress: string, codeId: number, migrateMsg: Record<string, unknown>): Promise<MigrateResponse> => {
|
|
return new Promise((resolve, reject) => {
|
|
if (!badgeHub) {
|
|
reject(new Error('Contract is not initialized.'))
|
|
return
|
|
}
|
|
console.log(wallet.address, contractAddress, codeId)
|
|
badgeHub.migrate(wallet.address, contractAddress, codeId, migrateMsg).then(resolve).catch(reject)
|
|
})
|
|
},
|
|
[badgeHub, wallet],
|
|
)
|
|
|
|
const use = useCallback(
|
|
(customAddress = ''): BadgeHubInstance | undefined => {
|
|
return badgeHub?.use(address || customAddress)
|
|
},
|
|
[badgeHub, address],
|
|
)
|
|
|
|
const getContractAddress = (): string | undefined => {
|
|
return address
|
|
}
|
|
|
|
const messages = useCallback(
|
|
(customAddress = ''): BadgeHubMessages | undefined => {
|
|
return badgeHub?.messages(address || customAddress)
|
|
},
|
|
[badgeHub, address],
|
|
)
|
|
|
|
return {
|
|
instantiate,
|
|
use,
|
|
updateContractAddress,
|
|
getContractAddress,
|
|
messages,
|
|
migrate,
|
|
}
|
|
}
|