tf-leaflet-freedraw/src/helpers/Flags.js

Zoopla inspired freehand polygon creation using Leaflet.js.

import { DomUtil } from 'leaflet';
import { edgesKey, modesKey, instanceKey, notifyDeferredKey, polygons } from '../FreeDraw';
import { classesFor } from './Layer';

/**
 * @constant NONE
 * @type {Number}
 */
export const NONE = 0;

/**
 * @constant CREATE
 * @type {Number}
 */
export const CREATE = 1;

/**
 * @constant EDIT
 * @type {Number}
 */
export const EDIT = 2;

/**
 * @constant DELETE
 * @type {Number}
 */
export const DELETE = 4;

/**
 * @constant APPEND
 * @type {Number}
 */
export const APPEND = 8;

/**
 * @constant EDIT_APPEND
 * @type {Number}
 */
export const EDIT_APPEND = EDIT | APPEND;

/**
 * @constant ALL
 * @type {number}
 */
export const ALL = CREATE | EDIT | DELETE | APPEND;

/**
 * @method modeFor
 * @param {Object} map
 * @param {Number} mode
 * @param {Object} options
 * @return {Number}
 */
export const modeFor = (map, mode, options) => {

    // Update the mode.
    map[modesKey] = mode;

    // Fire the updated mode.
    map[instanceKey].fire('mode', { mode });

    // Disable the map if the `CREATE` mode is a default flag.
    if (mode & CREATE) {
        map.dragging.disable();
        map._container.style['touch-action'] = 'none';
    } else {
        map.dragging.enable();
        map._container.style.removeProperty('touch-action');
    }

    Array.from(polygons.get(map)).forEach(polygon => {

        polygon[edgesKey].forEach(edge => {

            // Modify the edge class names based on whether edit mode is enabled.
            mode & EDIT ? DomUtil.removeClass(edge._icon, 'disabled') : DomUtil.addClass(edge._icon, 'disabled');

        });

    });

    // Apply the conditional class names to the map container.
    classesFor(map, mode);

    // Fire the event for having manipulated the polygons if the `hasManipulated` is `true` and the
    // `notifyAfterEditExit` option is equal to `true`, and then reset the `notifyDeferredKey`.
    options.notifyAfterEditExit && map[notifyDeferredKey]();
    map[notifyDeferredKey] = () => {};

    return mode;

};