@shipfox/api-runners 5.0.0 → 7.0.1

package/dist/db/runner-instances.js

@@ -0,0 +1,497 @@
+import { logger } from '@shipfox/node-opentelemetry';
+import { canonicalizeLabels } from '@shipfox/runner-labels';
+import { and, asc, desc, eq, exists, gt, inArray, isNotNull, isNull, lt, notExists, notInArray, or, sql } from 'drizzle-orm';
+import { alias } from 'drizzle-orm/pg-core';
+import { db } from './db.js';
+import { listRunningJobExecutionsByRunnerInstanceTx } from './job-executions.js';
+import { releaseReservationUnits } from './reservations.js';
+import { ephemeralRegistrationTokens } from './schema/ephemeral-registration-tokens.js';
+import { provisionerTokens } from './schema/provisioner-tokens.js';
+import { providerRunners, toRunnerInstance } from './schema/runner-instances.js';
+import { runnerSessions } from './schema/runner-sessions.js';
+import { runningJobExecutions } from './schema/running-job-executions.js';
+export const terminalStates = [
+    'stopped',
+    'failed',
+    'terminated'
+];
+export const activeStates = [
+    'starting',
+    'running',
+    'stopping'
+];
+export const divergenceCountStates = [
+    'starting',
+    'running'
+];
+export async function attachRunnerInstanceProviderId(params) {
+    const updated = await db().update(providerRunners).set({
+        providerRunnerId: params.providerRunnerId,
+        updatedAt: sql`now()`
+    }).where(and(eq(providerRunners.id, params.runnerInstanceId), eq(providerRunners.provisionerId, params.provisionerId), isNull(providerRunners.providerRunnerId), notInArray(providerRunners.state, [
+        ...terminalStates
+    ]))).returning({
+        id: providerRunners.id
+    });
+    return updated.length === 1;
+}
+export async function reportRunnerInstances(params) {
+    if (params.events.length === 0) return {
+        accepted: 0,
+        reservationsReleased: 0,
+        terminateIntentsHonored: []
+    };
+    return await db().transaction(async (tx)=>{
+        const receivedAt = new Date();
+        const aggregatedEvents = aggregateEvents(params.events, receivedAt);
+        const hasTerminalEvent = aggregatedEvents.some((event)=>isTerminalState(event.state));
+        if (hasTerminalEvent) {
+            await tx.execute(sql`select pg_advisory_xact_lock(hashtext(${params.workspaceId ?? params.provisionerId}))`);
+        }
+        const events = await hydrateRunnerSessionIdsFromConsumedTokens(tx, params, aggregatedEvents);
+        const terminateIntentsHonored = await listTerminateIntentsHonoredByTerminatedReportsTx(tx, params, events);
+        const values = events.map((event)=>({
+                workspaceId: params.workspaceId,
+                provisionerId: params.provisionerId,
+                providerRunnerId: event.providerRunnerId,
+                reservationId: event.reservationId,
+                templateKey: event.templateKey,
+                labels: [
+                    ...canonicalizeLabels(event.labels)
+                ],
+                state: event.state,
+                reason: event.reason,
+                runnerSessionId: event.runnerSessionId,
+                providerKind: event.providerKind,
+                reportedAt: event.reportedAt > receivedAt ? receivedAt : event.reportedAt,
+                startedAt: event.startedAt,
+                stoppingAt: event.stoppingAt,
+                stoppedAt: event.stoppedAt,
+                failedAt: event.failedAt,
+                terminatedAt: event.terminatedAt
+            }));
+        await tx.insert(providerRunners).values(values).onConflictDoUpdate({
+            target: [
+                providerRunners.provisionerId,
+                providerRunners.providerRunnerId
+            ],
+            targetWhere: isNotNull(providerRunners.providerRunnerId),
+            set: {
+                reservationId: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN coalesce(excluded.reservation_id, ${providerRunners.reservationId}) ELSE ${providerRunners.reservationId} END`,
+                templateKey: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN coalesce(excluded.template_key, ${providerRunners.templateKey}) ELSE ${providerRunners.templateKey} END`,
+                labels: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN excluded.labels ELSE ${providerRunners.labels} END`,
+                state: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN excluded.state ELSE ${providerRunners.state} END`,
+                reason: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN excluded.reason ELSE ${providerRunners.reason} END`,
+                runnerSessionId: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN coalesce(${providerRunners.runnerSessionId}, excluded.runner_session_id) ELSE ${providerRunners.runnerSessionId} END`,
+                providerKind: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN coalesce(excluded.provider_kind, ${providerRunners.providerKind}) ELSE ${providerRunners.providerKind} END`,
+                reportedAt: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN excluded.reported_at ELSE ${providerRunners.reportedAt} END`,
+                startedAt: firstObservedAt(providerRunners.startedAt, sql`excluded.started_at`),
+                stoppingAt: firstObservedAt(providerRunners.stoppingAt, sql`excluded.stopping_at`),
+                stoppedAt: firstObservedAt(providerRunners.stoppedAt, sql`excluded.stopped_at`),
+                failedAt: firstObservedAt(providerRunners.failedAt, sql`excluded.failed_at`),
+                terminatedAt: firstObservedAt(providerRunners.terminatedAt, sql`excluded.terminated_at`),
+                updatedAt: sql`now()`
+            },
+            setWhere: sql`
+          ${providerRunnerProjectionUpdateCondition()}
+          OR ${providerRunnerMilestoneUpdateCondition()}
+        `
+        });
+        const reservationsReleased = hasTerminalEvent ? await releaseTerminalRunnerInstanceReservations(tx, params, events) : 0;
+        return {
+            accepted: events.length,
+            reservationsReleased,
+            terminateIntentsHonored
+        };
+    });
+}
+export async function listActiveRunnerInstanceCountsByTemplateTx(tx, params) {
+    const rows = await tx.select({
+        templateKey: providerRunners.templateKey,
+        state: providerRunners.state,
+        count: sql`count(*)::int`
+    }).from(providerRunners).where(and(eq(providerRunners.workspaceId, params.workspaceId), eq(providerRunners.provisionerId, params.provisionerId), inArray(providerRunners.state, divergenceCountStates), isNotNull(providerRunners.templateKey))).groupBy(providerRunners.templateKey, providerRunners.state);
+    return rows.flatMap((row)=>row.templateKey && isDivergenceCountState(row.state) ? [
+            {
+                templateKey: row.templateKey,
+                state: row.state,
+                count: row.count
+            }
+        ] : []);
+}
+export async function listActiveRunnerInstances(params) {
+    const rows = await db().select().from(providerRunners).where(and(eq(providerRunners.workspaceId, params.workspaceId), inArray(providerRunners.state, activeStates), sql`${providerRunners.updatedAt} > now() - (${params.windowSeconds} || ' seconds')::interval`)).orderBy(desc(providerRunners.updatedAt), desc(providerRunners.id)).limit(params.limit ?? 1000);
+    return rows.map(toRunnerInstance);
+}
+export async function listProvisionerTerminateIntents(params) {
+    return await db().transaction(async (tx)=>{
+        const rows = await listProvisionerTerminateIntentRowsTx(tx, params);
+        return rows.map((row)=>row.providerRunnerId);
+    });
+}
+export async function listProvisionerTerminateIntentRowsTx(tx, params) {
+    const rows = await provisionerTerminateIntentsQuery(tx, params).orderBy(asc(providerRunners.providerRunnerId)).limit(params.limit + 1);
+    const truncated = rows.length > params.limit;
+    const returnedRows = truncated ? rows.slice(0, params.limit) : rows;
+    if (truncated) {
+        logger().warn({
+            workspaceId: params.workspaceId,
+            provisionerId: params.provisionerId,
+            limit: params.limit,
+            returnedCount: returnedRows.length
+        }, 'provisioner terminate intents truncated by poll-demand limit');
+    }
+    return returnedRows.flatMap((row)=>row.providerRunnerId ? [
+            {
+                ...row,
+                providerRunnerId: row.providerRunnerId
+            }
+        ] : []);
+}
+function provisionerTerminateIntentsQuery(tx, params) {
+    const newerRunningJobExecutions = alias(runningJobExecutions, 'newer_running_jobs');
+    return tx.select({
+        providerRunnerId: providerRunners.providerRunnerId,
+        reason: sql`'job-cancelled'`
+    }).from(runningJobExecutions).innerJoin(providerRunners, and(eq(providerRunners.workspaceId, runningJobExecutions.workspaceId), eq(providerRunners.provisionerId, runningJobExecutions.provisionerId), eq(providerRunners.providerRunnerId, runningJobExecutions.providerRunnerId))).where(and(eq(runningJobExecutions.workspaceId, params.workspaceId), eq(runningJobExecutions.provisionerId, params.provisionerId), isNotNull(providerRunners.providerRunnerId), isNotNull(runningJobExecutions.cancellationRequestedAt), inArray(providerRunners.state, activeStates), params.providerRunnerIds && params.providerRunnerIds.length > 0 ? inArray(providerRunners.providerRunnerId, params.providerRunnerIds) : undefined, notExists(tx.select({
+        id: newerRunningJobExecutions.id
+    }).from(newerRunningJobExecutions).where(and(eq(newerRunningJobExecutions.workspaceId, runningJobExecutions.workspaceId), eq(newerRunningJobExecutions.provisionerId, runningJobExecutions.provisionerId), eq(newerRunningJobExecutions.providerRunnerId, runningJobExecutions.providerRunnerId), or(gt(newerRunningJobExecutions.startedAt, runningJobExecutions.startedAt), and(eq(newerRunningJobExecutions.startedAt, runningJobExecutions.startedAt), gt(newerRunningJobExecutions.jobExecutionId, runningJobExecutions.jobExecutionId)))))))).groupBy(providerRunners.providerRunnerId);
+}
+async function listTerminateIntentsHonoredByTerminatedReportsTx(tx, params, events) {
+    if (!params.workspaceId) return [];
+    const terminatedRunnerInstanceIds = [
+        ...new Set(events.filter((event)=>event.state === 'terminated').map((event)=>event.providerRunnerId))
+    ];
+    if (terminatedRunnerInstanceIds.length === 0) return [];
+    const rows = await provisionerTerminateIntentsQuery(tx, {
+        workspaceId: params.workspaceId,
+        provisionerId: params.provisionerId,
+        providerRunnerIds: terminatedRunnerInstanceIds
+    }).orderBy(asc(providerRunners.providerRunnerId));
+    return rows.flatMap((row)=>row.providerRunnerId ? [
+            {
+                ...row,
+                providerRunnerId: row.providerRunnerId
+            }
+        ] : []);
+}
+export async function reconcileRunnerInstances(params) {
+    const observedRunnerInstanceIds = [
+        ...new Set(params.observedRunnerInstanceIds)
+    ];
+    return await db().transaction(async (tx)=>{
+        await tx.execute(sql`select pg_advisory_xact_lock(hashtext(${params.workspaceId ?? params.provisionerId}))`);
+        let absentIds = [];
+        let reservationsReleased = 0;
+        if (observedRunnerInstanceIds.length > 0) {
+            const staleAbsentRows = await tx.select({
+                id: providerRunners.id,
+                providerRunnerId: providerRunners.providerRunnerId
+            }).from(providerRunners).where(and(params.workspaceId ? eq(providerRunners.workspaceId, params.workspaceId) : isNull(providerRunners.workspaceId), eq(providerRunners.provisionerId, params.provisionerId), inArray(providerRunners.state, activeStates), lt(providerRunners.reportedAt, sql`now() - (${params.terminateGraceSeconds} || ' seconds')::interval`), notInArray(providerRunners.providerRunnerId, observedRunnerInstanceIds)));
+            if (staleAbsentRows.length > 0) {
+                const updated = await tx.update(providerRunners).set({
+                    state: 'terminated',
+                    terminatedAt: sql`coalesce(${providerRunners.terminatedAt}, now())`,
+                    updatedAt: sql`now()`
+                }).where(and(inArray(providerRunners.id, staleAbsentRows.map((row)=>row.id)), inArray(providerRunners.state, activeStates), lt(providerRunners.reportedAt, sql`now() - (${params.terminateGraceSeconds} || ' seconds')::interval`))).returning({
+                    providerRunnerId: providerRunners.providerRunnerId
+                });
+                absentIds = updated.flatMap((row)=>row.providerRunnerId ? [
+                        row.providerRunnerId
+                    ] : []);
+                if (params.workspaceId) {
+                    reservationsReleased = await releaseTerminalRunnerInstanceReservationsByIds(tx, {
+                        workspaceId: params.workspaceId,
+                        provisionerId: params.provisionerId,
+                        providerRunnerIds: absentIds
+                    });
+                }
+            }
+        }
+        const observedRows = observedRunnerInstanceIds.length === 0 ? [] : (await tx.select().from(providerRunners).where(and(params.workspaceId ? eq(providerRunners.workspaceId, params.workspaceId) : isNull(providerRunners.workspaceId), eq(providerRunners.provisionerId, params.provisionerId), inArray(providerRunners.providerRunnerId, observedRunnerInstanceIds)))).map(toRunnerInstance);
+        const boundJobExecutions = params.workspaceId ? await listRunningJobExecutionsByRunnerInstanceTx(tx, {
+            workspaceId: params.workspaceId,
+            provisionerId: params.provisionerId,
+            providerRunnerIds: observedRunnerInstanceIds
+        }) : [];
+        return {
+            observedRows,
+            boundJobExecutionsByRunnerInstanceId: new Map(boundJobExecutions.map((jobExecution)=>[
+                    jobExecution.providerRunnerId,
+                    jobExecution
+                ])),
+            absentIds,
+            reservationsReleased
+        };
+    });
+}
+export async function reapStaleRunnerInstances(params) {
+    const cutoff = staleRunnerInstanceCutoff(params.thresholdSeconds);
+    return await db().transaction(async (tx)=>{
+        const candidateRows = await tx.select({
+            id: providerRunners.id,
+            workspaceId: providerRunners.workspaceId,
+            provisionerId: providerRunners.provisionerId,
+            providerRunnerId: providerRunners.providerRunnerId
+        }).from(providerRunners).where(staleRunnerInstanceWhere(tx, cutoff)).orderBy(asc(providerRunners.updatedAt), asc(providerRunners.id)).limit(params.limit);
+        if (candidateRows.length === 0) return {
+            reaped: 0,
+            reservationsReleased: 0
+        };
+        const workspaceIds = [
+            ...new Set(candidateRows.flatMap((row)=>row.workspaceId ? [
+                    row.workspaceId
+                ] : []))
+        ].sort();
+        for (const workspaceId of workspaceIds){
+            await tx.execute(sql`select pg_advisory_xact_lock(hashtext(${workspaceId}))`);
+        }
+        const updatedRows = await tx.update(providerRunners).set({
+            state: 'failed',
+            reason: 'stale-provisioner',
+            failedAt: sql`coalesce(${providerRunners.failedAt}, now())`,
+            updatedAt: sql`now()`
+        }).where(and(inArray(providerRunners.id, candidateRows.map((row)=>row.id)), staleRunnerInstanceWhere(tx, cutoff))).returning({
+            workspaceId: providerRunners.workspaceId,
+            provisionerId: providerRunners.provisionerId,
+            providerRunnerId: providerRunners.providerRunnerId
+        });
+        let reservationsReleased = 0;
+        for (const group of groupRunnerInstanceIds(updatedRows)){
+            reservationsReleased += await releaseTerminalRunnerInstanceReservationsByIds(tx, {
+                workspaceId: group.workspaceId,
+                provisionerId: group.provisionerId,
+                providerRunnerIds: group.providerRunnerIds,
+                requireUnlinkedSession: false
+            });
+        }
+        return {
+            reaped: updatedRows.length,
+            reservationsReleased
+        };
+    });
+}
+async function releaseTerminalRunnerInstanceReservations(tx, params, events) {
+    const terminalEvents = events.filter((event)=>isTerminalState(event.state));
+    if (terminalEvents.length === 0 || !params.workspaceId) return 0;
+    return await releaseTerminalRunnerInstanceReservationsByIds(tx, {
+        workspaceId: params.workspaceId,
+        provisionerId: params.provisionerId,
+        providerRunnerIds: terminalEvents.map((event)=>event.providerRunnerId)
+    });
+}
+async function releaseTerminalRunnerInstanceReservationsByIds(tx, params) {
+    if (params.providerRunnerIds.length === 0) return 0;
+    const rows = await tx.select({
+        id: providerRunners.id,
+        reservationId: providerRunners.reservationId
+    }).from(providerRunners).where(and(eq(providerRunners.workspaceId, params.workspaceId), eq(providerRunners.provisionerId, params.provisionerId), inArray(providerRunners.providerRunnerId, params.providerRunnerIds), inArray(providerRunners.state, terminalStates), isNotNull(providerRunners.reservationId), params.requireUnlinkedSession === false ? undefined : isNull(providerRunners.runnerSessionId), isNull(providerRunners.reservationReleasedAt)));
+    if (rows.length === 0) return 0;
+    const updated = await tx.update(providerRunners).set({
+        reservationReleasedAt: sql`now()`,
+        updatedAt: sql`now()`
+    }).where(and(inArray(providerRunners.id, rows.map((row)=>row.id)), params.requireUnlinkedSession === false ? undefined : isNull(providerRunners.runnerSessionId), isNull(providerRunners.reservationReleasedAt))).returning({
+        reservationId: providerRunners.reservationId
+    });
+    const releasesByReservationId = new Map();
+    for (const row of updated){
+        if (!row.reservationId) continue;
+        releasesByReservationId.set(row.reservationId, (releasesByReservationId.get(row.reservationId) ?? 0) + 1);
+    }
+    if (releasesByReservationId.size === 0) return 0;
+    return await releaseReservationUnits(tx, {
+        workspaceId: params.workspaceId,
+        provisionerId: params.provisionerId,
+        releases: [
+            ...releasesByReservationId
+        ].map(([reservationId, count])=>({
+                reservationId,
+                count
+            }))
+    });
+}
+function staleRunnerInstanceCutoff(thresholdSeconds) {
+    return sql`now() - (${thresholdSeconds} || ' seconds')::interval`;
+}
+function staleRunnerInstanceWhere(tx, cutoff) {
+    return and(inArray(providerRunners.state, activeStates), lt(providerRunners.reportedAt, cutoff), lt(providerRunners.updatedAt, cutoff), exists(tx.select({
+        id: provisionerTokens.id
+    }).from(provisionerTokens).where(and(eq(provisionerTokens.id, providerRunners.provisionerId), or(isNull(provisionerTokens.lastSeenAt), lt(provisionerTokens.lastSeenAt, cutoff))))), notExists(tx.select({
+        id: runningJobExecutions.id
+    }).from(runningJobExecutions).where(and(eq(runningJobExecutions.workspaceId, providerRunners.workspaceId), eq(runningJobExecutions.provisionerId, providerRunners.provisionerId), eq(runningJobExecutions.providerRunnerId, providerRunners.providerRunnerId)))), notExists(tx.select({
+        id: runnerSessions.id
+    }).from(runnerSessions).where(and(eq(runnerSessions.workspaceId, providerRunners.workspaceId), eq(runnerSessions.provisionerId, providerRunners.provisionerId), eq(runnerSessions.providerRunnerId, providerRunners.providerRunnerId), sql`${runnerSessions.updatedAt} >= ${cutoff}`))));
+}
+function groupRunnerInstanceIds(rows) {
+    const groups = new Map();
+    for (const row of rows){
+        if (!row.workspaceId || !row.providerRunnerId) continue;
+        const key = `${row.workspaceId}:${row.provisionerId}`;
+        const group = groups.get(key) ?? {
+            workspaceId: row.workspaceId,
+            provisionerId: row.provisionerId,
+            providerRunnerIds: []
+        };
+        group.providerRunnerIds.push(row.providerRunnerId);
+        groups.set(key, group);
+    }
+    return [
+        ...groups.values()
+    ];
+}
+async function hydrateRunnerSessionIdsFromConsumedTokens(tx, params, events) {
+    if (!params.workspaceId) return events;
+    const providerRunnerIds = [
+        ...new Set(events.map((event)=>event.providerRunnerId))
+    ];
+    if (providerRunnerIds.length === 0) return events;
+    const tokenRows = await tx.select({
+        providerRunnerId: ephemeralRegistrationTokens.providerRunnerId,
+        consumedSessionId: ephemeralRegistrationTokens.consumedSessionId
+    }).from(ephemeralRegistrationTokens).where(and(eq(ephemeralRegistrationTokens.workspaceId, params.workspaceId), eq(ephemeralRegistrationTokens.provisionerId, params.provisionerId), inArray(ephemeralRegistrationTokens.providerRunnerId, providerRunnerIds), isNotNull(ephemeralRegistrationTokens.consumedSessionId))).orderBy(desc(ephemeralRegistrationTokens.consumedAt), desc(ephemeralRegistrationTokens.createdAt));
+    const consumedSessionIdsByRunnerInstanceId = new Map();
+    for (const row of tokenRows){
+        if (!row.consumedSessionId) continue;
+        if (consumedSessionIdsByRunnerInstanceId.has(row.providerRunnerId)) continue;
+        consumedSessionIdsByRunnerInstanceId.set(row.providerRunnerId, row.consumedSessionId);
+    }
+    if (consumedSessionIdsByRunnerInstanceId.size === 0) return events;
+    return events.map((event)=>{
+        const consumedSessionId = consumedSessionIdsByRunnerInstanceId.get(event.providerRunnerId);
+        if (!consumedSessionId || event.runnerSessionId === consumedSessionId) return event;
+        return {
+            ...event,
+            runnerSessionId: consumedSessionId
+        };
+    });
+}
+function aggregateEvents(events, receivedAt) {
+    const byRunnerInstanceId = new Map();
+    for (const rawEvent of events){
+        const event = toRunnerInstanceReportRow(rawEvent, receivedAt);
+        const existing = byRunnerInstanceId.get(event.providerRunnerId);
+        if (existing) mergeMilestones(existing, event);
+        if (!existing || compareRunnerInstanceReportEvents(event, existing) > 0) {
+            byRunnerInstanceId.set(event.providerRunnerId, existing ? mergeProjectionMetadata(event, existing) : event);
+        }
+    }
+    return [
+        ...byRunnerInstanceId.values()
+    ];
+}
+function toRunnerInstanceReportRow(event, receivedAt) {
+    const reportedAt = event.reportedAt > receivedAt ? receivedAt : event.reportedAt;
+    return {
+        ...event,
+        reportedAt,
+        startedAt: event.state === 'running' ? reportedAt : null,
+        stoppingAt: event.state === 'stopping' ? reportedAt : null,
+        stoppedAt: event.state === 'stopped' ? reportedAt : null,
+        failedAt: event.state === 'failed' ? reportedAt : null,
+        terminatedAt: event.state === 'terminated' ? reportedAt : null
+    };
+}
+function mergeMilestones(target, source) {
+    target.startedAt = earliestDate(target.startedAt, source.startedAt);
+    target.stoppingAt = earliestDate(target.stoppingAt, source.stoppingAt);
+    target.stoppedAt = earliestDate(target.stoppedAt, source.stoppedAt);
+    target.failedAt = earliestDate(target.failedAt, source.failedAt);
+    target.terminatedAt = earliestDate(target.terminatedAt, source.terminatedAt);
+}
+function mergeProjectionMetadata(event, existing) {
+    return {
+        ...event,
+        reservationId: event.reservationId ?? existing.reservationId,
+        templateKey: event.templateKey ?? existing.templateKey,
+        runnerSessionId: event.runnerSessionId ?? existing.runnerSessionId,
+        providerKind: event.providerKind ?? existing.providerKind,
+        ...pickMilestones(existing)
+    };
+}
+function pickMilestones(event) {
+    return {
+        startedAt: event.startedAt,
+        stoppingAt: event.stoppingAt,
+        stoppedAt: event.stoppedAt,
+        failedAt: event.failedAt,
+        terminatedAt: event.terminatedAt
+    };
+}
+function earliestDate(a, b) {
+    if (!a) return b;
+    if (!b) return a;
+    return a < b ? a : b;
+}
+function compareRunnerInstanceReportEvents(a, b) {
+    const timeDelta = a.reportedAt.getTime() - b.reportedAt.getTime();
+    const rankDelta = getRunnerInstanceStateRank(a.state) - getRunnerInstanceStateRank(b.state);
+    if (rankDelta !== 0) return rankDelta;
+    return timeDelta;
+}
+function getRunnerInstanceStateRank(state) {
+    switch(state){
+        case 'starting':
+            return 1;
+        case 'running':
+            return 2;
+        case 'stopping':
+            return 3;
+        case 'stopped':
+            return 4;
+        case 'failed':
+            return 5;
+        case 'terminated':
+            return 6;
+    }
+}
+function providerRunnerStateRank(state) {
+    return sql`
+    CASE ${state}
+      WHEN 'starting' THEN 1
+      WHEN 'running' THEN 2
+      WHEN 'stopping' THEN 3
+      WHEN 'stopped' THEN 4
+      WHEN 'failed' THEN 5
+      WHEN 'terminated' THEN 6
+      ELSE 0
+    END
+  `;
+}
+function providerRunnerProjectionUpdateCondition() {
+    return sql`
+    ${providerRunnerStateRank(sql`excluded.state`)} > ${providerRunnerStateRank(providerRunners.state)}
+    OR (
+      ${providerRunnerStateRank(sql`excluded.state`)} = ${providerRunnerStateRank(providerRunners.state)}
+      AND excluded.reported_at >= ${providerRunners.reportedAt}
+    )
+  `;
+}
+function providerRunnerMilestoneUpdateCondition() {
+    return sql`
+    ${milestoneNeedsUpdate(providerRunners.startedAt, sql`excluded.started_at`)}
+    OR ${milestoneNeedsUpdate(providerRunners.stoppingAt, sql`excluded.stopping_at`)}
+    OR ${milestoneNeedsUpdate(providerRunners.stoppedAt, sql`excluded.stopped_at`)}
+    OR ${milestoneNeedsUpdate(providerRunners.failedAt, sql`excluded.failed_at`)}
+    OR ${milestoneNeedsUpdate(providerRunners.terminatedAt, sql`excluded.terminated_at`)}
+  `;
+}
+function milestoneNeedsUpdate(current, incoming) {
+    return sql`${incoming} IS NOT NULL AND (${current} IS NULL OR ${incoming} < ${current})`;
+}
+function firstObservedAt(current, incoming) {
+    return sql`
+    CASE
+      WHEN ${incoming} IS NULL THEN ${current}
+      WHEN ${current} IS NULL THEN ${incoming}
+      ELSE least(${current}, ${incoming})
+    END
+  `;
+}
+export function isTerminalState(state) {
+    return terminalStates.includes(state);
+}
+function isDivergenceCountState(state) {
+    return divergenceCountStates.includes(state);
+}
+
+//# sourceMappingURL=runner-instances.js.map

package/dist/db/runner-instances.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/db/runner-instances.ts"],"sourcesContent":["import {logger} from '@shipfox/node-opentelemetry';\nimport {canonicalizeLabels} from '@shipfox/runner-labels';\nimport {\n  and,\n  asc,\n  desc,\n  eq,\n  exists,\n  gt,\n  inArray,\n  isNotNull,\n  isNull,\n  lt,\n  notExists,\n  notInArray,\n  or,\n  type SQL,\n  sql,\n} from 'drizzle-orm';\nimport {alias} from 'drizzle-orm/pg-core';\nimport type {RunnerInstance, RunnerInstanceState} from '#core/entities/runner-instance.js';\nimport type {Tx} from './db.js';\nimport {db} from './db.js';\nimport {\n  listRunningJobExecutionsByRunnerInstanceTx,\n  type RunnerInstanceBoundJobExecution,\n} from './job-executions.js';\nimport {releaseReservationUnits} from './reservations.js';\nimport {ephemeralRegistrationTokens} from './schema/ephemeral-registration-tokens.js';\nimport {provisionerTokens} from './schema/provisioner-tokens.js';\nimport {providerRunners, toRunnerInstance} from './schema/runner-instances.js';\nimport {runnerSessions} from './schema/runner-sessions.js';\nimport {runningJobExecutions} from './schema/running-job-executions.js';\n\nexport const terminalStates = [\n  'stopped',\n  'failed',\n  'terminated',\n] as const satisfies readonly RunnerInstanceState[];\nexport const activeStates = [\n  'starting',\n  'running',\n  'stopping',\n] as const satisfies readonly RunnerInstanceState[];\nexport const divergenceCountStates = ['starting', 'running'] as const satisfies readonly Extract<\n  RunnerInstanceState,\n  'starting' | 'running'\n>[];\n\nexport type RunnerInstanceTerminateIntentReason = 'job-cancelled';\n\nexport interface RunnerInstanceTerminateIntent {\n  providerRunnerId: string;\n  reason: RunnerInstanceTerminateIntentReason;\n}\n\nexport interface ActiveRunnerInstanceTemplateCount {\n  templateKey: string;\n  state: (typeof divergenceCountStates)[number];\n  count: number;\n}\n\nexport interface RunnerInstanceReportEvent {\n  providerRunnerId: string;\n  reservationId: string | null;\n  templateKey: string | null;\n  labels: string[];\n  state: RunnerInstanceState;\n  reason: string | null;\n  runnerSessionId: string | null;\n  providerKind: string | null;\n  reportedAt: Date;\n}\n\nexport interface ReportRunnerInstancesParams {\n  workspaceId: string | null;\n  provisionerId: string;\n  events: RunnerInstanceReportEvent[];\n}\n\nexport interface ReconcileRunnerInstancesParams {\n  workspaceId: string | null;\n  provisionerId: string;\n  observedRunnerInstanceIds: string[];\n  terminateGraceSeconds: number;\n}\n\nexport interface ReconcileRunnerInstancesDbResult {\n  observedRows: RunnerInstance[];\n  boundJobExecutionsByRunnerInstanceId: Map<string, RunnerInstanceBoundJobExecution>;\n  absentIds: string[];\n  reservationsReleased: number;\n}\n\nexport interface ReapStaleRunnerInstancesResult {\n  reaped: number;\n  reservationsReleased: number;\n}\n\nexport async function attachRunnerInstanceProviderId(params: {\n  runnerInstanceId: string;\n  provisionerId: string;\n  providerRunnerId: string;\n}): Promise<boolean> {\n  const updated = await db()\n    .update(providerRunners)\n    .set({providerRunnerId: params.providerRunnerId, updatedAt: sql`now()`})\n    .where(\n      and(\n        eq(providerRunners.id, params.runnerInstanceId),\n        eq(providerRunners.provisionerId, params.provisionerId),\n        isNull(providerRunners.providerRunnerId),\n        notInArray(providerRunners.state, [...terminalStates]),\n      ),\n    )\n    .returning({id: providerRunners.id});\n  return updated.length === 1;\n}\n\ninterface RunnerInstanceReportRow extends RunnerInstanceReportEvent {\n  startedAt: Date | null;\n  stoppingAt: Date | null;\n  stoppedAt: Date | null;\n  failedAt: Date | null;\n  terminatedAt: Date | null;\n}\n\ntype RunnerInstanceMilestoneColumn =\n  | typeof providerRunners.startedAt\n  | typeof providerRunners.stoppingAt\n  | typeof providerRunners.stoppedAt\n  | typeof providerRunners.failedAt\n  | typeof providerRunners.terminatedAt;\n\nexport async function reportRunnerInstances(params: ReportRunnerInstancesParams): Promise<{\n  accepted: number;\n  reservationsReleased: number;\n  terminateIntentsHonored: RunnerInstanceTerminateIntent[];\n}> {\n  if (params.events.length === 0)\n    return {accepted: 0, reservationsReleased: 0, terminateIntentsHonored: []};\n\n  return await db().transaction(async (tx) => {\n    const receivedAt = new Date();\n    const aggregatedEvents = aggregateEvents(params.events, receivedAt);\n    const hasTerminalEvent = aggregatedEvents.some((event) => isTerminalState(event.state));\n\n    if (hasTerminalEvent) {\n      await tx.execute(\n        sql`select pg_advisory_xact_lock(hashtext(${params.workspaceId ?? params.provisionerId}))`,\n      );\n    }\n\n    const events = await hydrateRunnerSessionIdsFromConsumedTokens(tx, params, aggregatedEvents);\n    const terminateIntentsHonored = await listTerminateIntentsHonoredByTerminatedReportsTx(\n      tx,\n      params,\n      events,\n    );\n\n    const values = events.map((event) => ({\n      workspaceId: params.workspaceId,\n      provisionerId: params.provisionerId,\n      providerRunnerId: event.providerRunnerId,\n      reservationId: event.reservationId,\n      templateKey: event.templateKey,\n      labels: [...canonicalizeLabels(event.labels)],\n      state: event.state,\n      reason: event.reason,\n      runnerSessionId: event.runnerSessionId,\n      providerKind: event.providerKind,\n      reportedAt: event.reportedAt > receivedAt ? receivedAt : event.reportedAt,\n      startedAt: event.startedAt,\n      stoppingAt: event.stoppingAt,\n      stoppedAt: event.stoppedAt,\n      failedAt: event.failedAt,\n      terminatedAt: event.terminatedAt,\n    }));\n\n    await tx\n      .insert(providerRunners)\n      .values(values)\n      .onConflictDoUpdate({\n        target: [providerRunners.provisionerId, providerRunners.providerRunnerId],\n        targetWhere: isNotNull(providerRunners.providerRunnerId),\n        set: {\n          reservationId: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN coalesce(excluded.reservation_id, ${providerRunners.reservationId}) ELSE ${providerRunners.reservationId} END`,\n          templateKey: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN coalesce(excluded.template_key, ${providerRunners.templateKey}) ELSE ${providerRunners.templateKey} END`,\n          labels: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN excluded.labels ELSE ${providerRunners.labels} END`,\n          state: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN excluded.state ELSE ${providerRunners.state} END`,\n          reason: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN excluded.reason ELSE ${providerRunners.reason} END`,\n          runnerSessionId: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN coalesce(${providerRunners.runnerSessionId}, excluded.runner_session_id) ELSE ${providerRunners.runnerSessionId} END`,\n          providerKind: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN coalesce(excluded.provider_kind, ${providerRunners.providerKind}) ELSE ${providerRunners.providerKind} END`,\n          reportedAt: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN excluded.reported_at ELSE ${providerRunners.reportedAt} END`,\n          startedAt: firstObservedAt(providerRunners.startedAt, sql`excluded.started_at`),\n          stoppingAt: firstObservedAt(providerRunners.stoppingAt, sql`excluded.stopping_at`),\n          stoppedAt: firstObservedAt(providerRunners.stoppedAt, sql`excluded.stopped_at`),\n          failedAt: firstObservedAt(providerRunners.failedAt, sql`excluded.failed_at`),\n          terminatedAt: firstObservedAt(providerRunners.terminatedAt, sql`excluded.terminated_at`),\n          updatedAt: sql`now()`,\n        },\n        setWhere: sql`\n          ${providerRunnerProjectionUpdateCondition()}\n          OR ${providerRunnerMilestoneUpdateCondition()}\n        `,\n      });\n\n    const reservationsReleased = hasTerminalEvent\n      ? await releaseTerminalRunnerInstanceReservations(tx, params, events)\n      : 0;\n\n    return {accepted: events.length, reservationsReleased, terminateIntentsHonored};\n  });\n}\n\nexport async function listActiveRunnerInstanceCountsByTemplateTx(\n  tx: Tx,\n  params: {workspaceId: string; provisionerId: string},\n): Promise<ActiveRunnerInstanceTemplateCount[]> {\n  const rows = await tx\n    .select({\n      templateKey: providerRunners.templateKey,\n      state: providerRunners.state,\n      count: sql<number>`count(*)::int`,\n    })\n    .from(providerRunners)\n    .where(\n      and(\n        eq(providerRunners.workspaceId, params.workspaceId),\n        eq(providerRunners.provisionerId, params.provisionerId),\n        inArray(providerRunners.state, divergenceCountStates),\n        isNotNull(providerRunners.templateKey),\n      ),\n    )\n    .groupBy(providerRunners.templateKey, providerRunners.state);\n\n  return rows.flatMap((row) =>\n    row.templateKey && isDivergenceCountState(row.state)\n      ? [{templateKey: row.templateKey, state: row.state, count: row.count}]\n      : [],\n  );\n}\n\nexport async function listActiveRunnerInstances(params: {\n  workspaceId: string;\n  windowSeconds: number;\n  limit?: number;\n}): Promise<RunnerInstance[]> {\n  const rows = await db()\n    .select()\n    .from(providerRunners)\n    .where(\n      and(\n        eq(providerRunners.workspaceId, params.workspaceId),\n        inArray(providerRunners.state, activeStates),\n        sql`${providerRunners.updatedAt} > now() - (${params.windowSeconds} || ' seconds')::interval`,\n      ),\n    )\n    .orderBy(desc(providerRunners.updatedAt), desc(providerRunners.id))\n    .limit(params.limit ?? 1000);\n\n  return rows.map(toRunnerInstance);\n}\n\nexport async function listProvisionerTerminateIntents(params: {\n  workspaceId: string;\n  provisionerId: string;\n  limit: number;\n}): Promise<string[]> {\n  return await db().transaction(async (tx) => {\n    const rows = await listProvisionerTerminateIntentRowsTx(tx, params);\n    return rows.map((row) => row.providerRunnerId);\n  });\n}\n\nexport async function listProvisionerTerminateIntentRowsTx(\n  tx: Tx,\n  params: {\n    workspaceId: string;\n    provisionerId: string;\n    limit: number;\n  },\n): Promise<RunnerInstanceTerminateIntent[]> {\n  const rows = await provisionerTerminateIntentsQuery(tx, params)\n    .orderBy(asc(providerRunners.providerRunnerId))\n    .limit(params.limit + 1);\n\n  const truncated = rows.length > params.limit;\n  const returnedRows = truncated ? rows.slice(0, params.limit) : rows;\n  if (truncated) {\n    logger().warn(\n      {\n        workspaceId: params.workspaceId,\n        provisionerId: params.provisionerId,\n        limit: params.limit,\n        returnedCount: returnedRows.length,\n      },\n      'provisioner terminate intents truncated by poll-demand limit',\n    );\n  }\n\n  return returnedRows.flatMap((row) =>\n    row.providerRunnerId ? [{...row, providerRunnerId: row.providerRunnerId}] : [],\n  );\n}\n\nfunction provisionerTerminateIntentsQuery(\n  tx: Tx,\n  params: {workspaceId: string; provisionerId: string; providerRunnerIds?: string[]},\n) {\n  const newerRunningJobExecutions = alias(runningJobExecutions, 'newer_running_jobs');\n\n  return tx\n    .select({\n      providerRunnerId: providerRunners.providerRunnerId,\n      reason: sql<RunnerInstanceTerminateIntentReason>`'job-cancelled'`,\n    })\n    .from(runningJobExecutions)\n    .innerJoin(\n      providerRunners,\n      and(\n        eq(providerRunners.workspaceId, runningJobExecutions.workspaceId),\n        eq(providerRunners.provisionerId, runningJobExecutions.provisionerId),\n        eq(providerRunners.providerRunnerId, runningJobExecutions.providerRunnerId),\n      ),\n    )\n    .where(\n      and(\n        eq(runningJobExecutions.workspaceId, params.workspaceId),\n        eq(runningJobExecutions.provisionerId, params.provisionerId),\n        isNotNull(providerRunners.providerRunnerId),\n        isNotNull(runningJobExecutions.cancellationRequestedAt),\n        inArray(providerRunners.state, activeStates),\n        params.providerRunnerIds && params.providerRunnerIds.length > 0\n          ? inArray(providerRunners.providerRunnerId, params.providerRunnerIds)\n          : undefined,\n        notExists(\n          tx\n            .select({id: newerRunningJobExecutions.id})\n            .from(newerRunningJobExecutions)\n            .where(\n              and(\n                eq(newerRunningJobExecutions.workspaceId, runningJobExecutions.workspaceId),\n                eq(newerRunningJobExecutions.provisionerId, runningJobExecutions.provisionerId),\n                eq(\n                  newerRunningJobExecutions.providerRunnerId,\n                  runningJobExecutions.providerRunnerId,\n                ),\n                or(\n                  gt(newerRunningJobExecutions.startedAt, runningJobExecutions.startedAt),\n                  and(\n                    eq(newerRunningJobExecutions.startedAt, runningJobExecutions.startedAt),\n                    gt(\n                      newerRunningJobExecutions.jobExecutionId,\n                      runningJobExecutions.jobExecutionId,\n                    ),\n                  ),\n                ),\n              ),\n            ),\n        ),\n      ),\n    )\n    .groupBy(providerRunners.providerRunnerId);\n}\n\nasync function listTerminateIntentsHonoredByTerminatedReportsTx(\n  tx: Tx,\n  params: ReportRunnerInstancesParams,\n  events: RunnerInstanceReportEvent[],\n): Promise<RunnerInstanceTerminateIntent[]> {\n  if (!params.workspaceId) return [];\n  const terminatedRunnerInstanceIds = [\n    ...new Set(\n      events.filter((event) => event.state === 'terminated').map((event) => event.providerRunnerId),\n    ),\n  ];\n  if (terminatedRunnerInstanceIds.length === 0) return [];\n\n  const rows = await provisionerTerminateIntentsQuery(tx, {\n    workspaceId: params.workspaceId,\n    provisionerId: params.provisionerId,\n    providerRunnerIds: terminatedRunnerInstanceIds,\n  }).orderBy(asc(providerRunners.providerRunnerId));\n  return rows.flatMap((row) =>\n    row.providerRunnerId ? [{...row, providerRunnerId: row.providerRunnerId}] : [],\n  );\n}\n\nexport async function reconcileRunnerInstances(\n  params: ReconcileRunnerInstancesParams,\n): Promise<ReconcileRunnerInstancesDbResult> {\n  const observedRunnerInstanceIds = [...new Set(params.observedRunnerInstanceIds)];\n\n  return await db().transaction(async (tx) => {\n    await tx.execute(\n      sql`select pg_advisory_xact_lock(hashtext(${params.workspaceId ?? params.provisionerId}))`,\n    );\n\n    let absentIds: string[] = [];\n    let reservationsReleased = 0;\n    if (observedRunnerInstanceIds.length > 0) {\n      const staleAbsentRows = await tx\n        .select({\n          id: providerRunners.id,\n          providerRunnerId: providerRunners.providerRunnerId,\n        })\n        .from(providerRunners)\n        .where(\n          and(\n            params.workspaceId\n              ? eq(providerRunners.workspaceId, params.workspaceId)\n              : isNull(providerRunners.workspaceId),\n            eq(providerRunners.provisionerId, params.provisionerId),\n            inArray(providerRunners.state, activeStates),\n            lt(\n              providerRunners.reportedAt,\n              sql`now() - (${params.terminateGraceSeconds} || ' seconds')::interval`,\n            ),\n            notInArray(providerRunners.providerRunnerId, observedRunnerInstanceIds),\n          ),\n        );\n\n      if (staleAbsentRows.length > 0) {\n        const updated = await tx\n          .update(providerRunners)\n          .set({\n            state: 'terminated',\n            terminatedAt: sql`coalesce(${providerRunners.terminatedAt}, now())`,\n            updatedAt: sql`now()`,\n          })\n          .where(\n            and(\n              inArray(\n                providerRunners.id,\n                staleAbsentRows.map((row) => row.id),\n              ),\n              inArray(providerRunners.state, activeStates),\n              lt(\n                providerRunners.reportedAt,\n                sql`now() - (${params.terminateGraceSeconds} || ' seconds')::interval`,\n              ),\n            ),\n          )\n          .returning({providerRunnerId: providerRunners.providerRunnerId});\n\n        absentIds = updated.flatMap((row) => (row.providerRunnerId ? [row.providerRunnerId] : []));\n        if (params.workspaceId) {\n          reservationsReleased = await releaseTerminalRunnerInstanceReservationsByIds(tx, {\n            workspaceId: params.workspaceId,\n            provisionerId: params.provisionerId,\n            providerRunnerIds: absentIds,\n          });\n        }\n      }\n    }\n\n    const observedRows =\n      observedRunnerInstanceIds.length === 0\n        ? []\n        : (\n            await tx\n              .select()\n              .from(providerRunners)\n              .where(\n                and(\n                  params.workspaceId\n                    ? eq(providerRunners.workspaceId, params.workspaceId)\n                    : isNull(providerRunners.workspaceId),\n                  eq(providerRunners.provisionerId, params.provisionerId),\n                  inArray(providerRunners.providerRunnerId, observedRunnerInstanceIds),\n                ),\n              )\n          ).map(toRunnerInstance);\n\n    const boundJobExecutions = params.workspaceId\n      ? await listRunningJobExecutionsByRunnerInstanceTx(tx, {\n          workspaceId: params.workspaceId,\n          provisionerId: params.provisionerId,\n          providerRunnerIds: observedRunnerInstanceIds,\n        })\n      : [];\n\n    return {\n      observedRows,\n      boundJobExecutionsByRunnerInstanceId: new Map(\n        boundJobExecutions.map((jobExecution) => [jobExecution.providerRunnerId, jobExecution]),\n      ),\n      absentIds,\n      reservationsReleased,\n    };\n  });\n}\n\nexport async function reapStaleRunnerInstances(params: {\n  thresholdSeconds: number;\n  limit: number;\n}): Promise<ReapStaleRunnerInstancesResult> {\n  const cutoff = staleRunnerInstanceCutoff(params.thresholdSeconds);\n\n  return await db().transaction(async (tx) => {\n    const candidateRows = await tx\n      .select({\n        id: providerRunners.id,\n        workspaceId: providerRunners.workspaceId,\n        provisionerId: providerRunners.provisionerId,\n        providerRunnerId: providerRunners.providerRunnerId,\n      })\n      .from(providerRunners)\n      .where(staleRunnerInstanceWhere(tx, cutoff))\n      .orderBy(asc(providerRunners.updatedAt), asc(providerRunners.id))\n      .limit(params.limit);\n\n    if (candidateRows.length === 0) return {reaped: 0, reservationsReleased: 0};\n\n    const workspaceIds = [\n      ...new Set(candidateRows.flatMap((row) => (row.workspaceId ? [row.workspaceId] : []))),\n    ].sort();\n    for (const workspaceId of workspaceIds) {\n      await tx.execute(sql`select pg_advisory_xact_lock(hashtext(${workspaceId}))`);\n    }\n\n    const updatedRows = await tx\n      .update(providerRunners)\n      .set({\n        state: 'failed',\n        reason: 'stale-provisioner',\n        failedAt: sql`coalesce(${providerRunners.failedAt}, now())`,\n        updatedAt: sql`now()`,\n      })\n      .where(\n        and(\n          inArray(\n            providerRunners.id,\n            candidateRows.map((row) => row.id),\n          ),\n          staleRunnerInstanceWhere(tx, cutoff),\n        ),\n      )\n      .returning({\n        workspaceId: providerRunners.workspaceId,\n        provisionerId: providerRunners.provisionerId,\n        providerRunnerId: providerRunners.providerRunnerId,\n      });\n\n    let reservationsReleased = 0;\n    for (const group of groupRunnerInstanceIds(updatedRows)) {\n      reservationsReleased += await releaseTerminalRunnerInstanceReservationsByIds(tx, {\n        workspaceId: group.workspaceId,\n        provisionerId: group.provisionerId,\n        providerRunnerIds: group.providerRunnerIds,\n        requireUnlinkedSession: false,\n      });\n    }\n\n    return {reaped: updatedRows.length, reservationsReleased};\n  });\n}\n\nasync function releaseTerminalRunnerInstanceReservations(\n  tx: Tx,\n  params: ReportRunnerInstancesParams,\n  events: RunnerInstanceReportEvent[],\n): Promise<number> {\n  const terminalEvents = events.filter((event) => isTerminalState(event.state));\n  if (terminalEvents.length === 0 || !params.workspaceId) return 0;\n\n  return await releaseTerminalRunnerInstanceReservationsByIds(tx, {\n    workspaceId: params.workspaceId,\n    provisionerId: params.provisionerId,\n    providerRunnerIds: terminalEvents.map((event) => event.providerRunnerId),\n  });\n}\n\nasync function releaseTerminalRunnerInstanceReservationsByIds(\n  tx: Tx,\n  params: {\n    workspaceId: string;\n    provisionerId: string;\n    providerRunnerIds: string[];\n    requireUnlinkedSession?: boolean;\n  },\n): Promise<number> {\n  if (params.providerRunnerIds.length === 0) return 0;\n\n  const rows = await tx\n    .select({\n      id: providerRunners.id,\n      reservationId: providerRunners.reservationId,\n    })\n    .from(providerRunners)\n    .where(\n      and(\n        eq(providerRunners.workspaceId, params.workspaceId),\n        eq(providerRunners.provisionerId, params.provisionerId),\n        inArray(providerRunners.providerRunnerId, params.providerRunnerIds),\n        inArray(providerRunners.state, terminalStates),\n        isNotNull(providerRunners.reservationId),\n        params.requireUnlinkedSession === false\n          ? undefined\n          : isNull(providerRunners.runnerSessionId),\n        isNull(providerRunners.reservationReleasedAt),\n      ),\n    );\n\n  if (rows.length === 0) return 0;\n\n  const updated = await tx\n    .update(providerRunners)\n    .set({reservationReleasedAt: sql`now()`, updatedAt: sql`now()`})\n    .where(\n      and(\n        inArray(\n          providerRunners.id,\n          rows.map((row) => row.id),\n        ),\n        params.requireUnlinkedSession === false\n          ? undefined\n          : isNull(providerRunners.runnerSessionId),\n        isNull(providerRunners.reservationReleasedAt),\n      ),\n    )\n    .returning({reservationId: providerRunners.reservationId});\n\n  const releasesByReservationId = new Map<string, number>();\n  for (const row of updated) {\n    if (!row.reservationId) continue;\n    releasesByReservationId.set(\n      row.reservationId,\n      (releasesByReservationId.get(row.reservationId) ?? 0) + 1,\n    );\n  }\n\n  if (releasesByReservationId.size === 0) return 0;\n\n  return await releaseReservationUnits(tx, {\n    workspaceId: params.workspaceId,\n    provisionerId: params.provisionerId,\n    releases: [...releasesByReservationId].map(([reservationId, count]) => ({\n      reservationId,\n      count,\n    })),\n  });\n}\n\nfunction staleRunnerInstanceCutoff(thresholdSeconds: number): SQL {\n  return sql`now() - (${thresholdSeconds} || ' seconds')::interval`;\n}\n\nfunction staleRunnerInstanceWhere(tx: Tx, cutoff: SQL): SQL<boolean> {\n  return and(\n    inArray(providerRunners.state, activeStates),\n    lt(providerRunners.reportedAt, cutoff),\n    lt(providerRunners.updatedAt, cutoff),\n    exists(\n      tx\n        .select({id: provisionerTokens.id})\n        .from(provisionerTokens)\n        .where(\n          and(\n            eq(provisionerTokens.id, providerRunners.provisionerId),\n            or(isNull(provisionerTokens.lastSeenAt), lt(provisionerTokens.lastSeenAt, cutoff)),\n          ),\n        ),\n    ),\n    notExists(\n      tx\n        .select({id: runningJobExecutions.id})\n        .from(runningJobExecutions)\n        .where(\n          and(\n            eq(runningJobExecutions.workspaceId, providerRunners.workspaceId),\n            eq(runningJobExecutions.provisionerId, providerRunners.provisionerId),\n            eq(runningJobExecutions.providerRunnerId, providerRunners.providerRunnerId),\n          ),\n        ),\n    ),\n    notExists(\n      tx\n        .select({id: runnerSessions.id})\n        .from(runnerSessions)\n        .where(\n          and(\n            eq(runnerSessions.workspaceId, providerRunners.workspaceId),\n            eq(runnerSessions.provisionerId, providerRunners.provisionerId),\n            eq(runnerSessions.providerRunnerId, providerRunners.providerRunnerId),\n            sql`${runnerSessions.updatedAt} >= ${cutoff}`,\n          ),\n        ),\n    ),\n  ) as SQL<boolean>;\n}\n\nfunction groupRunnerInstanceIds(\n  rows: Array<{\n    workspaceId: string | null;\n    provisionerId: string;\n    providerRunnerId: string | null;\n  }>,\n): Array<{workspaceId: string; provisionerId: string; providerRunnerIds: string[]}> {\n  const groups = new Map<\n    string,\n    {workspaceId: string; provisionerId: string; providerRunnerIds: string[]}\n  >();\n  for (const row of rows) {\n    if (!row.workspaceId || !row.providerRunnerId) continue;\n    const key = `${row.workspaceId}:${row.provisionerId}`;\n    const group = groups.get(key) ?? {\n      workspaceId: row.workspaceId,\n      provisionerId: row.provisionerId,\n      providerRunnerIds: [],\n    };\n    group.providerRunnerIds.push(row.providerRunnerId);\n    groups.set(key, group);\n  }\n  return [...groups.values()];\n}\n\nasync function hydrateRunnerSessionIdsFromConsumedTokens(\n  tx: Tx,\n  params: ReportRunnerInstancesParams,\n  events: RunnerInstanceReportRow[],\n): Promise<RunnerInstanceReportRow[]> {\n  if (!params.workspaceId) return events;\n  const providerRunnerIds = [...new Set(events.map((event) => event.providerRunnerId))];\n  if (providerRunnerIds.length === 0) return events;\n\n  const tokenRows = await tx\n    .select({\n      providerRunnerId: ephemeralRegistrationTokens.providerRunnerId,\n      consumedSessionId: ephemeralRegistrationTokens.consumedSessionId,\n    })\n    .from(ephemeralRegistrationTokens)\n    .where(\n      and(\n        eq(ephemeralRegistrationTokens.workspaceId, params.workspaceId),\n        eq(ephemeralRegistrationTokens.provisionerId, params.provisionerId),\n        inArray(ephemeralRegistrationTokens.providerRunnerId, providerRunnerIds),\n        isNotNull(ephemeralRegistrationTokens.consumedSessionId),\n      ),\n    )\n    .orderBy(\n      desc(ephemeralRegistrationTokens.consumedAt),\n      desc(ephemeralRegistrationTokens.createdAt),\n    );\n\n  const consumedSessionIdsByRunnerInstanceId = new Map<string, string>();\n  for (const row of tokenRows) {\n    if (!row.consumedSessionId) continue;\n    if (consumedSessionIdsByRunnerInstanceId.has(row.providerRunnerId)) continue;\n    consumedSessionIdsByRunnerInstanceId.set(row.providerRunnerId, row.consumedSessionId);\n  }\n\n  if (consumedSessionIdsByRunnerInstanceId.size === 0) return events;\n\n  return events.map((event) => {\n    const consumedSessionId = consumedSessionIdsByRunnerInstanceId.get(event.providerRunnerId);\n    if (!consumedSessionId || event.runnerSessionId === consumedSessionId) return event;\n    return {...event, runnerSessionId: consumedSessionId};\n  });\n}\n\nfunction aggregateEvents(\n  events: RunnerInstanceReportEvent[],\n  receivedAt: Date,\n): RunnerInstanceReportRow[] {\n  const byRunnerInstanceId = new Map<string, RunnerInstanceReportRow>();\n  for (const rawEvent of events) {\n    const event = toRunnerInstanceReportRow(rawEvent, receivedAt);\n    const existing = byRunnerInstanceId.get(event.providerRunnerId);\n    if (existing) mergeMilestones(existing, event);\n    if (!existing || compareRunnerInstanceReportEvents(event, existing) > 0) {\n      byRunnerInstanceId.set(\n        event.providerRunnerId,\n        existing ? mergeProjectionMetadata(event, existing) : event,\n      );\n    }\n  }\n  return [...byRunnerInstanceId.values()];\n}\n\nfunction toRunnerInstanceReportRow(\n  event: RunnerInstanceReportEvent,\n  receivedAt: Date,\n): RunnerInstanceReportRow {\n  const reportedAt = event.reportedAt > receivedAt ? receivedAt : event.reportedAt;\n  return {\n    ...event,\n    reportedAt,\n    startedAt: event.state === 'running' ? reportedAt : null,\n    stoppingAt: event.state === 'stopping' ? reportedAt : null,\n    stoppedAt: event.state === 'stopped' ? reportedAt : null,\n    failedAt: event.state === 'failed' ? reportedAt : null,\n    terminatedAt: event.state === 'terminated' ? reportedAt : null,\n  };\n}\n\nfunction mergeMilestones(target: RunnerInstanceReportRow, source: RunnerInstanceReportRow) {\n  target.startedAt = earliestDate(target.startedAt, source.startedAt);\n  target.stoppingAt = earliestDate(target.stoppingAt, source.stoppingAt);\n  target.stoppedAt = earliestDate(target.stoppedAt, source.stoppedAt);\n  target.failedAt = earliestDate(target.failedAt, source.failedAt);\n  target.terminatedAt = earliestDate(target.terminatedAt, source.terminatedAt);\n}\n\nfunction mergeProjectionMetadata(\n  event: RunnerInstanceReportRow,\n  existing: RunnerInstanceReportRow,\n): RunnerInstanceReportRow {\n  return {\n    ...event,\n    reservationId: event.reservationId ?? existing.reservationId,\n    templateKey: event.templateKey ?? existing.templateKey,\n    runnerSessionId: event.runnerSessionId ?? existing.runnerSessionId,\n    providerKind: event.providerKind ?? existing.providerKind,\n    ...pickMilestones(existing),\n  };\n}\n\nfunction pickMilestones(event: RunnerInstanceReportRow) {\n  return {\n    startedAt: event.startedAt,\n    stoppingAt: event.stoppingAt,\n    stoppedAt: event.stoppedAt,\n    failedAt: event.failedAt,\n    terminatedAt: event.terminatedAt,\n  };\n}\n\nfunction earliestDate(a: Date | null, b: Date | null): Date | null {\n  if (!a) return b;\n  if (!b) return a;\n  return a < b ? a : b;\n}\n\nfunction compareRunnerInstanceReportEvents(\n  a: RunnerInstanceReportEvent,\n  b: RunnerInstanceReportEvent,\n): number {\n  const timeDelta = a.reportedAt.getTime() - b.reportedAt.getTime();\n  const rankDelta = getRunnerInstanceStateRank(a.state) - getRunnerInstanceStateRank(b.state);\n  if (rankDelta !== 0) return rankDelta;\n  return timeDelta;\n}\n\nfunction getRunnerInstanceStateRank(state: RunnerInstanceState): number {\n  switch (state) {\n    case 'starting':\n      return 1;\n    case 'running':\n      return 2;\n    case 'stopping':\n      return 3;\n    case 'stopped':\n      return 4;\n    case 'failed':\n      return 5;\n    case 'terminated':\n      return 6;\n  }\n}\n\nfunction providerRunnerStateRank(state: SQL | typeof providerRunners.state): SQL<number> {\n  return sql<number>`\n    CASE ${state}\n      WHEN 'starting' THEN 1\n      WHEN 'running' THEN 2\n      WHEN 'stopping' THEN 3\n      WHEN 'stopped' THEN 4\n      WHEN 'failed' THEN 5\n      WHEN 'terminated' THEN 6\n      ELSE 0\n    END\n  `;\n}\n\nfunction providerRunnerProjectionUpdateCondition(): SQL<boolean> {\n  return sql<boolean>`\n    ${providerRunnerStateRank(sql`excluded.state`)} > ${providerRunnerStateRank(providerRunners.state)}\n    OR (\n      ${providerRunnerStateRank(sql`excluded.state`)} = ${providerRunnerStateRank(providerRunners.state)}\n      AND excluded.reported_at >= ${providerRunners.reportedAt}\n    )\n  `;\n}\n\nfunction providerRunnerMilestoneUpdateCondition(): SQL<boolean> {\n  return sql<boolean>`\n    ${milestoneNeedsUpdate(providerRunners.startedAt, sql`excluded.started_at`)}\n    OR ${milestoneNeedsUpdate(providerRunners.stoppingAt, sql`excluded.stopping_at`)}\n    OR ${milestoneNeedsUpdate(providerRunners.stoppedAt, sql`excluded.stopped_at`)}\n    OR ${milestoneNeedsUpdate(providerRunners.failedAt, sql`excluded.failed_at`)}\n    OR ${milestoneNeedsUpdate(providerRunners.terminatedAt, sql`excluded.terminated_at`)}\n  `;\n}\n\nfunction milestoneNeedsUpdate(\n  current: SQL | RunnerInstanceMilestoneColumn,\n  incoming: SQL,\n): SQL<boolean> {\n  return sql<boolean>`${incoming} IS NOT NULL AND (${current} IS NULL OR ${incoming} < ${current})`;\n}\n\nfunction firstObservedAt(current: SQL | RunnerInstanceMilestoneColumn, incoming: SQL) {\n  return sql`\n    CASE\n      WHEN ${incoming} IS NULL THEN ${current}\n      WHEN ${current} IS NULL THEN ${incoming}\n      ELSE least(${current}, ${incoming})\n    END\n  `;\n}\n\nexport function isTerminalState(state: RunnerInstanceState): boolean {\n  return terminalStates.includes(state as (typeof terminalStates)[number]);\n}\n\nfunction isDivergenceCountState(\n  state: RunnerInstanceState,\n): state is (typeof divergenceCountStates)[number] {\n  return divergenceCountStates.includes(state as (typeof divergenceCountStates)[number]);\n}\n"],"names":["logger","canonicalizeLabels","and","asc","desc","eq","exists","gt","inArray","isNotNull","isNull","lt","notExists","notInArray","or","sql","alias","db","listRunningJobExecutionsByRunnerInstanceTx","releaseReservationUnits","ephemeralRegistrationTokens","provisionerTokens","providerRunners","toRunnerInstance","runnerSessions","runningJobExecutions","terminalStates","activeStates","divergenceCountStates","attachRunnerInstanceProviderId","params","updated","update","set","providerRunnerId","updatedAt","where","id","runnerInstanceId","provisionerId","state","returning","length","reportRunnerInstances","events","accepted","reservationsReleased","terminateIntentsHonored","transaction","tx","receivedAt","Date","aggregatedEvents","aggregateEvents","hasTerminalEvent","some","event","isTerminalState","execute","workspaceId","hydrateRunnerSessionIdsFromConsumedTokens","listTerminateIntentsHonoredByTerminatedReportsTx","values","map","reservationId","templateKey","labels","reason","runnerSessionId","providerKind","reportedAt","startedAt","stoppingAt","stoppedAt","failedAt","terminatedAt","insert","onConflictDoUpdate","target","targetWhere","providerRunnerProjectionUpdateCondition","firstObservedAt","setWhere","providerRunnerMilestoneUpdateCondition","releaseTerminalRunnerInstanceReservations","listActiveRunnerInstanceCountsByTemplateTx","rows","select","count","from","groupBy","flatMap","row","isDivergenceCountState","listActiveRunnerInstances","windowSeconds","orderBy","limit","listProvisionerTerminateIntents","listProvisionerTerminateIntentRowsTx","provisionerTerminateIntentsQuery","truncated","returnedRows","slice","warn","returnedCount","newerRunningJobExecutions","innerJoin","cancellationRequestedAt","providerRunnerIds","undefined","jobExecutionId","terminatedRunnerInstanceIds","Set","filter","reconcileRunnerInstances","observedRunnerInstanceIds","absentIds","staleAbsentRows","terminateGraceSeconds","releaseTerminalRunnerInstanceReservationsByIds","observedRows","boundJobExecutions","boundJobExecutionsByRunnerInstanceId","Map","jobExecution","reapStaleRunnerInstances","cutoff","staleRunnerInstanceCutoff","thresholdSeconds","candidateRows","staleRunnerInstanceWhere","reaped","workspaceIds","sort","updatedRows","group","groupRunnerInstanceIds","requireUnlinkedSession","terminalEvents","reservationReleasedAt","releasesByReservationId","get","size","releases","lastSeenAt","groups","key","push","tokenRows","consumedSessionId","consumedAt","createdAt","consumedSessionIdsByRunnerInstanceId","has","byRunnerInstanceId","rawEvent","toRunnerInstanceReportRow","existing","mergeMilestones","compareRunnerInstanceReportEvents","mergeProjectionMetadata","source","earliestDate","pickMilestones","a","b","timeDelta","getTime","rankDelta","getRunnerInstanceStateRank","providerRunnerStateRank","milestoneNeedsUpdate","current","incoming","includes"],"mappings":"AAAA,SAAQA,MAAM,QAAO,8BAA8B;AACnD,SAAQC,kBAAkB,QAAO,yBAAyB;AAC1D,SACEC,GAAG,EACHC,GAAG,EACHC,IAAI,EACJC,EAAE,EACFC,MAAM,EACNC,EAAE,EACFC,OAAO,EACPC,SAAS,EACTC,MAAM,EACNC,EAAE,EACFC,SAAS,EACTC,UAAU,EACVC,EAAE,EAEFC,GAAG,QACE,cAAc;AACrB,SAAQC,KAAK,QAAO,sBAAsB;AAG1C,SAAQC,EAAE,QAAO,UAAU;AAC3B,SACEC,0CAA0C,QAErC,sBAAsB;AAC7B,SAAQC,uBAAuB,QAAO,oBAAoB;AAC1D,SAAQC,2BAA2B,QAAO,4CAA4C;AACtF,SAAQC,iBAAiB,QAAO,iCAAiC;AACjE,SAAQC,eAAe,EAAEC,gBAAgB,QAAO,+BAA+B;AAC/E,SAAQC,cAAc,QAAO,8BAA8B;AAC3D,SAAQC,oBAAoB,QAAO,qCAAqC;AAExE,OAAO,MAAMC,iBAAiB;IAC5B;IACA;IACA;CACD,CAAmD;AACpD,OAAO,MAAMC,eAAe;IAC1B;IACA;IACA;CACD,CAAmD;AACpD,OAAO,MAAMC,wBAAwB;IAAC;IAAY;CAAU,CAGxD;AAoDJ,OAAO,eAAeC,+BAA+BC,MAIpD;IACC,MAAMC,UAAU,MAAMd,KACnBe,MAAM,CAACV,iBACPW,GAAG,CAAC;QAACC,kBAAkBJ,OAAOI,gBAAgB;QAAEC,WAAWpB,GAAG,CAAC,KAAK,CAAC;IAAA,GACrEqB,KAAK,CACJlC,IACEG,GAAGiB,gBAAgBe,EAAE,EAAEP,OAAOQ,gBAAgB,GAC9CjC,GAAGiB,gBAAgBiB,aAAa,EAAET,OAAOS,aAAa,GACtD7B,OAAOY,gBAAgBY,gBAAgB,GACvCrB,WAAWS,gBAAgBkB,KAAK,EAAE;WAAId;KAAe,IAGxDe,SAAS,CAAC;QAACJ,IAAIf,gBAAgBe,EAAE;IAAA;IACpC,OAAON,QAAQW,MAAM,KAAK;AAC5B;AAiBA,OAAO,eAAeC,sBAAsBb,MAAmC;IAK7E,IAAIA,OAAOc,MAAM,CAACF,MAAM,KAAK,GAC3B,OAAO;QAACG,UAAU;QAAGC,sBAAsB;QAAGC,yBAAyB,EAAE;IAAA;IAE3E,OAAO,MAAM9B,KAAK+B,WAAW,CAAC,OAAOC;QACnC,MAAMC,aAAa,IAAIC;QACvB,MAAMC,mBAAmBC,gBAAgBvB,OAAOc,MAAM,EAAEM;QACxD,MAAMI,mBAAmBF,iBAAiBG,IAAI,CAAC,CAACC,QAAUC,gBAAgBD,MAAMhB,KAAK;QAErF,IAAIc,kBAAkB;YACpB,MAAML,GAAGS,OAAO,CACd3C,GAAG,CAAC,sCAAsC,EAAEe,OAAO6B,WAAW,IAAI7B,OAAOS,aAAa,CAAC,EAAE,CAAC;QAE9F;QAEA,MAAMK,SAAS,MAAMgB,0CAA0CX,IAAInB,QAAQsB;QAC3E,MAAML,0BAA0B,MAAMc,iDACpCZ,IACAnB,QACAc;QAGF,MAAMkB,SAASlB,OAAOmB,GAAG,CAAC,CAACP,QAAW,CAAA;gBACpCG,aAAa7B,OAAO6B,WAAW;gBAC/BpB,eAAeT,OAAOS,aAAa;gBACnCL,kBAAkBsB,MAAMtB,gBAAgB;gBACxC8B,eAAeR,MAAMQ,aAAa;gBAClCC,aAAaT,MAAMS,WAAW;gBAC9BC,QAAQ;uBAAIjE,mBAAmBuD,MAAMU,MAAM;iBAAE;gBAC7C1B,OAAOgB,MAAMhB,KAAK;gBAClB2B,QAAQX,MAAMW,MAAM;gBACpBC,iBAAiBZ,MAAMY,eAAe;gBACtCC,cAAcb,MAAMa,YAAY;gBAChCC,YAAYd,MAAMc,UAAU,GAAGpB,aAAaA,aAAaM,MAAMc,UAAU;gBACzEC,WAAWf,MAAMe,SAAS;gBAC1BC,YAAYhB,MAAMgB,UAAU;gBAC5BC,WAAWjB,MAAMiB,SAAS;gBAC1BC,UAAUlB,MAAMkB,QAAQ;gBACxBC,cAAcnB,MAAMmB,YAAY;YAClC,CAAA;QAEA,MAAM1B,GACH2B,MAAM,CAACtD,iBACPwC,MAAM,CAACA,QACPe,kBAAkB,CAAC;YAClBC,QAAQ;gBAACxD,gBAAgBiB,aAAa;gBAAEjB,gBAAgBY,gBAAgB;aAAC;YACzE6C,aAAatE,UAAUa,gBAAgBY,gBAAgB;YACvDD,KAAK;gBACH+B,eAAejD,GAAG,CAAC,UAAU,EAAEiE,0CAA0C,wCAAwC,EAAE1D,gBAAgB0C,aAAa,CAAC,OAAO,EAAE1C,gBAAgB0C,aAAa,CAAC,IAAI,CAAC;gBAC7LC,aAAalD,GAAG,CAAC,UAAU,EAAEiE,0CAA0C,sCAAsC,EAAE1D,gBAAgB2C,WAAW,CAAC,OAAO,EAAE3C,gBAAgB2C,WAAW,CAAC,IAAI,CAAC;gBACrLC,QAAQnD,GAAG,CAAC,UAAU,EAAEiE,0CAA0C,2BAA2B,EAAE1D,gBAAgB4C,MAAM,CAAC,IAAI,CAAC;gBAC3H1B,OAAOzB,GAAG,CAAC,UAAU,EAAEiE,0CAA0C,0BAA0B,EAAE1D,gBAAgBkB,KAAK,CAAC,IAAI,CAAC;gBACxH2B,QAAQpD,GAAG,CAAC,UAAU,EAAEiE,0CAA0C,2BAA2B,EAAE1D,gBAAgB6C,MAAM,CAAC,IAAI,CAAC;gBAC3HC,iBAAiBrD,GAAG,CAAC,UAAU,EAAEiE,0CAA0C,eAAe,EAAE1D,gBAAgB8C,eAAe,CAAC,mCAAmC,EAAE9C,gBAAgB8C,eAAe,CAAC,IAAI,CAAC;gBACtMC,cAActD,GAAG,CAAC,UAAU,EAAEiE,0CAA0C,uCAAuC,EAAE1D,gBAAgB+C,YAAY,CAAC,OAAO,EAAE/C,gBAAgB+C,YAAY,CAAC,IAAI,CAAC;gBACzLC,YAAYvD,GAAG,CAAC,UAAU,EAAEiE,0CAA0C,gCAAgC,EAAE1D,gBAAgBgD,UAAU,CAAC,IAAI,CAAC;gBACxIC,WAAWU,gBAAgB3D,gBAAgBiD,SAAS,EAAExD,GAAG,CAAC,mBAAmB,CAAC;gBAC9EyD,YAAYS,gBAAgB3D,gBAAgBkD,UAAU,EAAEzD,GAAG,CAAC,oBAAoB,CAAC;gBACjF0D,WAAWQ,gBAAgB3D,gBAAgBmD,SAAS,EAAE1D,GAAG,CAAC,mBAAmB,CAAC;gBAC9E2D,UAAUO,gBAAgB3D,gBAAgBoD,QAAQ,EAAE3D,GAAG,CAAC,kBAAkB,CAAC;gBAC3E4D,cAAcM,gBAAgB3D,gBAAgBqD,YAAY,EAAE5D,GAAG,CAAC,sBAAsB,CAAC;gBACvFoB,WAAWpB,GAAG,CAAC,KAAK,CAAC;YACvB;YACAmE,UAAUnE,GAAG,CAAC;UACZ,EAAEiE,0CAA0C;aACzC,EAAEG,yCAAyC;QAChD,CAAC;QACH;QAEF,MAAMrC,uBAAuBQ,mBACzB,MAAM8B,0CAA0CnC,IAAInB,QAAQc,UAC5D;QAEJ,OAAO;YAACC,UAAUD,OAAOF,MAAM;YAAEI;YAAsBC;QAAuB;IAChF;AACF;AAEA,OAAO,eAAesC,2CACpBpC,EAAM,EACNnB,MAAoD;IAEpD,MAAMwD,OAAO,MAAMrC,GAChBsC,MAAM,CAAC;QACNtB,aAAa3C,gBAAgB2C,WAAW;QACxCzB,OAAOlB,gBAAgBkB,KAAK;QAC5BgD,OAAOzE,GAAW,CAAC,aAAa,CAAC;IACnC,GACC0E,IAAI,CAACnE,iBACLc,KAAK,CACJlC,IACEG,GAAGiB,gBAAgBqC,WAAW,EAAE7B,OAAO6B,WAAW,GAClDtD,GAAGiB,gBAAgBiB,aAAa,EAAET,OAAOS,aAAa,GACtD/B,QAAQc,gBAAgBkB,KAAK,EAAEZ,wBAC/BnB,UAAUa,gBAAgB2C,WAAW,IAGxCyB,OAAO,CAACpE,gBAAgB2C,WAAW,EAAE3C,gBAAgBkB,KAAK;IAE7D,OAAO8C,KAAKK,OAAO,CAAC,CAACC,MACnBA,IAAI3B,WAAW,IAAI4B,uBAAuBD,IAAIpD,KAAK,IAC/C;YAAC;gBAACyB,aAAa2B,IAAI3B,WAAW;gBAAEzB,OAAOoD,IAAIpD,KAAK;gBAAEgD,OAAOI,IAAIJ,KAAK;YAAA;SAAE,GACpE,EAAE;AAEV;AAEA,OAAO,eAAeM,0BAA0BhE,MAI/C;IACC,MAAMwD,OAAO,MAAMrE,KAChBsE,MAAM,GACNE,IAAI,CAACnE,iBACLc,KAAK,CACJlC,IACEG,GAAGiB,gBAAgBqC,WAAW,EAAE7B,OAAO6B,WAAW,GAClDnD,QAAQc,gBAAgBkB,KAAK,EAAEb,eAC/BZ,GAAG,CAAC,EAAEO,gBAAgBa,SAAS,CAAC,YAAY,EAAEL,OAAOiE,aAAa,CAAC,yBAAyB,CAAC,GAGhGC,OAAO,CAAC5F,KAAKkB,gBAAgBa,SAAS,GAAG/B,KAAKkB,gBAAgBe,EAAE,GAChE4D,KAAK,CAACnE,OAAOmE,KAAK,IAAI;IAEzB,OAAOX,KAAKvB,GAAG,CAACxC;AAClB;AAEA,OAAO,eAAe2E,gCAAgCpE,MAIrD;IACC,OAAO,MAAMb,KAAK+B,WAAW,CAAC,OAAOC;QACnC,MAAMqC,OAAO,MAAMa,qCAAqClD,IAAInB;QAC5D,OAAOwD,KAAKvB,GAAG,CAAC,CAAC6B,MAAQA,IAAI1D,gBAAgB;IAC/C;AACF;AAEA,OAAO,eAAeiE,qCACpBlD,EAAM,EACNnB,MAIC;IAED,MAAMwD,OAAO,MAAMc,iCAAiCnD,IAAInB,QACrDkE,OAAO,CAAC7F,IAAImB,gBAAgBY,gBAAgB,GAC5C+D,KAAK,CAACnE,OAAOmE,KAAK,GAAG;IAExB,MAAMI,YAAYf,KAAK5C,MAAM,GAAGZ,OAAOmE,KAAK;IAC5C,MAAMK,eAAeD,YAAYf,KAAKiB,KAAK,CAAC,GAAGzE,OAAOmE,KAAK,IAAIX;IAC/D,IAAIe,WAAW;QACbrG,SAASwG,IAAI,CACX;YACE7C,aAAa7B,OAAO6B,WAAW;YAC/BpB,eAAeT,OAAOS,aAAa;YACnC0D,OAAOnE,OAAOmE,KAAK;YACnBQ,eAAeH,aAAa5D,MAAM;QACpC,GACA;IAEJ;IAEA,OAAO4D,aAAaX,OAAO,CAAC,CAACC,MAC3BA,IAAI1D,gBAAgB,GAAG;YAAC;gBAAC,GAAG0D,GAAG;gBAAE1D,kBAAkB0D,IAAI1D,gBAAgB;YAAA;SAAE,GAAG,EAAE;AAElF;AAEA,SAASkE,iCACPnD,EAAM,EACNnB,MAAkF;IAElF,MAAM4E,4BAA4B1F,MAAMS,sBAAsB;IAE9D,OAAOwB,GACJsC,MAAM,CAAC;QACNrD,kBAAkBZ,gBAAgBY,gBAAgB;QAClDiC,QAAQpD,GAAwC,CAAC,eAAe,CAAC;IACnE,GACC0E,IAAI,CAAChE,sBACLkF,SAAS,CACRrF,iBACApB,IACEG,GAAGiB,gBAAgBqC,WAAW,EAAElC,qBAAqBkC,WAAW,GAChEtD,GAAGiB,gBAAgBiB,aAAa,EAAEd,qBAAqBc,aAAa,GACpElC,GAAGiB,gBAAgBY,gBAAgB,EAAET,qBAAqBS,gBAAgB,IAG7EE,KAAK,CACJlC,IACEG,GAAGoB,qBAAqBkC,WAAW,EAAE7B,OAAO6B,WAAW,GACvDtD,GAAGoB,qBAAqBc,aAAa,EAAET,OAAOS,aAAa,GAC3D9B,UAAUa,gBAAgBY,gBAAgB,GAC1CzB,UAAUgB,qBAAqBmF,uBAAuB,GACtDpG,QAAQc,gBAAgBkB,KAAK,EAAEb,eAC/BG,OAAO+E,iBAAiB,IAAI/E,OAAO+E,iBAAiB,CAACnE,MAAM,GAAG,IAC1DlC,QAAQc,gBAAgBY,gBAAgB,EAAEJ,OAAO+E,iBAAiB,IAClEC,WACJlG,UACEqC,GACGsC,MAAM,CAAC;QAAClD,IAAIqE,0BAA0BrE,EAAE;IAAA,GACxCoD,IAAI,CAACiB,2BACLtE,KAAK,CACJlC,IACEG,GAAGqG,0BAA0B/C,WAAW,EAAElC,qBAAqBkC,WAAW,GAC1EtD,GAAGqG,0BAA0BnE,aAAa,EAAEd,qBAAqBc,aAAa,GAC9ElC,GACEqG,0BAA0BxE,gBAAgB,EAC1CT,qBAAqBS,gBAAgB,GAEvCpB,GACEP,GAAGmG,0BAA0BnC,SAAS,EAAE9C,qBAAqB8C,SAAS,GACtErE,IACEG,GAAGqG,0BAA0BnC,SAAS,EAAE9C,qBAAqB8C,SAAS,GACtEhE,GACEmG,0BAA0BK,cAAc,EACxCtF,qBAAqBsF,cAAc,SASpDrB,OAAO,CAACpE,gBAAgBY,gBAAgB;AAC7C;AAEA,eAAe2B,iDACbZ,EAAM,EACNnB,MAAmC,EACnCc,MAAmC;IAEnC,IAAI,CAACd,OAAO6B,WAAW,EAAE,OAAO,EAAE;IAClC,MAAMqD,8BAA8B;WAC/B,IAAIC,IACLrE,OAAOsE,MAAM,CAAC,CAAC1D,QAAUA,MAAMhB,KAAK,KAAK,cAAcuB,GAAG,CAAC,CAACP,QAAUA,MAAMtB,gBAAgB;KAE/F;IACD,IAAI8E,4BAA4BtE,MAAM,KAAK,GAAG,OAAO,EAAE;IAEvD,MAAM4C,OAAO,MAAMc,iCAAiCnD,IAAI;QACtDU,aAAa7B,OAAO6B,WAAW;QAC/BpB,eAAeT,OAAOS,aAAa;QACnCsE,mBAAmBG;IACrB,GAAGhB,OAAO,CAAC7F,IAAImB,gBAAgBY,gBAAgB;IAC/C,OAAOoD,KAAKK,OAAO,CAAC,CAACC,MACnBA,IAAI1D,gBAAgB,GAAG;YAAC;gBAAC,GAAG0D,GAAG;gBAAE1D,kBAAkB0D,IAAI1D,gBAAgB;YAAA;SAAE,GAAG,EAAE;AAElF;AAEA,OAAO,eAAeiF,yBACpBrF,MAAsC;IAEtC,MAAMsF,4BAA4B;WAAI,IAAIH,IAAInF,OAAOsF,yBAAyB;KAAE;IAEhF,OAAO,MAAMnG,KAAK+B,WAAW,CAAC,OAAOC;QACnC,MAAMA,GAAGS,OAAO,CACd3C,GAAG,CAAC,sCAAsC,EAAEe,OAAO6B,WAAW,IAAI7B,OAAOS,aAAa,CAAC,EAAE,CAAC;QAG5F,IAAI8E,YAAsB,EAAE;QAC5B,IAAIvE,uBAAuB;QAC3B,IAAIsE,0BAA0B1E,MAAM,GAAG,GAAG;YACxC,MAAM4E,kBAAkB,MAAMrE,GAC3BsC,MAAM,CAAC;gBACNlD,IAAIf,gBAAgBe,EAAE;gBACtBH,kBAAkBZ,gBAAgBY,gBAAgB;YACpD,GACCuD,IAAI,CAACnE,iBACLc,KAAK,CACJlC,IACE4B,OAAO6B,WAAW,GACdtD,GAAGiB,gBAAgBqC,WAAW,EAAE7B,OAAO6B,WAAW,IAClDjD,OAAOY,gBAAgBqC,WAAW,GACtCtD,GAAGiB,gBAAgBiB,aAAa,EAAET,OAAOS,aAAa,GACtD/B,QAAQc,gBAAgBkB,KAAK,EAAEb,eAC/BhB,GACEW,gBAAgBgD,UAAU,EAC1BvD,GAAG,CAAC,SAAS,EAAEe,OAAOyF,qBAAqB,CAAC,yBAAyB,CAAC,GAExE1G,WAAWS,gBAAgBY,gBAAgB,EAAEkF;YAInD,IAAIE,gBAAgB5E,MAAM,GAAG,GAAG;gBAC9B,MAAMX,UAAU,MAAMkB,GACnBjB,MAAM,CAACV,iBACPW,GAAG,CAAC;oBACHO,OAAO;oBACPmC,cAAc5D,GAAG,CAAC,SAAS,EAAEO,gBAAgBqD,YAAY,CAAC,QAAQ,CAAC;oBACnExC,WAAWpB,GAAG,CAAC,KAAK,CAAC;gBACvB,GACCqB,KAAK,CACJlC,IACEM,QACEc,gBAAgBe,EAAE,EAClBiF,gBAAgBvD,GAAG,CAAC,CAAC6B,MAAQA,IAAIvD,EAAE,IAErC7B,QAAQc,gBAAgBkB,KAAK,EAAEb,eAC/BhB,GACEW,gBAAgBgD,UAAU,EAC1BvD,GAAG,CAAC,SAAS,EAAEe,OAAOyF,qBAAqB,CAAC,yBAAyB,CAAC,IAI3E9E,SAAS,CAAC;oBAACP,kBAAkBZ,gBAAgBY,gBAAgB;gBAAA;gBAEhEmF,YAAYtF,QAAQ4D,OAAO,CAAC,CAACC,MAASA,IAAI1D,gBAAgB,GAAG;wBAAC0D,IAAI1D,gBAAgB;qBAAC,GAAG,EAAE;gBACxF,IAAIJ,OAAO6B,WAAW,EAAE;oBACtBb,uBAAuB,MAAM0E,+CAA+CvE,IAAI;wBAC9EU,aAAa7B,OAAO6B,WAAW;wBAC/BpB,eAAeT,OAAOS,aAAa;wBACnCsE,mBAAmBQ;oBACrB;gBACF;YACF;QACF;QAEA,MAAMI,eACJL,0BAA0B1E,MAAM,KAAK,IACjC,EAAE,GACF,AACE,CAAA,MAAMO,GACHsC,MAAM,GACNE,IAAI,CAACnE,iBACLc,KAAK,CACJlC,IACE4B,OAAO6B,WAAW,GACdtD,GAAGiB,gBAAgBqC,WAAW,EAAE7B,OAAO6B,WAAW,IAClDjD,OAAOY,gBAAgBqC,WAAW,GACtCtD,GAAGiB,gBAAgBiB,aAAa,EAAET,OAAOS,aAAa,GACtD/B,QAAQc,gBAAgBY,gBAAgB,EAAEkF,4BAE9C,EACFrD,GAAG,CAACxC;QAEZ,MAAMmG,qBAAqB5F,OAAO6B,WAAW,GACzC,MAAMzC,2CAA2C+B,IAAI;YACnDU,aAAa7B,OAAO6B,WAAW;YAC/BpB,eAAeT,OAAOS,aAAa;YACnCsE,mBAAmBO;QACrB,KACA,EAAE;QAEN,OAAO;YACLK;YACAE,sCAAsC,IAAIC,IACxCF,mBAAmB3D,GAAG,CAAC,CAAC8D,eAAiB;oBAACA,aAAa3F,gBAAgB;oBAAE2F;iBAAa;YAExFR;YACAvE;QACF;IACF;AACF;AAEA,OAAO,eAAegF,yBAAyBhG,MAG9C;IACC,MAAMiG,SAASC,0BAA0BlG,OAAOmG,gBAAgB;IAEhE,OAAO,MAAMhH,KAAK+B,WAAW,CAAC,OAAOC;QACnC,MAAMiF,gBAAgB,MAAMjF,GACzBsC,MAAM,CAAC;YACNlD,IAAIf,gBAAgBe,EAAE;YACtBsB,aAAarC,gBAAgBqC,WAAW;YACxCpB,eAAejB,gBAAgBiB,aAAa;YAC5CL,kBAAkBZ,gBAAgBY,gBAAgB;QACpD,GACCuD,IAAI,CAACnE,iBACLc,KAAK,CAAC+F,yBAAyBlF,IAAI8E,SACnC/B,OAAO,CAAC7F,IAAImB,gBAAgBa,SAAS,GAAGhC,IAAImB,gBAAgBe,EAAE,GAC9D4D,KAAK,CAACnE,OAAOmE,KAAK;QAErB,IAAIiC,cAAcxF,MAAM,KAAK,GAAG,OAAO;YAAC0F,QAAQ;YAAGtF,sBAAsB;QAAC;QAE1E,MAAMuF,eAAe;eAChB,IAAIpB,IAAIiB,cAAcvC,OAAO,CAAC,CAACC,MAASA,IAAIjC,WAAW,GAAG;oBAACiC,IAAIjC,WAAW;iBAAC,GAAG,EAAE;SACpF,CAAC2E,IAAI;QACN,KAAK,MAAM3E,eAAe0E,aAAc;YACtC,MAAMpF,GAAGS,OAAO,CAAC3C,GAAG,CAAC,sCAAsC,EAAE4C,YAAY,EAAE,CAAC;QAC9E;QAEA,MAAM4E,cAAc,MAAMtF,GACvBjB,MAAM,CAACV,iBACPW,GAAG,CAAC;YACHO,OAAO;YACP2B,QAAQ;YACRO,UAAU3D,GAAG,CAAC,SAAS,EAAEO,gBAAgBoD,QAAQ,CAAC,QAAQ,CAAC;YAC3DvC,WAAWpB,GAAG,CAAC,KAAK,CAAC;QACvB,GACCqB,KAAK,CACJlC,IACEM,QACEc,gBAAgBe,EAAE,EAClB6F,cAAcnE,GAAG,CAAC,CAAC6B,MAAQA,IAAIvD,EAAE,IAEnC8F,yBAAyBlF,IAAI8E,UAGhCtF,SAAS,CAAC;YACTkB,aAAarC,gBAAgBqC,WAAW;YACxCpB,eAAejB,gBAAgBiB,aAAa;YAC5CL,kBAAkBZ,gBAAgBY,gBAAgB;QACpD;QAEF,IAAIY,uBAAuB;QAC3B,KAAK,MAAM0F,SAASC,uBAAuBF,aAAc;YACvDzF,wBAAwB,MAAM0E,+CAA+CvE,IAAI;gBAC/EU,aAAa6E,MAAM7E,WAAW;gBAC9BpB,eAAeiG,MAAMjG,aAAa;gBAClCsE,mBAAmB2B,MAAM3B,iBAAiB;gBAC1C6B,wBAAwB;YAC1B;QACF;QAEA,OAAO;YAACN,QAAQG,YAAY7F,MAAM;YAAEI;QAAoB;IAC1D;AACF;AAEA,eAAesC,0CACbnC,EAAM,EACNnB,MAAmC,EACnCc,MAAmC;IAEnC,MAAM+F,iBAAiB/F,OAAOsE,MAAM,CAAC,CAAC1D,QAAUC,gBAAgBD,MAAMhB,KAAK;IAC3E,IAAImG,eAAejG,MAAM,KAAK,KAAK,CAACZ,OAAO6B,WAAW,EAAE,OAAO;IAE/D,OAAO,MAAM6D,+CAA+CvE,IAAI;QAC9DU,aAAa7B,OAAO6B,WAAW;QAC/BpB,eAAeT,OAAOS,aAAa;QACnCsE,mBAAmB8B,eAAe5E,GAAG,CAAC,CAACP,QAAUA,MAAMtB,gBAAgB;IACzE;AACF;AAEA,eAAesF,+CACbvE,EAAM,EACNnB,MAKC;IAED,IAAIA,OAAO+E,iBAAiB,CAACnE,MAAM,KAAK,GAAG,OAAO;IAElD,MAAM4C,OAAO,MAAMrC,GAChBsC,MAAM,CAAC;QACNlD,IAAIf,gBAAgBe,EAAE;QACtB2B,eAAe1C,gBAAgB0C,aAAa;IAC9C,GACCyB,IAAI,CAACnE,iBACLc,KAAK,CACJlC,IACEG,GAAGiB,gBAAgBqC,WAAW,EAAE7B,OAAO6B,WAAW,GAClDtD,GAAGiB,gBAAgBiB,aAAa,EAAET,OAAOS,aAAa,GACtD/B,QAAQc,gBAAgBY,gBAAgB,EAAEJ,OAAO+E,iBAAiB,GAClErG,QAAQc,gBAAgBkB,KAAK,EAAEd,iBAC/BjB,UAAUa,gBAAgB0C,aAAa,GACvClC,OAAO4G,sBAAsB,KAAK,QAC9B5B,YACApG,OAAOY,gBAAgB8C,eAAe,GAC1C1D,OAAOY,gBAAgBsH,qBAAqB;IAIlD,IAAItD,KAAK5C,MAAM,KAAK,GAAG,OAAO;IAE9B,MAAMX,UAAU,MAAMkB,GACnBjB,MAAM,CAACV,iBACPW,GAAG,CAAC;QAAC2G,uBAAuB7H,GAAG,CAAC,KAAK,CAAC;QAAEoB,WAAWpB,GAAG,CAAC,KAAK,CAAC;IAAA,GAC7DqB,KAAK,CACJlC,IACEM,QACEc,gBAAgBe,EAAE,EAClBiD,KAAKvB,GAAG,CAAC,CAAC6B,MAAQA,IAAIvD,EAAE,IAE1BP,OAAO4G,sBAAsB,KAAK,QAC9B5B,YACApG,OAAOY,gBAAgB8C,eAAe,GAC1C1D,OAAOY,gBAAgBsH,qBAAqB,IAG/CnG,SAAS,CAAC;QAACuB,eAAe1C,gBAAgB0C,aAAa;IAAA;IAE1D,MAAM6E,0BAA0B,IAAIjB;IACpC,KAAK,MAAMhC,OAAO7D,QAAS;QACzB,IAAI,CAAC6D,IAAI5B,aAAa,EAAE;QACxB6E,wBAAwB5G,GAAG,CACzB2D,IAAI5B,aAAa,EACjB,AAAC6E,CAAAA,wBAAwBC,GAAG,CAAClD,IAAI5B,aAAa,KAAK,CAAA,IAAK;IAE5D;IAEA,IAAI6E,wBAAwBE,IAAI,KAAK,GAAG,OAAO;IAE/C,OAAO,MAAM5H,wBAAwB8B,IAAI;QACvCU,aAAa7B,OAAO6B,WAAW;QAC/BpB,eAAeT,OAAOS,aAAa;QACnCyG,UAAU;eAAIH;SAAwB,CAAC9E,GAAG,CAAC,CAAC,CAACC,eAAewB,MAAM,GAAM,CAAA;gBACtExB;gBACAwB;YACF,CAAA;IACF;AACF;AAEA,SAASwC,0BAA0BC,gBAAwB;IACzD,OAAOlH,GAAG,CAAC,SAAS,EAAEkH,iBAAiB,yBAAyB,CAAC;AACnE;AAEA,SAASE,yBAAyBlF,EAAM,EAAE8E,MAAW;IACnD,OAAO7H,IACLM,QAAQc,gBAAgBkB,KAAK,EAAEb,eAC/BhB,GAAGW,gBAAgBgD,UAAU,EAAEyD,SAC/BpH,GAAGW,gBAAgBa,SAAS,EAAE4F,SAC9BzH,OACE2C,GACGsC,MAAM,CAAC;QAAClD,IAAIhB,kBAAkBgB,EAAE;IAAA,GAChCoD,IAAI,CAACpE,mBACLe,KAAK,CACJlC,IACEG,GAAGgB,kBAAkBgB,EAAE,EAAEf,gBAAgBiB,aAAa,GACtDzB,GAAGJ,OAAOW,kBAAkB4H,UAAU,GAAGtI,GAAGU,kBAAkB4H,UAAU,EAAElB,aAIlFnH,UACEqC,GACGsC,MAAM,CAAC;QAAClD,IAAIZ,qBAAqBY,EAAE;IAAA,GACnCoD,IAAI,CAAChE,sBACLW,KAAK,CACJlC,IACEG,GAAGoB,qBAAqBkC,WAAW,EAAErC,gBAAgBqC,WAAW,GAChEtD,GAAGoB,qBAAqBc,aAAa,EAAEjB,gBAAgBiB,aAAa,GACpElC,GAAGoB,qBAAqBS,gBAAgB,EAAEZ,gBAAgBY,gBAAgB,MAIlFtB,UACEqC,GACGsC,MAAM,CAAC;QAAClD,IAAIb,eAAea,EAAE;IAAA,GAC7BoD,IAAI,CAACjE,gBACLY,KAAK,CACJlC,IACEG,GAAGmB,eAAemC,WAAW,EAAErC,gBAAgBqC,WAAW,GAC1DtD,GAAGmB,eAAee,aAAa,EAAEjB,gBAAgBiB,aAAa,GAC9DlC,GAAGmB,eAAeU,gBAAgB,EAAEZ,gBAAgBY,gBAAgB,GACpEnB,GAAG,CAAC,EAAES,eAAeW,SAAS,CAAC,IAAI,EAAE4F,OAAO,CAAC;AAKzD;AAEA,SAASU,uBACPnD,IAIE;IAEF,MAAM4D,SAAS,IAAItB;IAInB,KAAK,MAAMhC,OAAON,KAAM;QACtB,IAAI,CAACM,IAAIjC,WAAW,IAAI,CAACiC,IAAI1D,gBAAgB,EAAE;QAC/C,MAAMiH,MAAM,GAAGvD,IAAIjC,WAAW,CAAC,CAAC,EAAEiC,IAAIrD,aAAa,EAAE;QACrD,MAAMiG,QAAQU,OAAOJ,GAAG,CAACK,QAAQ;YAC/BxF,aAAaiC,IAAIjC,WAAW;YAC5BpB,eAAeqD,IAAIrD,aAAa;YAChCsE,mBAAmB,EAAE;QACvB;QACA2B,MAAM3B,iBAAiB,CAACuC,IAAI,CAACxD,IAAI1D,gBAAgB;QACjDgH,OAAOjH,GAAG,CAACkH,KAAKX;IAClB;IACA,OAAO;WAAIU,OAAOpF,MAAM;KAAG;AAC7B;AAEA,eAAeF,0CACbX,EAAM,EACNnB,MAAmC,EACnCc,MAAiC;IAEjC,IAAI,CAACd,OAAO6B,WAAW,EAAE,OAAOf;IAChC,MAAMiE,oBAAoB;WAAI,IAAII,IAAIrE,OAAOmB,GAAG,CAAC,CAACP,QAAUA,MAAMtB,gBAAgB;KAAG;IACrF,IAAI2E,kBAAkBnE,MAAM,KAAK,GAAG,OAAOE;IAE3C,MAAMyG,YAAY,MAAMpG,GACrBsC,MAAM,CAAC;QACNrD,kBAAkBd,4BAA4Bc,gBAAgB;QAC9DoH,mBAAmBlI,4BAA4BkI,iBAAiB;IAClE,GACC7D,IAAI,CAACrE,6BACLgB,KAAK,CACJlC,IACEG,GAAGe,4BAA4BuC,WAAW,EAAE7B,OAAO6B,WAAW,GAC9DtD,GAAGe,4BAA4BmB,aAAa,EAAET,OAAOS,aAAa,GAClE/B,QAAQY,4BAA4Bc,gBAAgB,EAAE2E,oBACtDpG,UAAUW,4BAA4BkI,iBAAiB,IAG1DtD,OAAO,CACN5F,KAAKgB,4BAA4BmI,UAAU,GAC3CnJ,KAAKgB,4BAA4BoI,SAAS;IAG9C,MAAMC,uCAAuC,IAAI7B;IACjD,KAAK,MAAMhC,OAAOyD,UAAW;QAC3B,IAAI,CAACzD,IAAI0D,iBAAiB,EAAE;QAC5B,IAAIG,qCAAqCC,GAAG,CAAC9D,IAAI1D,gBAAgB,GAAG;QACpEuH,qCAAqCxH,GAAG,CAAC2D,IAAI1D,gBAAgB,EAAE0D,IAAI0D,iBAAiB;IACtF;IAEA,IAAIG,qCAAqCV,IAAI,KAAK,GAAG,OAAOnG;IAE5D,OAAOA,OAAOmB,GAAG,CAAC,CAACP;QACjB,MAAM8F,oBAAoBG,qCAAqCX,GAAG,CAACtF,MAAMtB,gBAAgB;QACzF,IAAI,CAACoH,qBAAqB9F,MAAMY,eAAe,KAAKkF,mBAAmB,OAAO9F;QAC9E,OAAO;YAAC,GAAGA,KAAK;YAAEY,iBAAiBkF;QAAiB;IACtD;AACF;AAEA,SAASjG,gBACPT,MAAmC,EACnCM,UAAgB;IAEhB,MAAMyG,qBAAqB,IAAI/B;IAC/B,KAAK,MAAMgC,YAAYhH,OAAQ;QAC7B,MAAMY,QAAQqG,0BAA0BD,UAAU1G;QAClD,MAAM4G,WAAWH,mBAAmBb,GAAG,CAACtF,MAAMtB,gBAAgB;QAC9D,IAAI4H,UAAUC,gBAAgBD,UAAUtG;QACxC,IAAI,CAACsG,YAAYE,kCAAkCxG,OAAOsG,YAAY,GAAG;YACvEH,mBAAmB1H,GAAG,CACpBuB,MAAMtB,gBAAgB,EACtB4H,WAAWG,wBAAwBzG,OAAOsG,YAAYtG;QAE1D;IACF;IACA,OAAO;WAAImG,mBAAmB7F,MAAM;KAAG;AACzC;AAEA,SAAS+F,0BACPrG,KAAgC,EAChCN,UAAgB;IAEhB,MAAMoB,aAAad,MAAMc,UAAU,GAAGpB,aAAaA,aAAaM,MAAMc,UAAU;IAChF,OAAO;QACL,GAAGd,KAAK;QACRc;QACAC,WAAWf,MAAMhB,KAAK,KAAK,YAAY8B,aAAa;QACpDE,YAAYhB,MAAMhB,KAAK,KAAK,aAAa8B,aAAa;QACtDG,WAAWjB,MAAMhB,KAAK,KAAK,YAAY8B,aAAa;QACpDI,UAAUlB,MAAMhB,KAAK,KAAK,WAAW8B,aAAa;QAClDK,cAAcnB,MAAMhB,KAAK,KAAK,eAAe8B,aAAa;IAC5D;AACF;AAEA,SAASyF,gBAAgBjF,MAA+B,EAAEoF,MAA+B;IACvFpF,OAAOP,SAAS,GAAG4F,aAAarF,OAAOP,SAAS,EAAE2F,OAAO3F,SAAS;IAClEO,OAAON,UAAU,GAAG2F,aAAarF,OAAON,UAAU,EAAE0F,OAAO1F,UAAU;IACrEM,OAAOL,SAAS,GAAG0F,aAAarF,OAAOL,SAAS,EAAEyF,OAAOzF,SAAS;IAClEK,OAAOJ,QAAQ,GAAGyF,aAAarF,OAAOJ,QAAQ,EAAEwF,OAAOxF,QAAQ;IAC/DI,OAAOH,YAAY,GAAGwF,aAAarF,OAAOH,YAAY,EAAEuF,OAAOvF,YAAY;AAC7E;AAEA,SAASsF,wBACPzG,KAA8B,EAC9BsG,QAAiC;IAEjC,OAAO;QACL,GAAGtG,KAAK;QACRQ,eAAeR,MAAMQ,aAAa,IAAI8F,SAAS9F,aAAa;QAC5DC,aAAaT,MAAMS,WAAW,IAAI6F,SAAS7F,WAAW;QACtDG,iBAAiBZ,MAAMY,eAAe,IAAI0F,SAAS1F,eAAe;QAClEC,cAAcb,MAAMa,YAAY,IAAIyF,SAASzF,YAAY;QACzD,GAAG+F,eAAeN,SAAS;IAC7B;AACF;AAEA,SAASM,eAAe5G,KAA8B;IACpD,OAAO;QACLe,WAAWf,MAAMe,SAAS;QAC1BC,YAAYhB,MAAMgB,UAAU;QAC5BC,WAAWjB,MAAMiB,SAAS;QAC1BC,UAAUlB,MAAMkB,QAAQ;QACxBC,cAAcnB,MAAMmB,YAAY;IAClC;AACF;AAEA,SAASwF,aAAaE,CAAc,EAAEC,CAAc;IAClD,IAAI,CAACD,GAAG,OAAOC;IACf,IAAI,CAACA,GAAG,OAAOD;IACf,OAAOA,IAAIC,IAAID,IAAIC;AACrB;AAEA,SAASN,kCACPK,CAA4B,EAC5BC,CAA4B;IAE5B,MAAMC,YAAYF,EAAE/F,UAAU,CAACkG,OAAO,KAAKF,EAAEhG,UAAU,CAACkG,OAAO;IAC/D,MAAMC,YAAYC,2BAA2BL,EAAE7H,KAAK,IAAIkI,2BAA2BJ,EAAE9H,KAAK;IAC1F,IAAIiI,cAAc,GAAG,OAAOA;IAC5B,OAAOF;AACT;AAEA,SAASG,2BAA2BlI,KAA0B;IAC5D,OAAQA;QACN,KAAK;YACH,OAAO;QACT,KAAK;YACH,OAAO;QACT,KAAK;YACH,OAAO;QACT,KAAK;YACH,OAAO;QACT,KAAK;YACH,OAAO;QACT,KAAK;YACH,OAAO;IACX;AACF;AAEA,SAASmI,wBAAwBnI,KAAyC;IACxE,OAAOzB,GAAW,CAAC;SACZ,EAAEyB,MAAM;;;;;;;;;EASf,CAAC;AACH;AAEA,SAASwC;IACP,OAAOjE,GAAY,CAAC;IAClB,EAAE4J,wBAAwB5J,GAAG,CAAC,cAAc,CAAC,EAAE,GAAG,EAAE4J,wBAAwBrJ,gBAAgBkB,KAAK,EAAE;;MAEjG,EAAEmI,wBAAwB5J,GAAG,CAAC,cAAc,CAAC,EAAE,GAAG,EAAE4J,wBAAwBrJ,gBAAgBkB,KAAK,EAAE;kCACvE,EAAElB,gBAAgBgD,UAAU,CAAC;;EAE7D,CAAC;AACH;AAEA,SAASa;IACP,OAAOpE,GAAY,CAAC;IAClB,EAAE6J,qBAAqBtJ,gBAAgBiD,SAAS,EAAExD,GAAG,CAAC,mBAAmB,CAAC,EAAE;OACzE,EAAE6J,qBAAqBtJ,gBAAgBkD,UAAU,EAAEzD,GAAG,CAAC,oBAAoB,CAAC,EAAE;OAC9E,EAAE6J,qBAAqBtJ,gBAAgBmD,SAAS,EAAE1D,GAAG,CAAC,mBAAmB,CAAC,EAAE;OAC5E,EAAE6J,qBAAqBtJ,gBAAgBoD,QAAQ,EAAE3D,GAAG,CAAC,kBAAkB,CAAC,EAAE;OAC1E,EAAE6J,qBAAqBtJ,gBAAgBqD,YAAY,EAAE5D,GAAG,CAAC,sBAAsB,CAAC,EAAE;EACvF,CAAC;AACH;AAEA,SAAS6J,qBACPC,OAA4C,EAC5CC,QAAa;IAEb,OAAO/J,GAAY,CAAC,EAAE+J,SAAS,kBAAkB,EAAED,QAAQ,YAAY,EAAEC,SAAS,GAAG,EAAED,QAAQ,CAAC,CAAC;AACnG;AAEA,SAAS5F,gBAAgB4F,OAA4C,EAAEC,QAAa;IAClF,OAAO/J,GAAG,CAAC;;WAEF,EAAE+J,SAAS,cAAc,EAAED,QAAQ;WACnC,EAAEA,QAAQ,cAAc,EAAEC,SAAS;iBAC7B,EAAED,QAAQ,EAAE,EAAEC,SAAS;;EAEtC,CAAC;AACH;AAEA,OAAO,SAASrH,gBAAgBjB,KAA0B;IACxD,OAAOd,eAAeqJ,QAAQ,CAACvI;AACjC;AAEA,SAASqD,uBACPrD,KAA0B;IAE1B,OAAOZ,sBAAsBmJ,QAAQ,CAACvI;AACxC"}

package/dist/db/runner-sessions.d.ts

@@ -9,6 +9,11 @@ export interface CreateRunnerSessionParams {
 }
 export declare function createRunnerSession(params: CreateRunnerSessionParams): Promise<RunnerSession>;
 export declare function getRunnerSessionById(runnerSessionId: string): Promise<RunnerSession | null>;
+export declare function createRunnerSessionConsumingActivationToken(params: {
+    activationTokenId: string;
+    labels: string[];
+    toolCapabilities?: RunnerToolCapabilitiesDto | null;
+}): Promise<RunnerSession>;
 export interface DeleteExpiredRunnerSessionsParams {
     manualRetentionDays: number;
     ephemeralRetentionDays: number;

package/dist/db/runner-sessions.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"runner-sessions.d.ts","sourceRoot":"","sources":["../../src/db/runner-sessions.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,yBAAyB,EAAC,MAAM,0BAA0B,CAAC;AAExE,OAAO,KAAK,EAAC,aAAa,EAAC,MAAM,kCAAkC,CAAC;AAKpE,MAAM,WAAW,yBAAyB;IACxC,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,WAAW,CAAC;IACnB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,gBAAgB,CAAC,EAAE,yBAAyB,GAAG,IAAI,CAAC;CACrD;AAED,wBAAsB,mBAAmB,CACvC,MAAM,EAAE,yBAAyB,GAChC,OAAO,CAAC,aAAa,CAAC,CAmBxB;AAED,wBAAsB,oBAAoB,CAAC,eAAe,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC,CASjG;AAED,MAAM,WAAW,iCAAiC;IAChD,mBAAmB,EAAE,MAAM,CAAC;IAC5B,sBAAsB,EAAE,MAAM,CAAC;IAC/B,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,wBAAsB,2BAA2B,CAC/C,MAAM,EAAE,iCAAiC,GACxC,OAAO,CAAC,MAAM,CAAC,CAuCjB"}
+{"version":3,"file":"runner-sessions.d.ts","sourceRoot":"","sources":["../../src/db/runner-sessions.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,yBAAyB,EAAC,MAAM,0BAA0B,CAAC;AAExE,OAAO,KAAK,EAAC,aAAa,EAAC,MAAM,kCAAkC,CAAC;AAQpE,MAAM,WAAW,yBAAyB;IACxC,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,WAAW,CAAC;IACnB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,gBAAgB,CAAC,EAAE,yBAAyB,GAAG,IAAI,CAAC;CACrD;AAED,wBAAsB,mBAAmB,CACvC,MAAM,EAAE,yBAAyB,GAChC,OAAO,CAAC,aAAa,CAAC,CAmBxB;AAED,wBAAsB,oBAAoB,CAAC,eAAe,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC,CASjG;AAED,wBAAsB,2CAA2C,CAAC,MAAM,EAAE;IACxE,iBAAiB,EAAE,MAAM,CAAC;IAC1B,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,gBAAgB,CAAC,EAAE,yBAAyB,GAAG,IAAI,CAAC;CACrD,0BAmEA;AAED,MAAM,WAAW,iCAAiC;IAChD,mBAAmB,EAAE,MAAM,CAAC;IAC5B,sBAAsB,EAAE,MAAM,CAAC;IAC/B,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,wBAAsB,2BAA2B,CAC/C,MAAM,EAAE,iCAAiC,GACxC,OAAO,CAAC,MAAM,CAAC,CAuCjB"}