feat: add SharePoint picker support

This commit is contained in:
Dustin Healy 2025-09-05 19:37:53 -07:00
parent d65accddc1
commit 600641d02f

View file

@ -1,8 +1,12 @@
import * as Ariakit from '@ariakit/react'; import * as Ariakit from '@ariakit/react';
import { EToolResources } from 'librechat-data-provider'; import { EToolResources } from 'librechat-data-provider';
import React, { useRef, useState, useMemo, useCallback } from 'react'; import React, { useRef, useState, useMemo, useCallback } from 'react';
import { FileUpload, DropdownPopup, AttachmentIcon } from '@librechat/client';
import { FileSearch, ImageUpIcon, TerminalSquareIcon, FileType2Icon } from 'lucide-react'; 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'; import { useLocalize } from '~/hooks';
interface AttachFileButtonProps { interface AttachFileButtonProps {
@ -16,6 +20,13 @@ const AttachFileButton = ({ handleFileChange, disabled }: AttachFileButtonProps)
const inputRef = useRef<HTMLInputElement>(null); const inputRef = useRef<HTMLInputElement>(null);
const [isPopoverActive, setIsPopoverActive] = useState(false); const [isPopoverActive, setIsPopoverActive] = useState(false);
const [toolResource, setToolResource] = useState<EToolResources | undefined>(); 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) => { const handleUploadClick = useCallback((isImage?: boolean) => {
if (!inputRef.current) { if (!inputRef.current) {
@ -28,41 +39,61 @@ const AttachFileButton = ({ handleFileChange, disabled }: AttachFileButtonProps)
}, []); }, []);
const dropdownItems = useMemo(() => { const dropdownItems = useMemo(() => {
return [ const createMenuItems = (onAction: (isImage?: boolean) => void) => {
{ const items: MenuItemProps[] = [
label: localize('com_ui_upload_image_input'), {
onClick: () => { label: localize('com_ui_upload_image_input'),
setToolResource(EToolResources.image_edit); onClick: () => {
handleUploadClick(true); setToolResource(EToolResources.image_edit);
onAction(true);
},
icon: <ImageUpIcon className="icon-md" />,
}, },
icon: <ImageUpIcon className="icon-md" />, {
}, label: localize('com_ui_upload_ocr_text'),
{ onClick: () => {
label: localize('com_ui_upload_ocr_text'), setToolResource(EToolResources.ocr);
onClick: () => { onAction();
setToolResource(EToolResources.ocr); },
handleUploadClick(); icon: <FileType2Icon className="icon-md" />,
}, },
icon: <FileType2Icon className="icon-md" />, {
}, label: localize('com_ui_upload_file_search'),
{ onClick: () => {
label: localize('com_ui_upload_file_search'), setToolResource(EToolResources.file_search);
onClick: () => { onAction();
setToolResource(EToolResources.file_search); },
handleUploadClick(); icon: <FileSearch className="icon-md" />,
}, },
icon: <FileSearch className="icon-md" />, {
}, label: localize('com_ui_upload_code_files'),
{ onClick: () => {
label: localize('com_ui_upload_code_files'), setToolResource(EToolResources.execute_code);
onClick: () => { onAction();
setToolResource(EToolResources.execute_code); },
handleUploadClick(); icon: <TerminalSquareIcon className="icon-md" />,
}, },
icon: <TerminalSquareIcon className="icon-md" />, ];
}, return items;
]; };
}, [localize, handleUploadClick]);
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 = ( const menuTrigger = (
<Ariakit.MenuButton <Ariakit.MenuButton
@ -76,25 +107,43 @@ const AttachFileButton = ({ handleFileChange, disabled }: AttachFileButtonProps)
</Ariakit.MenuButton> </Ariakit.MenuButton>
); );
const handleSharePointFilesSelected = async (sharePointFiles: any[]) => {
try {
await handleSharePointFiles(sharePointFiles);
setIsSharePointDialogOpen(false);
} catch (error) {
console.error('SharePoint file processing error:', error);
}
};
return ( return (
<FileUpload <>
ref={inputRef} <FileUpload
handleFileChange={(e) => { ref={inputRef}
handleFileChange?.(e, toolResource); handleFileChange={(e) => {
}} handleFileChange?.(e, toolResource);
> }}
<DropdownPopup >
menuId="attach-file-button" <DropdownPopup
className="overflow-visible" menuId="attach-file-button"
isOpen={isPopoverActive} className="overflow-visible"
setIsOpen={setIsPopoverActive} isOpen={isPopoverActive}
modal={true} setIsOpen={setIsPopoverActive}
unmountOnHide={true} modal={true}
trigger={menuTrigger} unmountOnHide={true}
items={dropdownItems} trigger={menuTrigger}
iconClassName="mr-0" items={dropdownItems}
iconClassName="mr-0"
/>
</FileUpload>
<SharePointPickerDialog
isOpen={isSharePointDialogOpen}
onOpenChange={setIsSharePointDialogOpen}
onFilesSelected={handleSharePointFilesSelected}
isDownloading={isProcessing}
downloadProgress={downloadProgress}
/> />
</FileUpload> </>
); );
}; };