Files
supabase/apps/studio/components/interfaces/Settings/Database/DatabaseSettings/DatabaseConnectionString.tsx
Jonathan Summers-Muir f650b5ae67 Feat/connection string revamp (#30572)
* add new page

* moar

* moar

* added icons

* improve icons

* moved connect dialog to main header

* update text

* smaller screen support

* moar

* add python and sqlalchemyString

* moar

* add IPv4 warning

* moar

* Delete pooler-icons-v2.tsx

* tidy

* Delete DatabaseSettings.tsx

* tidy

* tidy

* moar. Session pooler is de-prioritized

* Update DatabaseConnectionString.tsx

* type issue

* moar

* Update DatabaseConnectionString.tsx

* Spelling

* Clean up LayoutHeader

* Clean up ConnectionPanel

* Clean up ConnectionParameters

* Last batch of clean up

* Fix loading state padding

* Shift old Connect files to new Connect folder outside of Home

* Final clean up

* Smol fix

* FIX

* Fix button link

* Fixes

* Lint

---------

Co-authored-by: Terry Sutton <saltcod@gmail.com>
Co-authored-by: Joshen Lim <joshenlimek@gmail.com>
2024-12-05 17:23:14 +08:00

331 lines
13 KiB
TypeScript

import { ChevronDown } from 'lucide-react'
import { useEffect, useRef, useState } from 'react'
import { useParams } from 'common'
import { getAddons } from 'components/interfaces/Billing/Subscription/Subscription.utils'
import { useProjectContext } from 'components/layouts/ProjectLayout/ProjectContext'
import AlertError from 'components/ui/AlertError'
import DatabaseSelector from 'components/ui/DatabaseSelector'
import { DocsButton } from 'components/ui/DocsButton'
import Panel from 'components/ui/Panel'
import ShimmeringLoader from 'components/ui/ShimmeringLoader'
import { usePoolingConfigurationQuery } from 'data/database/pooling-configuration-query'
import { useReadReplicasQuery } from 'data/read-replicas/replicas-query'
import { useProjectAddonsQuery } from 'data/subscriptions/project-addons-query'
import { useSendEventMutation } from 'data/telemetry/send-event-mutation'
import { useSelectedProject } from 'hooks/misc/useSelectedProject'
import { pluckObjectFields } from 'lib/helpers'
import { useDatabaseSelectorStateSnapshot } from 'state/database-selector'
import { useDatabaseSettingsStateSnapshot } from 'state/database-settings'
import {
CollapsibleContent_Shadcn_,
CollapsibleTrigger_Shadcn_,
Collapsible_Shadcn_,
Input,
Separator,
Tabs,
TooltipContent_Shadcn_,
TooltipTrigger_Shadcn_,
Tooltip_Shadcn_,
cn,
} from 'ui'
import {
DefaultSessionModeNotice,
IPv4AddonDirectConnectionNotice,
IPv4DeprecationNotice,
} from '../DatabaseConnectionNotices'
import { UsePoolerCheckbox } from '../UsePoolerCheckbox'
import {
constructConnStringSyntax,
getConnectionStrings,
getPoolerTld,
} from './DatabaseSettings.utils'
const CONNECTION_TYPES = [
{ id: 'uri', label: 'URI' },
{ id: 'psql', label: 'PSQL' },
{ id: 'golang', label: 'Golang' },
{ id: 'jdbc', label: 'JDBC' },
{ id: 'dotnet', label: '.NET' },
{ id: 'nodejs', label: 'Node.js' },
{ id: 'php', label: 'PHP' },
{ id: 'python', label: 'Python' },
]
interface DatabaseConnectionStringProps {
appearance: 'default' | 'minimal'
}
/**
* @deprecated Will be removed once `connectDialogUpdate` flag is persisted
*/
export const DatabaseConnectionString = ({ appearance }: DatabaseConnectionStringProps) => {
const project = useSelectedProject()
const { ref: projectRef, connectionString } = useParams()
const snap = useDatabaseSettingsStateSnapshot()
const state = useDatabaseSelectorStateSnapshot()
const { project: projectDetails, isLoading: isProjectLoading } = useProjectContext()
const connectionStringsRef = useRef<HTMLDivElement>(null)
const [poolingMode, setPoolingMode] = useState<'transaction' | 'session'>('transaction')
const [selectedTab, setSelectedTab] = useState<
'uri' | 'psql' | 'golang' | 'jdbc' | 'dotnet' | 'nodejs' | 'php' | 'python'
>('uri')
const { data: poolingInfo, isSuccess: isSuccessPoolingInfo } = usePoolingConfigurationQuery({
projectRef,
})
const poolingConfiguration = poolingInfo?.find((x) => x.identifier === state.selectedDatabaseId)
const defaultPoolingMode = poolingConfiguration?.pool_mode
const {
data: databases,
error: readReplicasError,
isLoading: isLoadingReadReplicas,
isError: isErrorReadReplicas,
isSuccess: isSuccessReadReplicas,
} = useReadReplicasQuery({ projectRef })
const selectedDatabase = (databases ?? []).find(
(db) => db.identifier === state.selectedDatabaseId
)
const { data: addons, isSuccess: isSuccessAddons } = useProjectAddonsQuery({ projectRef })
const { ipv4: ipv4Addon } = getAddons(addons?.selected_addons ?? [])
const { mutate: sendEvent } = useSendEventMutation()
const DB_FIELDS = ['db_host', 'db_name', 'db_port', 'db_user', 'inserted_at']
const emptyState = { db_user: '', db_host: '', db_port: '', db_name: '' }
const connectionInfo = pluckObjectFields(selectedDatabase || emptyState, DB_FIELDS)
const connectionTld =
projectDetails?.restUrl !== undefined
? new URL(projectDetails?.restUrl ?? '').hostname.split('.').pop() ?? 'co'
: 'co'
const handleCopy = (id: string) => {
const labelValue = CONNECTION_TYPES.find((type) => type.id === id)?.label
sendEvent({
category: 'settings',
action: 'copy_connection_string',
label: labelValue ? labelValue : '',
})
}
const connectionStrings =
isSuccessPoolingInfo && poolingConfiguration !== undefined
? getConnectionStrings(connectionInfo, poolingConfiguration, {
projectRef,
usePoolerConnection: snap.usePoolerConnection,
})
: { uri: '', psql: '', golang: '', jdbc: '', dotnet: '', nodejs: '', php: '', python: '' }
const poolerTld =
isSuccessPoolingInfo && poolingConfiguration !== undefined
? getPoolerTld(poolingConfiguration?.connectionString)
: 'com'
const poolerConnStringSyntax =
isSuccessPoolingInfo && poolingConfiguration !== undefined
? constructConnStringSyntax(poolingConfiguration?.connectionString, {
selectedTab,
usePoolerConnection: snap.usePoolerConnection,
ref: projectRef as string,
cloudProvider: isProjectLoading ? '' : project?.cloud_provider || '',
region: isProjectLoading ? '' : project?.region || '',
tld: snap.usePoolerConnection ? poolerTld : connectionTld,
portNumber: snap.usePoolerConnection
? poolingMode === 'transaction'
? poolingConfiguration?.db_port.toString()
: '5432'
: connectionInfo.db_port.toString(),
})
: []
useEffect(() => {
if (connectionString !== undefined && connectionStringsRef.current !== undefined) {
state.setSelectedDatabaseId(connectionString)
connectionStringsRef.current?.scrollIntoView({ block: 'center', behavior: 'smooth' })
}
}, [connectionString])
useEffect(() => {
if (poolingConfiguration?.pool_mode === 'session') {
setPoolingMode(poolingConfiguration.pool_mode)
}
}, [poolingConfiguration?.pool_mode])
return (
<div id="connection-string" className="w-full">
<Panel
className={cn(
'!m-0 [&>div:nth-child(1)]:!border-0 [&>div:nth-child(1)]:!p-0',
appearance === 'minimal' && 'border-0 shadow-none bg-transparent'
)}
titleClasses={cn(appearance === 'minimal' && 'bg-transparent')}
title={
<div
ref={connectionStringsRef}
className={cn('w-full flex flex-col', appearance === 'default' ? 'pt-4' : 'pt-0')}
>
<div
className={cn(
'flex items-center justify-between mb-2',
appearance === 'default' && 'px-6'
)}
>
{appearance === 'default' && (
<h5 key="panel-title" className="mb-0">
Connection string
</h5>
)}
<div
className={cn(
'flex items-center gap-x-2 ml-auto',
appearance === 'minimal' ? 'absolute -top-1 right-0' : ''
)}
>
<DatabaseSelector />
<DocsButton href="https://supabase.com/docs/guides/database/connecting-to-postgres" />
</div>
</div>
<Tabs
type="underlined"
size="tiny"
activeId={selectedTab}
baseClassNames={cn(
'!space-y-0 px-6 -mb-[1px]',
appearance === 'minimal' && '-mt-1 px-0'
)}
onChange={setSelectedTab}
>
{CONNECTION_TYPES.map((type) => (
<Tabs.Panel key={type.id} id={type.id} label={type.label} />
))}
</Tabs>
<Separator />
</div>
}
>
<Panel.Content className={appearance === 'minimal' && 'px-0'}>
{isLoadingReadReplicas && <ShimmeringLoader className="h-8 w-full" />}
{isErrorReadReplicas && (
<AlertError error={readReplicasError} subject="Failed to retrieve database settings" />
)}
{isSuccessReadReplicas && (
<div className="flex flex-col gap-y-4 pt-2">
<UsePoolerCheckbox
id="connection-string"
checked={snap.usePoolerConnection}
poolingMode={poolingMode}
ipv4AddonAdded={!!ipv4Addon}
onCheckedChange={snap.setUsePoolerConnection}
onSelectPoolingMode={setPoolingMode}
/>
{defaultPoolingMode === 'session' && poolingMode === 'transaction' && (
<DefaultSessionModeNotice />
)}
{isSuccessAddons &&
poolingMode === 'session' &&
ipv4Addon !== undefined &&
snap.usePoolerConnection && <IPv4AddonDirectConnectionNotice />}
{ipv4Addon === undefined && !snap.usePoolerConnection && <IPv4DeprecationNotice />}
<Input
copy
readOnly
disabled
className="input-mono [&>div>div>div>input]:text-xs [&>div>div>div>input]:opacity-100"
value={
poolingMode === 'transaction'
? connectionStrings[selectedTab]
: connectionStrings[selectedTab].replace('6543', '5432')
}
onCopy={() => handleCopy(selectedTab)}
/>
</div>
)}
</Panel.Content>
{poolerConnStringSyntax.length > 0 && (
<>
<Separator />
<Panel.Content className={cn('!py-3 space-y-2', appearance === 'minimal' && 'px-0')}>
<Collapsible_Shadcn_>
<CollapsibleTrigger_Shadcn_ className="group [&[data-state=open]>div>svg]:!-rotate-180">
<div className="flex items-center gap-x-2 w-full">
<p className="text-xs text-foreground-light group-hover:text-foreground transition">
How to connect to a different database or switch to another user
</p>
<ChevronDown
className="transition-transform duration-200"
strokeWidth={1.5}
size={14}
/>
</div>
</CollapsibleTrigger_Shadcn_>
<CollapsibleContent_Shadcn_ className="my-2">
<div className="text-foreground-light">
<p className="text-xs">
You can use the following URI format to switch to a different database or user
{snap.usePoolerConnection ? ' when using connection pooling' : ''}.
</p>
<p className="text-sm tracking-tight text-foreground-lighter">
{poolerConnStringSyntax.map((x, idx) => {
if (x.tooltip) {
return (
<Tooltip_Shadcn_ key={`syntax-${idx}`}>
<TooltipTrigger_Shadcn_ asChild>
<span className="text-foreground text-xs font-mono">{x.value}</span>
</TooltipTrigger_Shadcn_>
<TooltipContent_Shadcn_ side="bottom">
{x.tooltip}
</TooltipContent_Shadcn_>
</Tooltip_Shadcn_>
)
} else {
return (
<span key={`syntax-${idx}`} className="text-xs font-mono">
{x.value}
</span>
)
}
})}
</p>
</div>
</CollapsibleContent_Shadcn_>
</Collapsible_Shadcn_>
{selectedTab === 'python' && (
<Collapsible_Shadcn_>
<CollapsibleTrigger_Shadcn_ className="group [&[data-state=open]>div>svg]:!-rotate-180">
<div className="flex items-center gap-x-2 w-full">
<p className="text-xs text-foreground-light group-hover:text-foreground transition">
Connecting to SQL Alchemy
</p>
<ChevronDown
className="transition-transform duration-200"
strokeWidth={1.5}
size={14}
/>
</div>
</CollapsibleTrigger_Shadcn_>
<CollapsibleContent_Shadcn_ className="my-2">
<div className="text-foreground-light text-xs grid gap-2">
<p>
Please use <code>postgresql://</code> instead of <code>postgres://</code> as
your dialect when connecting via SQLAlchemy.
</p>
<p>
Example:
<code>create_engine("postgresql+psycopg2://...")</code>
</p>
<p className="text-sm font-mono tracking-tight text-foreground-lighter"></p>
</div>
</CollapsibleContent_Shadcn_>
</Collapsible_Shadcn_>
)}
</Panel.Content>
</>
)}
</Panel>
</div>
)
}