import * as React from "react"; import BaseWidget, { WidgetProps, WidgetState } from "./BaseWidget"; import { WidgetType } from "constants/WidgetConstants"; import Uppy from "@uppy/core"; import GoogleDrive from "@uppy/google-drive"; import Webcam from "@uppy/webcam"; import Url from "@uppy/url"; import OneDrive from "@uppy/onedrive"; import FilePickerComponent from "components/designSystems/appsmith/FilePickerComponent"; import { WidgetPropertyValidationType } from "utils/ValidationFactory"; import { VALIDATION_TYPES } from "constants/WidgetValidation"; import { EventType } from "constants/ActionConstants"; import { TriggerPropertiesMap } from "utils/WidgetFactory"; class FilePickerWidget extends BaseWidget { uppy: any; filePickerRef: React.RefObject; constructor(props: FilePickerWidgetProps) { super(props); this.refreshUppy(props); this.filePickerRef = React.createRef(); } static getPropertyValidationMap(): WidgetPropertyValidationType { return { label: VALIDATION_TYPES.TEXT, maxNumFiles: VALIDATION_TYPES.NUMBER, allowedFileTypes: VALIDATION_TYPES.ARRAY, }; } refreshUppy = (props: FilePickerWidgetProps) => { this.uppy = Uppy({ id: this.props.widgetId, autoProceed: false, allowMultipleUploads: true, debug: false, restrictions: { maxFileSize: null, maxNumberOfFiles: props.maxNumFiles, minNumberOfFiles: null, allowedFileTypes: props.allowedFileTypes, }, }) .use(GoogleDrive, { companionUrl: "https://companion.uppy.io" }) .use(Url, { companionUrl: "https://companion.uppy.io" }) .use(OneDrive, { companionUrl: "https://companion.uppy.io/", }) .use(Webcam, { onBeforeSnapshot: () => Promise.resolve(), countdown: false, mirror: true, facingMode: "user", locale: {}, }); this.uppy.on("file-added", (file: any) => { this.updateWidgetMetaProperty("file", file); }); this.uppy.on("upload", (uppyFile: any) => { const files = this.uppy.getFiles(); const fileArray: any = []; if (files.length === 0) { this.updateWidgetMetaProperty("files", []); } else { files.map((file: any) => { const reader = new FileReader(); reader.readAsDataURL(file.data); reader.onloadend = () => { const base64data = reader.result; fileArray.push(base64data); if (fileArray.length === files.length) { if (this.filePickerRef.current) this.filePickerRef.current.closeModal(); this.updateWidgetMetaProperty("files", fileArray); this.onFilesSelected(); } }; }); } return true; }); }; static getTriggerPropertyMap(): TriggerPropertiesMap { return { onFilesSelected: true, }; } onFilesSelected() { if (this.props.onFilesSelected) { this.executeAction({ dynamicString: this.props.onFilesSelected, event: { type: EventType.ON_FILES_SELECTED, }, }); } } componentDidUpdate(prevProps: FilePickerWidgetProps) { super.componentDidUpdate(prevProps); if ( prevProps.allowedFileTypes !== this.props.allowedFileTypes || prevProps.maxNumFiles !== this.props.maxNumFiles ) { this.refreshUppy(this.props); } } componentDidMount() { super.componentDidMount(); this.refreshUppy(this.props); } componentWillUnmount() { this.uppy.close(); } getPageView() { return ( ); } getWidgetType(): WidgetType { return "FILE_PICKER_WIDGET"; } } export interface FilePickerWidgetProps extends WidgetProps { label: string; maxNumFiles?: number; files: any[]; allowedFileTypes: string[]; onFilesSelected?: string; } export default FilePickerWidget;