import { useEffect } from 'react'

import config from 'config'
import { useSharedTiDBContext } from 'dbaas/screens/NextGen/tidb-mgmt-container/shared/tidb-context'

import { CapacityEdit } from './Edit'
import { PremiumCapacityEdit } from './premium/Edit'

export const TiDBCapacityEdit = () => {
  const tidb = useSharedTiDBContext()

  useEffect(() => {
    document.title = `${config.PAGE_TITLE_PREFIX}Update Capacity`
  }, [])

  return tidb.computed.isPremium ? <PremiumCapacityEdit /> : <CapacityEdit />
}