mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-13 07:39:10 +08:00
14fa342da3
* chore(deps): update dependency eslint to v8 * fix eslint errors * fix eslint errors Co-authored-by: Renovate Bot <bot@renovateapp.com> Co-authored-by: afc163 <afc163@gmail.com>
252 lines
7.5 KiB
TypeScript
252 lines
7.5 KiB
TypeScript
import React, { Component } from 'react';
|
|
import { Upload, Tooltip, Popover, Modal, Progress, message, Spin, Result } from 'antd';
|
|
import CopyToClipboard from 'react-copy-to-clipboard';
|
|
import { injectIntl } from 'react-intl';
|
|
import * as AntdIcons from '@ant-design/icons';
|
|
|
|
const allIcons: {
|
|
[key: string]: any;
|
|
} = AntdIcons;
|
|
|
|
const { Dragger } = Upload;
|
|
interface AntdIconClassifier {
|
|
load: Function;
|
|
predict: Function;
|
|
}
|
|
declare global {
|
|
interface Window {
|
|
antdIconClassifier: AntdIconClassifier;
|
|
}
|
|
}
|
|
|
|
interface PicSearcherProps {
|
|
intl: any;
|
|
}
|
|
|
|
interface PicSearcherState {
|
|
loading: boolean;
|
|
modalVisible: boolean;
|
|
popoverVisible: boolean;
|
|
icons: Array<string>;
|
|
fileList: Array<any>;
|
|
error: boolean;
|
|
modelLoaded: boolean;
|
|
}
|
|
|
|
interface iconObject {
|
|
type: string;
|
|
score: number;
|
|
}
|
|
|
|
class PicSearcher extends Component<PicSearcherProps, PicSearcherState> {
|
|
state = {
|
|
loading: false,
|
|
modalVisible: false,
|
|
popoverVisible: false,
|
|
icons: [],
|
|
fileList: [],
|
|
error: false,
|
|
modelLoaded: false,
|
|
};
|
|
|
|
componentDidMount() {
|
|
this.loadModel();
|
|
this.setState({ popoverVisible: !localStorage.getItem('disableIconTip') });
|
|
}
|
|
|
|
componentWillUnmount() {
|
|
document.removeEventListener('paste', this.onPaste);
|
|
}
|
|
|
|
loadModel = () => {
|
|
const script = document.createElement('script');
|
|
script.onload = async () => {
|
|
await window.antdIconClassifier.load();
|
|
this.setState({ modelLoaded: true });
|
|
document.addEventListener('paste', this.onPaste);
|
|
};
|
|
script.src = 'https://cdn.jsdelivr.net/gh/lewis617/antd-icon-classifier@0.0/dist/main.js';
|
|
document.head.appendChild(script);
|
|
};
|
|
|
|
onPaste = (event: ClipboardEvent) => {
|
|
const items = event.clipboardData && event.clipboardData.items;
|
|
let file = null;
|
|
if (items && items.length) {
|
|
for (let i = 0; i < items.length; i += 1) {
|
|
if (items[i].type.indexOf('image') !== -1) {
|
|
file = items[i].getAsFile();
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
if (file) this.uploadFile(file);
|
|
};
|
|
|
|
uploadFile = (file: File) => {
|
|
this.setState(() => ({ loading: true }));
|
|
const reader: FileReader = new FileReader();
|
|
reader.onload = () => {
|
|
this.toImage(reader.result).then(this.predict);
|
|
this.setState(() => ({
|
|
fileList: [{ uid: 1, name: file.name, status: 'done', url: reader.result }],
|
|
}));
|
|
};
|
|
reader.readAsDataURL(file);
|
|
};
|
|
|
|
// eslint-disable-next-line class-methods-use-this
|
|
toImage = (url: any) =>
|
|
new Promise(resolve => {
|
|
const img = new Image();
|
|
img.setAttribute('crossOrigin', 'anonymous');
|
|
img.src = url;
|
|
img.onload = function onload() {
|
|
resolve(img);
|
|
};
|
|
});
|
|
|
|
predict = (imgEl: any) => {
|
|
try {
|
|
let icons = window.antdIconClassifier.predict(imgEl);
|
|
if (gtag && icons.length >= 1) {
|
|
gtag('event', 'icon', {
|
|
event_category: 'search-by-image',
|
|
event_label: icons[0].className,
|
|
});
|
|
}
|
|
icons = icons.map((i: any) => ({ score: i.score, type: i.className.replace(/\s/g, '-') }));
|
|
this.setState(() => ({ icons, loading: false, error: false }));
|
|
} catch (err) {
|
|
this.setState(() => ({ loading: false, error: true }));
|
|
}
|
|
};
|
|
|
|
toggleModal = () => {
|
|
this.setState(prev => ({
|
|
modalVisible: !prev.modalVisible,
|
|
popoverVisible: false,
|
|
fileList: [],
|
|
icons: [],
|
|
}));
|
|
if (!localStorage.getItem('disableIconTip')) {
|
|
localStorage.setItem('disableIconTip', 'true');
|
|
}
|
|
};
|
|
|
|
// eslint-disable-next-line class-methods-use-this
|
|
onCopied = (text: string) => {
|
|
message.success(
|
|
<span>
|
|
<code className="copied-code">{text}</code> copied 🎉
|
|
</span>,
|
|
);
|
|
};
|
|
|
|
render() {
|
|
const {
|
|
intl: { messages },
|
|
} = this.props;
|
|
const { modalVisible, popoverVisible, icons, fileList, loading, modelLoaded, error } =
|
|
this.state;
|
|
return (
|
|
<div className="icon-pic-searcher">
|
|
<Popover
|
|
content={messages[`app.docs.components.icon.pic-searcher.intro`]}
|
|
visible={popoverVisible}
|
|
>
|
|
<AntdIcons.CameraOutlined className="icon-pic-btn" onClick={this.toggleModal} />
|
|
</Popover>
|
|
<Modal
|
|
title={messages[`app.docs.components.icon.pic-searcher.title`]}
|
|
visible={modalVisible}
|
|
onCancel={this.toggleModal}
|
|
footer={null}
|
|
>
|
|
{modelLoaded || (
|
|
<Spin
|
|
spinning={!modelLoaded}
|
|
tip={messages['app.docs.components.icon.pic-searcher.modelloading']}
|
|
>
|
|
<div style={{ height: 100 }} />
|
|
</Spin>
|
|
)}
|
|
{modelLoaded && (
|
|
<Dragger
|
|
accept="image/jpeg, image/png"
|
|
listType="picture"
|
|
customRequest={(o: any) => this.uploadFile(o.file)}
|
|
fileList={fileList}
|
|
showUploadList={{ showPreviewIcon: false, showRemoveIcon: false }}
|
|
>
|
|
<p className="ant-upload-drag-icon">
|
|
<AntdIcons.InboxOutlined />
|
|
</p>
|
|
<p className="ant-upload-text">
|
|
{messages['app.docs.components.icon.pic-searcher.upload-text']}
|
|
</p>
|
|
<p className="ant-upload-hint">
|
|
{messages['app.docs.components.icon.pic-searcher.upload-hint']}
|
|
</p>
|
|
</Dragger>
|
|
)}
|
|
<Spin spinning={loading} tip={messages['app.docs.components.icon.pic-searcher.matching']}>
|
|
<div className="icon-pic-search-result">
|
|
{icons.length > 0 && (
|
|
<div className="result-tip">
|
|
{messages['app.docs.components.icon.pic-searcher.result-tip']}
|
|
</div>
|
|
)}
|
|
<table>
|
|
{icons.length > 0 && (
|
|
<thead>
|
|
<tr>
|
|
<th className="col-icon">
|
|
{messages['app.docs.components.icon.pic-searcher.th-icon']}
|
|
</th>
|
|
<th>{messages['app.docs.components.icon.pic-searcher.th-score']}</th>
|
|
</tr>
|
|
</thead>
|
|
)}
|
|
<tbody>
|
|
{icons.map((icon: iconObject) => {
|
|
const { type } = icon;
|
|
const iconName = `${type
|
|
.split('-')
|
|
.map(str => `${str[0].toUpperCase()}${str.slice(1)}`)
|
|
.join('')}Outlined`;
|
|
|
|
return (
|
|
<tr key={iconName}>
|
|
<td className="col-icon">
|
|
<CopyToClipboard text={`<${iconName} />`} onCopy={this.onCopied}>
|
|
<Tooltip title={icon.type} placement="right">
|
|
{React.createElement(allIcons[iconName])}
|
|
</Tooltip>
|
|
</CopyToClipboard>
|
|
</td>
|
|
<td>
|
|
<Progress percent={Math.ceil(icon.score * 100)} />
|
|
</td>
|
|
</tr>
|
|
);
|
|
})}
|
|
</tbody>
|
|
</table>
|
|
{error && (
|
|
<Result
|
|
status="500"
|
|
title="503"
|
|
subTitle={messages['app.docs.components.icon.pic-searcher.server-error']}
|
|
/>
|
|
)}
|
|
</div>
|
|
</Spin>
|
|
</Modal>
|
|
</div>
|
|
);
|
|
}
|
|
}
|
|
|
|
export default injectIntl(PicSearcher);
|