Skip to content

Commit

Permalink
feat: filtering logic
Browse files Browse the repository at this point in the history
  • Loading branch information
cpvalente committed Jan 12, 2025
1 parent 74026f0 commit e707826
Show file tree
Hide file tree
Showing 4 changed files with 214 additions and 24 deletions.
Original file line number Diff line number Diff line change
@@ -1,13 +1,15 @@
import { TimerLifeCycle } from 'ontime-types';
import { PlayableEvent, TimerLifeCycle } from 'ontime-types';

import { deleteAllAutomations, addAutomation, addBlueprint } from '../automation.dao.js';
import { triggerAutomations } from '../automation.service.js';

import { makeOSCAction, makeHTTPAction } from './testUtils.js';
import { makeRuntimeStateData } from '../../../stores/__mocks__/runtimeState.mocks.js';
import { makeOntimeEvent } from '../../../services/rundown-service/__mocks__/rundown.mocks.js';

import { deleteAllAutomations, addAutomation, addBlueprint } from '../automation.dao.js';
import { testConditions, triggerAutomations } from '../automation.service.js';
import * as oscClient from '../clients/osc.client.js';
import * as httpClient from '../clients/http.client.js';

import { makeOSCAction, makeHTTPAction } from './testUtils.js';

beforeAll(() => {
vi.mock('../../../classes/data-provider/DataProvider.js', () => {
// Create a small mock store
Expand Down Expand Up @@ -67,26 +69,196 @@ describe('triggerAction()', () => {
});

it('should trigger automations for a given action', () => {
triggerAutomations(TimerLifeCycle.onLoad, {});
const state = makeRuntimeStateData();
triggerAutomations(TimerLifeCycle.onLoad, state);
expect(oscSpy).toHaveBeenCalledTimes(1);
expect(httpSpy).not.toBeCalled();
oscSpy.mockReset();
httpSpy.mockReset();

triggerAutomations(TimerLifeCycle.onStart, {});
triggerAutomations(TimerLifeCycle.onStart, state);
expect(oscClient.emitOSC).not.toBeCalled();
expect(httpSpy).not.toBeCalled();
oscSpy.mockReset();
httpSpy.mockReset();

triggerAutomations(TimerLifeCycle.onFinish, {});
triggerAutomations(TimerLifeCycle.onFinish, state);
expect(oscSpy).not.toBeCalled();
expect(httpSpy).toHaveBeenCalledTimes(1);
oscSpy.mockReset();
httpSpy.mockReset();

triggerAutomations(TimerLifeCycle.onStop, {});
triggerAutomations(TimerLifeCycle.onStop, state);
expect(oscSpy).not.toBeCalled();
expect(httpSpy).not.toBeCalled();
});
});

describe('testConditions()', () => {
it('should return true when no filters are provided', () => {
const result = testConditions([], 'all', {});
expect(result).toBe(true);
});

it('should compare two equal values', () => {
const mockStore = makeRuntimeStateData({ clock: 10 });
const result = testConditions([{ field: 'clock', operator: 'equals', value: '10' }], 'all', mockStore);
expect(result).toBe(true);
});

it('should check if a value does not exist', () => {
const mockStore = makeRuntimeStateData({ eventNow: null });
const result = testConditions([{ field: 'eventNow.title', operator: 'equals', value: '' }], 'all', mockStore);
expect(result).toBe(true);
});

it('should check if two values are different', () => {
const mockStore = makeRuntimeStateData({ clock: 10 });
const result = testConditions([{ field: 'clock', operator: 'not_equals', value: '11' }], 'all', mockStore);
expect(result).toBe(true);
});

it('should check if the given value is smaller', () => {
const mockStore = makeRuntimeStateData({ clock: 10 });
const result = testConditions([{ field: 'clock', operator: 'greater_than', value: '9' }], 'all', mockStore);
expect(result).toBe(true);
});

it('should check if the given value is larger', () => {
const mockStore = makeRuntimeStateData({ clock: 10 });
const result = testConditions([{ field: 'clock', operator: 'less_than', value: '11' }], 'all', mockStore);
expect(result).toBe(true);
});

it('should check if value contains given string', () => {
const result = testConditions(
[{ field: 'eventNow.title', operator: 'contains', value: 'lighting' }],
'all',
makeRuntimeStateData({ eventNow: makeOntimeEvent({ title: 'test-lighting' }) as PlayableEvent }),
);
expect(result).toBe(true);

const result2 = testConditions(
[{ field: 'eventNow.title', operator: 'contains', value: 'sound' }],
'all',
makeRuntimeStateData({ eventNow: makeOntimeEvent({ title: 'test-lighting' }) as PlayableEvent }),
);
expect(result2).toBe(false);
});

it('should check if value does not contain given string', () => {
const result = testConditions(
[{ field: 'eventNow.title', operator: 'not_contains', value: 'lighting' }],
'all',
makeRuntimeStateData({ eventNow: makeOntimeEvent({ title: 'test-lighting' }) as PlayableEvent }),
);
expect(result).toBe(false);

const result2 = testConditions(
[{ field: 'eventNow.title', operator: 'not_contains', value: 'sound' }],
'all',
makeRuntimeStateData({ eventNow: makeOntimeEvent({ title: 'test-lighting' }) as PlayableEvent }),
);
expect(result2).toBe(true);
});

describe('for all filter rule', () => {
it('should return true when all filters are true', () => {
const mockStore = makeRuntimeStateData({
clock: 10,
eventNow: makeOntimeEvent({
title: 'test',
custom: { av: 'av-value' },
}) as PlayableEvent,
});
const result = testConditions(
[
{ field: 'clock', operator: 'equals', value: '10' },
{ field: 'eventNow.title', operator: 'equals', value: 'test' },
{ field: 'eventNow.custom.av', operator: 'equals', value: 'av-value' },
],
'all',
mockStore,
);
expect(result).toBe(true);
});

it('should return false if any filters are false', () => {
const mockStore = makeRuntimeStateData({
clock: 10,
eventNow: makeOntimeEvent({
title: 'test',
custom: { av: 'not-av-value' },
}) as PlayableEvent,
});
const result = testConditions(
[
{ field: 'clock', operator: 'equals', value: '11' },
{ field: 'eventNow.title', operator: 'equals', value: 'test' },
{ field: 'eventNow.custom.av', operator: 'equals', value: 'av-value' },
],
'all',
mockStore,
);
expect(result).toBe(false);
});
});

describe('for all filter rule', () => {
it('should return true when all filters are true', () => {
const mockStore = makeRuntimeStateData({
clock: 10,
eventNow: makeOntimeEvent({
title: 'test',
custom: { av: 'av-value' },
}) as PlayableEvent,
});
const result = testConditions(
[
{ field: 'clock', operator: 'equals', value: '10' },
{ field: 'eventNow.title', operator: 'equals', value: 'test' },
{ field: 'eventNow.custom.av', operator: 'equals', value: 'av-value' },
],
'any',
mockStore,
);
expect(result).toBe(true);
});

it('should return true if any filters are true', () => {
const mockStore = makeRuntimeStateData({
clock: 10,
eventNow: makeOntimeEvent({
title: 'not-test',
custom: { av: 'av-value' },
}) as PlayableEvent,
});
const result = testConditions(
[
{ field: 'clock', operator: 'equals', value: '11' },
{ field: 'eventNow.title', operator: 'equals', value: 'not-test' },
{ field: 'eventNow.custom.av', operator: 'equals', value: 'av-value' },
],
'any',
mockStore,
);
expect(result).toBe(true);
});

it('should return false if all filters are false', () => {
const mockStore = makeRuntimeStateData({
clock: 10,
eventNow: makeOntimeEvent({ title: 'test' }) as PlayableEvent,
});
const result = testConditions(
[
{ field: 'clock', operator: 'equals', value: '11' },
{ field: 'eventNow.title', operator: 'equals', value: 'not-test' },
],
'any',
mockStore,
);
expect(result).toBe(false);
});
});
});
26 changes: 17 additions & 9 deletions apps/server/src/api-data/automation/automation.service.ts
Original file line number Diff line number Diff line change
Expand Up @@ -4,16 +4,16 @@ import {
type AutomationFilter,
type AutomationOutput,
type FilterRule,
type RuntimeStore,
type TimerLifeCycle,
} from 'ontime-types';
import { getPropertyFromPath } from 'ontime-utils';

import { getState, type RuntimeState } from '../../stores/runtimeState.js';
import { isOntimeCloud } from '../../externals.js';

import { emitOSC } from './clients/osc.client.js';
import { emitHTTP } from './clients/http.client.js';
import { getAutomations, getBlueprints } from './automation.dao.js';
import { isOntimeCloud } from '../../externals.js';

/**
* Exposes a method for triggering actions based on a TimerLifeCycle event
Expand Down Expand Up @@ -42,6 +42,9 @@ export function triggerAutomations(event: TimerLifeCycle, state: RuntimeState) {
});
}

/**
* Exposes a method for bypassing the condition check and testing the sending of an output
*/
export function testOutput(payload: AutomationOutput) {
send([payload]);
}
Expand All @@ -52,28 +55,33 @@ export function testOutput(payload: AutomationOutput) {
export function testConditions(
filters: AutomationFilter[],
filterRule: FilterRule,
state: Partial<RuntimeStore>,
state: Partial<RuntimeState>,
): boolean {
if (filters.length === 0) {
return true;
}

if (filterRule === 'all') {
return filters.every((filter) => evaluateCondition(filter));
return filters.every(evaluateCondition);
}

return filters.some((filter) => evaluateCondition(filter));
return filters.some(evaluateCondition);

function evaluateCondition(filter: AutomationFilter): boolean {
const { field, operator, value } = filter;
const fieldValue = state[field];
const fieldValue = getPropertyFromPath(field, state);

// TODO: if value is empty string, the user could be meaning to check if the value does not exist
// if value is empty string, the user could be meaning to check if the value does not exist
// we use loose equality to be able to check for converted values (eg '10' == 10)
switch (operator) {
case 'equals':
return fieldValue === value;
// overload the edge case where we use empty string to check if a value does not exist
if (value === '' && fieldValue === undefined) {
return true;
}
return fieldValue == value;
case 'not_equals':
return fieldValue !== value;
return fieldValue != value;
case 'greater_than':
return fieldValue > value;
case 'less_than':
Expand Down
18 changes: 14 additions & 4 deletions apps/server/src/api-data/automation/automation.utils.ts
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
import { Argument } from 'node-osc';
import { FilterRule, MaybeNumber } from 'ontime-types';
import { millisToString, removeLeadingZero, splitWhitespace } from 'ontime-utils';
import { millisToString, removeLeadingZero, splitWhitespace, getPropertyFromPath } from 'ontime-utils';

import { Argument } from 'node-osc';

type FilterOperator = 'equals' | 'not_equals' | 'greater_than' | 'less_than' | 'contains';

Expand Down Expand Up @@ -76,8 +77,7 @@ export function parseTemplateNested(template: string, state: object, humanReadab
value = undefined;
}
} else {
// iterate through variable parts, and look for the property in the state object
value = variableParts.reduce((obj, key) => obj?.[key], state);
value = getPropertyFromPath(variableName, state);
}
if (value !== undefined) {
parsedTemplate = parsedTemplate.replace(match[0], value);
Expand All @@ -87,9 +87,14 @@ export function parseTemplateNested(template: string, state: object, humanReadab
return parsedTemplate;
}

/**
* Handles the specific case where the MaybeNumber is encoded in a string
* After parsed, the value is formatted to a human-readable string
*/
function formatDisplayFromString(value: string, hideZero = false): string {
let valueInNumber: MaybeNumber = null;

// the value will be a string, so we need to convert it to the Maybe type
if (value !== 'null') {
const parsedValue = Number(value);
if (!Number.isNaN(parsedValue)) {
Expand All @@ -104,6 +109,11 @@ function formatDisplayFromString(value: string, hideZero = false): string {
}

type AliasesDefinition = Record<string, { key: string; cb: (value: string) => string }>;

/**
* This object matches some common RuntimeState paths
* to a function that formats them to a human-readable string
*/
const quickAliases: AliasesDefinition = {
clock: { key: 'clock', cb: (value: string) => formatDisplayFromString(value) },
duration: { key: 'timer.duration', cb: (value: string) => formatDisplayFromString(value, true) },
Expand Down
4 changes: 2 additions & 2 deletions apps/server/src/stores/__mocks__/runtimeState.mocks.ts
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,6 @@ const baseState: RuntimeState = {
},
};

export function makeRuntimeStateData(patch?: Partial<RuntimeState>) {
return deepmerge(baseState, patch);
export function makeRuntimeStateData(patch?: Partial<RuntimeState>): RuntimeState {
return deepmerge(baseState, patch) as RuntimeState;
}

0 comments on commit e707826

Please sign in to comment.