storybook/dist/cli/tools/sdk/child-host.js

Storybook: Develop, document, and test UI components in isolation

import CJS_COMPAT_NODE_URL_qjinduibsyq from 'node:url';
import CJS_COMPAT_NODE_PATH_qjinduibsyq from 'node:path';
import CJS_COMPAT_NODE_MODULE_qjinduibsyq from "node:module";

var __filename = CJS_COMPAT_NODE_URL_qjinduibsyq.fileURLToPath(import.meta.url);
var __dirname = CJS_COMPAT_NODE_PATH_qjinduibsyq.dirname(__filename);
var require = CJS_COMPAT_NODE_MODULE_qjinduibsyq.createRequire(import.meta.url);

// ------------------------------------------------------------
// end of CJS compatibility banner, injected by Storybook's esbuild configuration
// ------------------------------------------------------------
import {
  createTools
} from "../../../_node-chunks/chunk-VT3VBFAX.js";
import "../../../_node-chunks/chunk-4CTPDGB7.js";
import {
  ToolsRuntimeError
} from "../../../_node-chunks/chunk-F5UFNM2U.js";
import "../../../_node-chunks/chunk-EEWXQXZP.js";
import "../../../_node-chunks/chunk-52XR72NR.js";
import {
  serializeError
} from "../../../_node-chunks/chunk-6WZPWJR4.js";
import "../../../_node-chunks/chunk-DW727PJG.js";
import "../../../_node-chunks/chunk-FE5AV6EP.js";
import "../../../_node-chunks/chunk-VGN5LGMI.js";
import "../../../_node-chunks/chunk-4US4PNS3.js";

// src/cli/tools/sdk/child-host.ts
import { pathToFileURL } from "node:url";
async function runChildHost({
  send = (message) => {
    process.send?.(message);
  },
  subscribe = (handler) => {
    process.on("message", handler);
  },
  cwd = () => process.cwd()
} = {}) {
  process.on("disconnect", () => {
    process.exit(0);
  }), process.env.STORYBOOK_TOOLS_CHILD_HOST = "true";
  let tools, controllers = /* @__PURE__ */ new Map(), handle = async (raw) => {
    let message = raw;
    switch (message.type) {
      case "init": {
        tools = await createTools({
          ...message.options,
          cwd: cwd(),
          mode: message.options.mode ?? "attached",
          autoSpawn: !1
        }), send({
          type: "hello",
          version: 1,
          storybook: tools.storybook,
          clientInfo: tools.clientInfo
        });
        return;
      }
      case "describe": {
        await reply(message.id, tools, () => tools.describe(message.options));
        return;
      }
      case "call": {
        let controller = new AbortController();
        controllers.set(message.id, controller);
        try {
          await reply(
            message.id,
            tools,
            () => tools.call(message.ref, message.input, {
              signal: controller.signal,
              telemetry: async (event, payload) => {
                send({ type: "telemetry", id: message.id, event, payload });
              }
            })
          );
        } finally {
          controllers.delete(message.id);
        }
        return;
      }
      case "cancel": {
        controllers.get(message.id)?.abort();
        return;
      }
      case "close": {
        await tools?.close(), process.exit(0);
        return;
      }
      default:
        throw message;
    }
  };
  async function reply(id, host, run) {
    if (!host) {
      send({
        type: "error",
        id,
        error: serializeError(
          new ToolsRuntimeError({
            reason: "command-unhandled",
            message: "The child host has not been initialized."
          })
        )
      });
      return;
    }
    try {
      send({ type: "result", id, value: await run() });
    } catch (error) {
      send({ type: "error", id, error: serializeError(error) });
    }
  }
  subscribe((message) => {
    handle(message).catch((error) => {
      let fallbackId = typeof message == "object" && message !== null && "id" in message && typeof message.id == "string" ? message.id : "init";
      send({ type: "error", id: fallbackId, error: serializeError(error) });
    });
  });
}
function isExecutedAsEntry() {
  let entry = process.argv[1];
  if (!entry)
    return !1;
  try {
    return import.meta.url === pathToFileURL(entry).href;
  } catch {
    return !1;
  }
}
isExecutedAsEntry() && runChildHost();
export {
  runChildHost
};