在Leaflet.PM中为自定义控件添加徽标的方法

前言

Leaflet.PM是一个强大的Leaflet插件，用于在地图上进行几何图形的绘制和编辑。在实际开发中，我们经常需要为工具栏中的自定义控件添加额外的视觉元素，比如徽标(badge)来提示用户新功能或重要操作。

两种实现方法

方法一：重写工具栏显示逻辑

第一种方法是通过重写Leaflet.PM工具栏的_showHideButtons方法，在控件渲染后动态添加徽标元素：

// 重写工具栏显示方法
map.pm.Toolbar._showHideButtons = function () {
    // 调用原始方法
    L.PM.Toolbar.prototype._showHideButtons.call(this);
    
    // 获取所有需要添加徽标的元素
    const btns = [...document.querySelectorAll('.newBadget')];
    
    // 为每个元素添加徽标样式和内容
    btns.forEach((btn)=>{
        btn.style.width = '35px';
        btn.style.height = '15px';
        btn.style.lineHeight = '15px';
        btn.style.marginLeft = '30px';
        btn.style.background = 'green';
        btn.innerHTML = '<span style="color: #fff">New</span>'
    })
}

// 创建自定义控件时指定类名
map.pm.Toolbar.createCustomControl({
    name: 'export',
    block: 'custom',
    title: 'Export',
    toggle: false,
    className: 'newBadget',
});

这种方法的核心思路是：

1. 继承并扩展原始工具栏的显示逻辑
2. 在控件渲染完成后，通过DOM操作添加徽标元素
3. 通过CSS样式控制徽标的外观和位置

方法二：利用MutationObserver监听DOM变化

第二种方法更加现代化，利用了MutationObserver API来监听DOM变化，在控件渲染完成后添加样式类：

// 创建自定义控件并指定类名
map.pm.Toolbar.createCustomControl({
    name: 'export',
    block: 'custom',
    title: 'Export',
    toggle: false,
    onClick: () => tools.export?.handler(),
    className: `${cc([styles.exportIcon, 'custom-export-control'])}`,
    actions: [{ text: 'New', onClick: () => tools.export?.handler() }],
});

// 使用MutationObserver监听工具栏DOM变化
const toolbar = document.querySelector('.leaflet-top.leaflet-left')
const observer = new MutationObserver(() => {
    const customControl = toolbar?.querySelector('.custom-export-control')
    const buttonContainer = customControl?.closest('div.button-container')
    buttonContainer?.classList.add('active', 'export-button-container')
    observer.disconnect()
})

if (toolbar) {
    observer.observe(toolbar, { childList: true, subtree: true })
}

这种方法的特点：

1. 使用MutationObserver监听DOM变化，更加可靠
2. 在控件渲染完成后添加样式类，避免直接操作样式
3. 可以结合CSS预处理器编写更复杂的样式

技术要点分析

1. DOM操作时机：两种方法都解决了控件渲染完成后的DOM操作问题，这是实现徽标的关键。

2. 样式控制：第一种方法直接操作内联样式，第二种方法通过添加类名实现样式控制，后者更符合现代前端开发的最佳实践。

3. 浏览器兼容性：MutationObserver在现代浏览器中支持良好，如果需要支持老旧浏览器，第一种方法可能更可靠。

4. 性能考虑：MutationObserver在监听大量DOM变化时可能会有性能影响，但在这个场景下影响可以忽略。

实际应用建议

在实际项目中，建议：

1. 如果项目已经使用了CSS模块化方案，优先考虑第二种方法，保持样式的一致性。

2. 如果需要快速实现且不涉及复杂样式，第一种方法更加简单直接。

3. 考虑徽标的交互功能，是否需要点击事件，以及如何与原有控件功能协调。

4. 对于复杂的徽标需求，可以考虑封装成可复用的组件或高阶函数。

通过这两种方法，开发者可以灵活地为Leaflet.PM的自定义控件添加各种视觉提示元素，提升用户体验和界面友好度。