feat: add SharePoint picker support

client/src/components/Prompts/Files/AttachFileButton.tsx

@@ -1,8 +1,12 @@
 import * as Ariakit from '@ariakit/react';
 import { EToolResources } from 'librechat-data-provider';
 import React, { useRef, useState, useMemo, useCallback } from 'react';
-import { FileUpload, DropdownPopup, AttachmentIcon } from '@librechat/client';
 import { FileSearch, ImageUpIcon, TerminalSquareIcon, FileType2Icon } from 'lucide-react';
+import { FileUpload, DropdownPopup, AttachmentIcon, SharePointIcon } from '@librechat/client';
+import useSharePointFileHandling from '~/hooks/Files/useSharePointFileHandling';
+import { SharePointPickerDialog } from '~/components/SharePoint';
+import { useGetStartupConfig } from '~/data-provider';
+import { MenuItemProps } from '~/common';
 import { useLocalize } from '~/hooks';
 
 interface AttachFileButtonProps {
@@ -16,6 +20,13 @@ const AttachFileButton = ({ handleFileChange, disabled }: AttachFileButtonProps)
   const inputRef = useRef<HTMLInputElement>(null);
   const [isPopoverActive, setIsPopoverActive] = useState(false);
   const [toolResource, setToolResource] = useState<EToolResources | undefined>();
+  const [isSharePointDialogOpen, setIsSharePointDialogOpen] = useState(false);
+
+  const { handleSharePointFiles, isProcessing, downloadProgress } = useSharePointFileHandling({
+    toolResource,
+  });
+  const { data: startupConfig } = useGetStartupConfig();
+  const sharePointEnabled = startupConfig?.sharePointFilePickerEnabled;
 
   const handleUploadClick = useCallback((isImage?: boolean) => {
     if (!inputRef.current) {
@@ -28,12 +39,13 @@ const AttachFileButton = ({ handleFileChange, disabled }: AttachFileButtonProps)
   }, []);
 
   const dropdownItems = useMemo(() => {
-    return [
+    const createMenuItems = (onAction: (isImage?: boolean) => void) => {
+      const items: MenuItemProps[] = [
         {
           label: localize('com_ui_upload_image_input'),
           onClick: () => {
             setToolResource(EToolResources.image_edit);
-          handleUploadClick(true);
+            onAction(true);
           },
           icon: <ImageUpIcon className="icon-md" />,
         },
@@ -41,7 +53,7 @@ const AttachFileButton = ({ handleFileChange, disabled }: AttachFileButtonProps)
           label: localize('com_ui_upload_ocr_text'),
           onClick: () => {
             setToolResource(EToolResources.ocr);
-          handleUploadClick();
+            onAction();
           },
           icon: <FileType2Icon className="icon-md" />,
         },
@@ -49,7 +61,7 @@ const AttachFileButton = ({ handleFileChange, disabled }: AttachFileButtonProps)
           label: localize('com_ui_upload_file_search'),
           onClick: () => {
             setToolResource(EToolResources.file_search);
-          handleUploadClick();
+            onAction();
           },
           icon: <FileSearch className="icon-md" />,
         },
@@ -57,12 +69,31 @@ const AttachFileButton = ({ handleFileChange, disabled }: AttachFileButtonProps)
           label: localize('com_ui_upload_code_files'),
           onClick: () => {
             setToolResource(EToolResources.execute_code);
-          handleUploadClick();
+            onAction();
           },
           icon: <TerminalSquareIcon className="icon-md" />,
         },
       ];
-  }, [localize, handleUploadClick]);
+      return items;
+    };
+
+    const localItems = createMenuItems(handleUploadClick);
+
+    if (sharePointEnabled) {
+      const sharePointItems = createMenuItems(() => {
+        setIsSharePointDialogOpen(true);
+      });
+      localItems.push({
+        label: localize('com_files_upload_sharepoint'),
+        onClick: () => {},
+        icon: <SharePointIcon className="icon-md" />,
+        subItems: sharePointItems,
+      });
+      return localItems;
+    }
+
+    return localItems;
+  }, [localize, handleUploadClick, sharePointEnabled, setIsSharePointDialogOpen]);
 
   const menuTrigger = (
     <Ariakit.MenuButton
@@ -76,7 +107,17 @@ const AttachFileButton = ({ handleFileChange, disabled }: AttachFileButtonProps)
     </Ariakit.MenuButton>
   );
 
+  const handleSharePointFilesSelected = async (sharePointFiles: any[]) => {
+    try {
+      await handleSharePointFiles(sharePointFiles);
+      setIsSharePointDialogOpen(false);
+    } catch (error) {
+      console.error('SharePoint file processing error:', error);
+    }
+  };
+
   return (
+    <>
       <FileUpload
         ref={inputRef}
         handleFileChange={(e) => {
@@ -95,6 +136,14 @@ const AttachFileButton = ({ handleFileChange, disabled }: AttachFileButtonProps)
           iconClassName="mr-0"
         />
       </FileUpload>
+      <SharePointPickerDialog
+        isOpen={isSharePointDialogOpen}
+        onOpenChange={setIsSharePointDialogOpen}
+        onFilesSelected={handleSharePointFilesSelected}
+        isDownloading={isProcessing}
+        downloadProgress={downloadProgress}
+      />
+    </>
   );
 };