'use client' import type { FC } from 'react' import React, { useEffect, useState } from 'react' import { useTranslation } from 'react-i18next' import { useDebounce, useGetState } from 'ahooks' import produce from 'immer' import { LinkExternal02, Settings01 } from '../../base/icons/src/vender/line/general' import type { Credential, CustomCollectionBackend, CustomParamSchema, Emoji } from '../types' import { AuthHeaderPrefix, AuthType } from '../types' import GetSchema from './get-schema' import ConfigCredentials from './config-credentials' import TestApi from './test-api' import cn from '@/utils/classnames' import Drawer from '@/app/components/base/drawer-plus' import Button from '@/app/components/base/button' import Input from '@/app/components/base/input' import Textarea from '@/app/components/base/textarea' import EmojiPicker from '@/app/components/base/emoji-picker' import AppIcon from '@/app/components/base/app-icon' import { parseParamsSchema } from '@/service/tools' import LabelSelector from '@/app/components/tools/labels/selector' import Toast from '@/app/components/base/toast' const fieldNameClassNames = 'py-2 leading-5 text-sm font-medium text-gray-900' type Props = { positionLeft?: boolean payload: any onHide: () => void onAdd?: (payload: CustomCollectionBackend) => void onRemove?: () => void onEdit?: (payload: CustomCollectionBackend) => void } // Add and Edit const EditCustomCollectionModal: FC<Props> = ({ positionLeft, payload, onHide, onAdd, onEdit, onRemove, }) => { const { t } = useTranslation() const isAdd = !payload const isEdit = !!payload const [editFirst, setEditFirst] = useState(!isAdd) const [paramsSchemas, setParamsSchemas] = useState<CustomParamSchema[]>(payload?.tools || []) const [customCollection, setCustomCollection, getCustomCollection] = useGetState<CustomCollectionBackend>(isAdd ? { provider: '', credentials: { auth_type: AuthType.none, api_key_header: 'Authorization', api_key_header_prefix: AuthHeaderPrefix.basic, }, icon: { content: '🕵️', background: '#FEF7C3', }, schema_type: '', schema: '', } : payload) const originalProvider = isEdit ? payload.provider : '' const [showEmojiPicker, setShowEmojiPicker] = useState(false) const emoji = customCollection.icon const setEmoji = (emoji: Emoji) => { const newCollection = produce(customCollection, (draft) => { draft.icon = emoji }) setCustomCollection(newCollection) } const schema = customCollection.schema const debouncedSchema = useDebounce(schema, { wait: 500 }) const setSchema = (schema: any) => { const newCollection = produce(customCollection, (draft) => { draft.schema = schema }) setCustomCollection(newCollection) } useEffect(() => { if (!debouncedSchema) return if (isEdit && editFirst) { setEditFirst(false) return } (async () => { try { const { parameters_schema, schema_type } = await parseParamsSchema(debouncedSchema) const customCollection = getCustomCollection() const newCollection = produce(customCollection, (draft) => { draft.schema_type = schema_type }) setCustomCollection(newCollection) setParamsSchemas(parameters_schema) } catch (e) { const customCollection = getCustomCollection() const newCollection = produce(customCollection, (draft) => { draft.schema_type = '' }) setCustomCollection(newCollection) setParamsSchemas([]) } })() }, [debouncedSchema]) const [credentialsModalShow, setCredentialsModalShow] = useState(false) const credential = customCollection.credentials const setCredential = (credential: Credential) => { const newCollection = produce(customCollection, (draft) => { draft.credentials = credential }) setCustomCollection(newCollection) } const [currTool, setCurrTool] = useState<CustomParamSchema | null>(null) const [isShowTestApi, setIsShowTestApi] = useState(false) const [labels, setLabels] = useState<string[]>(payload?.labels || []) const handleLabelSelect = (value: string[]) => { setLabels(value) } const handleSave = () => { // const postData = clone(customCollection) const postData = produce(customCollection, (draft) => { delete draft.tools if (draft.credentials.auth_type === AuthType.none) { delete draft.credentials.api_key_header delete draft.credentials.api_key_header_prefix delete draft.credentials.api_key_value } draft.labels = labels }) let errorMessage = '' if (!postData.provider) errorMessage = t('common.errorMsg.fieldRequired', { field: t('tools.createTool.name') }) if (!postData.schema) errorMessage = t('common.errorMsg.fieldRequired', { field: t('tools.createTool.schema') }) if (errorMessage) { Toast.notify({ type: 'error', message: errorMessage, }) return } if (isAdd) { onAdd?.(postData) return } onEdit?.({ ...postData, original_provider: originalProvider, }) } const getPath = (url: string) => { if (!url) return '' try { const path = decodeURI(new URL(url).pathname) return path || '' } catch (e) { return url } } return ( <> <Drawer isShow positionCenter={isAdd && !positionLeft} onHide={onHide} title={t(`tools.createTool.${isAdd ? 'title' : 'editTitle'}`)!} panelClassName='mt-2 !w-[630px]' maxWidthClassName='!max-w-[630px]' height='calc(100vh - 16px)' headerClassName='!border-b-black/5' body={ <div className='flex flex-col h-full'> <div className='grow h-0 overflow-y-auto px-6 py-3 space-y-4'> <div> <div className={fieldNameClassNames}>{t('tools.createTool.name')} <span className='ml-1 text-red-500'>*</span></div> <div className='flex items-center justify-between gap-3'> <AppIcon size='large' onClick={() => { setShowEmojiPicker(true) }} className='cursor-pointer' icon={emoji.content} background={emoji.background} /> <Input className='h-10 grow' placeholder={t('tools.createTool.toolNamePlaceHolder')!} value={customCollection.provider} onChange={(e) => { const newCollection = produce(customCollection, (draft) => { draft.provider = e.target.value }) setCustomCollection(newCollection) }} /> </div> </div> {/* Schema */} <div className='select-none'> <div className='flex justify-between items-center'> <div className='flex items-center'> <div className={fieldNameClassNames}>{t('tools.createTool.schema')}<span className='ml-1 text-red-500'>*</span></div> <div className='mx-2 w-px h-3 bg-black/5'></div> <a href="https://swagger.io/specification/" target='_blank' rel='noopener noreferrer' className='flex items-center h-[18px] space-x-1 text-[#155EEF]' > <div className='text-xs font-normal'>{t('tools.createTool.viewSchemaSpec')}</div> <LinkExternal02 className='w-3 h-3' /> </a> </div> <GetSchema onChange={setSchema} /> </div> <Textarea className='h-[240px] resize-none' value={schema} onChange={e => setSchema(e.target.value)} placeholder={t('tools.createTool.schemaPlaceHolder')!} /> </div> {/* Available Tools */} <div> <div className={fieldNameClassNames}>{t('tools.createTool.availableTools.title')}</div> <div className='rounded-lg border border-gray-200 w-full overflow-x-auto'> <table className='w-full leading-[18px] text-xs text-gray-700 font-normal'> <thead className='text-gray-500 uppercase'> <tr className={cn(paramsSchemas.length > 0 && 'border-b', 'border-gray-200')}> <th className="p-2 pl-3 font-medium">{t('tools.createTool.availableTools.name')}</th> <th className="p-2 pl-3 font-medium w-[236px]">{t('tools.createTool.availableTools.description')}</th> <th className="p-2 pl-3 font-medium">{t('tools.createTool.availableTools.method')}</th> <th className="p-2 pl-3 font-medium">{t('tools.createTool.availableTools.path')}</th> <th className="p-2 pl-3 font-medium w-[54px]">{t('tools.createTool.availableTools.action')}</th> </tr> </thead> <tbody> {paramsSchemas.map((item, index) => ( <tr key={index} className='border-b last:border-0 border-gray-200'> <td className="p-2 pl-3">{item.operation_id}</td> <td className="p-2 pl-3 text-gray-500 w-[236px]">{item.summary}</td> <td className="p-2 pl-3">{item.method}</td> <td className="p-2 pl-3">{getPath(item.server_url)}</td> <td className="p-2 pl-3 w-[62px]"> <Button size='small' onClick={() => { setCurrTool(item) setIsShowTestApi(true) }} > {t('tools.createTool.availableTools.test')} </Button> </td> </tr> ))} </tbody> </table> </div> </div> {/* Authorization method */} <div> <div className={fieldNameClassNames}>{t('tools.createTool.authMethod.title')}</div> <div className='flex items-center h-9 justify-between px-2.5 bg-gray-100 rounded-lg cursor-pointer' onClick={() => setCredentialsModalShow(true)}> <div className='text-sm font-normal text-gray-900'>{t(`tools.createTool.authMethod.types.${credential.auth_type}`)}</div> <Settings01 className='w-4 h-4 text-gray-700 opacity-60' /> </div> </div> {/* Labels */} <div> <div className='py-2 leading-5 text-sm font-medium text-gray-900'>{t('tools.createTool.toolInput.label')}</div> <LabelSelector value={labels} onChange={handleLabelSelect} /> </div> {/* Privacy Policy */} <div> <div className={fieldNameClassNames}>{t('tools.createTool.privacyPolicy')}</div> <Input value={customCollection.privacy_policy} onChange={(e) => { const newCollection = produce(customCollection, (draft) => { draft.privacy_policy = e.target.value }) setCustomCollection(newCollection) }} className='h-10 grow' placeholder={t('tools.createTool.privacyPolicyPlaceholder') || ''} /> </div> <div> <div className={fieldNameClassNames}>{t('tools.createTool.customDisclaimer')}</div> <Input value={customCollection.custom_disclaimer} onChange={(e) => { const newCollection = produce(customCollection, (draft) => { draft.custom_disclaimer = e.target.value }) setCustomCollection(newCollection) }} className='h-10 grow' placeholder={t('tools.createTool.customDisclaimerPlaceholder') || ''} /> </div> </div> <div className={cn(isEdit ? 'justify-between' : 'justify-end', 'mt-2 shrink-0 flex py-4 px-6 rounded-b-[10px] bg-gray-50 border-t border-black/5')} > { isEdit && ( <Button onClick={onRemove} className='text-red-500 border-red-50 hover:border-red-500'>{t('common.operation.delete')}</Button> ) } <div className='flex space-x-2 '> <Button onClick={onHide}>{t('common.operation.cancel')}</Button> <Button variant='primary' onClick={handleSave}>{t('common.operation.save')}</Button> </div> </div> {showEmojiPicker && <EmojiPicker onSelect={(icon, icon_background) => { setEmoji({ content: icon, background: icon_background }) setShowEmojiPicker(false) }} onClose={() => { setShowEmojiPicker(false) }} />} {credentialsModalShow && ( <ConfigCredentials positionCenter={isAdd} credential={credential} onChange={setCredential} onHide={() => setCredentialsModalShow(false)} />) } {isShowTestApi && ( <TestApi positionCenter={isAdd} tool={currTool as CustomParamSchema} customCollection={customCollection} onHide={() => setIsShowTestApi(false)} /> )} </div> } isShowMask={true} clickOutsideNotOpen={true} /> </> ) } export default React.memo(EditCustomCollectionModal)