ant-design-vue/components/_util/store/connect.jsx

101 lines
2.9 KiB
Vue
Raw Normal View History

2020-08-14 10:24:30 +00:00
import shallowEqual from '../shallowequal';
import { inject, createVNode, watchEffect } from 'vue';
2019-01-12 03:33:27 +00:00
import omit from 'omit.js';
2020-06-16 14:55:02 +00:00
import { getOptionProps } from '../props-util';
2018-03-25 10:07:04 +00:00
2019-01-12 03:33:27 +00:00
function getDisplayName(WrappedComponent) {
return WrappedComponent.name || 'Component';
2018-03-25 10:07:04 +00:00
}
2019-01-12 03:33:27 +00:00
const defaultMapStateToProps = () => ({});
export default function connect(mapStateToProps) {
const shouldSubscribe = !!mapStateToProps;
const finalMapStateToProps = mapStateToProps || defaultMapStateToProps;
2019-01-12 03:33:27 +00:00
return function wrapWithConnect(WrappedComponent) {
const tempProps = omit(WrappedComponent.props || {}, ['store']);
2020-06-20 13:45:08 +00:00
const props = {};
2019-01-12 03:33:27 +00:00
Object.keys(tempProps).forEach(k => {
props[k] = { ...tempProps[k], required: false };
2019-01-12 03:33:27 +00:00
});
2018-03-25 10:07:04 +00:00
const Connect = {
name: `Connect_${getDisplayName(WrappedComponent)}`,
2020-06-16 14:55:02 +00:00
inheritAttrs: false,
2018-03-25 10:07:04 +00:00
props,
2020-06-16 14:55:02 +00:00
setup() {
return {
storeContext: inject('storeContext', {}),
};
2018-03-25 10:07:04 +00:00
},
2019-01-12 03:33:27 +00:00
data() {
this.store = this.storeContext.store;
2020-06-20 13:45:08 +00:00
this.preProps = getOptionProps(this);
2020-06-18 15:50:02 +00:00
watchEffect(() => {
if (mapStateToProps && mapStateToProps.length === 2) {
this.subscribed = finalMapStateToProps(this.store.getState(), this.$props);
}
2020-06-18 15:50:02 +00:00
});
return {
subscribed: finalMapStateToProps(this.store.getState(), this.$props),
2020-06-18 15:50:02 +00:00
};
2018-03-25 10:07:04 +00:00
},
2019-01-12 03:33:27 +00:00
mounted() {
this.trySubscribe();
2018-03-25 10:07:04 +00:00
},
2020-06-11 08:13:09 +00:00
beforeUnmount() {
2019-01-12 03:33:27 +00:00
this.tryUnsubscribe();
2018-03-25 10:07:04 +00:00
},
methods: {
2019-01-12 03:33:27 +00:00
handleChange() {
2018-03-25 10:07:04 +00:00
if (!this.unsubscribe) {
2019-01-12 03:33:27 +00:00
return;
2018-03-25 10:07:04 +00:00
}
2020-06-18 15:50:02 +00:00
const props = getOptionProps(this);
const nextSubscribed = finalMapStateToProps(this.store.getState(), props);
2019-01-12 03:33:27 +00:00
if (
!shallowEqual(this.preProps, props) ||
2020-08-14 10:24:30 +00:00
!shallowEqual(this.subscribed, nextSubscribed)
2019-01-12 03:33:27 +00:00
) {
this.subscribed = nextSubscribed;
2018-03-25 10:07:04 +00:00
}
},
2019-01-12 03:33:27 +00:00
trySubscribe() {
2018-03-25 10:07:04 +00:00
if (shouldSubscribe) {
2019-01-12 03:33:27 +00:00
this.unsubscribe = this.store.subscribe(this.handleChange);
this.handleChange();
2018-03-25 10:07:04 +00:00
}
},
2019-01-12 03:33:27 +00:00
tryUnsubscribe() {
2018-03-25 10:07:04 +00:00
if (this.unsubscribe) {
2019-01-12 03:33:27 +00:00
this.unsubscribe();
this.unsubscribe = null;
2018-03-25 10:07:04 +00:00
}
},
2019-01-12 03:33:27 +00:00
getWrappedInstance() {
return this.$refs.wrappedInstance;
update to antd3.8.3 (#159) * refactor: align * feat: update align to 2.4.3 * feat: update trigger 2.5.4 * feat: update tooltip 3.7.2 * fix: align * feat: update vc-calendar to 9.6.2 * feat: update vc-checkbox to 2.1.5 * feat: update vc-dialog to 7.1.8 * feat: update vc-from to 2.2.1 * feat: update vc-notification to 3.1.1 * test: update snapshots * feat: update vc-tree to 1.12.6 * feat: update vc-table to 6.2.8 * feat: update vc-upload to 2.5.1 * feat: update vc-input-number to 4.0.12 * feat: update vc-tabs to 9.2.6 * refactor: vc-menu * refactor: update vc-menu to 7.0.5 * style: remove unused * feat: update pagination to 1.16.5 * feat: add vc-progress 2.2.5 tag * feat: add vc-rate 2.4.0 tag * feat: update vc-slider to 8.6.1 * fix: tooltip error * style: delete conosle * feat: update vc-steps to 3.1.1 * add vc-switch tag 1.6.0 * feat: update upload to 2.5.1 * fix: update vc-menu * fix: update store * fix: add ref dir * fix: trigger mock shouldComponentUpdate * fix: update vc-select * revert: trigger lazyrenderbox * fix: update vc-select * fix: update vc-select * fix: update vc-select * fix: update vc-menu * fix: update vc-slick ref * update style to 3.8.2 * test: update snapshots * update vc-select * update util & affix * feat: add drawer * fix: support title add slot mode * test: update affix test * update alert * update anchor * update snapshots * fix: doc and vc-drawer * update select & auto-complete * update back-top & grid * feractor: avatar * test: add drawer test * update badge * update button * update card * update divider * feat: update vc-tabs to 9.3.6 and tabs * add afterEnter callback * update form * fix: update drawer * test: update snapshots * update modal & notification * test: update snapshots * update message * update locale-provider * update dropdown * update layout popconfirm popover * update time-picker * update menu * update date-picker * docs: update input docs * update input * update snapshots * update table * update test snapshots * feat: update progress * update checkbox * feat: update spin * update radio * docs: slider steps timeline * update list * update transfer * update collapse * update cascader * update upload
2018-09-05 13:28:54 +00:00
},
2018-03-25 10:07:04 +00:00
},
2019-01-12 03:33:27 +00:00
render() {
2020-06-16 14:55:02 +00:00
const { $slots = {}, subscribed, store, $attrs } = this;
2019-01-12 03:33:27 +00:00
const props = getOptionProps(this);
2020-06-18 15:50:02 +00:00
this.preProps = { ...props };
2018-03-25 10:07:04 +00:00
const wrapProps = {
2020-06-16 14:55:02 +00:00
...props,
...subscribed,
...$attrs,
store,
ref: 'wrappedInstance',
2019-01-12 03:33:27 +00:00
};
2020-07-23 10:34:24 +00:00
// const slots = {};
// for (let [key, value] of Object.entries($slots)) {
// slots[key] = () => value();
// }
return createVNode(WrappedComponent, wrapProps, $slots);
2018-03-25 10:07:04 +00:00
},
2019-01-12 03:33:27 +00:00
};
2020-06-18 15:50:02 +00:00
return Connect;
2019-01-12 03:33:27 +00:00
};
2018-03-25 10:07:04 +00:00
}