pangu/src/browser/task-scheduler.ts

Paranoid text spacing for good readability, to automatically insert whitespace between CJK (Chinese, Japanese, Korean) and half-width characters (alphabetical letters, numerical digits and symbols).

export interface TaskSchedulerConfig {
  enabled: boolean;
  chunkSize: number;
  timeout: number;
}

export class TaskQueue {
  private queue: (() => void)[] = [];
  private isProcessing = false;
  private onComplete?: () => void;

  add(task: () => void) {
    this.queue.push(task);
    this.scheduleProcessing();
  }

  clear() {
    this.queue.length = 0;
    this.onComplete = undefined;
  }

  setOnComplete(onComplete?: () => void) {
    this.onComplete = onComplete;
  }

  get length() {
    return this.queue.length;
  }

  private scheduleProcessing() {
    if (!this.isProcessing && this.queue.length > 0) {
      this.isProcessing = true;
      requestIdleCallback((deadline) => this.process(deadline), { timeout: 5000 });
    }
  }

  private process(deadline: IdleDeadline) {
    while (deadline.timeRemaining() > 0 && this.queue.length > 0) {
      const task = this.queue.shift();
      task?.();
    }

    this.isProcessing = false;

    if (this.queue.length > 0) {
      this.scheduleProcessing();
    } else {
      // All task completed, call completion callback
      this.onComplete?.();
    }
  }
}

/**
 * Schedules and executes text spacing operations during browser idle time to avoid blocking the UI.
 * Uses requestIdleCallback to process task in chunks when the browser has spare time,
 * ensuring smooth user experience even when processing large amounts of text.
 */
export class TaskScheduler {
  public readonly config: TaskSchedulerConfig = {
    enabled: true,
    chunkSize: 40, // Process 40 text nodes per idle cycle
    timeout: 2000, // 2 second timeout for idle processing
  };

  private taskQueue = new TaskQueue();

  get queue() {
    return this.taskQueue;
  }

  processInChunks<T>(items: T[], processor: (chunk: T[]) => void, onComplete?: () => void) {
    if (!this.config.enabled) {
      // Process synchronously if idle processing is disabled
      processor(items);
      onComplete?.();
      return;
    }

    if (items.length === 0) {
      onComplete?.();
      return;
    }

    // Set up completion callback
    if (onComplete) {
      this.taskQueue.setOnComplete(onComplete);
    }

    // Split items into chunks
    const chunks: T[][] = [];
    for (let i = 0; i < items.length; i += this.config.chunkSize) {
      chunks.push(items.slice(i, i + this.config.chunkSize));
    }

    // Add each chunk as a task item to the task queue
    for (const chunk of chunks) {
      this.taskQueue.add(() => {
        processor(chunk);
      });
    }
  }

  clear() {
    this.taskQueue.clear();
  }

  updateConfig(config: Partial<TaskSchedulerConfig>) {
    Object.assign(this.config, config);
  }
}