Briven/apps/studio/components/interfaces/Integrations/Integration/IntegrationOverviewTabV2/InstallIntegrationSheet/AdvancedSettings.tsx

import { type Dispatch, type SetStateAction } from 'react'
import {
  Accordion,
  AccordionContent,
  AccordionItem,
  AccordionTrigger,
  Badge,
  Select,
  SelectContent,
  SelectItem,
  SelectSeparator,
  SelectTrigger,
  SelectValue,
  SheetSection,
} from 'ui'
import { Input } from 'ui-patterns/DataInputs/Input'
import { FormItemLayout } from 'ui-patterns/form/FormItemLayout/FormItemLayout'

import { getExtensionDefaultSchema } from '../IntegrationOverviewTabV2.utils'
import { type ExtensionsSchema, type InstallIntegrationSheetProps } from './InstallIntegrationSheet'
import { extensionsWithRecommendedSchemas } from '@/components/interfaces/Database/Extensions/Extensions.constants'
import { useDatabaseExtensionsQuery } from '@/data/database-extensions/database-extensions-query'
import { useSchemasQuery } from '@/data/database/schemas-query'
import { useSelectedProjectQuery } from '@/hooks/misc/useSelectedProject'
import { useProtectedSchemas } from '@/hooks/useProtectedSchemas'

type AdvancedSettingsProps = InstallIntegrationSheetProps & {
  extensionsSchema: ExtensionsSchema
  setExtensionsSchema: Dispatch<SetStateAction<ExtensionsSchema>>
}

export const AdvancedSettings = ({
  integration,
  extensionsSchema,
  setExtensionsSchema,
}: AdvancedSettingsProps) => {
  const { data: project } = useSelectedProjectQuery()
  const { data: protectedSchemas } = useProtectedSchemas({ excludeSchemas: ['extensions'] })
  const { requiredExtensions: requiredExtensionNames } = integration

  const involvesExtensions = requiredExtensionNames.length > 0
  const { data: extensions = [] } = useDatabaseExtensionsQuery(
    { projectRef: project?.ref, connectionString: project?.connectionString },
    { enabled: involvesExtensions }
  )

  const { data: schemas = [] } = useSchemasQuery(
    { projectRef: project?.ref, connectionString: project?.connectionString },
    { enabled: involvesExtensions }
  )
  const availableSchemas = schemas.filter(
    (schema) => !protectedSchemas.some((protectedSchema) => protectedSchema.name === schema.name)
  )

  return (
    <SheetSection>
      <Accordion type="single" collapsible>
        <AccordionItem value="advanced-settings" className="border-none">
          <AccordionTrigger className="font-normal gap-2 py-0 justify-between text-sm hover:no-underline">
            Advanced settings
          </AccordionTrigger>
          <AccordionContent className="pb-0! pt-3 [&>div]:flex [&>div]:flex-col [&>div]:gap-y-4">
            <p className="text-foreground-light">
              Select which schemas to install the database extensions under
            </p>
            {requiredExtensionNames.map((extName) => {
              const ext = extensions.find((x) => x.name === extName)
              const extMeta = extensionsSchema[extName]
              const { schema, value } = extMeta
              const recommendedSchema = extensionsWithRecommendedSchemas[extName]
              const defaultSchema = getExtensionDefaultSchema(ext)

              return (
                <FormItemLayout
                  key={extName}
                  isReactForm={false}
                  layout="horizontal"
                  label={extName}
                  description={
                    ext?.installed_version ? (
                      <>
                        Installed in <code className="text-code-inline">{ext.schema}</code> schema
                      </>
                    ) : defaultSchema ? (
                      <>
                        Must be installed in the{' '}
                        <code className="text-code-inline">{defaultSchema}</code> schema
                      </>
                    ) : recommendedSchema ? (
                      <>
                        Use the <code className="text-code-inline">{recommendedSchema}</code> schema
                        for full compatibility with related features
                      </>
                    ) : undefined
                  }
                >
                  <Select
                    disabled={!!defaultSchema || !!ext?.installed_version}
                    value={schema}
                    onValueChange={(schema) =>
                      setExtensionsSchema((prev) => ({
                        ...prev,
                        [extName]: {
                          schema,
                          value: schema === 'custom' ? extName : undefined,
                        },
                      }))
                    }
                  >
                    <SelectTrigger>
                      <SelectValue placeholder="Select a schema" />
                    </SelectTrigger>
                    <SelectContent>
                      <SelectItem value="custom">Create a new schema</SelectItem>
                      <SelectSeparator />
                      {availableSchemas.map((schema) => {
                        return (
                          <SelectItem key={schema.id} value={schema.name}>
                            {schema.name}
                            {schema.name === recommendedSchema ? (
                              <Badge className="ml-2" variant="success">
                                Recommended
                              </Badge>
                            ) : schema.name === 'extensions' ? (
                              <Badge className="ml-2">Default</Badge>
                            ) : null}
                          </SelectItem>
                        )
                      })}
                      {defaultSchema &&
                      !availableSchemas.some((schema) => schema.name === defaultSchema) ? (
                        <SelectItem key={defaultSchema} value={defaultSchema}>
                          {defaultSchema}
                          <Badge className="ml-2">Default</Badge>
                        </SelectItem>
                      ) : null}
                    </SelectContent>
                  </Select>

                  {schema === 'custom' && (
                    <FormItemLayout
                      isReactForm={false}
                      className="mt-2"
                      label="Provide a name for your new schema"
                    >
                      <Input
                        value={value}
                        onChange={(e) =>
                          setExtensionsSchema((prev) => ({
                            ...prev,
                            [extName]: {
                              schema: prev[extName].schema,
                              value: e.target.value,
                            },
                          }))
                        }
                        placeholder="Provide a name for your schema"
                      />
                    </FormItemLayout>
                  )}
                </FormItemLayout>
              )
            })}
          </AccordionContent>
        </AccordionItem>
      </Accordion>
    </SheetSection>
  )
}