supabase/studio/components/interfaces/Settings/Database/NetworkRestrictions/AddRestrictionModal.tsx

import { FC, useState } from 'react'
import { Address4 } from 'ip-address'
import { Button, Form, IconHelpCircle, Input, Modal } from 'ui'
import * as Tooltip from '@radix-ui/react-tooltip'

import { useStore, useParams } from 'hooks'
import InformationBox from 'components/ui/InformationBox'
import { checkIfPrivate, getAddressEndRange } from './NetworkRestrictions.utils'
import { useNetworkRestrictionsApplyMutation } from 'data/network-restrictions/network-retrictions-apply-mutation'

interface Props {
  restrictedIps: string[]
  visible: boolean
  hasOverachingRestriction: boolean
  onClose: () => void
}

const AddRestrictionModal: FC<Props> = ({
  restrictedIps,
  visible,
  hasOverachingRestriction,
  onClose,
}) => {
  const formId = 'add-restriction-form'
  const { ui } = useStore()
  const { ref } = useParams()
  const { mutateAsync: applyNetworkRestrictions } = useNetworkRestrictionsApplyMutation()

  const validate = (values: any) => {
    const errors: any = {}
    const { ipAddress, cidrBlockSize } = values

    // Validate CIDR block size
    const isOutOfCidrSizeRange = cidrBlockSize < 0 || cidrBlockSize > 32
    if (isOutOfCidrSizeRange) errors.cidrBlockSize = 'Size has to be between 0 to 32'

    // Validate IP address
    const isValid = Address4.isValid(ipAddress)
    if (!isValid) {
      errors.ipAddress = 'Please enter a valid IP address'
      return errors
    }

    const isPrivate = checkIfPrivate(ipAddress)
    if (isPrivate) errors.ipAddress = 'Private IP addresses are not supported'

    return errors
  }

  const onSubmit = async (values: any, { setSubmitting }: any) => {
    if (!ref) return console.error('Project ref is required')

    setSubmitting(true)
    const cidr = `${values.ipAddress}/${values.cidrBlockSize}`
    const alreadyExists = restrictedIps.includes(cidr)
    if (alreadyExists) {
      return ui.setNotification({
        category: 'info',
        message: `The address ${cidr} is already restricted`,
      })
    }

    const dbAllowedCidrs = hasOverachingRestriction ? [cidr] : [...restrictedIps, cidr]

    try {
      await applyNetworkRestrictions({ projectRef: ref, dbAllowedCidrs })
      onClose()
    } catch (error: any) {
      ui.setNotification({
        category: 'error',
        message: `Failed to add restriction: ${error.message}`,
      })
    } finally {
      setSubmitting(false)
    }
  }

  return (
    <Modal
      closable
      hideFooter
      size="medium"
      visible={visible}
      onCancel={onClose}
      header="Add a new restriction"
    >
      <Form
        validateOnBlur
        id={formId}
        className="!border-t-0"
        initialValues={{ ipAddress: '', cidrBlockSize: undefined }}
        validate={validate}
        onSubmit={onSubmit}
      >
        {({ isSubmitting, values, resetForm }: any) => {
          const isPrivate = Address4.isValid(values.ipAddress)
            ? checkIfPrivate(values.ipAddress)
            : false
          const isValidBlockSize =
            values.cidrBlockSize !== '' && values.cidrBlockSize >= 0 && values.cidrBlockSize <= 32
          const availableAddresses = Math.pow(2, 32 - (values?.cidrBlockSize ?? 0))
          const addressRange = getAddressEndRange(`${values.ipAddress}/${values.cidrBlockSize}`)

          const isValidCIDR = isValidBlockSize && !isPrivate && addressRange !== undefined

          const [isFetchingAddress, setIsFetchingAddress] = useState(false)

          const getClientIpAddress = async () => {
            setIsFetchingAddress(true)
            const res = await fetch('http://www.geoplugin.net/json.gp', { method: 'GET' })
            const { geoplugin_request } = await res.json()

            if (geoplugin_request) {
              const updatedValues = { ...values, ipAddress: geoplugin_request, cidrBlockSize: 32 }
              resetForm({ initialValues: updatedValues, values: updatedValues })
            } else {
              ui.setNotification({
                category: 'error',
                duration: 4000,
                message: 'Failed to retrieve client IP address, please enter address manually',
              })
            }
            setIsFetchingAddress(false)
          }

          return (
            <>
              <Modal.Content>
                <div className="py-6 space-y-4">
                  <p className="text-sm text-scale-1100">
                    This will add an IP address range to a list of allowed ranges that can access
                    your database. Only IPv4 addresses are supported at the moment.
                  </p>
                  <InformationBox
                    title="Note: Restrictions only apply to your database and PgBouncer"
                    description="They do not currently apply to Supabase services such as PostgREST, Storage, or Authentication"
                  />
                  <div className="flex space-x-4">
                    <div className="w-[55%]">
                      <Input
                        label="IPv4 address"
                        id="ipAddress"
                        name="ipAddress"
                        placeholder="0.0.0.0"
                        className=""
                      />
                    </div>
                    <div className="flex-grow">
                      <Input
                        label={
                          <div className="flex items-center space-x-2">
                            <p>CIDR Block Size</p>
                            <Tooltip.Root delayDuration={0}>
                              <Tooltip.Trigger>
                                <IconHelpCircle size="tiny" strokeWidth={2} />
                              </Tooltip.Trigger>
                              <Tooltip.Content side="bottom">
                                <Tooltip.Arrow className="radix-tooltip-arrow" />
                                <div
                                  className={[
                                    'rounded bg-scale-100 py-1 px-2 leading-none shadow',
                                    'border border-scale-200 w-[300px]',
                                  ].join(' ')}
                                >
                                  <span className="text-xs text-scale-1200">
                                    Classless inter-domain routing (CIDR) notation is the notation
                                    used to identify networks and hosts in the networks. The block
                                    size tells us how many bits we need to take for the network
                                    prefix, and is a value between 0 to 32.
                                  </span>
                                </div>
                              </Tooltip.Content>
                            </Tooltip.Root>
                          </div>
                        }
                        id="cidrBlockSize"
                        name="cidrBlockSize"
                        type="number"
                        placeholder="0"
                        min={0}
                        max={32}
                      />
                    </div>
                  </div>
                  <Button
                    type="default"
                    loading={isFetchingAddress}
                    disabled={isFetchingAddress}
                    onClick={() => getClientIpAddress()}
                  >
                    Use my IP address
                  </Button>
                </div>
              </Modal.Content>
              <Modal.Separator />
              {isValidCIDR ? (
                <Modal.Content>
                  <div className="space-y-1 pt-2 pb-4">
                    <p className="text-sm">
                      The address range{' '}
                      <code className="text-xs">
                        {values.ipAddress}/{values.cidrBlockSize}
                      </code>{' '}
                      will be restricted
                    </p>
                    <p className="text-sm text-scale-1000">
                      Selected address space: <code className="text-xs">{addressRange.start}</code>{' '}
                      to <code className="text-xs">{addressRange.end}</code>{' '}
                    </p>
                    <p className="text-sm text-scale-1000">
                      Number of addresses: {availableAddresses}
                    </p>
                  </div>
                </Modal.Content>
              ) : (
                <Modal.Content>
                  <div className="pt-2 pb-4">
                    <div className="h-[68px] flex items-center">
                      <p className="text-sm text-scale-1000">
                        A summary of your restriction will be shown here after entering a valid IP
                        address and CIDR block size
                      </p>
                    </div>
                  </div>
                </Modal.Content>
              )}
              <div className="flex items-center justify-end px-6 py-4 border-t space-x-2">
                <Button type="default" disabled={isSubmitting} onClick={() => onClose()}>
                  Cancel
                </Button>
                <Button htmlType="submit" loading={isSubmitting} disabled={isSubmitting}>
                  Save restriction
                </Button>
              </div>
            </>
          )
        }}
      </Form>
    </Modal>
  )
}

export default AddRestrictionModal