camunda-bpmn-js-behaviors/lib/camunda-cloud/CleanUpTimerExpressionBehavior.js

Camunda Platform 7 and 8 behaviors for bpmn-js

import {
  is
} from 'bpmn-js/lib/util/ModelUtil';

import CommandInterceptor from 'diagram-js/lib/command/CommandInterceptor';

import {
  getTimerEventDefinition,
  isTimerExpressionTypeSupported,
  TIMER_PROPERTIES
} from './util/TimerUtil';

/**
 * Zeebe BPMN behavior ensuring that bpmn:TimerEventDefinition has only allowed time properties of:
 * - timeCycle
 * - timeDate
 * - timeDuration
 */
export default class CleanUpTimerExpressionBehavior extends CommandInterceptor {
  constructor(eventBus, modeling) {
    super(eventBus);

    this._modeling = modeling;

    this.postExecuted([
      'shape.move',
      'shape.replace',
      'shape.create',
      'element.updateProperties',
      'element.updateModdleProperties'
    ], context => {
      const element = context.shape || context.newShape || context.element;
      this.cleanUpTimerProperties(element);
    }, true);
  }

  /**
  * Remove unsupported timer properties.
  */
  cleanUpTimerProperties(element) {

    if (!is(element, 'bpmn:Event')) {
      return;
    }

    const timerEventDefinition = getTimerEventDefinition(element);

    if (!timerEventDefinition) {
      return;
    }

    const propertiesUpdate = TIMER_PROPERTIES.reduce((acc, type) => {
      if (timerEventDefinition.get(type) && !isTimerExpressionTypeSupported(type, element)) {
        acc[type] = undefined;
      }
      return acc;
    }, {});

    if (Object.keys(propertiesUpdate).length) {
      this._modeling.updateModdleProperties(element, timerEventDefinition, propertiesUpdate);
    }
  }
}

CleanUpTimerExpressionBehavior.$inject = [
  'eventBus',
  'modeling'
];