80 lines
3.4 KiB
TypeScript
80 lines
3.4 KiB
TypeScript
import { type ComponentInternalInstance, getCurrentInstance } from 'vue';
|
||
|
||
/**
|
||
* 将事件名转换为驼峰格式
|
||
* @param str 需要转换的字符串
|
||
* @description 例如：on-form-change -> onFormChange
|
||
* @returns
|
||
*/
|
||
function formatToCamelCase(str: string): string {
|
||
return str.replace(/-([a-z])/g, function (g) {
|
||
return g[1].toUpperCase();
|
||
});
|
||
}
|
||
|
||
export function useEmitter(name: string) {
|
||
const instance: ComponentInternalInstance | null | undefined = getCurrentInstance();
|
||
|
||
/** * 向上查找父组件并派发事件
|
||
* @param instance 当前组件实例（setup中可用getCurrentInstance()）
|
||
* @param componentName 目标组件名
|
||
* @param eventName 事件名
|
||
* @param params 参数
|
||
*/
|
||
function dispatch(componentName: string, eventName: string, ...params: any[]) {
|
||
let parent = instance && (instance.parent as ComponentInternalInstance | null | undefined);
|
||
while (parent) {
|
||
const name = (parent.type as any)?.name as string | undefined;
|
||
if (name === componentName) {
|
||
// 找到目标组件，派发事件
|
||
// Vue3未解决，目标组件事件监听失效，待优化，暂时使用下面的方式解决，如果你有好的方式也可以告诉我或者提PR
|
||
parent.emit && parent.emit(eventName, ...params);
|
||
// 如果有对应的方法，执行方法
|
||
// 这里可以考虑将 eventName 转换为驼峰格式
|
||
// 例如：on-form-change -> onFormChange
|
||
parent.exposed?.[formatToCamelCase(eventName)] &&
|
||
parent.exposed[formatToCamelCase(eventName)](...params);
|
||
break;
|
||
}
|
||
parent = parent.parent;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 向下递归查找子组件并广播事件
|
||
* @param instance 当前组件实例（setup中可用getCurrentInstance()）
|
||
* @param componentName 目标组件名
|
||
* @param eventName 事件名
|
||
* @param params 参数
|
||
*/
|
||
function broadcast(componentName: string, eventName: string, ...params: any[]) {
|
||
if (!instance) return;
|
||
const subTree = (instance.subTree as any)?.children || [];
|
||
const children = Array.isArray(subTree) ? subTree : [subTree];
|
||
children.forEach((vnode: any) => {
|
||
const child = vnode.component as ComponentInternalInstance | undefined;
|
||
|
||
if (child) {
|
||
const name = (child.type as any)?.name as string | undefined;
|
||
if (name === componentName) {
|
||
// 找到目标组件，广播事件
|
||
// Vue3未解决，目标组件事件监听失效，待优化，暂时使用下面的方式解决，如果你有好的方式也可以告诉我或者提PR
|
||
child.emit && child.emit(eventName, ...params);
|
||
// 如果有对应的方法，执行方法
|
||
// 这里可以考虑将 eventName 转换为驼峰格式
|
||
// 例如：on-form-change -> onFormChange
|
||
child.exposed?.[formatToCamelCase(eventName)] &&
|
||
child.exposed[formatToCamelCase(eventName)](...params);
|
||
} else {
|
||
broadcast.call(child, componentName, eventName, ...params);
|
||
}
|
||
}
|
||
});
|
||
}
|
||
|
||
return {
|
||
dispatch,
|
||
broadcast
|
||
};
|
||
}
|