78 lines
2.3 KiB
TypeScript
78 lines
2.3 KiB
TypeScript
import pgMeta from '@supabase/pg-meta'
|
|
import { ident } from '@supabase/pg-meta/src/pg-format'
|
|
import { useMutation, UseMutationOptions, useQueryClient } from '@tanstack/react-query'
|
|
import { toast } from 'sonner'
|
|
|
|
import { executeSql } from 'data/sql/execute-sql-query'
|
|
import { configKeys } from 'data/config/keys'
|
|
import type { ResponseError } from 'types'
|
|
import { databaseExtensionsKeys } from './keys'
|
|
|
|
export type DatabaseExtensionEnableVariables = {
|
|
projectRef: string
|
|
connectionString?: string | null
|
|
schema: string
|
|
name: string
|
|
version: string
|
|
cascade?: boolean
|
|
createSchema?: boolean
|
|
}
|
|
|
|
export async function enableDatabaseExtension({
|
|
projectRef,
|
|
connectionString,
|
|
schema,
|
|
name,
|
|
version,
|
|
cascade = false,
|
|
createSchema = false,
|
|
}: DatabaseExtensionEnableVariables) {
|
|
let headers = new Headers()
|
|
if (connectionString) headers.set('x-connection-encrypted', connectionString)
|
|
|
|
const { sql } = pgMeta.extensions.create({ schema, name, version, cascade })
|
|
const { result } = await executeSql({
|
|
projectRef,
|
|
connectionString,
|
|
sql: createSchema ? `create schema if not exists ${ident(schema)}; ${sql}` : sql,
|
|
queryKey: ['extension', 'create'],
|
|
})
|
|
|
|
return result
|
|
}
|
|
|
|
type DatabaseExtensionEnableData = Awaited<ReturnType<typeof enableDatabaseExtension>>
|
|
|
|
export const useDatabaseExtensionEnableMutation = ({
|
|
onSuccess,
|
|
onError,
|
|
...options
|
|
}: Omit<
|
|
UseMutationOptions<DatabaseExtensionEnableData, ResponseError, DatabaseExtensionEnableVariables>,
|
|
'mutationFn'
|
|
> = {}) => {
|
|
const queryClient = useQueryClient()
|
|
|
|
return useMutation<DatabaseExtensionEnableData, ResponseError, DatabaseExtensionEnableVariables>(
|
|
(vars) => enableDatabaseExtension(vars),
|
|
{
|
|
async onSuccess(data, variables, context) {
|
|
const { projectRef } = variables
|
|
await Promise.all([
|
|
queryClient.invalidateQueries(databaseExtensionsKeys.list(projectRef)),
|
|
queryClient.invalidateQueries(configKeys.upgradeEligibility(projectRef)),
|
|
])
|
|
await onSuccess?.(data, variables, context)
|
|
},
|
|
async onError(data, variables, context) {
|
|
if (onError === undefined) {
|
|
toast.error(`Failed to enable database extension: ${data.message}`)
|
|
} else {
|
|
onError(data, variables, context)
|
|
}
|
|
},
|
|
...options,
|
|
}
|
|
)
|
|
}
|