nx

Version:
The core Nx plugin contains the core functionality of Nx like the project graph, nx commands and task orchestration.
88 lines (87 loc) • 3.15 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.isPluginWorkerNotification = isPluginWorkerNotification;
exports.isPluginWorkerMessage = isPluginWorkerMessage;
exports.isPluginWorkerResult = isPluginWorkerResult;
exports.consumeMessage = consumeMessage;
exports.sendMessageOverSocket = sendMessageOverSocket;
const socket_utils_1 = require("../../../daemon/socket-utils");
const workspace_root_1 = require("../../../utils/workspace-root");
const NOTIFICATION_TYPES = [
    'emitLog',
];
function isPluginWorkerNotification(message) {
    return (typeof message === 'object' &&
        message !== null &&
        'type' in message &&
        typeof message.type === 'string' &&
        NOTIFICATION_TYPES.includes(message.type));
}
// =============================================================================
// TYPE GUARDS
// =============================================================================
const MESSAGE_TYPES = [
    'load',
    'createNodes',
    'createDependencies',
    'createMetadata',
    'preTasksExecution',
    'postTasksExecution',
    'setWorkerEnv',
];
const RESULT_TYPES = [
    'loadResult',
    'createNodesResult',
    'createDependenciesResult',
    'createMetadataResult',
    'preTasksExecutionResult',
    'postTasksExecutionResult',
    'setWorkerEnvResult',
];
function isPluginWorkerMessage(message) {
    return (typeof message === 'object' &&
        message !== null &&
        'type' in message &&
        typeof message.type === 'string' &&
        MESSAGE_TYPES.includes(message.type));
}
function isPluginWorkerResult(message) {
    return (typeof message === 'object' &&
        message !== null &&
        'type' in message &&
        typeof message.type === 'string' &&
        RESULT_TYPES.includes(message.type));
}
// =============================================================================
// MESSAGE HANDLING
// =============================================================================
/**
 * Consumes a message and dispatches to the appropriate handler.
 * If the handler returns a value, it's automatically wrapped in a result message
 * with the correct type and transaction ID.
 *
 * Handlers return just the result payload - the infrastructure handles wrapping.
 */
async function consumeMessage(socket, raw, handlers) {
    const message = raw;
    const type = message.type;
    const handler = handlers[type];
    // Type widening for dynamic dispatch - safe because types guarantee
    // message.type always indexes into the matching handler
    const resultPayload = await handler(message.payload);
    if (resultPayload !== undefined && resultPayload !== null) {
        sendMessageOverSocket(socket, {
            type: `${type}Result`,
            payload: resultPayload,
            tx: message.tx,
        });
    }
}
/**
 * Sends a message over the socket with proper formatting.
 */
function sendMessageOverSocket(socket, message) {
    // Stamped here rather than in each constructor, mirroring the daemon.
    message.workspaceRoot = workspace_root_1.workspaceRoot;
    (0, socket_utils_1.sendMessage)(socket, message);
}