@dhis2/gis-api/src/layers/FeatureGroup.js

Maps API for DHIS2 based on Leaflet

import L from 'leaflet';
import layerMixin from './layerMixin';
import { Buffer } from './Buffer';
import { LabelGroup } from './LabelGroup';
import { toLngLatBounds } from '../utils/geometry';

// Layer group with support for labels and buffers
export const FeatureGroup = L.FeatureGroup.extend({
    ...layerMixin,

    initialize(options) {
        L.FeatureGroup.prototype.initialize.call(this, null, {
            ...options,
            pane: options.id,
        });
    },

    addBuffers() {
        const { buffer, bufferStyle, data, pane, opacity, isVisible } = this.options;

        if (buffer) {
            const bufferLayer = new Buffer({
                pane: `${pane}-buffer`,
                buffer,
                isVisible,
                opacity,
                ...(bufferStyle && { style: bufferStyle }),
                data,
            });

            this.addLayer(bufferLayer);
        }
    },

    addLabels() {
        const { data, label, labelStyle, pane } = this.options;

        if (label) {
            this.addLayer(
                new LabelGroup({
                    pane: `${pane}-label`,
                    style: labelStyle,
                    label,
                    data,
                })
            );
        }
    },

    createPane(map) {
        this.invoke('createPane', map);
    },

    setIndex(index) {
        const zIndex = 200 + (index * 10);

        this.getLayers().forEach((layer, i) => {
            layer.getPane().style.zIndex = zIndex + i;
        });

        this.options.index = index;
    },

    setOpacity(opacity) {
        this.invoke('setOpacity', opacity);
    },

    setVisibility(isVisible) {
        this.invoke('setVisibility', isVisible);
    },

    // Convert bounds before returning
    getBounds() {
        const bounds = L.FeatureGroup.prototype.getBounds.call(this);

        if (bounds.isValid()) {
            return toLngLatBounds(bounds);
        }
    },

    onAdd(map) {
        L.FeatureGroup.prototype.onAdd.call(this, map);
        this.setIndex(this.options.index);
    },
});

export default FeatureGroup;