unpackWorkerAsync.ts

// Copyright (c) Microsoft Corporation. All rights reserved. Licensed under the MIT license.
// See LICENSE in the project root for license information.

import type { Worker } from 'node:worker_threads';

import type {
  IWorkerToHostMessage,
  IHostToWorkerMessage,
  IZipSyncUnpackWorkerResult,
  IZipSyncUnpackOptions
} from './unpackWorker';

export type { IZipSyncUnpackWorkerResult } from './unpackWorker';

export async function unpackWorkerAsync(
  options: Omit<IZipSyncUnpackOptions, 'terminal'>
): Promise<IZipSyncUnpackWorkerResult> {
  const { Worker } = await import('node:worker_threads');

  const worker: Worker = new Worker(require.resolve('./unpackWorker'));

  return new Promise<IZipSyncUnpackWorkerResult>((resolve, reject) => {
    worker.on('message', (message: IWorkerToHostMessage) => {
      switch (message.type) {
        case 'zipsync-unpack': {
          resolve(message.result);
          break;
        }
        case 'error': {
          const error: Error = new Error(message.args.message);
          error.stack = message.args.stack;
          reject(error);
          break;
        }
        default: {
          const exhaustiveCheck: never = message;
          throw new Error(`Unexpected message type: ${JSON.stringify(exhaustiveCheck)}`);
        }
      }
    });

    worker.on('error', (err) => {
      reject(err);
    });

    worker.on('exit', (code) => {
      if (code !== 0) {
        reject(new Error(`Worker stopped with exit code ${code}`));
      }
    });

    const commandMessage: IHostToWorkerMessage = {
      type: 'zipsync-unpack',
      id: 0,
      options
    };
    worker.postMessage(commandMessage);
  }).finally(() => {
    worker.postMessage(false);
  });
}