From 49b7acf52eb31e5e647174aefaf916287112d4be Mon Sep 17 00:00:00 2001
From: StyleZhang <jasonapring2015@outlook.com>
Date: Thu, 19 Sep 2024 14:52:22 +0800
Subject: [PATCH] fix: file uploader

---
 .../file-uploader-in-attachment/index.tsx           | 13 +++++++------
 .../file-uploader-in-chat-input/index.tsx           |  3 +++
 2 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/web/app/components/base/file-uploader/file-uploader-in-attachment/index.tsx b/web/app/components/base/file-uploader/file-uploader-in-attachment/index.tsx
index 54182413f1..facc037b03 100644
--- a/web/app/components/base/file-uploader/file-uploader-in-attachment/index.tsx
+++ b/web/app/components/base/file-uploader/file-uploader-in-attachment/index.tsx
@@ -18,6 +18,7 @@ import FileItem from './file-item'
 import Button from '@/app/components/base/button'
 import cn from '@/utils/classnames'
 import type { FileUpload } from '@/app/components/base/features/types'
+import { TransferMethod } from '@/types/app'
 
 type Option = {
   value: string
@@ -38,12 +39,12 @@ const FileUploaderInAttachment = ({
   } = useFile(fileConfig)
   const options = [
     {
-      value: 'local',
+      value: TransferMethod.local_file,
       label: t('common.fileUploader.uploadFromComputer'),
       icon: <RiUploadCloud2Line className='w-4 h-4' />,
     },
     {
-      value: 'link',
+      value: TransferMethod.remote_url,
       label: t('common.fileUploader.pasteFileLink'),
       icon: <RiLink className='w-4 h-4' />,
     },
@@ -54,13 +55,13 @@ const FileUploaderInAttachment = ({
       <Button
         key={option.value}
         variant='tertiary'
-        className={cn('basis-1/2 relative', open && 'bg-components-button-tertiary-bg-hover')}
+        className={cn('grow relative', open && 'bg-components-button-tertiary-bg-hover')}
         disabled={!!(fileConfig.number_limits && files.length >= fileConfig.number_limits)}
       >
         {option.icon}
         <span className='ml-1'>{option.label}</span>
         {
-          option.value === 'local' && (
+          option.value === TransferMethod.local_file && (
             <FileInput fileConfig={fileConfig} />
           )
         }
@@ -71,10 +72,10 @@ const FileUploaderInAttachment = ({
     return (open: boolean) => renderButton(option, open)
   }, [renderButton])
   const renderOption = useCallback((option: Option) => {
-    if (option.value === 'local')
+    if (option.value === TransferMethod.local_file && fileConfig?.allowed_file_upload_methods?.includes(TransferMethod.local_file))
       return renderButton(option)
 
-    if (option.value === 'link') {
+    if (option.value === TransferMethod.remote_url && fileConfig?.allowed_file_upload_methods?.includes(TransferMethod.remote_url)) {
       return (
         <FileFromLinkOrLocal
           key={option.value}
diff --git a/web/app/components/base/file-uploader/file-uploader-in-chat-input/index.tsx b/web/app/components/base/file-uploader/file-uploader-in-chat-input/index.tsx
index 6b8c4efce3..e20a119aac 100644
--- a/web/app/components/base/file-uploader/file-uploader-in-chat-input/index.tsx
+++ b/web/app/components/base/file-uploader/file-uploader-in-chat-input/index.tsx
@@ -9,6 +9,7 @@ import FileFromLinkOrLocal from '../file-from-link-or-local'
 import ActionButton from '@/app/components/base/action-button'
 import cn from '@/utils/classnames'
 import type { FileUpload } from '@/app/components/base/features/types'
+import { TransferMethod } from '@/types/app'
 
 type FileUploaderInChatInputProps = {
   fileConfig: FileUpload
@@ -31,6 +32,8 @@ const FileUploaderInChatInput = ({
     <FileFromLinkOrLocal
       trigger={renderTrigger}
       fileConfig={fileConfig}
+      showFromLocal={fileConfig?.allowed_file_upload_methods?.includes(TransferMethod.local_file)}
+      showFromLink={fileConfig?.allowed_file_upload_methods?.includes(TransferMethod.remote_url)}
     />
   )
 }