Skip to content

Commit 6a1b481

Browse files
committed
feat: include transitions definitions inside states definitions
1 parent 4b39424 commit 6a1b481

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

packages/automation/src/lib/state-machine.ts

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ import {
2121
BaseStateMachineParams,
2222
PKPInfo,
2323
StateMachineDefinition,
24+
TransitionDefinition,
2425
TransitionParams,
2526
} from './types';
2627

@@ -94,8 +95,9 @@ export class StateMachine {
9495
pkp,
9596
});
9697

97-
machineConfig.states.forEach((state) => {
98-
const { litAction, transaction } = state;
98+
const stateTransitions = [] as TransitionDefinition[];
99+
states.forEach((state) => {
100+
const { litAction, transaction, transitions = [] } = state;
99101

100102
const stateConfig: StateParams = {
101103
key: state.key,
@@ -214,7 +216,7 @@ export class StateMachine {
214216
stateMachine.addState(stateConfig);
215217
});
216218

217-
machineConfig.transitions.forEach((transition) => {
219+
[...stateTransitions, ...transitions].forEach((transition) => {
218220
const { balances, evmContractEvent, fromState, timer, toState } =
219221
transition;
220222

0 commit comments

Comments
 (0)