mango-token-sale/hooks/useIpAddress.tsx

61 lines
2.0 KiB
TypeScript

import { useEffect, useState } from 'react'
const SANCTIONED_COUNTRIES = [
{ country: 'Antigua and Barbuda', code: 'AG' },
{ country: 'Algeria', code: 'DZ' },
{ country: 'Bangladesh', code: 'BD' },
{ country: 'Bolivia', code: 'BO' },
{ country: 'Belarus', code: 'BY' },
{ country: 'Burundi', code: 'BI' },
{ country: 'Burma (Myanmar)', code: 'MM' },
{ country: "Cote D'Ivoire (Ivory Coast)", code: 'CI' },
{ country: 'Cuba', code: 'CU' },
{ country: 'Democratic Republic of Congo', code: 'CD' },
{ country: 'Ecuador', code: 'EC' },
{ country: 'Iran', code: 'IR' },
{ country: 'Iraq', code: 'IQ' },
{ country: 'Liberia', code: 'LR' },
{ country: 'Libya', code: 'LY' },
{ country: 'Mali', code: 'ML' },
{ country: 'Morocco', code: 'MA' },
{ country: 'Nepal', code: 'NP' },
{ country: 'North Korea', code: 'KP' },
{ country: 'Somalia', code: 'SO' },
{ country: 'Sudan', code: 'SD' },
{ country: 'Syria', code: 'SY' },
{ country: 'Venezuela', code: 'VE' },
{ country: 'Yemen', code: 'YE' },
{ country: 'Zimbabwe', code: 'ZW' },
{ country: 'United States', code: 'US' },
]
const SANCTIONED_COUNTRY_CODES = SANCTIONED_COUNTRIES.map(({ code }) => code)
export default function useIpAddress() {
const [ipAllowed, setIpAllowed] = useState(true)
useEffect(() => {
const checkIpLocation = async () => {
let ipCountryCode
try {
const response = await fetch(`https://www.cloudflare.com/cdn-cgi/trace`)
const parsedResponse = await response.text()
const ipLocation = parsedResponse.match(/loc=(.+)/)
ipCountryCode = ipLocation ? ipLocation[1] : ''
} catch {
const response = await fetch(`https://countrycode.bonfida.workers.dev/`)
const parsedResponse = await response.json()
ipCountryCode = parsedResponse.countryCode
}
if (ipCountryCode) {
setIpAllowed(!SANCTIONED_COUNTRY_CODES.includes(ipCountryCode))
}
}
checkIpLocation()
}, [])
return { ipAllowed }
}