@toolproof-core/lib 1.0.31 → 1.0.33

package/src/utils/strategyState.ts

@@ -0,0 +1,56 @@
+import type {
+    CreationContext,
+    Strategy,
+    StrategyStateInputEntry,
+    StrategyStateInputEntryByRoleName,
+} from '@toolproof-core/schema';
+
+export type StrategyState = Strategy['strategyState'];
+
+export function getStrategyStateBucket(
+    strategyState: StrategyState,
+    toolStepPath: CreationContext['toolStepPath'],
+): StrategyStateInputEntryByRoleName | undefined {
+    return strategyState[toolStepPath];
+}
+
+export function getStrategyStateEntry(
+    strategyState: StrategyState,
+    context: CreationContext,
+): StrategyStateInputEntry | undefined {
+    return getStrategyStateBucket(strategyState, context.toolStepPath)?.[context.roleName];
+}
+
+export function setStrategyStateEntry(
+    strategyState: StrategyState,
+    context: CreationContext,
+    entry: StrategyStateInputEntry,
+): StrategyState {
+    const bucket = getStrategyStateBucket(strategyState, context.toolStepPath) ?? {};
+
+    return {
+        ...strategyState,
+        [context.toolStepPath]: {
+            ...bucket,
+            [context.roleName]: entry,
+        },
+    };
+}
+
+export function clearStrategyStateEntry(
+    strategyState: StrategyState,
+    context: CreationContext,
+): StrategyState {
+    const bucket = getStrategyStateBucket(strategyState, context.toolStepPath);
+    if (!bucket || !(context.roleName in bucket)) {
+        return strategyState;
+    }
+
+    const nextBucket = { ...bucket };
+    delete nextBucket[context.roleName];
+
+    return {
+        ...strategyState,
+        [context.toolStepPath]: nextBucket,
+    };
+}