ant-design-vue/components/table/createBodyRow.tsx
2020-10-23 14:29:39 +08:00

74 lines
1.7 KiB
TypeScript

import PropTypes from '../_util/vue-types';
import { defineComponent } from 'vue';
import { Store } from './createStore';
import { getSlot } from '../_util/props-util';
import omit from 'omit.js';
const BodyRowProps = {
store: Store,
rowKey: PropTypes.oneOfType([PropTypes.string, PropTypes.number]),
prefixCls: PropTypes.string,
};
export default function createBodyRow(Component = 'tr') {
const BodyRow = defineComponent({
name: 'BodyRow',
inheritAttrs: false,
props: BodyRowProps,
data() {
const { selectedRowKeys } = this.store.getState();
return {
selected: selectedRowKeys.indexOf(this.rowKey) >= 0,
};
},
setup() {
return {
unsubscribe: null,
};
},
mounted() {
this.subscribe();
},
beforeUnmount() {
if (this.unsubscribe) {
this.unsubscribe();
}
},
methods: {
subscribe() {
const { store, rowKey } = this;
this.unsubscribe = store.subscribe(() => {
const { selectedRowKeys } = this.store.getState();
const selected = selectedRowKeys.indexOf(rowKey) >= 0;
if (selected !== this.selected) {
this.selected = selected;
}
});
},
},
render() {
const rowProps = omit({ ...this.$props, ...this.$attrs }, [
'prefixCls',
'rowKey',
'store',
'class',
]);
const className = {
[`${this.prefixCls}-row-selected`]: this.selected,
[this.$attrs.class as string]: !!this.$attrs.class,
};
return (
<Component class={className} {...rowProps}>
{getSlot(this)}
</Component>
);
},
});
return BodyRow;
}