dify/web/app/components/header/account-setting/api-based-extension-page/selector.tsx

import type { FC } from 'react'
import {
  RiAddLine,
  RiArrowDownSLine,
} from '@remixicon/react'
import { useState } from 'react'
import { useTranslation } from 'react-i18next'
import {
  ArrowUpRight,
} from '@/app/components/base/icons/src/vender/line/arrows'
import {
  PortalToFollowElem,
  PortalToFollowElemContent,
  PortalToFollowElemTrigger,
} from '@/app/components/base/portal-to-follow-elem'
import { ACCOUNT_SETTING_TAB } from '@/app/components/header/account-setting/constants'
import { useModalContext } from '@/context/modal-context'
import { useApiBasedExtensions } from '@/service/use-common'

type ApiBasedExtensionSelectorProps = {
  value: string
  onChange: (value: string) => void
}

const ApiBasedExtensionSelector: FC<ApiBasedExtensionSelectorProps> = ({
  value,
  onChange,
}) => {
  const { t } = useTranslation()
  const [open, setOpen] = useState(false)
  const {
    setShowAccountSettingModal,
    setShowApiBasedExtensionModal,
  } = useModalContext()
  const { data, refetch: mutate } = useApiBasedExtensions()
  const handleSelect = (id: string) => {
    onChange(id)
    setOpen(false)
  }

  const currentItem = data?.find(item => item.id === value)

  return (
    <PortalToFollowElem
      open={open}
      onOpenChange={setOpen}
      placement="bottom-start"
      offset={4}
    >
      <PortalToFollowElemTrigger onClick={() => setOpen(v => !v)} className="w-full">
        {
          currentItem
            ? (
                <div className="flex h-9 cursor-pointer items-center justify-between rounded-lg bg-components-input-bg-normal pl-3 pr-2.5">
                  <div className="text-sm text-text-primary">{currentItem.name}</div>
                  <div className="flex items-center">
                    <div className="mr-1.5 w-[270px] truncate text-right text-xs text-text-quaternary">
                      {currentItem.api_endpoint}
                    </div>
                    <RiArrowDownSLine className={`h-4 w-4 text-text-secondary ${!open && 'opacity-60'}`} />
                  </div>
                </div>
              )
            : (
                <div className="flex h-9 cursor-pointer items-center justify-between rounded-lg bg-components-input-bg-normal pl-3 pr-2.5 text-sm text-text-quaternary">
                  {t('apiBasedExtension.selector.placeholder', { ns: 'common' })}
                  <RiArrowDownSLine className={`h-4 w-4 text-text-secondary ${!open && 'opacity-60'}`} />
                </div>
              )
        }
      </PortalToFollowElemTrigger>
      <PortalToFollowElemContent className="z-[102] w-[calc(100%-32px)] max-w-[576px]">
        <div className="z-10 w-full rounded-lg border-[0.5px] border-components-panel-border bg-components-panel-bg shadow-lg">
          <div className="p-1">
            <div className="flex items-center justify-between px-3 pb-1 pt-2">
              <div className="text-xs font-medium text-text-tertiary">
                {t('apiBasedExtension.selector.title', { ns: 'common' })}
              </div>
              <div
                className="flex cursor-pointer items-center text-xs text-text-accent"
                onClick={() => {
                  setOpen(false)
                  setShowAccountSettingModal({ payload: ACCOUNT_SETTING_TAB.API_BASED_EXTENSION })
                }}
              >
                {t('apiBasedExtension.selector.manage', { ns: 'common' })}
                <ArrowUpRight className="ml-0.5 h-3 w-3" />
              </div>
            </div>
            <div className="max-h-[250px] overflow-y-auto">
              {
                data?.map(item => (
                  <div
                    key={item.id}
                    className="w-full cursor-pointer rounded-md px-3 py-1.5 text-left hover:stroke-state-base-hover"
                    onClick={() => handleSelect(item.id!)}
                  >
                    <div className="text-sm text-text-primary">{item.name}</div>
                    <div className="text-xs text-text-tertiary">{item.api_endpoint}</div>
                  </div>
                ))
              }
            </div>
          </div>
          <div className="h-px bg-divider-regular" />
          <div className="p-1">
            <div
              className="flex h-8 cursor-pointer items-center px-3 text-sm text-text-accent"
              onClick={() => {
                setOpen(false)
                setShowApiBasedExtensionModal({ payload: {}, onSaveCallback: () => mutate() })
              }}
            >
              <RiAddLine className="mr-2 h-4 w-4" />
              {t('operation.add', { ns: 'common' })}
            </div>
          </div>
        </div>
      </PortalToFollowElemContent>
    </PortalToFollowElem>
  )
}

export default ApiBasedExtensionSelector