oceanprotocol/market

import { isCID } from '@utils/ipfs'
import isUrl from 'is-url-superb'
import * as Yup from 'yup'
import { ethers } from 'ethers'
import { isGoogleUrl } from './url/index'

export function testLinks(isEdit?: boolean) {
  return Yup.string().test((value, context) => {
    const { type } = context.parent
    let validField
    let errorMessage

    switch (type) {
      // we allow submit if the type input is hidden as will be ignore
      case 'hidden':
        validField = true
        break
      case 'url':
      case 'graphql':
        validField = isUrl(value?.toString() || '')
        // if we're in publish, the field must be valid
        if (!validField) {
          validField = false
          errorMessage = 'Must be a valid url.'
        }
        // we allow submit on empty sample field
        if (
          !value?.toString() &&
          (context.path === 'links[0].url' ||
            context.path === 'services[0].links[0].url')
        ) {
          validField = true
        }
        // if the url has google drive, we need to block the user from submit
        if (isGoogleUrl(value?.toString())) {
          validField = false
          errorMessage =
            'Google Drive is not a supported hosting service. Please use an alternative.'
        }
        break
      case 'ipfs':
        validField = isCID(value?.toString())
        errorMessage = !value?.toString() ? 'CID required.' : 'CID not valid.'
        break
      case 'arweave':
        validField = value && !value?.toString().includes('http')
        errorMessage = !value?.toString()
          ? 'Transaction ID required.'
          : 'Transaction ID not valid.'
        break
      case 'smartcontract':
        validField = ethers.utils.isAddress(value?.toString())
        errorMessage = !value?.toString()
          ? 'Address required.'
          : 'Address not valid.'
        break
    }

    if (!validField) {
      return context.createError({
        message: errorMessage
      })
    }

    return true
  })
}