Events as a function

medic · Oct 12, 2023 · 00de5c5 · 00de5c5
1 parent f54486e
commit 00de5c5
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 7 deletions.
diff --git a/src/lib/validate-declarative-schema.js b/src/lib/validate-declarative-schema.js
@@ -117,11 +117,14 @@ const TaskSchema = joi.array().items(
           'should define property "resolvedIf" as: function(contact, report) { ... }.'
         )
       ),
-    events: joi.alternatives().conditional('events', {
-      is: joi.array().length(1),
-      then: joi.array().items(EventSchema('optional')).min(1).required(),
-      otherwise: joi.array().items(EventSchema('required')).unique('id').required(),
-    }),
+    events: joi.alternatives().try(
+      joi.function(),
+      joi.alternatives().conditional('events', {
+        is: joi.array().length(1),
+        then: joi.array().items(EventSchema('optional')).min(1),
+        otherwise: joi.array().items(EventSchema('required')).unique('id'),
+      })
+    ).required(),
     priority: joi.alternatives().try(
       joi.object({
         level: joi.string().valid('high', 'medium').optional(),

diff --git a/src/nools/task-emitter.js b/src/nools/task-emitter.js
@@ -90,8 +90,19 @@ function emitTasks(taskDefinition, Utils, Task, emit, c, r) {
 
   function emitForEvents(scheduledTaskIdx) {
     var i, dueDate = null, event, priority, task;
-    for (i = 0; i < taskDefinition.events.length; i++) {
-      event = taskDefinition.events[i];
+
+    var events;
+    if (typeof taskDefinition.events === 'function') {
+      events = taskDefinition.events(c, r);
+      if (!Array.isArray(events)) {
+        throw Error('events did not return an array');
+      }
+    } else {
+      events = taskDefinition.events;
+    }
+
+    for (i = 0; i < events.length; i++) {
+      event = events[i];
 
       if (event.dueDate) {
         dueDate = event.dueDate(event, c, r, scheduledTaskIdx);

diff --git a/test/lib/validate-declarative-schema.spec.js b/test/lib/validate-declarative-schema.spec.js
@@ -123,6 +123,13 @@ describe('validate-declarative-schema', () => {
       ]);
     });
 
+    it('events can be function', () => {
+      const task = buildTaskWithAction('contact');
+      task.events = () => [];
+      const actual = validate([task], TaskSchema);
+      expect(actual).to.be.empty;
+    });
+
     it('array.unique internal', () => {
       const schema = joi.array().items(joi.object({
         event: joi.array().items(joi.object()).unique('id'),