import React, { Component } from 'react';
import { Tooltip } from 'antd';
import classNames from 'classnames';
import styles from './index.less';
/* eslint react/no-did-mount-set-state: 0 */
/* eslint no-param-reassign: 0 */
const EllipsisText = ({ text, length, tooltip, ...other }) => {
if (typeof text !== 'string') {
throw new Error('Ellipsis children must be string.');
}
if (text.length <= length || length < 0) {
return {text};
}
const tail = '...';
let displayText;
if (length - tail.length <= 0) {
displayText = '';
} else {
displayText = text.slice(0, (length - tail.length));
}
if (tooltip) {
return {displayText}{tail};
}
return (
{displayText}{tail}
);
};
export default class Ellipsis extends Component {
state = {
lineHeight: 0,
text: '',
targetCount: 0,
}
componentDidMount() {
const { lines, cover } = this.props;
if (this.node) {
if (lines && cover) {
this.setState({
lineHeight: parseInt(window.getComputedStyle(this.node).lineHeight, 10),
});
}
this.computeLine();
}
}
componentWillReceiveProps(nextProps) {
if (this.props.lines !== nextProps.lines || this.props.cover !== nextProps.cover) {
this.setState({
lineHeight: parseInt(window.getComputedStyle(this.node).lineHeight, 10),
});
this.computeLine();
}
}
computeLine = () => {
const { lines, cover } = this.props;
if (lines && !cover) {
const fontSize = parseInt(window.getComputedStyle(this.node).fontSize, 10) || 14;
const text = this.shadowChildren.innerText;
const targetWidth = (this.node.offsetWidth || this.node.parentNode.offsetWidth) * lines;
const shadowNode = this.shadow.firstChild;
// bisection
const tw = (targetWidth - (lines * (fontSize / 2)) - fontSize);
const len = text.length;
const mid = Math.floor(len / 2);
const count = this.bisection(tw, mid, 0, len, text, shadowNode);
this.setState({
text,
targetCount: count,
});
}
}
bisection = (tw, m, b, e, text, shadowNode) => {
let mid = m;
let end = e;
let begin = b;
shadowNode.innerHTML = text.substring(0, mid);
let sw = shadowNode.offsetWidth;
if (sw < tw) {
shadowNode.innerHTML = text.substring(0, mid + 1);
sw = shadowNode.offsetWidth;
if (sw >= tw) {
return mid;
} else {
begin = mid;
mid = Math.floor((end - begin) / 2) + begin;
return this.bisection(tw, mid, begin, end, text, shadowNode);
}
} else {
if (mid - 1 < 0) {
return mid;
}
shadowNode.innerHTML = text.substring(0, mid - 1);
sw = shadowNode.offsetWidth;
if (sw <= tw) {
return mid;
} else {
end = mid;
mid = Math.floor((end - begin) / 2) + begin;
return this.bisection(tw, mid, begin, end, text, shadowNode);
}
}
}
handleRef = (n) => {
this.node = n;
}
handleShadow = (n) => {
this.shadow = n;
}
handleShadowChildren = (n) => {
this.shadowChildren = n;
}
render() {
const { text, targetCount, lineHeight } = this.state;
const {
children,
lines,
length,
cover = false,
suffixColor = '#fff',
suffixOffset = 0,
className,
tooltip,
...restProps
} = this.props;
const cls = classNames(styles.ellipsis, className, {
[styles.lines]: (lines && !cover),
[styles.linesCover]: (lines && cover),
});
if (!lines && !length) {
return ({children});
}
// length
if (!lines) {
return ();
}
// lines cover
if (cover) {
const id = `antd-pro-ellipsis-${`${new Date().getTime()}${Math.floor(Math.random() * 100)}`}`;
const style = `#${id}:before{background-color:${suffixColor};padding-left:${suffixOffset}px;}`;
return (
{children}
);
}
// lines no cover
const suffix = tooltip ? ... : '...';
return (
{
(targetCount > 0) && text.substring(0, targetCount)
}
{
(targetCount > 0) && (targetCount < text.length) && suffix
}
{children}
{text}
);
}
}