feat: add config for engines

bpmn-io · Dec 6, 2024 · 71b86f9 · barmac · Dec 6, 2024 · 71b86f9
1 parent c585fd4
commit 71b86f9
Show file tree

Hide file tree

Showing 3 changed files with 43 additions and 7 deletions.
diff --git a/src/cloud-element-templates/ElementTemplates.js b/src/cloud-element-templates/ElementTemplates.js
@@ -1,3 +1,8 @@
+import {
+  isArray,
+  isFunction
+} from 'min-dash';
+
 import {
   getTemplateId,
   getTemplateVersion
@@ -9,10 +14,17 @@ import { default as DefaultElementTemplates } from '../element-templates/Element
  * Registry for element templates.
  */
 export default class ElementTemplates extends DefaultElementTemplates {
-  constructor(templateElementFactory, commandStack, eventBus, modeling, injector) {
+  constructor(templateElementFactory, commandStack, eventBus, modeling, injector, config) {
     super(commandStack, eventBus, modeling, injector);
 
     this._templateElementFactory = templateElementFactory;
+    this._config = {};
+    this._engines = {};
+
+    if (!(isArray(this._config) || isFunction(this._config))) {
+      this._config = config || {};
+      this._engines = this._config.engines;
+    }
   }
 
   _getTemplateId(element) {
@@ -75,12 +87,21 @@ export default class ElementTemplates extends DefaultElementTemplates {
 
     return context.element;
   }
+
+  getEngines() {
+    return this._engines;
+  }
+
+  setEngines(engines) {
+    this._engines = engines;
+  }
 }
 
 ElementTemplates.$inject = [
   'templateElementFactory',
   'commandStack',
   'eventBus',
   'modeling',
-  'injector'
+  'injector',
+  'config.elementTemplates',
 ];
diff --git a/src/cloud-element-templates/ElementTemplatesLoader.js b/src/cloud-element-templates/ElementTemplatesLoader.js
@@ -2,10 +2,16 @@ import { Validator } from './Validator';
 
 import { default as TemplatesLoader } from '../element-templates/ElementTemplatesLoader';
 
+/**
+ * @param {Object|Array<TemplateDescriptor>|Function} config
+ * @param {EventBus} eventBus
+ * @param {ElementTemplates} elementTemplates
+ * @param {Moddle} moddle
+ */
 export default class ElementTemplatesLoader extends TemplatesLoader {
-  constructor(loadTemplates, eventBus, elementTemplates, moddle) {
+  constructor(config, eventBus, elementTemplates, moddle) {
 
-    super(loadTemplates, eventBus, elementTemplates, moddle);
+    super(config, eventBus, elementTemplates, moddle);
 
     this._elementTemplates = elementTemplates;
   }

diff --git a/src/element-templates/ElementTemplatesLoader.js b/src/element-templates/ElementTemplatesLoader.js
@@ -1,6 +1,7 @@
 import {
   isFunction,
-  isUndefined
+  isUndefined,
+  isArray,
 } from 'min-dash';
 
 import { Validator } from './Validator';
@@ -20,12 +21,20 @@ import { Validator } from './Validator';
  * @param {Moddle} moddle
  */
 export default class ElementTemplatesLoader {
-  constructor(loadTemplates, eventBus, elementTemplates, moddle) {
-    this._loadTemplates = loadTemplates;
+  constructor(config, eventBus, elementTemplates, moddle) {
+    this._loadTemplates;
     this._eventBus = eventBus;
     this._elementTemplates = elementTemplates;
     this._moddle = moddle;
 
+    if (isArray(config) || isFunction(config)) {
+      this._loadTemplates = config;
+    }
+
+    if (config && config.loadTemplates) {
+      this._loadTemplates = config.loadTemplates;
+    }
+
     eventBus.on('diagram.init', () => {
       this.reload();
     });