@notionhq/custom-blocks-dev-shell 0.1.33 → 0.1.35

package/dist-cli/main.js

@@ -1,271 +0,0 @@
-/**
- * Entry point for the published dev shell CLI (`npx`-run from a worker
- * project). Mirrors the repo's `scripts/dev.ts`: build the
- * worker, extract its manifest, serve one Vite dev server per custom block
- * (using the worker's own Vite install), and serve the prebuilt dev shell UI
- * with the block registry injected at runtime.
- */
-import { spawn } from "node:child_process";
-import { existsSync, readFileSync } from "node:fs";
-import { createRequire } from "node:module";
-import { basename, dirname, join, resolve } from "node:path";
-import { createInterface } from "node:readline";
-import { fileURLToPath } from "node:url";
-import { BLOCK_BASE_PORT, buildBlockRegistry, makeChangeLogger, SHELL_PORT, writeBlockViteConfig, } from "./block-server.js";
-import { runConvert } from "./convert.js";
-import { readDataSources } from "./data-sources.js";
-import { materializeWorkerSchemaDataSources } from "./materialize.js";
-import { copyPrebuiltDataSources } from "./prebuilt.js";
-import { serveUi } from "./serve-ui.js";
-import { blockCapabilities, findWorkerDir, generateWorkerManifest, } from "./worker-manifest.js";
-const __dirname = dirname(fileURLToPath(import.meta.url));
-const dim = "\x1b[2m";
-const bold = "\x1b[1m";
-const cyan = "\x1b[36m";
-const reset = "\x1b[0m";
-const label = (name) => `${cyan}[${name}]${reset}`;
-function parseCliArgs(argv) {
-    const args = {
-        worker: undefined,
-        shellPort: SHELL_PORT,
-        blockBasePort: BLOCK_BASE_PORT,
-    };
-    const takeValue = (name, index) => {
-        const value = argv[index];
-        if (value === undefined || value.startsWith("--")) {
-            throw new Error(`${name} requires a value.`);
-        }
-        return value;
-    };
-    const takePort = (name, raw) => {
-        const port = Number(raw);
-        if (!Number.isInteger(port) || port <= 0 || port > 65535) {
-            throw new Error(`${name} requires a port number, got "${raw}".`);
-        }
-        return port;
-    };
-    for (let index = 0; index < argv.length; index++) {
-        const arg = argv[index];
-        if (arg === "--worker") {
-            args.worker = takeValue("--worker", ++index);
-        }
-        else if (arg.startsWith("--worker=")) {
-            args.worker = arg.slice("--worker=".length);
-            if (args.worker.length === 0) {
-                throw new Error("--worker requires a path to a worker directory.");
-            }
-        }
-        else if (arg === "--port") {
-            args.shellPort = takePort("--port", takeValue("--port", ++index));
-        }
-        else if (arg.startsWith("--port=")) {
-            args.shellPort = takePort("--port", arg.slice("--port=".length));
-        }
-        else if (arg === "--block-base-port") {
-            args.blockBasePort = takePort("--block-base-port", takeValue("--block-base-port", ++index));
-        }
-        else if (arg.startsWith("--block-base-port=")) {
-            args.blockBasePort = takePort("--block-base-port", arg.slice("--block-base-port=".length));
-        }
-    }
-    return args;
-}
-function resolveWorkerDir(workerArg) {
-    if (workerArg !== undefined) {
-        return resolve(process.cwd(), workerArg);
-    }
-    const detected = findWorkerDir(process.cwd());
-    if (detected !== undefined) {
-        console.log(`Detected a worker at ${detected}.`);
-        return detected;
-    }
-    throw new Error("No worker found: run from inside a worker directory, or pass --worker <dir>.");
-}
-/**
- * The worker's own Vite binary. Blocks are served with the worker's Vite (and
- * plugins) rather than anything bundled here, matching how the block builds in
- * production.
- */
-function resolveViteBin(workerDir) {
-    let vitePkgPath;
-    try {
-        const workerRequire = createRequire(join(workerDir, "package.json"));
-        vitePkgPath = workerRequire.resolve("vite/package.json");
-    }
-    catch {
-        throw new Error(`Could not resolve "vite" from ${workerDir}. Add vite to the worker's ` +
-            `devDependencies and reinstall.`);
-    }
-    const vitePkg = JSON.parse(readFileSync(vitePkgPath, "utf-8"));
-    const bin = typeof vitePkg.bin === "string" ? vitePkg.bin : vitePkg.bin?.vite;
-    if (bin === undefined) {
-        throw new Error(`The vite package at ${vitePkgPath} exposes no bin.`);
-    }
-    return resolve(dirname(vitePkgPath), bin);
-}
-const procs = [];
-async function main() {
-    const argv = process.argv.slice(2);
-    // One-shot subcommands return before any dev-server machinery starts;
-    // with no subcommand the CLI is the dev shell, as it always was.
-    if (argv[0] === "convert") {
-        await runConvert(argv.slice(1));
-        return;
-    }
-    const cliArgs = parseCliArgs(argv);
-    const workerDir = resolveWorkerDir(cliArgs.worker);
-    if (!existsSync(resolve(workerDir, "node_modules"))) {
-        throw new Error(`No node_modules in ${workerDir}. Install the worker's dependencies first ` +
-            `(e.g. \`npm install\`), then rerun.`);
-    }
-    console.log(`${label(basename(workerDir))} Extracting worker manifest...`);
-    const { manifest, manifestPath } = await generateWorkerManifest(workerDir);
-    console.log(`${label(basename(workerDir))} Wrote ${manifestPath}`);
-    const blocks = blockCapabilities(manifest);
-    if (blocks.length === 0) {
-        // Not an error — start the shell anyway; it shows "None" under Blocks.
-        console.log(`${label(basename(workerDir))} Worker declares no custom blocks.`);
-    }
-    // Materialize schema-only files for the worker's declared sources, then
-    // load the directory. Files are validated before injection; a malformed
-    // file fails spin-up with the problem named.
-    const dataDir = resolve(workerDir, "src/data");
-    materializeWorkerSchemaDataSources(manifest, dataDir);
-    const dataSources = readDataSources(dataDir);
-    console.log(dataSources.length > 0
-        ? `${label(basename(workerDir))} Data sources from src/data: ${dataSources
-            .map(source => source.name)
-            .join(", ")}`
-        : `${label(basename(workerDir))} No data sources — create src/data/<key>.json files ` +
-            `(format: node_modules/@notionhq/custom-blocks-dev-shell/docs/data-sources.md).`);
-    if (blocks.length > 0 &&
-        cliArgs.shellPort >= cliArgs.blockBasePort &&
-        cliArgs.shellPort < cliArgs.blockBasePort + blocks.length) {
-        throw new Error(`--port ${cliArgs.shellPort} collides with the block server ports ` +
-            `(${cliArgs.blockBasePort}–${cliArgs.blockBasePort + blocks.length - 1}); ` +
-            `pick a port outside that range or move --block-base-port.`);
-    }
-    const viteBin = blocks.length > 0 ? resolveViteBin(workerDir) : undefined;
-    const registry = buildBlockRegistry(blocks, cliArgs.blockBasePort);
-    for (const [index, capability] of blocks.entries()) {
-        const blockDir = resolve(workerDir, capability.config.source.path);
-        const configFile = writeBlockViteConfig(workerDir, blockDir, capability);
-        const port = cliArgs.blockBasePort + index;
-        const proc = spawn(process.execPath, [
-            viteBin,
-            "--config",
-            configFile,
-            "--port",
-            String(port),
-            "--strictPort",
-        ], {
-            cwd: workerDir,
-            stdio: ["ignore", "pipe", "inherit"],
-            // Process groups (and negative-PID kills) are POSIX-only; on
-            // Windows children are killed individually in shutdown().
-            detached: process.platform !== "win32",
-        });
-        if (proc.stdout !== null) {
-            createInterface({ input: proc.stdout }).on("line", makeChangeLogger(message => console.log(`${label(capability.key)} ${message}`)));
-        }
-        proc.on("exit", code => {
-            if (shuttingDown || code === 0 || code === null) {
-                return;
-            }
-            console.error(`${label(capability.key)} dev server exited with code ${code}`);
-            process.exitCode = code;
-            shutdown("SIGTERM");
-        });
-        procs.push(proc);
-    }
-    // The published layout is dist/ next to cli/; index.html must be prebuilt.
-    const distDir = resolve(__dirname, "..", "dist");
-    if (!existsSync(join(distDir, "index.html"))) {
-        throw new Error(`No prebuilt UI found at ${distDir}. This package was not assembled ` +
-            `correctly; reinstall it.`);
-    }
-    const prebuiltDir = resolve(__dirname, "..", "data");
-    try {
-        await serveUi(distDir, cliArgs.shellPort, {
-            blocks: registry,
-            dataSources,
-            prebuiltFilenames: readDataSources(prebuiltDir).map(source => source.filename),
-        }, () => {
-            const written = copyPrebuiltDataSources(prebuiltDir, dataDir);
-            if (written.length > 0) {
-                console.log(`${label(basename(workerDir))} Added pre-built data sources: ` +
-                    written.join(", "));
-            }
-            return readDataSources(dataDir);
-        });
-    }
-    catch (error) {
-        const code = error.code;
-        if (code === "EADDRINUSE") {
-            throw new Error(`Port ${cliArgs.shellPort} is already in use. Stop whatever holds it ` +
-                `or rerun with --port <port> (and --block-base-port <port> for the ` +
-                `block servers).`);
-        }
-        throw error;
-    }
-    console.log("");
-    console.log(`${bold}Dev shell${reset}`);
-    console.log(`  ${label("dev-shell")} ${dim}http://localhost:${cliArgs.shellPort}${reset}`);
-    if (blocks.length > 0) {
-        console.log("");
-        console.log(`${bold}Blocks${reset}`);
-        for (const [index, entry] of registry.entries()) {
-            console.log(`  ${label(entry.key)} ${dim}http://localhost:${cliArgs.blockBasePort + index}${reset}`);
-        }
-    }
-    console.log("");
-}
-function killProc(p, signal) {
-    if (p.pid === undefined || p.killed) {
-        return;
-    }
-    try {
-        if (process.platform === "win32") {
-            p.kill(signal);
-        }
-        else {
-            process.kill(-p.pid, signal);
-        }
-    }
-    catch { }
-}
-let shuttingDown = false;
-function shutdown(signal) {
-    if (shuttingDown) {
-        return;
-    }
-    shuttingDown = true;
-    if (signal !== "exit" && procs.length > 0) {
-        console.log(`\n${dim}Shutting down...${reset}`);
-    }
-    for (const p of procs) {
-        killProc(p, "SIGTERM");
-    }
-    setTimeout(() => {
-        for (const p of procs) {
-            killProc(p, "SIGKILL");
-        }
-        // Preserve a failure exit code set before shutdown (startup errors,
-        // crashed block servers); plain signal shutdowns still exit 0.
-        process.exit(process.exitCode ?? 0);
-    }, 1500).unref();
-}
-for (const signal of ["SIGINT", "SIGTERM", "SIGHUP"]) {
-    process.on(signal, () => shutdown(signal));
-}
-process.on("exit", () => shutdown("exit"));
-process.on("uncaughtException", err => {
-    console.error(err);
-    process.exitCode = 1;
-    shutdown("SIGTERM");
-});
-main().catch(err => {
-    console.error(err instanceof Error ? err.message : err);
-    process.exitCode = 1;
-    shutdown("SIGTERM");
-});