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
    };
}