2020-06-10 18:21:16 +08:00
|
|
|
|
import { filterEmpty } from './props-util';
|
2022-03-20 10:57:44 +08:00
|
|
|
|
import type { VNode, VNodeProps } from 'vue';
|
2020-10-02 15:44:10 +08:00
|
|
|
|
import { cloneVNode } from 'vue';
|
2020-07-20 16:29:46 +08:00
|
|
|
|
import warning from './warning';
|
2022-03-20 10:57:44 +08:00
|
|
|
|
import type { RefObject } from './createRef';
|
2022-12-15 10:28:33 +08:00
|
|
|
|
type NodeProps = Record<string, any> &
|
|
|
|
|
Omit<VNodeProps, 'ref'> & { ref?: VNodeProps['ref'] | RefObject };
|
2019-07-24 19:21:27 +08:00
|
|
|
|
|
2022-03-20 10:57:44 +08:00
|
|
|
|
export function cloneElement<T, U>(
|
|
|
|
|
vnode: VNode<T, U> | VNode<T, U>[],
|
2022-12-15 10:28:33 +08:00
|
|
|
|
nodeProps: NodeProps = {},
|
2022-03-20 10:57:44 +08:00
|
|
|
|
override = true,
|
|
|
|
|
mergeRef = false,
|
|
|
|
|
): VNode<T, U> {
|
2020-07-06 22:31:07 +08:00
|
|
|
|
let ele = vnode;
|
|
|
|
|
if (Array.isArray(vnode)) {
|
|
|
|
|
ele = filterEmpty(vnode)[0];
|
2018-01-29 18:57:20 +08:00
|
|
|
|
}
|
|
|
|
|
if (!ele) {
|
2019-01-12 11:33:27 +08:00
|
|
|
|
return null;
|
2018-01-29 18:57:20 +08:00
|
|
|
|
}
|
2022-03-20 10:57:44 +08:00
|
|
|
|
const node = cloneVNode(ele as VNode<T, U>, nodeProps as any, mergeRef);
|
2018-03-03 19:14:03 +08:00
|
|
|
|
|
2020-06-10 18:21:16 +08:00
|
|
|
|
// cloneVNode内部是合并属性,这里改成覆盖属性
|
2022-03-20 10:57:44 +08:00
|
|
|
|
node.props = (override ? { ...node.props, ...nodeProps } : node.props) as any;
|
2020-10-02 15:44:10 +08:00
|
|
|
|
warning(typeof node.props.class !== 'object', 'class must be string');
|
2019-01-12 11:33:27 +08:00
|
|
|
|
return node;
|
2017-12-14 12:13:15 +08:00
|
|
|
|
}
|
2020-07-06 22:31:07 +08:00
|
|
|
|
|
2020-10-02 15:44:10 +08:00
|
|
|
|
export function cloneVNodes(vnodes, nodeProps = {}, override = true) {
|
2020-07-06 22:31:07 +08:00
|
|
|
|
return vnodes.map(vnode => cloneElement(vnode, nodeProps, override));
|
|
|
|
|
}
|
2022-12-15 10:28:33 +08:00
|
|
|
|
|
|
|
|
|
export function deepCloneElement<T, U>(
|
|
|
|
|
vnode: VNode<T, U> | VNode<T, U>[],
|
|
|
|
|
nodeProps: NodeProps = {},
|
|
|
|
|
override = true,
|
|
|
|
|
mergeRef = false,
|
|
|
|
|
) {
|
|
|
|
|
if (Array.isArray(vnode)) {
|
|
|
|
|
return vnode.map(item => deepCloneElement(item, nodeProps, override, mergeRef));
|
|
|
|
|
} else {
|
|
|
|
|
const cloned = cloneElement(vnode, nodeProps, override, mergeRef);
|
|
|
|
|
if (Array.isArray(cloned.children)) {
|
|
|
|
|
cloned.children = deepCloneElement(cloned.children as VNode<T, U>[]);
|
|
|
|
|
}
|
|
|
|
|
return cloned;
|
|
|
|
|
}
|
|
|
|
|
}
|