@oneuptime/common 10.2.22 → 10.3.0

package/build/dist/Server/Services/RunbookAgentJobService.js

@@ -0,0 +1,393 @@
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+import DatabaseService from "./DatabaseService";
+import ObjectID from "../../Types/ObjectID";
+import BadDataException from "../../Types/Exception/BadDataException";
+import Model from "../../Models/DatabaseModels/RunbookAgentJob";
+import RunbookAgentJobStatus from "../../Types/Runbook/RunbookAgentJobStatus";
+import RunbookStepType from "../../Types/Runbook/RunbookStepType";
+import OneUptimeDate from "../../Types/Date";
+import PostgresAppInstance from "../Infrastructure/PostgresDatabase";
+import logger from "../Utils/Logger";
+import Sleep from "../../Types/Sleep";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
+/*
+ * Lease the Worker grants on each successful claim. While the lease is fresh
+ * the job belongs to that agent; once it elapses, another agent (or the
+ * Worker's poll loop) can mark the job TimedOut.
+ */
+const DEFAULT_LEASE_MS = 30000;
+/*
+ * If the agent's executor heartbeats faster than this, the Worker keeps
+ * extending the lease. The agent should call the job heartbeat endpoint at
+ * least once every (DEFAULT_LEASE_MS / 2) ms.
+ */
+const DEFAULT_CLAIM_TIMEOUT_MS = 2 * 60000;
+const POLL_INTERVAL_MS = 500;
+/*
+ * TypeORM's dataSource.query returns `[rows, rowCount]` for UPDATE/DELETE
+ * (even with RETURNING), and just `rows` for SELECT/INSERT. This helper
+ * normalises the response so callers always see the rows array.
+ */
+function unwrapRows(result) {
+    if (Array.isArray(result)) {
+        if (result.length === 2 &&
+            Array.isArray(result[0]) &&
+            typeof result[1] === "number") {
+            return result[0];
+        }
+        return result;
+    }
+    return [];
+}
+export class Service extends DatabaseService {
+    constructor() {
+        super(Model);
+    }
+    async enqueue(data) {
+        var _a;
+        if (!data.targetAgentId) {
+            throw new BadDataException("targetAgentId is required to dispatch a step to a Runbook Agent.");
+        }
+        if (data.stepType !== RunbookStepType.Bash &&
+            data.stepType !== RunbookStepType.JavaScript) {
+            throw new BadDataException(`RunbookAgent does not execute step type "${data.stepType}".`);
+        }
+        const claimDeadlineAt = OneUptimeDate.addRemoveSeconds(OneUptimeDate.getCurrentDate(), Math.ceil(((_a = data.claimTimeoutInMs) !== null && _a !== void 0 ? _a : DEFAULT_CLAIM_TIMEOUT_MS) / 1000));
+        const row = new Model();
+        row.projectId = data.projectId;
+        row.runbookExecutionId = data.runbookExecutionId;
+        row.stepId = data.stepId;
+        row.stepType = data.stepType;
+        row.targetAgentId = data.targetAgentId;
+        row.script = data.script;
+        row.timeoutInMs = data.timeoutInMs;
+        row.status = RunbookAgentJobStatus.Pending;
+        row.claimDeadlineAt = claimDeadlineAt;
+        return this.create({ data: row, props: { isRoot: true } });
+    }
+    /*
+     * Atomically claim the oldest Pending job in the agent's project targeted
+     * at this specific agent. Uses FOR UPDATE SKIP LOCKED so concurrent agents
+     * don't fight over the same row.
+     */
+    async claimNextJob(data) {
+        var _a;
+        const dataSource = PostgresAppInstance.getDataSource();
+        if (!dataSource) {
+            throw new BadDataException("Database is not connected");
+        }
+        const leaseMs = (_a = data.leaseMs) !== null && _a !== void 0 ? _a : DEFAULT_LEASE_MS;
+        const sql = `
+      WITH claimed AS (
+        SELECT "_id" FROM "RunbookAgentJob"
+        WHERE "projectId" = $1::uuid
+          AND "status" = $2
+          AND "targetAgentId" = $3::uuid
+          AND "claimDeadlineAt" > NOW()
+          AND "deletedAt" IS NULL
+        ORDER BY "createdAt" ASC
+        LIMIT 1
+        FOR UPDATE SKIP LOCKED
+      )
+      UPDATE "RunbookAgentJob" j
+      SET "status" = $4,
+          "assignedAgentId" = $3::uuid,
+          "claimedAt" = NOW(),
+          "leaseExpiresAt" = NOW() + ($5 || ' milliseconds')::interval,
+          "updatedAt" = NOW(),
+          "version" = j."version" + 1
+      FROM claimed
+      WHERE j."_id" = claimed."_id"
+      RETURNING j."_id", j."projectId", j."runbookExecutionId",
+                j."stepId", j."stepType", j."targetAgentId", j."script",
+                j."timeoutInMs", j."status", j."claimedAt", j."leaseExpiresAt";
+    `;
+        const rows = unwrapRows(await dataSource.query(sql, [
+            data.projectId.toString(),
+            RunbookAgentJobStatus.Pending,
+            data.agentId.toString(),
+            RunbookAgentJobStatus.Claimed,
+            leaseMs.toString(),
+        ]));
+        if (!rows || rows.length === 0) {
+            return null;
+        }
+        const r = rows[0];
+        const job = new Model();
+        job._id = String(r["_id"]);
+        job.projectId = new ObjectID(String(r["projectId"]));
+        job.runbookExecutionId = new ObjectID(String(r["runbookExecutionId"]));
+        job.stepId = String(r["stepId"]);
+        job.stepType = r["stepType"];
+        job.targetAgentId = new ObjectID(String(r["targetAgentId"]));
+        job.script = String(r["script"]);
+        job.timeoutInMs = Number(r["timeoutInMs"]);
+        job.status = r["status"];
+        if (r["claimedAt"]) {
+            job.claimedAt = new Date(String(r["claimedAt"]));
+        }
+        if (r["leaseExpiresAt"]) {
+            job.leaseExpiresAt = new Date(String(r["leaseExpiresAt"]));
+        }
+        job.assignedAgentId = data.agentId;
+        return job;
+    }
+    async heartbeatJob(data) {
+        var _a;
+        const dataSource = PostgresAppInstance.getDataSource();
+        if (!dataSource) {
+            throw new BadDataException("Database is not connected");
+        }
+        const leaseMs = (_a = data.leaseMs) !== null && _a !== void 0 ? _a : DEFAULT_LEASE_MS;
+        /*
+         * Only refresh the lease if this agent still owns the job and it hasn't
+         * reached a terminal state. If another agent reclaimed it, the agent
+         * should stop executing.
+         */
+        const sql = `
+      UPDATE "RunbookAgentJob"
+      SET "status" = CASE WHEN "status" = $4 THEN $5 ELSE "status" END,
+          "startedAt" = COALESCE("startedAt", NOW()),
+          "leaseExpiresAt" = NOW() + ($1 || ' milliseconds')::interval,
+          "updatedAt" = NOW(),
+          "version" = "version" + 1
+      WHERE "_id" = $2::uuid
+        AND "assignedAgentId" = $3::uuid
+        AND "status" IN ($4, $5)
+      RETURNING "_id";
+    `;
+        const rows = unwrapRows(await dataSource.query(sql, [
+            leaseMs.toString(),
+            data.jobId.toString(),
+            data.agentId.toString(),
+            RunbookAgentJobStatus.Claimed,
+            RunbookAgentJobStatus.Running,
+        ]));
+        return rows.length > 0;
+    }
+    async submitResult(data) {
+        var _a, _b, _c;
+        const dataSource = PostgresAppInstance.getDataSource();
+        if (!dataSource) {
+            throw new BadDataException("Database is not connected");
+        }
+        const status = data.success
+            ? RunbookAgentJobStatus.Succeeded
+            : RunbookAgentJobStatus.Failed;
+        /*
+         * Only the agent that holds the lease is allowed to write the terminal
+         * result. If the lease has already moved on (Worker timed out and
+         * reclaimed), this is a no-op.
+         */
+        const sql = `
+      UPDATE "RunbookAgentJob"
+      SET "status" = $1,
+          "output" = $2,
+          "exitCode" = $3,
+          "errorMessage" = $4,
+          "completedAt" = NOW(),
+          "updatedAt" = NOW(),
+          "version" = "version" + 1
+      WHERE "_id" = $5::uuid
+        AND "assignedAgentId" = $6::uuid
+        AND "status" IN ($7, $8)
+      RETURNING "_id";
+    `;
+        const rows = unwrapRows(await dataSource.query(sql, [
+            status,
+            (_a = data.output) !== null && _a !== void 0 ? _a : null,
+            (_b = data.exitCode) !== null && _b !== void 0 ? _b : null,
+            (_c = data.errorMessage) !== null && _c !== void 0 ? _c : null,
+            data.jobId.toString(),
+            data.agentId.toString(),
+            RunbookAgentJobStatus.Claimed,
+            RunbookAgentJobStatus.Running,
+        ]));
+        return rows.length > 0;
+    }
+    /*
+     * Called by the Worker after enqueueing. Polls the job row every
+     * POLL_INTERVAL_MS until the job reaches a terminal status, or until the
+     * combined claim + execution window is exhausted (in which case we mark
+     * the job TimedOut ourselves and return it).
+     */
+    async pollUntilTerminal(data) {
+        const overallDeadline = OneUptimeDate.addRemoveSeconds(OneUptimeDate.getCurrentDate(), Math.ceil((data.claimTimeoutInMs + data.executionTimeoutInMs + 5000) / 1000));
+        while (true) {
+            const job = await this.findOneById({
+                id: data.jobId,
+                select: {
+                    _id: true,
+                    status: true,
+                    output: true,
+                    exitCode: true,
+                    errorMessage: true,
+                    claimDeadlineAt: true,
+                    leaseExpiresAt: true,
+                    claimedAt: true,
+                    startedAt: true,
+                    completedAt: true,
+                    assignedAgentId: true,
+                },
+                props: { isRoot: true },
+            });
+            if (!job) {
+                throw new BadDataException(`RunbookAgentJob ${data.jobId.toString()} disappeared while waiting.`);
+            }
+            if (job.status === RunbookAgentJobStatus.Succeeded ||
+                job.status === RunbookAgentJobStatus.Failed ||
+                job.status === RunbookAgentJobStatus.TimedOut ||
+                job.status === RunbookAgentJobStatus.Cancelled) {
+                return job;
+            }
+            const now = OneUptimeDate.getCurrentDate();
+            // Pending with claim deadline elapsed -> no agent picked it up.
+            if (job.status === RunbookAgentJobStatus.Pending &&
+                job.claimDeadlineAt &&
+                now > job.claimDeadlineAt) {
+                return this.timeoutJob({
+                    jobId: data.jobId,
+                    reason: "No runbook agent picked up this step before the wait window expired. The agent may be offline — check that it is running and reachable, then try again.",
+                });
+            }
+            // Claimed/Running with lease elapsed -> agent went silent.
+            if ((job.status === RunbookAgentJobStatus.Claimed ||
+                job.status === RunbookAgentJobStatus.Running) &&
+                job.leaseExpiresAt &&
+                now > job.leaseExpiresAt) {
+                return this.timeoutJob({
+                    jobId: data.jobId,
+                    reason: "The runbook agent stopped responding while this step was running. The agent may have crashed or lost its network connection — check that it is still online, then try running the runbook again.",
+                });
+            }
+            if (now > overallDeadline) {
+                return this.timeoutJob({
+                    jobId: data.jobId,
+                    reason: "This step ran longer than the allowed execution window. Increase the timeout on the step or make the script complete faster.",
+                });
+            }
+            await Sleep.sleep(POLL_INTERVAL_MS);
+        }
+    }
+    async timeoutJob(data) {
+        const dataSource = PostgresAppInstance.getDataSource();
+        if (!dataSource) {
+            throw new BadDataException("Database is not connected");
+        }
+        /*
+         * Move to TimedOut only if the job hasn't already reached a terminal
+         * state. The same agent's late-arriving result must lose to this race.
+         */
+        const sql = `
+      UPDATE "RunbookAgentJob"
+      SET "status" = $1,
+          "errorMessage" = $2,
+          "completedAt" = NOW(),
+          "updatedAt" = NOW(),
+          "version" = "version" + 1
+      WHERE "_id" = $3::uuid
+        AND "status" NOT IN ($4, $5, $6, $7)
+      RETURNING "_id";
+    `;
+        await dataSource.query(sql, [
+            RunbookAgentJobStatus.TimedOut,
+            data.reason,
+            data.jobId.toString(),
+            RunbookAgentJobStatus.Succeeded,
+            RunbookAgentJobStatus.Failed,
+            RunbookAgentJobStatus.TimedOut,
+            RunbookAgentJobStatus.Cancelled,
+        ]);
+        const updated = await this.findOneById({
+            id: data.jobId,
+            select: {
+                _id: true,
+                status: true,
+                output: true,
+                exitCode: true,
+                errorMessage: true,
+                completedAt: true,
+            },
+            props: { isRoot: true },
+        });
+        if (!updated) {
+            throw new BadDataException(`RunbookAgentJob ${data.jobId.toString()} not found while timing out.`);
+        }
+        return updated;
+    }
+    async cancelJobsForExecution(data) {
+        const dataSource = PostgresAppInstance.getDataSource();
+        if (!dataSource) {
+            throw new BadDataException("Database is not connected");
+        }
+        try {
+            await dataSource.query(`UPDATE "RunbookAgentJob"
+         SET "status" = $1, "completedAt" = NOW(), "updatedAt" = NOW(), "version" = "version" + 1
+         WHERE "runbookExecutionId" = $2::uuid
+           AND "status" NOT IN ($3, $4, $5, $6)`, [
+                RunbookAgentJobStatus.Cancelled,
+                data.runbookExecutionId.toString(),
+                RunbookAgentJobStatus.Succeeded,
+                RunbookAgentJobStatus.Failed,
+                RunbookAgentJobStatus.TimedOut,
+                RunbookAgentJobStatus.Cancelled,
+            ]);
+        }
+        catch (err) {
+            logger.error("Failed to cancel RunbookAgentJobs for execution");
+            logger.error(err);
+        }
+    }
+}
+__decorate([
+    CaptureSpan(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], Service.prototype, "enqueue", null);
+__decorate([
+    CaptureSpan(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], Service.prototype, "claimNextJob", null);
+__decorate([
+    CaptureSpan(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], Service.prototype, "heartbeatJob", null);
+__decorate([
+    CaptureSpan(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], Service.prototype, "submitResult", null);
+__decorate([
+    CaptureSpan(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], Service.prototype, "pollUntilTerminal", null);
+__decorate([
+    CaptureSpan(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], Service.prototype, "timeoutJob", null);
+__decorate([
+    CaptureSpan(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], Service.prototype, "cancelJobsForExecution", null);
+export default new Service();
+//# sourceMappingURL=RunbookAgentJobService.js.map

package/build/dist/Server/Services/RunbookAgentJobService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"RunbookAgentJobService.js","sourceRoot":"","sources":["../../../../Server/Services/RunbookAgentJobService.ts"],"names":[],"mappings":";;;;;;;;;AAAA,OAAO,eAAe,MAAM,mBAAmB,CAAC;AAChD,OAAO,QAAQ,MAAM,sBAAsB,CAAC;AAC5C,OAAO,gBAAgB,MAAM,wCAAwC,CAAC;AACtE,OAAO,KAAK,MAAM,6CAA6C,CAAC;AAChE,OAAO,qBAAqB,MAAM,2CAA2C,CAAC;AAC9E,OAAO,eAAe,MAAM,qCAAqC,CAAC;AAClE,OAAO,aAAa,MAAM,kBAAkB,CAAC;AAE7C,OAAO,mBAAmB,MAAM,oCAAoC,CAAC;AACrE,OAAO,MAAM,MAAM,iBAAiB,CAAC;AACrC,OAAO,KAAK,MAAM,mBAAmB,CAAC;AACtC,OAAO,WAAW,MAAM,gCAAgC,CAAC;AAEzD;;;;GAIG;AACH,MAAM,gBAAgB,GAAW,KAAM,CAAC;AACxC;;;;GAIG;AACH,MAAM,wBAAwB,GAAW,CAAC,GAAG,KAAM,CAAC;AAEpD,MAAM,gBAAgB,GAAW,GAAG,CAAC;AAErC;;;;GAIG;AACH,SAAS,UAAU,CAAC,MAAe;IACjC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,IACE,MAAM,CAAC,MAAM,KAAK,CAAC;YACnB,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YACxB,OAAO,MAAM,CAAC,CAAC,CAAC,KAAK,QAAQ,EAC7B,CAAC;YACD,OAAO,MAAM,CAAC,CAAC,CAAsB,CAAC;QACxC,CAAC;QACD,OAAO,MAA2B,CAAC;IACrC,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,MAAM,OAAO,OAAQ,SAAQ,eAAsB;IACjD;QACE,KAAK,CAAC,KAAK,CAAC,CAAC;IACf,CAAC;IAGY,AAAN,KAAK,CAAC,OAAO,CAAC,IASpB;;QACC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;YACxB,MAAM,IAAI,gBAAgB,CACxB,kEAAkE,CACnE,CAAC;QACJ,CAAC;QAED,IACE,IAAI,CAAC,QAAQ,KAAK,eAAe,CAAC,IAAI;YACtC,IAAI,CAAC,QAAQ,KAAK,eAAe,CAAC,UAAU,EAC5C,CAAC;YACD,MAAM,IAAI,gBAAgB,CACxB,4CAA4C,IAAI,CAAC,QAAQ,IAAI,CAC9D,CAAC;QACJ,CAAC;QAED,MAAM,eAAe,GAAS,aAAa,CAAC,gBAAgB,CAC1D,aAAa,CAAC,cAAc,EAAE,EAC9B,IAAI,CAAC,IAAI,CAAC,CAAC,MAAA,IAAI,CAAC,gBAAgB,mCAAI,wBAAwB,CAAC,GAAG,IAAI,CAAC,CACtE,CAAC;QAEF,MAAM,GAAG,GAAU,IAAI,KAAK,EAAE,CAAC;QAC/B,GAAG,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QAC/B,GAAG,CAAC,kBAAkB,GAAG,IAAI,CAAC,kBAAkB,CAAC;QACjD,GAAG,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QACzB,GAAG,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC7B,GAAG,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;QACvC,GAAG,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QACzB,GAAG,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;QACnC,GAAG,CAAC,MAAM,GAAG,qBAAqB,CAAC,OAAO,CAAC;QAC3C,GAAG,CAAC,eAAe,GAAG,eAAe,CAAC;QAEtC,OAAO,IAAI,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED;;;;OAIG;IAEU,AAAN,KAAK,CAAC,YAAY,CAAC,IAIzB;;QACC,MAAM,UAAU,GACd,mBAAmB,CAAC,aAAa,EAAE,CAAC;QAEtC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,MAAM,IAAI,gBAAgB,CAAC,2BAA2B,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,OAAO,GAAW,MAAA,IAAI,CAAC,OAAO,mCAAI,gBAAgB,CAAC;QAEzD,MAAM,GAAG,GAAW;;;;;;;;;;;;;;;;;;;;;;;;KAwBnB,CAAC;QAEF,MAAM,IAAI,GAAsB,UAAU,CACxC,MAAM,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE;YAC1B,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE;YACzB,qBAAqB,CAAC,OAAO;YAC7B,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE;YACvB,qBAAqB,CAAC,OAAO;YAC7B,OAAO,CAAC,QAAQ,EAAE;SACnB,CAAC,CACH,CAAC;QAEF,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,CAAC,GAAe,IAAI,CAAC,CAAC,CAAE,CAAC;QAC/B,MAAM,GAAG,GAAU,IAAI,KAAK,EAAE,CAAC;QAC/B,GAAG,CAAC,GAAG,GAAG,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;QAC3B,GAAG,CAAC,SAAS,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;QACrD,GAAG,CAAC,kBAAkB,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC;QACvE,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;QACjC,GAAG,CAAC,QAAQ,GAAG,CAAC,CAAC,UAAU,CAAoB,CAAC;QAChD,GAAG,CAAC,aAAa,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC7D,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;QACjC,GAAG,CAAC,WAAW,GAAG,MAAM,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;QAC3C,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,QAAQ,CAA0B,CAAC;QAClD,IAAI,CAAC,CAAC,WAAW,CAAC,EAAE,CAAC;YACnB,GAAG,CAAC,SAAS,GAAG,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;QACnD,CAAC;QACD,IAAI,CAAC,CAAC,gBAAgB,CAAC,EAAE,CAAC;YACxB,GAAG,CAAC,cAAc,GAAG,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;QAC7D,CAAC;QACD,GAAG,CAAC,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC;QACnC,OAAO,GAAG,CAAC;IACb,CAAC;IAGY,AAAN,KAAK,CAAC,YAAY,CAAC,IAIzB;;QACC,MAAM,UAAU,GACd,mBAAmB,CAAC,aAAa,EAAE,CAAC;QACtC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,MAAM,IAAI,gBAAgB,CAAC,2BAA2B,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,OAAO,GAAW,MAAA,IAAI,CAAC,OAAO,mCAAI,gBAAgB,CAAC;QAEzD;;;;WAIG;QACH,MAAM,GAAG,GAAW;;;;;;;;;;;KAWnB,CAAC;QACF,MAAM,IAAI,GAAsB,UAAU,CACxC,MAAM,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE;YAC1B,OAAO,CAAC,QAAQ,EAAE;YAClB,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE;YACrB,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE;YACvB,qBAAqB,CAAC,OAAO;YAC7B,qBAAqB,CAAC,OAAO;SAC9B,CAAC,CACH,CAAC;QAEF,OAAO,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;IACzB,CAAC;IAGY,AAAN,KAAK,CAAC,YAAY,CAAC,IAOzB;;QACC,MAAM,UAAU,GACd,mBAAmB,CAAC,aAAa,EAAE,CAAC;QACtC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,MAAM,IAAI,gBAAgB,CAAC,2BAA2B,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,MAAM,GAA0B,IAAI,CAAC,OAAO;YAChD,CAAC,CAAC,qBAAqB,CAAC,SAAS;YACjC,CAAC,CAAC,qBAAqB,CAAC,MAAM,CAAC;QAEjC;;;;WAIG;QACH,MAAM,GAAG,GAAW;;;;;;;;;;;;;KAanB,CAAC;QACF,MAAM,IAAI,GAAsB,UAAU,CACxC,MAAM,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE;YAC1B,MAAM;YACN,MAAA,IAAI,CAAC,MAAM,mCAAI,IAAI;YACnB,MAAA,IAAI,CAAC,QAAQ,mCAAI,IAAI;YACrB,MAAA,IAAI,CAAC,YAAY,mCAAI,IAAI;YACzB,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE;YACrB,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE;YACvB,qBAAqB,CAAC,OAAO;YAC7B,qBAAqB,CAAC,OAAO;SAC9B,CAAC,CACH,CAAC;QAEF,OAAO,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;IACzB,CAAC;IAED;;;;;OAKG;IAEU,AAAN,KAAK,CAAC,iBAAiB,CAAC,IAI9B;QACC,MAAM,eAAe,GAAS,aAAa,CAAC,gBAAgB,CAC1D,aAAa,CAAC,cAAc,EAAE,EAC9B,IAAI,CAAC,IAAI,CACP,CAAC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,oBAAoB,GAAG,IAAK,CAAC,GAAG,IAAI,CACnE,CACF,CAAC;QAEF,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,GAAG,GAAiB,MAAM,IAAI,CAAC,WAAW,CAAC;gBAC/C,EAAE,EAAE,IAAI,CAAC,KAAK;gBACd,MAAM,EAAE;oBACN,GAAG,EAAE,IAAI;oBACT,MAAM,EAAE,IAAI;oBACZ,MAAM,EAAE,IAAI;oBACZ,QAAQ,EAAE,IAAI;oBACd,YAAY,EAAE,IAAI;oBAClB,eAAe,EAAE,IAAI;oBACrB,cAAc,EAAE,IAAI;oBACpB,SAAS,EAAE,IAAI;oBACf,SAAS,EAAE,IAAI;oBACf,WAAW,EAAE,IAAI;oBACjB,eAAe,EAAE,IAAI;iBACtB;gBACD,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;aACxB,CAAC,CAAC;YAEH,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,MAAM,IAAI,gBAAgB,CACxB,mBAAmB,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,6BAA6B,CACtE,CAAC;YACJ,CAAC;YAED,IACE,GAAG,CAAC,MAAM,KAAK,qBAAqB,CAAC,SAAS;gBAC9C,GAAG,CAAC,MAAM,KAAK,qBAAqB,CAAC,MAAM;gBAC3C,GAAG,CAAC,MAAM,KAAK,qBAAqB,CAAC,QAAQ;gBAC7C,GAAG,CAAC,MAAM,KAAK,qBAAqB,CAAC,SAAS,EAC9C,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC;YAED,MAAM,GAAG,GAAS,aAAa,CAAC,cAAc,EAAE,CAAC;YAEjD,gEAAgE;YAChE,IACE,GAAG,CAAC,MAAM,KAAK,qBAAqB,CAAC,OAAO;gBAC5C,GAAG,CAAC,eAAe;gBACnB,GAAG,GAAG,GAAG,CAAC,eAAe,EACzB,CAAC;gBACD,OAAO,IAAI,CAAC,UAAU,CAAC;oBACrB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,MAAM,EACJ,yJAAyJ;iBAC5J,CAAC,CAAC;YACL,CAAC;YAED,2DAA2D;YAC3D,IACE,CAAC,GAAG,CAAC,MAAM,KAAK,qBAAqB,CAAC,OAAO;gBAC3C,GAAG,CAAC,MAAM,KAAK,qBAAqB,CAAC,OAAO,CAAC;gBAC/C,GAAG,CAAC,cAAc;gBAClB,GAAG,GAAG,GAAG,CAAC,cAAc,EACxB,CAAC;gBACD,OAAO,IAAI,CAAC,UAAU,CAAC;oBACrB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,MAAM,EACJ,kMAAkM;iBACrM,CAAC,CAAC;YACL,CAAC;YAED,IAAI,GAAG,GAAG,eAAe,EAAE,CAAC;gBAC1B,OAAO,IAAI,CAAC,UAAU,CAAC;oBACrB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,MAAM,EACJ,8HAA8H;iBACjI,CAAC,CAAC;YACL,CAAC;YAED,MAAM,KAAK,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAGY,AAAN,KAAK,CAAC,UAAU,CAAC,IAGvB;QACC,MAAM,UAAU,GACd,mBAAmB,CAAC,aAAa,EAAE,CAAC;QACtC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,MAAM,IAAI,gBAAgB,CAAC,2BAA2B,CAAC,CAAC;QAC1D,CAAC;QAED;;;WAGG;QACH,MAAM,GAAG,GAAW;;;;;;;;;;KAUnB,CAAC;QACF,MAAM,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE;YAC1B,qBAAqB,CAAC,QAAQ;YAC9B,IAAI,CAAC,MAAM;YACX,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE;YACrB,qBAAqB,CAAC,SAAS;YAC/B,qBAAqB,CAAC,MAAM;YAC5B,qBAAqB,CAAC,QAAQ;YAC9B,qBAAqB,CAAC,SAAS;SAChC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAiB,MAAM,IAAI,CAAC,WAAW,CAAC;YACnD,EAAE,EAAE,IAAI,CAAC,KAAK;YACd,MAAM,EAAE;gBACN,GAAG,EAAE,IAAI;gBACT,MAAM,EAAE,IAAI;gBACZ,MAAM,EAAE,IAAI;gBACZ,QAAQ,EAAE,IAAI;gBACd,YAAY,EAAE,IAAI;gBAClB,WAAW,EAAE,IAAI;aAClB;YACD,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;SACxB,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,gBAAgB,CACxB,mBAAmB,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,8BAA8B,CACvE,CAAC;QACJ,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAGY,AAAN,KAAK,CAAC,sBAAsB,CAAC,IAEnC;QACC,MAAM,UAAU,GACd,mBAAmB,CAAC,aAAa,EAAE,CAAC;QACtC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,MAAM,IAAI,gBAAgB,CAAC,2BAA2B,CAAC,CAAC;QAC1D,CAAC;QACD,IAAI,CAAC;YACH,MAAM,UAAU,CAAC,KAAK,CACpB;;;gDAGwC,EACxC;gBACE,qBAAqB,CAAC,SAAS;gBAC/B,IAAI,CAAC,kBAAkB,CAAC,QAAQ,EAAE;gBAClC,qBAAqB,CAAC,SAAS;gBAC/B,qBAAqB,CAAC,MAAM;gBAC5B,qBAAqB,CAAC,QAAQ;gBAC9B,qBAAqB,CAAC,SAAS;aAChC,CACF,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,CAAC,KAAK,CAAC,iDAAiD,CAAC,CAAC;YAChE,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACpB,CAAC;IACH,CAAC;CACF;AApZc;IADZ,WAAW,EAAE;;;;sCA2Cb;AAQY;IADZ,WAAW,EAAE;;;;2CA0Eb;AAGY;IADZ,WAAW,EAAE;;;;2CA0Cb;AAGY;IADZ,WAAW,EAAE;;;;2CAoDb;AASY;IADZ,WAAW,EAAE;;;;gDAsFb;AAGY;IADZ,WAAW,EAAE;;;;yCAwDb;AAGY;IADZ,WAAW,EAAE;;;;qDA4Bb;AAGH,eAAe,IAAI,OAAO,EAAE,CAAC"}

package/build/dist/Server/Services/RunbookAgentOwnerTeamService.js

@@ -0,0 +1,9 @@
+import DatabaseService from "./DatabaseService";
+import Model from "../../Models/DatabaseModels/RunbookAgentOwnerTeam";
+export class Service extends DatabaseService {
+    constructor() {
+        super(Model);
+    }
+}
+export default new Service();
+//# sourceMappingURL=RunbookAgentOwnerTeamService.js.map

package/build/dist/Server/Services/RunbookAgentOwnerTeamService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"RunbookAgentOwnerTeamService.js","sourceRoot":"","sources":["../../../../Server/Services/RunbookAgentOwnerTeamService.ts"],"names":[],"mappings":"AAAA,OAAO,eAAe,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,MAAM,mDAAmD,CAAC;AAEtE,MAAM,OAAO,OAAQ,SAAQ,eAAsB;IACjD;QACE,KAAK,CAAC,KAAK,CAAC,CAAC;IACf,CAAC;CACF;AAED,eAAe,IAAI,OAAO,EAAE,CAAC"}

package/build/dist/Server/Services/RunbookAgentOwnerUserService.js

@@ -0,0 +1,9 @@
+import DatabaseService from "./DatabaseService";
+import Model from "../../Models/DatabaseModels/RunbookAgentOwnerUser";
+export class Service extends DatabaseService {
+    constructor() {
+        super(Model);
+    }
+}
+export default new Service();
+//# sourceMappingURL=RunbookAgentOwnerUserService.js.map

package/build/dist/Server/Services/RunbookAgentOwnerUserService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"RunbookAgentOwnerUserService.js","sourceRoot":"","sources":["../../../../Server/Services/RunbookAgentOwnerUserService.ts"],"names":[],"mappings":"AAAA,OAAO,eAAe,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,MAAM,mDAAmD,CAAC;AAEtE,MAAM,OAAO,OAAQ,SAAQ,eAAsB;IACjD;QACE,KAAK,CAAC,KAAK,CAAC,CAAC;IACf,CAAC;CACF;AAED,eAAe,IAAI,OAAO,EAAE,CAAC"}

package/build/dist/Server/Services/RunbookAgentService.js

@@ -0,0 +1,91 @@
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+import DatabaseService from "./DatabaseService";
+import ObjectID from "../../Types/ObjectID";
+import Version from "../../Types/Version";
+import BadDataException from "../../Types/Exception/BadDataException";
+import Model, { RunbookAgentConnectionStatus, } from "../../Models/DatabaseModels/RunbookAgent";
+import OneUptimeDate from "../../Types/Date";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
+export class Service extends DatabaseService {
+    constructor() {
+        super(Model);
+    }
+    async onBeforeCreate(createBy) {
+        if (!createBy.data.key) {
+            createBy.data.key = ObjectID.generate().toString();
+        }
+        if (!createBy.data.agentVersion) {
+            createBy.data.agentVersion = new Version("1.0.0");
+        }
+        if (!createBy.data.connectionStatus) {
+            createBy.data.connectionStatus =
+                RunbookAgentConnectionStatus.Disconnected;
+        }
+        return { createBy, carryForward: [] };
+    }
+    async findByIdAndKey(data) {
+        if (!data.agentId || !data.agentKey) {
+            return null;
+        }
+        return this.findOneBy({
+            query: {
+                _id: data.agentId.toString(),
+                key: data.agentKey,
+            },
+            select: {
+                _id: true,
+                projectId: true,
+                name: true,
+            },
+            props: { isRoot: true },
+        });
+    }
+    async heartbeat(data) {
+        if (!data.agentId) {
+            throw new BadDataException("agentId is required");
+        }
+        const update = {
+            lastAlive: OneUptimeDate.getCurrentDate(),
+            connectionStatus: RunbookAgentConnectionStatus.Connected,
+        };
+        if (data.agentVersion) {
+            update["agentVersion"] = data.agentVersion;
+        }
+        if (data.hostInfo) {
+            update["hostInfo"] = data.hostInfo;
+        }
+        await this.updateOneById({
+            id: data.agentId,
+            data: update,
+            props: { isRoot: true },
+        });
+    }
+}
+__decorate([
+    CaptureSpan(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], Service.prototype, "onBeforeCreate", null);
+__decorate([
+    CaptureSpan(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], Service.prototype, "findByIdAndKey", null);
+__decorate([
+    CaptureSpan(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], Service.prototype, "heartbeat", null);
+export default new Service();
+//# sourceMappingURL=RunbookAgentService.js.map

package/build/dist/Server/Services/RunbookAgentService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"RunbookAgentService.js","sourceRoot":"","sources":["../../../../Server/Services/RunbookAgentService.ts"],"names":[],"mappings":";;;;;;;;;AAEA,OAAO,eAAe,MAAM,mBAAmB,CAAC;AAChD,OAAO,QAAQ,MAAM,sBAAsB,CAAC;AAC5C,OAAO,OAAO,MAAM,qBAAqB,CAAC;AAC1C,OAAO,gBAAgB,MAAM,wCAAwC,CAAC;AACtE,OAAO,KAAK,EAAE,EACZ,4BAA4B,GAC7B,MAAM,0CAA0C,CAAC;AAClD,OAAO,aAAa,MAAM,kBAAkB,CAAC;AAE7C,OAAO,WAAW,MAAM,gCAAgC,CAAC;AAEzD,MAAM,OAAO,OAAQ,SAAQ,eAAsB;IACjD;QACE,KAAK,CAAC,KAAK,CAAC,CAAC;IACf,CAAC;IAGwB,AAAN,KAAK,CAAC,cAAc,CACrC,QAAyB;QAEzB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;YACvB,QAAQ,CAAC,IAAI,CAAC,GAAG,GAAG,QAAQ,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC;QACrD,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YAChC,QAAQ,CAAC,IAAI,CAAC,YAAY,GAAG,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC;QACpD,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACpC,QAAQ,CAAC,IAAI,CAAC,gBAAgB;gBAC5B,4BAA4B,CAAC,YAAY,CAAC;QAC9C,CAAC;QAED,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;IACxC,CAAC;IAGY,AAAN,KAAK,CAAC,cAAc,CAAC,IAG3B;QACC,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACpC,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC;YACpB,KAAK,EAAE;gBACL,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE;gBAC5B,GAAG,EAAE,IAAI,CAAC,QAAQ;aACnB;YACD,MAAM,EAAE;gBACN,GAAG,EAAE,IAAI;gBACT,SAAS,EAAE,IAAI;gBACf,IAAI,EAAE,IAAI;aACX;YACD,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;SACxB,CAAC,CAAC;IACL,CAAC;IAGY,AAAN,KAAK,CAAC,SAAS,CAAC,IAItB;QACC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,MAAM,IAAI,gBAAgB,CAAC,qBAAqB,CAAC,CAAC;QACpD,CAAC;QAED,MAAM,MAAM,GAAe;YACzB,SAAS,EAAE,aAAa,CAAC,cAAc,EAAE;YACzC,gBAAgB,EAAE,4BAA4B,CAAC,SAAS;SACzD,CAAC;QAEF,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,MAAM,CAAC,cAAc,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC;QAC7C,CAAC;QAED,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC;QACrC,CAAC;QAED,MAAM,IAAI,CAAC,aAAa,CAAC;YACvB,EAAE,EAAE,IAAI,CAAC,OAAO;YAChB,IAAI,EAAE,MAAe;YACrB,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;SACxB,CAAC,CAAC;IACL,CAAC;CACF;AAvE0B;IADxB,WAAW,EAAE;;;;6CAkBb;AAGY;IADZ,WAAW,EAAE;;;;6CAqBb;AAGY;IADZ,WAAW,EAAE;;;;wCA4Bb;AAGH,eAAe,IAAI,OAAO,EAAE,CAAC"}

package/build/dist/Server/Services/RunbookExecutionService.js

@@ -0,0 +1,9 @@
+import DatabaseService from "./DatabaseService";
+import Model from "../../Models/DatabaseModels/RunbookExecution";
+export class Service extends DatabaseService {
+    constructor() {
+        super(Model);
+    }
+}
+export default new Service();
+//# sourceMappingURL=RunbookExecutionService.js.map

package/build/dist/Server/Services/RunbookExecutionService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"RunbookExecutionService.js","sourceRoot":"","sources":["../../../../Server/Services/RunbookExecutionService.ts"],"names":[],"mappings":"AAAA,OAAO,eAAe,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,MAAM,8CAA8C,CAAC;AAEjE,MAAM,OAAO,OAAQ,SAAQ,eAAsB;IACjD;QACE,KAAK,CAAC,KAAK,CAAC,CAAC;IACf,CAAC;CACF;AAED,eAAe,IAAI,OAAO,EAAE,CAAC"}