113 lines
3.3 KiB
TypeScript
113 lines
3.3 KiB
TypeScript
/* eslint-disable eslint-comments/disable-enable-pair */
|
|
|
|
import { useCallback, useEffect, useState } from 'react'
|
|
import { useWallet } from 'utils/wallet'
|
|
|
|
import type {
|
|
InstantiateResponse,
|
|
MigrateResponse,
|
|
RoyaltyRegistryContract,
|
|
RoyaltyRegistryInstance,
|
|
RoyaltyRegistryMessages,
|
|
} from './contract'
|
|
import { RoyaltyRegistry as initContract } from './contract'
|
|
|
|
export interface UseRoyaltyRegistryContractProps {
|
|
instantiate: (
|
|
codeId: number,
|
|
initMsg: Record<string, unknown>,
|
|
label: string,
|
|
admin?: string,
|
|
) => Promise<InstantiateResponse>
|
|
|
|
migrate: (contractAddress: string, codeId: number, migrateMsg: Record<string, unknown>) => Promise<MigrateResponse>
|
|
|
|
use: (customAddress?: string) => RoyaltyRegistryInstance | undefined
|
|
|
|
updateContractAddress: (contractAddress: string) => void
|
|
|
|
messages: (contractAddress: string) => RoyaltyRegistryMessages | undefined
|
|
}
|
|
|
|
export function useRoyaltyRegistryContract(): UseRoyaltyRegistryContractProps {
|
|
const wallet = useWallet()
|
|
|
|
const [address, setAddress] = useState<string>('')
|
|
const [royaltyRegistry, setRoyaltyRegistry] = useState<RoyaltyRegistryContract>()
|
|
|
|
useEffect(() => {
|
|
setAddress(localStorage.getItem('contract_address') || '')
|
|
}, [])
|
|
|
|
useEffect(() => {
|
|
if (!wallet.isWalletConnected) {
|
|
return
|
|
}
|
|
|
|
const load = async () => {
|
|
const client = await wallet.getSigningCosmWasmClient()
|
|
const royaltyRegistryContract = initContract(client, wallet.address || '')
|
|
setRoyaltyRegistry(royaltyRegistryContract)
|
|
}
|
|
|
|
load().catch(console.error)
|
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
|
}, [wallet.isWalletConnected, wallet.address])
|
|
|
|
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 (!royaltyRegistry) {
|
|
reject(new Error('Contract is not initialized.'))
|
|
return
|
|
}
|
|
royaltyRegistry.instantiate(codeId, initMsg, label, admin).then(resolve).catch(reject)
|
|
})
|
|
},
|
|
[royaltyRegistry],
|
|
)
|
|
|
|
const migrate = useCallback(
|
|
(contractAddress: string, codeId: number, migrateMsg: Record<string, unknown>): Promise<MigrateResponse> => {
|
|
return new Promise((resolve, reject) => {
|
|
if (!royaltyRegistry) {
|
|
reject(new Error('Contract is not initialized.'))
|
|
return
|
|
}
|
|
console.log(wallet.address, contractAddress, codeId)
|
|
royaltyRegistry
|
|
.migrate(wallet.address || '', contractAddress, codeId, migrateMsg)
|
|
.then(resolve)
|
|
.catch(reject)
|
|
})
|
|
},
|
|
[royaltyRegistry, wallet],
|
|
)
|
|
|
|
const use = useCallback(
|
|
(customAddress = ''): RoyaltyRegistryInstance | undefined => {
|
|
return royaltyRegistry?.use(address || customAddress)
|
|
},
|
|
[royaltyRegistry, address],
|
|
)
|
|
|
|
const messages = useCallback(
|
|
(customAddress = ''): RoyaltyRegistryMessages | undefined => {
|
|
return royaltyRegistry?.messages(address || customAddress)
|
|
},
|
|
[royaltyRegistry, address],
|
|
)
|
|
|
|
return {
|
|
instantiate,
|
|
migrate,
|
|
use,
|
|
updateContractAddress,
|
|
messages,
|
|
}
|
|
}
|