added task features

Author: matthew-sessions

packages/jobs/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@durable-effect/jobs",
-  "version": "0.0.1-next.7",
+  "version": "0.0.1-next.8",
   "type": "module",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",

packages/jobs/src/handlers/task/context.ts

@@ -199,6 +199,7 @@ export function createTaskExecuteContext<S>(
 
 /**
  * Create TaskIdleContext for onIdle handler.
+ * Full parity with TaskEventContext/TaskExecuteContext.
  */
 export function createTaskIdleContext<S>(
   stateHolder: TaskStateHolder<S>,
@@ -212,15 +213,37 @@ export function createTaskIdleContext<S>(
   const scheduling = createSchedulingEffects(scheduleHolder, getScheduledFromStorage);
 
   return {
+    // State access
     state: stateHolder.dirty
       ? Effect.succeed(stateHolder.current)
       : getState(),
 
-    schedule: scheduling.schedule,
+    // State mutations
+    setState: (state: S): Effect.Effect<void, never, never> =>
+      Effect.sync(() => {
+        stateHolder.current = state;
+        stateHolder.dirty = true;
+      }),
 
+    updateState: (fn: (current: S) => S): Effect.Effect<void, never, never> =>
+      Effect.gen(function* () {
+        const current = stateHolder.dirty
+          ? stateHolder.current
+          : yield* getState();
+        if (current !== null) {
+          stateHolder.current = fn(current);
+          stateHolder.dirty = true;
+        }
+      }),
+
+    // Scheduling (full access)
+    ...scheduling,
+
+    // Cleanup
     terminate: (): Effect.Effect<never, never, never> =>
       Effect.fail(new TerminateSignal({ reason: "terminate", purgeState: true })) as Effect.Effect<never, never, never>,
 
+    // Metadata
     instanceId,
     jobName,
     idleReason,
@@ -229,6 +252,7 @@ export function createTaskIdleContext<S>(
 
 /**
  * Create TaskErrorContext for onError handler.
+ * Full parity with TaskEventContext/TaskExecuteContext.
  */
 export function createTaskErrorContext<S>(
   stateHolder: TaskStateHolder<S>,
@@ -242,10 +266,18 @@ export function createTaskErrorContext<S>(
   const scheduling = createSchedulingEffects(scheduleHolder, getScheduledFromStorage);
 
   return {
+    // State access
     state: stateHolder.dirty
       ? Effect.succeed(stateHolder.current)
       : getState(),
 
+    // State mutations
+    setState: (state: S): Effect.Effect<void, never, never> =>
+      Effect.sync(() => {
+        stateHolder.current = state;
+        stateHolder.dirty = true;
+      }),
+
     updateState: (fn: (current: S) => S): Effect.Effect<void, never, never> =>
       Effect.gen(function* () {
         const current = stateHolder.dirty
@@ -257,11 +289,14 @@ export function createTaskErrorContext<S>(
         }
       }),
 
-    schedule: scheduling.schedule,
+    // Scheduling (full access)
+    ...scheduling,
 
+    // Cleanup
     terminate: (): Effect.Effect<never, never, never> =>
       Effect.fail(new TerminateSignal({ reason: "terminate", purgeState: true })) as Effect.Effect<never, never, never>,
 
+    // Metadata
     instanceId,
     jobName,
     errorSource,

packages/jobs/src/registry/types.ts

@@ -613,16 +613,32 @@ export interface TaskExecuteContext<S> {
  * Context provided to task onIdle handler.
  *
  * Called when either onEvent or execute completes without scheduling
- * another execution. Use this to schedule cleanup or maintenance.
+ * another execution. Use this to update state, schedule cleanup, or terminate.
  */
 export interface TaskIdleContext<S> {
+  // State access
   /** Get current state */
   readonly state: Effect.Effect<S | null, never, never>;
+
+  // State mutations
+  /** Replace the entire state */
+  readonly setState: (state: S) => Effect.Effect<void, never, never>;
+  /** Update state via transformation function (no-op if state is null) */
+  readonly updateState: (fn: (current: S) => S) => Effect.Effect<void, never, never>;
+
+  // Scheduling
   /** Schedule execution (e.g., for delayed cleanup) */
   readonly schedule: (when: Duration.DurationInput | number | Date) => Effect.Effect<void, never, never>;
+  /** Cancel any scheduled execution */
+  readonly cancelSchedule: () => Effect.Effect<void, never, never>;
+  /** Get the currently scheduled execution time (null if none) */
+  readonly getScheduledTime: () => Effect.Effect<number | null, never, never>;
+
+  // Cleanup
   /** Terminate the task immediately - cancel alarms and delete all state */
   readonly terminate: () => Effect.Effect<never, never, never>;
 
+  // Metadata
   /** The unique instance ID for this task */
   readonly instanceId: string;
   /** The name of this job (as registered) */
@@ -635,18 +651,32 @@ export interface TaskIdleContext<S> {
  * Context provided to task onError handler.
  *
  * Called when onEvent or execute throws an error.
- * Use this to log errors, update state, or schedule retries.
+ * Use this to log errors, update state, schedule retries, or terminate.
  */
 export interface TaskErrorContext<S> {
+  // State access
   /** Get current state */
   readonly state: Effect.Effect<S | null, never, never>;
+
+  // State mutations
+  /** Replace the entire state */
+  readonly setState: (state: S) => Effect.Effect<void, never, never>;
   /** Update state (e.g., to track error count) */
   readonly updateState: (fn: (current: S) => S) => Effect.Effect<void, never, never>;
+
+  // Scheduling
   /** Schedule execution (e.g., for retry) */
   readonly schedule: (when: Duration.DurationInput | number | Date) => Effect.Effect<void, never, never>;
+  /** Cancel any scheduled execution */
+  readonly cancelSchedule: () => Effect.Effect<void, never, never>;
+  /** Get the currently scheduled execution time (null if none) */
+  readonly getScheduledTime: () => Effect.Effect<number | null, never, never>;
+
+  // Cleanup
   /** Terminate the task immediately - cancel alarms and delete all state */
   readonly terminate: () => Effect.Effect<never, never, never>;
 
+  // Metadata
   /** The unique instance ID for this task */
   readonly instanceId: string;
   /** The name of this job (as registered) */