Fix Schedule option types

pmndrs · Jun 3, 2024 · 10d4dc6 · 10d4dc6
1 parent 06dc5ef
commit 10d4dc6
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/packages/scheduler/src/class/schedule.ts b/packages/scheduler/src/class/schedule.ts
@@ -23,7 +23,7 @@ export class Schedule<T extends Scheduler.Context = Scheduler.Context> {
         this.symbols = new Map<symbol | string, Runnable<T>>();
     }
 
-    add(runnable: Runnable<T>, options?: OptionsObject): Schedule<T> {
+    add(runnable: Runnable<T>, options?: OptionsObject<T>): Schedule<T> {
         const optionsFns = createOptionsFns<T>(options);
         add(this, runnable, ...optionsFns);
 
@@ -50,7 +50,7 @@ export class Schedule<T extends Scheduler.Context = Scheduler.Context> {
         return this;
     }
 
-    createTag(id: symbol | string, options?: OptionsObject): Schedule<T> {
+    createTag(id: symbol | string, options?: OptionsObject<T>): Schedule<T> {
         const optionsFns = createOptionsFns<T>(options);
         createTag(this, id, ...optionsFns);
         return this;

diff --git a/packages/scheduler/src/class/types.ts b/packages/scheduler/src/class/types.ts
@@ -1,8 +1,8 @@
 import { Runnable } from '../scheduler-types';
 
-export type OptionsObject = {
+export type OptionsObject<T extends Scheduler.Context = Scheduler.Context> = {
     id?: symbol | string;
-    before?: symbol | string | Runnable | (symbol | string | Runnable)[];
-    after?: symbol | string | Runnable | (symbol | string | Runnable)[];
+    before?: symbol | string | Runnable<T> | (symbol | string | Runnable<T>)[];
+    after?: symbol | string | Runnable<T> | (symbol | string | Runnable<T>)[];
     tag?: symbol | string | (symbol | string)[];
 };
diff --git a/packages/scheduler/src/class/utils/create-options-fns.ts b/packages/scheduler/src/class/utils/create-options-fns.ts
@@ -9,7 +9,7 @@ import { OptionsObject } from '../types';
 
 export function createOptionsFns<
     T extends Scheduler.Context = Scheduler.Context
->(options: OptionsObject | undefined): OptionsFn<T>[] {
+>(options: OptionsObject<any> | undefined): OptionsFn<T>[] {
     const optionsFns: OptionsFn[] = [];
 
     if (options?.id) {