Scheduler

declare class ControlledScheduler implements Scheduler {
  constructor();
  deferred: boolean;
  tasks: PriorityBuckets<Task>;
  scheduleTask(task: Task, priority: number, fiber?: RuntimeFiber<unknown, unknown>): void;
  shouldYield(fiber: RuntimeFiber<unknown, unknown>): number | false;
  step(): void;
}

declare function make(
  scheduleTask: (task: Task, priority: number, fiber?: RuntimeFiber<unknown, unknown>) => void,
  shouldYield: (fiber: RuntimeFiber<unknown, unknown>) => number | false,
): Scheduler;

declare function makeBatched(
  callback: (runBatch: () => void) => void,
  shouldYield: (fiber: RuntimeFiber<unknown, unknown>) => number | false,
): Scheduler;

declare function makeMatrix(...record: Array<[number, Scheduler]>): Scheduler;

declare class MixedScheduler implements Scheduler {
  constructor(maxNextTickBeforeTimer: number);
  readonly maxNextTickBeforeTimer: number;
  scheduleTask(task: Task, priority: number, fiber?: RuntimeFiber<unknown, unknown>): void;
  shouldYield(fiber: RuntimeFiber<unknown, unknown>): number | false;
}

declare class SyncScheduler implements Scheduler {
  constructor();
  deferred: boolean;
  tasks: PriorityBuckets<Task>;
  flush(): void;
  scheduleTask(task: Task, priority: number, fiber?: RuntimeFiber<unknown, unknown>): void;
  shouldYield(fiber: RuntimeFiber<unknown, unknown>): number | false;
}

declare function timer(
  ms: number,
  shouldYield: (fiber: RuntimeFiber<unknown, unknown>) => number | false,
): Scheduler;

declare function timerBatched(
  ms: number,
  shouldYield: (fiber: RuntimeFiber<unknown, unknown>) => number | false,
): Scheduler;

interface Scheduler {
  scheduleTask(task: Task, priority: number, fiber?: RuntimeFiber<unknown, unknown>): void;
  shouldYield(fiber: RuntimeFiber<unknown, unknown>): number | false;
}

declare class SchedulerRunner {
  constructor(scheduleDrain: (depth: number, drain: (depth: number) => void) => void);
  running: boolean;
  readonly scheduleDrain: (depth: number, drain: (depth: number) => void) => void;
  tasks: PriorityBuckets<Task>;
  scheduleTask(task: Task, priority: number): void;
  static cached(
    scheduleDrain: (depth: number, drain: (depth: number) => void) => void,
  ): (fiber?: RuntimeFiber<unknown, unknown>) => SchedulerRunner;
}

type Task = () => void;

declare const defaultScheduler: Scheduler;

declare const defaultShouldYield: Scheduler["shouldYield"];

declare class PriorityBuckets<in out T = Task> {
  constructor<in out T = Task>();
  buckets: Array<[number, Array<T>]>;
  scheduleTask(task: T, priority: number): void;
}