import * as React from 'react'; import { polyfill } from 'react-lifecycles-compat'; import RcUpload from 'rc-upload'; import classNames from 'classnames'; import uniqBy from 'lodash/uniqBy'; import findIndex from 'lodash/findIndex'; import Dragger from './Dragger'; import UploadList from './UploadList'; import { RcFile, UploadProps, UploadState, UploadFile, UploadLocale, UploadChangeParam, UploadType, UploadListType, } from './interface'; import { T, fileToObject, genPercentAdd, getFileItem, removeFileItem } from './utils'; import LocaleReceiver from '../locale-provider/LocaleReceiver'; import defaultLocale from '../locale-provider/default'; import { ConfigConsumer, ConfigConsumerProps } from '../config-provider'; export { UploadProps }; class Upload extends React.Component { static Dragger: typeof Dragger; static defaultProps = { type: 'select' as UploadType, multiple: false, action: '', data: {}, accept: '', beforeUpload: T, showUploadList: true, listType: 'text' as UploadListType, // or picture className: '', disabled: false, supportServerRender: true, }; static getDerivedStateFromProps(nextProps: UploadProps) { if ('fileList' in nextProps) { return { fileList: nextProps.fileList || [], }; } return null; } recentUploadStatus: boolean | PromiseLike; progressTimer: any; upload: any; constructor(props: UploadProps) { super(props); this.state = { fileList: props.fileList || props.defaultFileList || [], dragState: 'drop', }; } componentWillUnmount() { this.clearProgressTimer(); } onStart = (file: RcFile) => { const targetItem = fileToObject(file); targetItem.status = 'uploading'; const nextFileList = this.state.fileList.concat(); const fileIndex = findIndex(nextFileList, ({ uid }: UploadFile) => uid === targetItem.uid); if (fileIndex === -1) { nextFileList.push(targetItem); } else { nextFileList[fileIndex] = targetItem; } this.onChange({ file: targetItem, fileList: nextFileList, }); // fix ie progress if (!(window as any).FormData) { this.autoUpdateProgress(0, targetItem); } }; autoUpdateProgress(_: any, file: UploadFile) { const getPercent = genPercentAdd(); let curPercent = 0; this.clearProgressTimer(); this.progressTimer = setInterval(() => { curPercent = getPercent(curPercent); this.onProgress( { percent: curPercent * 100, }, file, ); }, 200); } onSuccess = (response: any, file: UploadFile) => { this.clearProgressTimer(); try { if (typeof response === 'string') { response = JSON.parse(response); } } catch (e) { /* do nothing */ } const fileList = this.state.fileList; const targetItem = getFileItem(file, fileList); // removed if (!targetItem) { return; } targetItem.status = 'done'; targetItem.response = response; this.onChange({ file: { ...targetItem }, fileList, }); }; onProgress = (e: { percent: number }, file: UploadFile) => { const fileList = this.state.fileList; const targetItem = getFileItem(file, fileList); // removed if (!targetItem) { return; } targetItem.percent = e.percent; this.onChange({ event: e, file: { ...targetItem }, fileList: this.state.fileList, }); }; onError = (error: Error, response: any, file: UploadFile) => { this.clearProgressTimer(); const fileList = this.state.fileList; const targetItem = getFileItem(file, fileList); // removed if (!targetItem) { return; } targetItem.error = error; targetItem.response = response; targetItem.status = 'error'; this.onChange({ file: { ...targetItem }, fileList, }); }; handleRemove(file: UploadFile) { const { onRemove } = this.props; const { status } = file; file.status = 'removed'; // eslint-disable-line Promise.resolve(typeof onRemove === 'function' ? onRemove(file) : onRemove).then(ret => { // Prevent removing file if (ret === false) { file.status = status; return; } const removedFileList = removeFileItem(file, this.state.fileList); if (removedFileList) { this.onChange({ file, fileList: removedFileList, }); } }); } handleManualRemove = (file: UploadFile) => { if (this.upload) { this.upload.abort(file); } this.handleRemove(file); }; onChange = (info: UploadChangeParam) => { if (!('fileList' in this.props)) { this.setState({ fileList: info.fileList }); } const { onChange } = this.props; if (onChange) { onChange(info); } }; onFileDrop = (e: React.DragEvent) => { this.setState({ dragState: e.type, }); }; beforeUpload = (file: RcFile, fileList: RcFile[]) => { if (!this.props.beforeUpload) { return true; } const result = this.props.beforeUpload(file, fileList); if (result === false) { this.onChange({ file, fileList: uniqBy( this.state.fileList.concat(fileList.map(fileToObject)), (item: UploadFile) => item.uid, ), }); return false; } if (result && (result as PromiseLike).then) { return result; } return true; }; clearProgressTimer() { clearInterval(this.progressTimer); } saveUpload = (node: typeof RcUpload) => { this.upload = node; }; renderUploadList = (locale: UploadLocale) => { const { showUploadList, listType, onPreview, previewFile, disabled } = this.props; const { showRemoveIcon, showPreviewIcon } = showUploadList as any; return ( ); }; renderUpload = ({ getPrefixCls }: ConfigConsumerProps) => { const { prefixCls: customizePrefixCls, className, showUploadList, listType, type, disabled, children, } = this.props; const prefixCls = getPrefixCls('upload', customizePrefixCls); const rcUploadProps = { onStart: this.onStart, onError: this.onError, onProgress: this.onProgress, onSuccess: this.onSuccess, ...this.props, prefixCls, beforeUpload: this.beforeUpload, }; delete rcUploadProps.className; const uploadList = showUploadList ? ( {this.renderUploadList} ) : null; if (type === 'drag') { const dragCls = classNames(prefixCls, { [`${prefixCls}-drag`]: true, [`${prefixCls}-drag-uploading`]: this.state.fileList.some( file => file.status === 'uploading', ), [`${prefixCls}-drag-hover`]: this.state.dragState === 'dragover', [`${prefixCls}-disabled`]: disabled, }); return (
{children}
{uploadList}
); } const uploadButtonCls = classNames(prefixCls, { [`${prefixCls}-select`]: true, [`${prefixCls}-select-${listType}`]: true, [`${prefixCls}-disabled`]: disabled, }); // Remove id to avoid open by label when trigger is hidden // https://github.com/ant-design/ant-design/issues/14298 // https://github.com/ant-design/ant-design/issues/16478 if (!children || disabled) { delete rcUploadProps.id; } const uploadButton = (
); if (listType === 'picture-card') { return ( {uploadList} {uploadButton} ); } return ( {uploadButton} {uploadList} ); }; render() { return {this.renderUpload}; } } polyfill(Upload); export default Upload;