stargaze-studio/components/AirdropUpload.tsx
2023-10-11 16:48:20 -07:00

125 lines
4.5 KiB
TypeScript

import { toUtf8 } from '@cosmjs/encoding'
import clsx from 'clsx'
import React, { useState } from 'react'
import { toast } from 'react-hot-toast'
import { SG721_NAME_ADDRESS } from 'utils/constants'
import { csvToArray } from 'utils/csvToArray'
import type { AirdropAllocation } from 'utils/isValidAccountsFile'
import { isValidAccountsFile } from 'utils/isValidAccountsFile'
import { isValidAddress } from 'utils/isValidAddress'
import { useWallet } from 'utils/wallet'
interface AirdropUploadProps {
onChange: (data: AirdropAllocation[]) => void
}
export const AirdropUpload = ({ onChange }: AirdropUploadProps) => {
const wallet = useWallet()
const [resolvedAllocationData, setResolvedAllocationData] = useState<AirdropAllocation[]>([])
const resolveAllocationData = async (allocationData: AirdropAllocation[]) => {
if (!allocationData.length) return []
await new Promise((resolve) => {
let i = 0
allocationData.map(async (data) => {
if (!wallet.isWalletConnected) throw new Error('Wallet not connected')
await (
await wallet.getCosmWasmClient()
)
.queryContractRaw(
SG721_NAME_ADDRESS,
toUtf8(
Buffer.from(
`0006${Buffer.from('tokens').toString('hex')}${Buffer.from(
data.address.trim().substring(0, data.address.lastIndexOf('.stars')),
).toString('hex')}`,
'hex',
).toString(),
),
)
.then((res) => {
const tokenUri = JSON.parse(new TextDecoder().decode(res as Uint8Array)).token_uri
if (tokenUri && isValidAddress(tokenUri))
resolvedAllocationData.push({ address: tokenUri, amount: data.amount, tokenId: data.tokenId })
else toast.error(`Resolved address is empty or invalid for the name: ${data.address}`)
})
.catch((e) => {
console.log(e)
toast.error(`Error resolving address for the name: ${data.address}`)
})
i++
if (i === allocationData.length) resolve(resolvedAllocationData)
})
})
return resolvedAllocationData
}
const onFileChange = (event: React.ChangeEvent<HTMLInputElement>) => {
setResolvedAllocationData([])
if (!event.target.files) return toast.error('Error opening file')
// eslint-disable-next-line @typescript-eslint/no-unnecessary-condition
if (!event.target.files[0]?.name.endsWith('.csv')) {
toast.error('Please select a .csv file!')
return onChange([])
}
const reader = new FileReader()
reader.onload = async (e: ProgressEvent<FileReader>) => {
try {
if (!e.target?.result) return toast.error('Error parsing file.')
// eslint-disable-next-line @typescript-eslint/no-base-to-string
const accountsData = csvToArray(e.target.result.toString())
console.log(accountsData)
if (!isValidAccountsFile(accountsData)) {
event.target.value = ''
return onChange([])
}
await resolveAllocationData(accountsData.filter((data) => data.address.trim().endsWith('.stars'))).finally(
() => {
return onChange(
accountsData
.filter((data) => data.address.startsWith('stars') && !data.address.endsWith('.stars'))
.map((data) => ({
address: data.address.trim(),
amount: data.amount,
tokenId: data.tokenId,
}))
.concat(
resolvedAllocationData.map((data) => ({
address: data.address,
amount: data.amount,
tokenId: data.tokenId,
})),
),
)
},
)
} catch (error: any) {
toast.error(error.message, { style: { maxWidth: 'none' } })
}
}
reader.readAsText(event.target.files[0])
}
return (
<div
className={clsx(
'flex relative justify-center items-center mt-2 space-y-4 w-full h-32',
'rounded border-2 border-white/20 border-dashed',
)}
>
<input
accept=".csv"
className={clsx(
'file:py-2 file:px-4 file:mr-4 file:bg-plumbus-light file:rounded file:border-0 cursor-pointer',
'before:absolute before:inset-0 before:hover:bg-white/5 before:transition',
)}
id="airdrop-file"
multiple
onChange={onFileChange}
type="file"
/>
</div>
)
}