@versori/run 0.5.0-alpha.6 → 0.5.0

package/script/src/internal/supervisor.js

@@ -1,275 +0,0 @@
-"use strict";
-/*
- * Copyright (c) 2025 Versori Group Inc
- *
- * Use of this software is governed by the Business Source License 1.1
- * included in the LICENSE file at the root of this repository.
- *
- * Change Date: 2029-01-01
- * Change License: Apache License, Version 2.0
- *
- * As of the Change Date, in accordance with the Business Source License,
- * use of this software will be governed by the Apache License, Version 2.0.
- */
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Fake = exports.enabled = void 0;
-exports.signalReady = signalReady;
-exports.registerWorkflow = registerWorkflow;
-exports.sendTaskStartEvent = sendTaskStartEvent;
-exports.mockAPIRequest = mockAPIRequest;
-const core_1 = require("@opentelemetry/core");
-const ConsoleLogger_js_1 = require("../observability/logging/ConsoleLogger.js");
-const ChainTask_js_1 = require("../dsl/tasks/ChainTask.js");
-const ScheduleTrigger_js_1 = require("../dsl/triggers/ScheduleTrigger.js");
-const WebhookTrigger_js_1 = require("../dsl/triggers/WebhookTrigger.js");
-const DurableTrigger_js_1 = require("../dsl/triggers/DurableTrigger.js");
-const Task_js_1 = require("../dsl/Task.js");
-const CatchTask_js_1 = require("../dsl/tasks/CatchTask.js");
-const FnTask_js_1 = require("../dsl/tasks/FnTask.js");
-const HttpTask_js_1 = require("../dsl/tasks/HttpTask.js");
-const BackgroundTask_js_1 = require("../dsl/tasks/BackgroundTask.js");
-const ParallelTask_js_1 = require("../dsl/tasks/ParallelTask.js");
-const supervisorURL = (function () {
-    return Deno.env.get('RUN_SUPERVISOR_URL') || '';
-})();
-exports.enabled = (function () {
-    return Deno.env.get('RUN_SUPERVISOR_URL') ? true : false;
-})();
-const flows = (function () {
-    try {
-        const _cwd = Deno.cwd();
-        const flowsFile = Deno.readTextFileSync(_cwd + '/versori-flow.json');
-        const flows = JSON.parse(flowsFile);
-        const nodes = {};
-        for (const node of flows.nodes) {
-            nodes[node.id] = {
-                description: node?.data?.description,
-                steps: node?.data?.steps,
-                files: node?.data?.files[0],
-            };
-        }
-        return nodes;
-    }
-    catch {
-        return {};
-    }
-})();
-const logger = new ConsoleLogger_js_1.ConsoleLogger('debug');
-function getTriggerData(workflow) {
-    if (workflow.trigger instanceof WebhookTrigger_js_1.WebhookTrigger) {
-        return {
-            id: workflow.trigger.id,
-            type: 'webhook',
-            method: workflow.trigger.options.method || 'post',
-        };
-    }
-    else if (workflow.trigger instanceof ScheduleTrigger_js_1.ScheduleTrigger) {
-        return {
-            id: workflow.trigger.id,
-            type: 'schedule',
-            schedule: workflow.trigger.schedule,
-        };
-    }
-    else if (workflow.trigger instanceof DurableTrigger_js_1.DurableWorkflowTrigger) {
-        return {
-            id: workflow.trigger.id,
-            type: 'durable',
-        };
-    }
-    return null;
-}
-function getTaskData(task) {
-    let taskData = [];
-    if (task instanceof ChainTask_js_1.ChainTask) {
-        taskData = task.tasks.map((t) => getTaskData(t)).flat();
-    }
-    else if (task instanceof CatchTask_js_1.CatchTask) {
-        taskData = [...getTaskData(task._base), ...getTaskData(task._errorHandler)];
-    }
-    else if (task instanceof BackgroundTask_js_1.BackgroundTask) {
-        taskData = getTaskData(task._base);
-    }
-    else if (task instanceof ParallelTask_js_1.ParallelTask) {
-        // I'm not sure if this is correct, but I don't think there are any ParallelTasks out in the wild yet
-        taskData = [...getTaskData(task._base), ...getTaskData(task._each)];
-    }
-    else if (task instanceof FnTask_js_1.FnTask) {
-        taskData.push({
-            taskId: task.id,
-            type: task[Task_js_1.TaskType].toLowerCase(),
-        });
-    }
-    else if (task instanceof HttpTask_js_1.HttpTaskImpl) {
-        taskData.push({
-            taskId: task.id,
-            type: task[Task_js_1.TaskType].toLowerCase(),
-        });
-    }
-    else {
-        // yolo
-        taskData.push({
-            taskId: String(task.id || 'unknown'),
-            type: task[Task_js_1.TaskType].toLowerCase(),
-        });
-    }
-    return taskData;
-}
-async function signalReady() {
-    if (!exports.enabled) {
-        return;
-    }
-    try {
-        await fetch(`${supervisorURL}/ready`, {
-            method: 'POST',
-        });
-    }
-    catch (err) {
-        logger.error('Failed to signal ready to supervisor', { error: err });
-    }
-}
-async function registerWorkflow(workflow) {
-    if (!exports.enabled) {
-        return;
-    }
-    const triggerData = getTriggerData(workflow);
-    const tasks = getTaskData(workflow.task);
-    try {
-        if (triggerData) {
-            const data = {
-                workflowId: triggerData.id,
-                trigger: triggerData,
-                tasks,
-            };
-            await fetch(`${supervisorURL}/register-workflow`, {
-                method: 'POST',
-                body: JSON.stringify(data),
-                headers: {
-                    'Content-Type': 'application/json',
-                },
-            });
-        }
-    }
-    catch (err) {
-        logger.error('Failed to register workflow with supervisor', { error: err });
-    }
-}
-async function sendTaskStartEvent(taskId, ctx, span) {
-    if (!exports.enabled) {
-        return ctx;
-    }
-    const data = {
-        taskId,
-        spanId: span.spanContext().spanId,
-        traceId: span.spanContext().traceId,
-        startTime: new Date().toISOString(), // assuming current time as start time
-        executionId: ctx.executionId,
-        workflowId: ctx.workflowId || '',
-    };
-    try {
-        const resp = await fetch(`${supervisorURL}/tasks/start`, {
-            method: 'POST',
-            body: JSON.stringify(data),
-            headers: {
-                'Content-Type': 'application/json',
-            },
-        });
-        if (resp.ok) {
-            const bodyString = await resp.text();
-            if (!bodyString) {
-                return ctx;
-            }
-            // TODO: XML, CSV yada yada
-            const respData = JSON.parse(bodyString);
-            return ctx.withData(respData);
-        }
-    }
-    catch (err) {
-        logger.error('Failed to send task start event to supervisor', { error: err });
-    }
-    return ctx;
-}
-async function mockAPIRequest(info, cnx, options) {
-    if (!exports.enabled) {
-        return new Response();
-    }
-    const mockRequest = {
-        url: info.toString(),
-        payload: options?.body || '',
-        method: options?.method || 'GET',
-        headers: options?.headers,
-        connection: cnx?.name,
-        connectionId: cnx?.id,
-    };
-    try {
-        const response = await fetch(`${supervisorURL}/ai/fake-fetch-call`, {
-            method: 'POST',
-            body: JSON.stringify(mockRequest),
-            headers: {
-                'Content-Type': 'application/json',
-            },
-        });
-        if (!response.ok) {
-            logger.error('Failed to mock api request', { response });
-            return new Response();
-        }
-        const text = await response.text();
-        return new Response(text);
-    }
-    catch (err) {
-        logger.error('Failed to mock api request', { error: err });
-        return new Response();
-    }
-}
-function printableSpan(s) {
-    const span = {
-        name: s.name,
-        startTime: (0, core_1.hrTimeToTimeStamp)(s.startTime),
-        endTime: (0, core_1.hrTimeToTimeStamp)(s.endTime),
-        duration: (0, core_1.hrTimeToMilliseconds)(s.duration),
-        attributes: s.attributes,
-        status: s.status,
-        ended: s.ended,
-        events: s.events,
-        parentSpanContext: s.parentSpanContext,
-        spanContext: s.spanContext(),
-    };
-    const nodeDetail = flows[span.attributes['task.id']]; // I know I have coded all of those as string
-    if (nodeDetail) {
-        span.attributes['description'] = nodeDetail?.description || '';
-        span.attributes['steps'] = nodeDetail?.steps || '';
-        span.attributes['files'] = nodeDetail?.files || '';
-    }
-    return span;
-}
-class Fake {
-    constructor() {
-        this.export = this.export.bind(this);
-        this.shutdown = this.shutdown.bind(this);
-        this.forceFlush = this.forceFlush.bind(this);
-    }
-    export(spans, resultCallback) {
-        const ss = spans.map(printableSpan);
-        try {
-            fetch(`${supervisorURL}/receive-spans`, {
-                method: 'POST',
-                body: JSON.stringify({ spans: ss }),
-                headers: {
-                    'Content-Type': 'application/json',
-                },
-            }).catch((err) => {
-                logger.error('Failed to send spans to supervisor', { error: err });
-            });
-        }
-        catch (err) {
-            logger.error('Failed to send spans to supervisor', { error: err });
-        }
-        resultCallback({ code: core_1.ExportResultCode.SUCCESS });
-    }
-    shutdown() {
-        return Promise.resolve();
-    }
-    forceFlush() {
-        return Promise.resolve();
-    }
-}
-exports.Fake = Fake;