ant-design/components/table/SelectionBox.tsx

71 lines
1.6 KiB
TypeScript
Raw Normal View History

import * as React from 'react';
import Checkbox from '../checkbox';
import Radio from '../radio';
2017-11-21 13:48:37 +08:00
import { SelectionBoxProps, SelectionBoxState } from './interface';
2017-11-21 13:48:37 +08:00
export default class SelectionBox extends React.Component<SelectionBoxProps, SelectionBoxState> {
unsubscribe: () => void;
2017-11-21 13:48:37 +08:00
constructor(props: SelectionBoxProps) {
super(props);
this.state = {
checked: this.getCheckState(props),
};
}
componentDidMount() {
this.subscribe();
}
componentWillUnmount() {
if (this.unsubscribe) {
this.unsubscribe();
}
}
subscribe() {
const { store } = this.props;
this.unsubscribe = store.subscribe(() => {
const checked = this.getCheckState(this.props);
2016-11-28 13:35:05 +08:00
this.setState({ checked });
});
}
2017-11-21 13:48:37 +08:00
getCheckState(props: SelectionBoxProps) {
const { store, defaultSelection, rowIndex } = props;
let checked = false;
if (store.getState().selectionDirty) {
checked = store.getState().selectedRowKeys.indexOf(rowIndex) >= 0;
} else {
checked = (store.getState().selectedRowKeys.indexOf(rowIndex) >= 0 ||
defaultSelection.indexOf(rowIndex) >= 0);
}
return checked;
}
render() {
const { type, rowIndex, disabled, onChange } = this.props;
const { checked } = this.state;
if (type === 'radio') {
return (
<Radio
disabled={disabled}
onChange={onChange}
value={rowIndex}
checked={checked}
/>
);
}
return (
<Checkbox
checked={checked}
disabled={disabled}
onChange={onChange}
/>
);
}
}