feat: add ElementTemplates#applyTemplate API

src/provider/cloud-element-templates/ElementTemplates.js

@@ -9,9 +9,10 @@ import { default as DefaultElementTemplates } from '../element-templates/Element
  * Registry for element templates.
  */
 export default class ElementTemplates extends DefaultElementTemplates {
-  constructor(templateElementFactory) {
-    super();
-    this._templates = {};
+  constructor(templateElementFactory, commandStack) {
+    super(commandStack);
+
+    this._commandStack = commandStack;
     this._templateElementFactory = templateElementFactory;
   }
 
@@ -39,6 +40,26 @@ export default class ElementTemplates extends DefaultElementTemplates {
     return element;
   }
 
+  /**
+   * Apply element template to a given element.
+   *
+   * @param {djs.model.Base} element
+   * @param {ElementTemplate} newTemplate
+   *
+   * @return {djs.model.Base} the updated element
+   */
+  applyTemplate(element, newTemplate) {
+
+    const oldTemplate = this.get(element);
+
+    this._commandStack.execute('propertiesPanel.zeebe.changeTemplate', {
+      element: element,
+      newTemplate,
+      oldTemplate
+    });
+
+    return element;
+  }
 }
 
-ElementTemplates.$inject = [ 'templateElementFactory' ];
+ElementTemplates.$inject = [ 'templateElementFactory', 'commandStack' ];

src/provider/cloud-element-templates/util/templateUtil.js

@@ -13,14 +13,7 @@ export function unlinkTemplate(element, injector) {
 }
 
 export function updateTemplate(element, newTemplate, injector) {
-  const commandStack = injector.get('commandStack'),
-        elementTemplates = injector.get('elementTemplates');
+  const elementTemplates = injector.get('elementTemplates');
 
-  const oldTemplate = elementTemplates.get(element);
-
-  commandStack.execute('propertiesPanel.zeebe.changeTemplate', {
-    element: element,
-    newTemplate,
-    oldTemplate
-  });
+  return elementTemplates.applyTemplate(element, newTemplate);
 }

src/provider/element-templates/ElementTemplates.js

@@ -17,7 +17,9 @@ import { isAny } from 'bpmn-js/lib/features/modeling/util/ModelingUtil';
  * Registry for element templates.
  */
 export default class ElementTemplates {
-  constructor() {
+  constructor(commandStack) {
+    this._commandStack = commandStack;
+
     this._templates = {};
   }
 
@@ -109,4 +111,27 @@ export default class ElementTemplates {
   _getTemplateVersion(element) {
     return getTemplateVersion(element);
   }
+
+  /**
+   * Apply element template to a given element.
+   *
+   * @param {djs.model.Base} element
+   * @param {ElementTemplate} newTemplate
+   *
+   * @return {djs.model.Base} the updated element
+   */
+  applyTemplate(element, newTemplate) {
+
+    const oldTemplate = this.get(element);
+
+    this._commandStack.execute('propertiesPanel.camunda.changeTemplate', {
+      element: element,
+      newTemplate,
+      oldTemplate
+    });
+
+    return element;
+  }
 }
+
+ElementTemplates.$inject = [ 'commandStack' ];

src/provider/element-templates/util/templateUtil.js

@@ -30,16 +30,9 @@ export function removeTemplate(element, injector) {
 }
 
 export function updateTemplate(element, newTemplate, injector) {
-  const commandStack = injector.get('commandStack'),
-        elementTemplates = injector.get('elementTemplates');
+  const elementTemplates = injector.get('elementTemplates');
 
-  const oldTemplate = elementTemplates.get(element);
-
-  commandStack.execute('propertiesPanel.camunda.changeTemplate', {
-    element: element,
-    newTemplate,
-    oldTemplate
-  });
+  return elementTemplates.applyTemplate(element, newTemplate);
 }
 
 export function getVersionOrDateFromTemplate(template) {

test/spec/provider/cloud-element-templates/ElementTemplates.spec.js

@@ -1,5 +1,9 @@
 import TestContainer from 'mocha-test-container-support';
 
+import {
+  isAny
+} from 'bpmn-js/lib/util/ModelUtil';
+
 import {
   bootstrapModeler,
   createCanvasEvent as canvasEvent,
@@ -191,4 +195,29 @@ describe('provider/cloud-element-templates - ElementTemplates', function() {
 
   });
 
+
+  describe('applyTemplate', function() {
+
+    it('should set template on element', inject(function(elementRegistry, elementTemplates) {
+
+      // given
+      const task = elementRegistry.get('Task_1');
+
+      const template = elementTemplates.getAll().find(
+        t => isAny(task, t.appliesTo)
+      );
+
+      // assume
+      expect(template).to.exist;
+
+      // when
+      const updatedTask = elementTemplates.applyTemplate(task, template);
+
+      // then
+      expect(updatedTask).to.exist;
+      expect(elementTemplates.get(updatedTask)).to.equal(template);
+    }));
+
+  });
+
 });

test/spec/provider/element-templates/ElementTemplates.spec.js

@@ -1,5 +1,9 @@
 import TestContainer from 'mocha-test-container-support';
 
+import {
+  isAny
+} from 'bpmn-js/lib/util/ModelUtil';
+
 import { bootstrapModeler, inject } from 'test/TestHelper';
 
 import coreModule from 'bpmn-js/lib/core';
@@ -197,4 +201,29 @@ describe('provider/element-templates - ElementTemplates', function() {
 
   });
 
+
+  describe('applyTemplate', function() {
+
+    it('should set template on element', inject(function(elementRegistry, elementTemplates) {
+
+      // given
+      const task = elementRegistry.get('Task_1');
+
+      const template = elementTemplates.getAll().find(
+        t => isAny(task, t.appliesTo)
+      );
+
+      // assume
+      expect(template).to.exist;
+
+      // when
+      const updatedTask = elementTemplates.applyTemplate(task, template);
+
+      // then
+      expect(updatedTask).to.exist;
+      expect(elementTemplates.get(updatedTask)).to.equal(template);
+    }));
+
+  });
+
 });