import * as React from 'react'; import RcUpload from 'rc-upload'; import classNames from 'classnames'; import uniqBy from 'lodash/uniqBy'; import LocaleReceiver from '../locale-provider/LocaleReceiver'; import defaultLocale from '../locale-provider/default'; 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'; export { UploadProps }; export default class Upload extends React.Component { static Dragger: typeof Dragger; static defaultProps = { prefixCls: 'ant-upload', type: 'select' as UploadType, multiple: false, action: '', data: {}, accept: '', beforeUpload: T, showUploadList: true, listType: 'text' as UploadListType, // or pictrue className: '', disabled: false, supportServerRender: true, }; recentUploadStatus: boolean | PromiseLike; progressTimer: any; private upload: any; constructor(props: UploadProps) { super(props); this.state = { fileList: props.fileList || props.defaultFileList || [], dragState: 'drop', }; } componentWillUnmount() { this.clearProgressTimer(); } onStart = (file: RcFile) => { let targetItem; let nextFileList = this.state.fileList.concat(); targetItem = fileToObject(file); targetItem.status = 'uploading'; nextFileList.push(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, }, file); }, 200); } onSuccess = (response: any, file: UploadFile) => { this.clearProgressTimer(); try { if (typeof response === 'string') { response = JSON.parse(response); } } catch (e) { /* do nothing */ } let fileList = this.state.fileList; let 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) => { let fileList = this.state.fileList; let 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(); let fileList = this.state.fileList; let 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; Promise.resolve(typeof onRemove === 'function' ? onRemove(file) : onRemove).then(ret => { // Prevent removing file if (ret === false) { return; } const removedFileList = removeFileItem(file, this.state.fileList); if (removedFileList) { this.onChange({ file, fileList: removedFileList, }); } }); } handleManualRemove = (file: UploadFile) => { this.upload.abort(file); file.status = 'removed'; // eslint-disable-line this.handleRemove(file); } onChange = (info: UploadChangeParam) => { if (!('fileList' in this.props)) { this.setState({ fileList: info.fileList }); } const { onChange } = this.props; if (onChange) { onChange(info); } } componentWillReceiveProps(nextProps: UploadProps) { if ('fileList' in nextProps) { this.setState({ fileList: nextProps.fileList || [], }); } } 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(fileList.map(fileToObject).concat(this.state.fileList), (item: UploadFile) => item.uid), }); return false; } else 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 } = this.props; const { showRemoveIcon, showPreviewIcon } = showUploadList as any; return ( ); } render() { const { prefixCls = '', className, showUploadList, listType, type, disabled, children, } = this.props; const rcUploadProps = { onStart: this.onStart, onError: this.onError, onProgress: this.onProgress, onSuccess: this.onSuccess, ...this.props, 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, }); const uploadButton = (
); if (listType === 'picture-card') { return ( {uploadList} {uploadButton} ); } return ( {uploadButton} {uploadList} ); } }