@shipfox/api-runners 12.1.0 → 12.3.0

package/dist/db/index.js

@@ -10,7 +10,7 @@ export { createInstallationProvisionerTokenWithAudit, createProvisionerToken, li
 export { consumeRunnersRateLimit, pruneExpiredRunnersRateLimits } from './rate-limits.js';
 export { deleteExpiredReservations, deleteReservationsByIds, pollDemandAndReserve, releaseReservationUnits } from './reservations.js';
 export { assignRunnerInstances } from './runner-assignments.js';
-export { attachRunnerInstanceProviderId, isTerminalState, listActiveRunnerInstanceCountsByTemplateTx, listActiveRunnerInstances, listProvisionerTerminateIntentRowsTx, listProvisionerTerminateIntents, reapStaleRunnerInstances, reconcileRunnerInstances, reportRunnerInstances } from './runner-instances.js';
+export { attachRunnerInstanceProviderId, countStaleEnrolledRunnerInstances, isTerminalState, listActiveRunnerInstanceCountsByTemplateTx, listActiveRunnerInstances, listProvisionerTerminateIntentRowsTx, listProvisionerTerminateIntents, reapStaleRunnerInstances, reconcileRunnerInstances, reportRunnerInstances } from './runner-instances.js';
 export { createRunnerSession, deleteExpiredRunnerSessions } from './runner-sessions.js';
 export { runnersOutbox } from './schema/outbox.js';
 export { runnerBootstrapTokens, runnerControlSessions } from './schema/runner-control-sessions.js';

package/dist/db/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/db/index.ts"],"sourcesContent":["import {dirname, resolve} from 'node:path';\nimport {fileURLToPath} from 'node:url';\n\nexport type {\n  AdministratorRunnerAssignmentFilter,\n  AdministratorRunnerInstanceRow,\n  ListAdministratorRunnerInstancesParams,\n  ListAdministratorRunnerInstancesResult,\n} from './admin-runner-instances.js';\nexport {listAdministratorRunnerInstances} from './admin-runner-instances.js';\nexport {closeDb, db, schema} from './db.js';\nexport type {\n  CreateEphemeralRegistrationTokenParams,\n  DeleteExpiredEphemeralRegistrationTokensParams,\n} from './ephemeral-registration-tokens.js';\nexport {\n  createEphemeralRegistrationToken,\n  createRunnerSessionConsumingEphemeralToken,\n  deleteExpiredEphemeralRegistrationTokens,\n  resolveEphemeralRegistrationTokenByHash,\n} from './ephemeral-registration-tokens.js';\nexport type {\n  ClaimedJobExecution,\n  EnqueueJobExecutionParams,\n  RunnerInstanceBoundJobExecution,\n} from './job-executions.js';\nexport {\n  cancelRunnerJobs,\n  claimPendingJobExecution,\n  enqueueJobExecution,\n  expireStuckJobExecutions,\n  getWorkspaceJobCounts,\n  isJobLeaseActive,\n  listActiveRunningJobExecutions,\n  listRunningJobExecutionsByRunnerInstanceTx,\n  recordHeartbeat,\n  releaseJobExecution,\n} from './job-executions.js';\nexport type {CreateManualRegistrationTokenParams} from './manual-registration-tokens.js';\nexport {\n  createManualRegistrationToken,\n  listUsableManualRegistrationTokensByWorkspaceId,\n  resolveManualRegistrationTokenByHash,\n  revokeManualRegistrationToken,\n} from './manual-registration-tokens.js';\nexport {\n  hasActiveWorkspaceProvisionerCapability,\n  listActiveWorkspaceProvisionerCapabilitySnapshots,\n  listStaleWorkspaceProvisionerCapabilitySnapshots,\n  publishWorkspaceProvisionerCapabilitySnapshot,\n} from './provisioner-capability-snapshots.js';\nexport type {CreateProvisionerTokenParams} from './provisioner-tokens.js';\nexport {\n  createInstallationProvisionerTokenWithAudit,\n  createProvisionerToken,\n  listActiveProvisionerTokens,\n  listInstallationProvisionerTokens,\n  listUsableProvisionerTokensByWorkspaceId,\n  resolveProvisionerTokenByHash,\n  revokeInstallationProvisionerTokenWithAudit,\n  revokeProvisionerToken,\n  touchProvisionerLastSeen,\n} from './provisioner-tokens.js';\nexport {\n  consumeRunnersRateLimit,\n  pruneExpiredRunnersRateLimits,\n} from './rate-limits.js';\nexport type {\n  DemandStat,\n  PollDemandAndReserveParams,\n  ReservationGrant,\n  ReservationTemplate,\n} from './reservations.js';\nexport {\n  deleteExpiredReservations,\n  deleteReservationsByIds,\n  pollDemandAndReserve,\n  releaseReservationUnits,\n} from './reservations.js';\nexport {assignRunnerInstances} from './runner-assignments.js';\nexport type {\n  ActiveRunnerInstanceTemplateCount,\n  ReapStaleRunnerInstancesResult,\n  ReconcileRunnerInstancesDbResult,\n  ReconcileRunnerInstancesParams,\n  ReportRunnerInstancesParams,\n  RunnerInstanceReportEvent,\n  RunnerInstanceTerminateIntent,\n  RunnerInstanceTerminateIntentReason,\n} from './runner-instances.js';\nexport {\n  attachRunnerInstanceProviderId,\n  isTerminalState,\n  listActiveRunnerInstanceCountsByTemplateTx,\n  listActiveRunnerInstances,\n  listProvisionerTerminateIntentRowsTx,\n  listProvisionerTerminateIntents,\n  reapStaleRunnerInstances,\n  reconcileRunnerInstances,\n  reportRunnerInstances,\n} from './runner-instances.js';\nexport type {\n  CreateRunnerSessionParams,\n  DeleteExpiredRunnerSessionsParams,\n} from './runner-sessions.js';\nexport {createRunnerSession, deleteExpiredRunnerSessions} from './runner-sessions.js';\nexport {runnersOutbox} from './schema/outbox.js';\nexport {runnerBootstrapTokens, runnerControlSessions} from './schema/runner-control-sessions.js';\n\nexport const migrationsPath = resolve(dirname(fileURLToPath(import.meta.url)), '../../drizzle');\n"],"names":["dirname","resolve","fileURLToPath","listAdministratorRunnerInstances","closeDb","db","schema","createEphemeralRegistrationToken","createRunnerSessionConsumingEphemeralToken","deleteExpiredEphemeralRegistrationTokens","resolveEphemeralRegistrationTokenByHash","cancelRunnerJobs","claimPendingJobExecution","enqueueJobExecution","expireStuckJobExecutions","getWorkspaceJobCounts","isJobLeaseActive","listActiveRunningJobExecutions","listRunningJobExecutionsByRunnerInstanceTx","recordHeartbeat","releaseJobExecution","createManualRegistrationToken","listUsableManualRegistrationTokensByWorkspaceId","resolveManualRegistrationTokenByHash","revokeManualRegistrationToken","hasActiveWorkspaceProvisionerCapability","listActiveWorkspaceProvisionerCapabilitySnapshots","listStaleWorkspaceProvisionerCapabilitySnapshots","publishWorkspaceProvisionerCapabilitySnapshot","createInstallationProvisionerTokenWithAudit","createProvisionerToken","listActiveProvisionerTokens","listInstallationProvisionerTokens","listUsableProvisionerTokensByWorkspaceId","resolveProvisionerTokenByHash","revokeInstallationProvisionerTokenWithAudit","revokeProvisionerToken","touchProvisionerLastSeen","consumeRunnersRateLimit","pruneExpiredRunnersRateLimits","deleteExpiredReservations","deleteReservationsByIds","pollDemandAndReserve","releaseReservationUnits","assignRunnerInstances","attachRunnerInstanceProviderId","isTerminalState","listActiveRunnerInstanceCountsByTemplateTx","listActiveRunnerInstances","listProvisionerTerminateIntentRowsTx","listProvisionerTerminateIntents","reapStaleRunnerInstances","reconcileRunnerInstances","reportRunnerInstances","createRunnerSession","deleteExpiredRunnerSessions","runnersOutbox","runnerBootstrapTokens","runnerControlSessions","migrationsPath","url"],"mappings":"AAAA,SAAQA,OAAO,EAAEC,OAAO,QAAO,YAAY;AAC3C,SAAQC,aAAa,QAAO,WAAW;AAQvC,SAAQC,gCAAgC,QAAO,8BAA8B;AAC7E,SAAQC,OAAO,EAAEC,EAAE,EAAEC,MAAM,QAAO,UAAU;AAK5C,SACEC,gCAAgC,EAChCC,0CAA0C,EAC1CC,wCAAwC,EACxCC,uCAAuC,QAClC,qCAAqC;AAM5C,SACEC,gBAAgB,EAChBC,wBAAwB,EACxBC,mBAAmB,EACnBC,wBAAwB,EACxBC,qBAAqB,EACrBC,gBAAgB,EAChBC,8BAA8B,EAC9BC,0CAA0C,EAC1CC,eAAe,EACfC,mBAAmB,QACd,sBAAsB;AAE7B,SACEC,6BAA6B,EAC7BC,+CAA+C,EAC/CC,oCAAoC,EACpCC,6BAA6B,QACxB,kCAAkC;AACzC,SACEC,uCAAuC,EACvCC,iDAAiD,EACjDC,gDAAgD,EAChDC,6CAA6C,QACxC,wCAAwC;AAE/C,SACEC,2CAA2C,EAC3CC,sBAAsB,EACtBC,2BAA2B,EAC3BC,iCAAiC,EACjCC,wCAAwC,EACxCC,6BAA6B,EAC7BC,2CAA2C,EAC3CC,sBAAsB,EACtBC,wBAAwB,QACnB,0BAA0B;AACjC,SACEC,uBAAuB,EACvBC,6BAA6B,QACxB,mBAAmB;AAO1B,SACEC,yBAAyB,EACzBC,uBAAuB,EACvBC,oBAAoB,EACpBC,uBAAuB,QAClB,oBAAoB;AAC3B,SAAQC,qBAAqB,QAAO,0BAA0B;AAW9D,SACEC,8BAA8B,EAC9BC,eAAe,EACfC,0CAA0C,EAC1CC,yBAAyB,EACzBC,oCAAoC,EACpCC,+BAA+B,EAC/BC,wBAAwB,EACxBC,wBAAwB,EACxBC,qBAAqB,QAChB,wBAAwB;AAK/B,SAAQC,mBAAmB,EAAEC,2BAA2B,QAAO,uBAAuB;AACtF,SAAQC,aAAa,QAAO,qBAAqB;AACjD,SAAQC,qBAAqB,EAAEC,qBAAqB,QAAO,sCAAsC;AAEjG,OAAO,MAAMC,iBAAiB1D,QAAQD,QAAQE,cAAc,YAAY0D,GAAG,IAAI,iBAAiB"}
+{"version":3,"sources":["../../src/db/index.ts"],"sourcesContent":["import {dirname, resolve} from 'node:path';\nimport {fileURLToPath} from 'node:url';\n\nexport type {\n  AdministratorRunnerAssignmentFilter,\n  AdministratorRunnerInstanceRow,\n  ListAdministratorRunnerInstancesParams,\n  ListAdministratorRunnerInstancesResult,\n} from './admin-runner-instances.js';\nexport {listAdministratorRunnerInstances} from './admin-runner-instances.js';\nexport {closeDb, db, schema} from './db.js';\nexport type {\n  CreateEphemeralRegistrationTokenParams,\n  DeleteExpiredEphemeralRegistrationTokensParams,\n} from './ephemeral-registration-tokens.js';\nexport {\n  createEphemeralRegistrationToken,\n  createRunnerSessionConsumingEphemeralToken,\n  deleteExpiredEphemeralRegistrationTokens,\n  resolveEphemeralRegistrationTokenByHash,\n} from './ephemeral-registration-tokens.js';\nexport type {\n  ClaimedJobExecution,\n  EnqueueJobExecutionParams,\n  RunnerInstanceBoundJobExecution,\n} from './job-executions.js';\nexport {\n  cancelRunnerJobs,\n  claimPendingJobExecution,\n  enqueueJobExecution,\n  expireStuckJobExecutions,\n  getWorkspaceJobCounts,\n  isJobLeaseActive,\n  listActiveRunningJobExecutions,\n  listRunningJobExecutionsByRunnerInstanceTx,\n  recordHeartbeat,\n  releaseJobExecution,\n} from './job-executions.js';\nexport type {CreateManualRegistrationTokenParams} from './manual-registration-tokens.js';\nexport {\n  createManualRegistrationToken,\n  listUsableManualRegistrationTokensByWorkspaceId,\n  resolveManualRegistrationTokenByHash,\n  revokeManualRegistrationToken,\n} from './manual-registration-tokens.js';\nexport {\n  hasActiveWorkspaceProvisionerCapability,\n  listActiveWorkspaceProvisionerCapabilitySnapshots,\n  listStaleWorkspaceProvisionerCapabilitySnapshots,\n  publishWorkspaceProvisionerCapabilitySnapshot,\n} from './provisioner-capability-snapshots.js';\nexport type {CreateProvisionerTokenParams} from './provisioner-tokens.js';\nexport {\n  createInstallationProvisionerTokenWithAudit,\n  createProvisionerToken,\n  listActiveProvisionerTokens,\n  listInstallationProvisionerTokens,\n  listUsableProvisionerTokensByWorkspaceId,\n  resolveProvisionerTokenByHash,\n  revokeInstallationProvisionerTokenWithAudit,\n  revokeProvisionerToken,\n  touchProvisionerLastSeen,\n} from './provisioner-tokens.js';\nexport {\n  consumeRunnersRateLimit,\n  pruneExpiredRunnersRateLimits,\n} from './rate-limits.js';\nexport type {\n  DemandStat,\n  PollDemandAndReserveParams,\n  ReservationGrant,\n  ReservationTemplate,\n} from './reservations.js';\nexport {\n  deleteExpiredReservations,\n  deleteReservationsByIds,\n  pollDemandAndReserve,\n  releaseReservationUnits,\n} from './reservations.js';\nexport {assignRunnerInstances} from './runner-assignments.js';\nexport type {\n  ActiveRunnerInstanceTemplateCount,\n  ReapStaleRunnerInstancesResult,\n  ReconcileRunnerInstancesDbResult,\n  ReconcileRunnerInstancesParams,\n  ReportRunnerInstancesParams,\n  RunnerInstanceReportEvent,\n  RunnerInstanceTerminateIntent,\n  RunnerInstanceTerminateIntentReason,\n} from './runner-instances.js';\nexport {\n  attachRunnerInstanceProviderId,\n  countStaleEnrolledRunnerInstances,\n  isTerminalState,\n  listActiveRunnerInstanceCountsByTemplateTx,\n  listActiveRunnerInstances,\n  listProvisionerTerminateIntentRowsTx,\n  listProvisionerTerminateIntents,\n  reapStaleRunnerInstances,\n  reconcileRunnerInstances,\n  reportRunnerInstances,\n} from './runner-instances.js';\nexport type {\n  CreateRunnerSessionParams,\n  DeleteExpiredRunnerSessionsParams,\n} from './runner-sessions.js';\nexport {createRunnerSession, deleteExpiredRunnerSessions} from './runner-sessions.js';\nexport {runnersOutbox} from './schema/outbox.js';\nexport {runnerBootstrapTokens, runnerControlSessions} from './schema/runner-control-sessions.js';\n\nexport const migrationsPath = resolve(dirname(fileURLToPath(import.meta.url)), '../../drizzle');\n"],"names":["dirname","resolve","fileURLToPath","listAdministratorRunnerInstances","closeDb","db","schema","createEphemeralRegistrationToken","createRunnerSessionConsumingEphemeralToken","deleteExpiredEphemeralRegistrationTokens","resolveEphemeralRegistrationTokenByHash","cancelRunnerJobs","claimPendingJobExecution","enqueueJobExecution","expireStuckJobExecutions","getWorkspaceJobCounts","isJobLeaseActive","listActiveRunningJobExecutions","listRunningJobExecutionsByRunnerInstanceTx","recordHeartbeat","releaseJobExecution","createManualRegistrationToken","listUsableManualRegistrationTokensByWorkspaceId","resolveManualRegistrationTokenByHash","revokeManualRegistrationToken","hasActiveWorkspaceProvisionerCapability","listActiveWorkspaceProvisionerCapabilitySnapshots","listStaleWorkspaceProvisionerCapabilitySnapshots","publishWorkspaceProvisionerCapabilitySnapshot","createInstallationProvisionerTokenWithAudit","createProvisionerToken","listActiveProvisionerTokens","listInstallationProvisionerTokens","listUsableProvisionerTokensByWorkspaceId","resolveProvisionerTokenByHash","revokeInstallationProvisionerTokenWithAudit","revokeProvisionerToken","touchProvisionerLastSeen","consumeRunnersRateLimit","pruneExpiredRunnersRateLimits","deleteExpiredReservations","deleteReservationsByIds","pollDemandAndReserve","releaseReservationUnits","assignRunnerInstances","attachRunnerInstanceProviderId","countStaleEnrolledRunnerInstances","isTerminalState","listActiveRunnerInstanceCountsByTemplateTx","listActiveRunnerInstances","listProvisionerTerminateIntentRowsTx","listProvisionerTerminateIntents","reapStaleRunnerInstances","reconcileRunnerInstances","reportRunnerInstances","createRunnerSession","deleteExpiredRunnerSessions","runnersOutbox","runnerBootstrapTokens","runnerControlSessions","migrationsPath","url"],"mappings":"AAAA,SAAQA,OAAO,EAAEC,OAAO,QAAO,YAAY;AAC3C,SAAQC,aAAa,QAAO,WAAW;AAQvC,SAAQC,gCAAgC,QAAO,8BAA8B;AAC7E,SAAQC,OAAO,EAAEC,EAAE,EAAEC,MAAM,QAAO,UAAU;AAK5C,SACEC,gCAAgC,EAChCC,0CAA0C,EAC1CC,wCAAwC,EACxCC,uCAAuC,QAClC,qCAAqC;AAM5C,SACEC,gBAAgB,EAChBC,wBAAwB,EACxBC,mBAAmB,EACnBC,wBAAwB,EACxBC,qBAAqB,EACrBC,gBAAgB,EAChBC,8BAA8B,EAC9BC,0CAA0C,EAC1CC,eAAe,EACfC,mBAAmB,QACd,sBAAsB;AAE7B,SACEC,6BAA6B,EAC7BC,+CAA+C,EAC/CC,oCAAoC,EACpCC,6BAA6B,QACxB,kCAAkC;AACzC,SACEC,uCAAuC,EACvCC,iDAAiD,EACjDC,gDAAgD,EAChDC,6CAA6C,QACxC,wCAAwC;AAE/C,SACEC,2CAA2C,EAC3CC,sBAAsB,EACtBC,2BAA2B,EAC3BC,iCAAiC,EACjCC,wCAAwC,EACxCC,6BAA6B,EAC7BC,2CAA2C,EAC3CC,sBAAsB,EACtBC,wBAAwB,QACnB,0BAA0B;AACjC,SACEC,uBAAuB,EACvBC,6BAA6B,QACxB,mBAAmB;AAO1B,SACEC,yBAAyB,EACzBC,uBAAuB,EACvBC,oBAAoB,EACpBC,uBAAuB,QAClB,oBAAoB;AAC3B,SAAQC,qBAAqB,QAAO,0BAA0B;AAW9D,SACEC,8BAA8B,EAC9BC,iCAAiC,EACjCC,eAAe,EACfC,0CAA0C,EAC1CC,yBAAyB,EACzBC,oCAAoC,EACpCC,+BAA+B,EAC/BC,wBAAwB,EACxBC,wBAAwB,EACxBC,qBAAqB,QAChB,wBAAwB;AAK/B,SAAQC,mBAAmB,EAAEC,2BAA2B,QAAO,uBAAuB;AACtF,SAAQC,aAAa,QAAO,qBAAqB;AACjD,SAAQC,qBAAqB,EAAEC,qBAAqB,QAAO,sCAAsC;AAEjG,OAAO,MAAMC,iBAAiB3D,QAAQD,QAAQE,cAAc,YAAY2D,GAAG,IAAI,iBAAiB"}

package/dist/db/reservations.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"reservations.d.ts","sourceRoot":"","sources":["../../src/db/reservations.ts"],"names":[],"mappings":"AAcA,OAAO,KAAK,EAAC,EAAE,EAAC,MAAM,SAAS,CAAC;AAUhC,MAAM,WAAW,mBAAmB;IAClC,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,cAAc,EAAE,MAAM,CAAC;IACvB,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,UAAU;IACzB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,cAAc,EAAE,IAAI,CAAC;CACtB;AAED,MAAM,WAAW,gBAAgB;IAC/B,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,IAAI,CAAC;CACjB;AAED,MAAM,WAAW,0BAA0B;IACzC,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,eAAe,EAAE,MAAM,CAAC;IACxB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,mBAAmB,EAAE,CAAC;IACjC,uBAAuB,CAAC,EAAE,MAAM,CAAC;CAClC;AAED,MAAM,WAAW,sCAAsC;IACrD,aAAa,EAAE,MAAM,CAAC;IACtB,eAAe,EAAE,MAAM,CAAC;IACxB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,mBAAmB,EAAE,CAAC;IACjC,uBAAuB,EAAE,MAAM,CAAC;IAChC,oBAAoB,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;IAC1C,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,cAAc,CAAC,EAAE,CAAC,YAAY,EAAE,gBAAgB,EAAE,KAAK,IAAI,CAAC;CAC7D;AAcD,wBAAsB,oBAAoB,CACxC,MAAM,EAAE,0BAA0B,GACjC,OAAO,CAAC;IAAC,KAAK,EAAE,UAAU,EAAE,CAAC;IAAC,YAAY,EAAE,gBAAgB,EAAE,CAAA;CAAC,CAAC,CAIlE;AAED,wBAAsB,sBAAsB,CAC1C,EAAE,EAAE,EAAE,EACN,MAAM,EAAE,0BAA0B,GACjC,OAAO,CAAC;IAAC,KAAK,EAAE,UAAU,EAAE,CAAC;IAAC,YAAY,EAAE,gBAAgB,EAAE,CAAA;CAAC,CAAC,CAGlE;AAED,wBAAsB,gCAAgC,CACpD,MAAM,EAAE,sCAAsC,GAC7C,OAAO,CAAC;IAAC,KAAK,EAAE,UAAU,EAAE,CAAC;IAAC,YAAY,EAAE,gBAAgB,EAAE,CAAA;CAAC,CAAC,CA2ClE;AAsOD,wBAAsB,4BAA4B,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC,CAMtE;AA0BD,wBAAsB,yBAAyB,CAAC,MAAM,CAAC,EAAE;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,CAc1F;AAED,wBAAsB,uBAAuB,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,CAM5E;AAyED,wBAAsB,uBAAuB,CAC3C,EAAE,EAAE,EAAE,EACN,MAAM,EAAE;IACN,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,KAAK,CAAC;QAAC,aAAa,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAC,CAAC,CAAC;CACzD,GACA,OAAO,CAAC,MAAM,CAAC,CA0CjB"}
+{"version":3,"file":"reservations.d.ts","sourceRoot":"","sources":["../../src/db/reservations.ts"],"names":[],"mappings":"AAgBA,OAAO,KAAK,EAAC,EAAE,EAAC,MAAM,SAAS,CAAC;AAUhC,MAAM,WAAW,mBAAmB;IAClC,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,cAAc,EAAE,MAAM,CAAC;IACvB,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,UAAU;IACzB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,cAAc,EAAE,IAAI,CAAC;CACtB;AAED,MAAM,WAAW,gBAAgB;IAC/B,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,IAAI,CAAC;CACjB;AAED,MAAM,WAAW,0BAA0B;IACzC,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,eAAe,EAAE,MAAM,CAAC;IACxB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,mBAAmB,EAAE,CAAC;IACjC,uBAAuB,CAAC,EAAE,MAAM,CAAC;CAClC;AAED,MAAM,WAAW,sCAAsC;IACrD,aAAa,EAAE,MAAM,CAAC;IACtB,eAAe,EAAE,MAAM,CAAC;IACxB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,mBAAmB,EAAE,CAAC;IACjC,uBAAuB,EAAE,MAAM,CAAC;IAChC,oBAAoB,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;IAC1C,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,cAAc,CAAC,EAAE,CAAC,YAAY,EAAE,gBAAgB,EAAE,KAAK,IAAI,CAAC;CAC7D;AAoBD,wBAAsB,oBAAoB,CACxC,MAAM,EAAE,0BAA0B,GACjC,OAAO,CAAC;IAAC,KAAK,EAAE,UAAU,EAAE,CAAC;IAAC,YAAY,EAAE,gBAAgB,EAAE,CAAA;CAAC,CAAC,CAIlE;AAED,wBAAsB,sBAAsB,CAC1C,EAAE,EAAE,EAAE,EACN,MAAM,EAAE,0BAA0B,GACjC,OAAO,CAAC;IAAC,KAAK,EAAE,UAAU,EAAE,CAAC;IAAC,YAAY,EAAE,gBAAgB,EAAE,CAAA;CAAC,CAAC,CAGlE;AAED,wBAAsB,gCAAgC,CACpD,MAAM,EAAE,sCAAsC,GAC7C,OAAO,CAAC;IAAC,KAAK,EAAE,UAAU,EAAE,CAAC;IAAC,YAAY,EAAE,gBAAgB,EAAE,CAAA;CAAC,CAAC,CA4ClE;AA+RD,wBAAsB,4BAA4B,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC,CAMtE;AA0BD,wBAAsB,yBAAyB,CAAC,MAAM,CAAC,EAAE;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,CAc1F;AAED,wBAAsB,uBAAuB,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,CAM5E;AA0GD,wBAAsB,uBAAuB,CAC3C,EAAE,EAAE,EAAE,EACN,MAAM,EAAE;IACN,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,KAAK,CAAC;QAAC,aAAa,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAC,CAAC,CAAC;CACzD,GACA,OAAO,CAAC,MAAM,CAAC,CA0CjB"}

package/dist/db/reservations.js

@@ -1,5 +1,5 @@
 import { canonicalizeLabels } from '@shipfox/runner-labels';
-import { and, arrayContains, asc, eq, exists, gt, inArray, isNotNull, isNull, lt, sql } from 'drizzle-orm';
+import { and, arrayContains, asc, eq, exists, gt, inArray, isNotNull, isNull, lt, notExists, or, sql } from 'drizzle-orm';
 import { db } from './db.js';
 import { pendingJobExecutions } from './schema/pending-job-executions.js';
 import { provisionerCapabilitySnapshots } from './schema/provisioner-capability-snapshots.js';
@@ -15,7 +15,10 @@ export async function pollDemandAndReserve(params) {
 }
 export async function pollDemandAndReserveTx(tx, params) {
     await tx.execute(sql`select pg_advisory_xact_lock(hashtext(${params.workspaceId}))`);
-    return await pollDemandAndReserveLockedTx(tx, params);
+    return await pollDemandAndReserveLockedTx(tx, {
+        ...params,
+        scope: 'workspace'
+    });
 }
 export async function pollInstallationDemandAndReserve(params) {
     const candidateWorkspaceIds = await listInstallationDemandWorkspaceIds(params.eligibleWorkspaceIds);
@@ -46,7 +49,8 @@ export async function pollInstallationDemandAndReserve(params) {
                         ...template,
                         availableSlots: template.remainingSlots
                     })),
-                capabilityWindowSeconds: params.capabilityWindowSeconds
+                capabilityWindowSeconds: params.capabilityWindowSeconds,
+                scope: 'installation'
             });
         });
         results.push(result);
@@ -134,7 +138,8 @@ async function pollDemandAndReserveLockedTx(tx, params) {
                 reservationId: inserted.id,
                 requiredLabels: demand.requiredLabels,
                 count: grant,
-                workspaceId: params.workspaceId
+                workspaceId: params.workspaceId,
+                scope: params.scope
             });
             reservedAfterGrant += grant;
             grants.push({
@@ -159,18 +164,29 @@ async function pollDemandAndReserveLockedTx(tx, params) {
     };
 }
 async function bindIdleRunnerInstancesTx(tx, params) {
+    // Reservation expiry is the activation grace deadline. A live reservation
+    // protects a runner that is still booting; an expired or missing row is stale.
+    const canBindRunner = ()=>and(or(and(isNull(providerRunners.workspaceId), isNull(providerRunners.reservationId)), and(isNotNull(providerRunners.reservationId), params.scope === 'installation' ? undefined : or(isNull(providerRunners.workspaceId), eq(providerRunners.workspaceId, params.workspaceId)), notExists(tx.select({
+            id: reservations.id
+        }).from(reservations).where(and(eq(reservations.id, providerRunners.reservationId), gt(reservations.expiresAt, sql`now()`)))))), or(isNull(providerRunners.intendedReservationId), notExists(tx.select({
+            id: reservations.id
+        }).from(reservations).where(and(eq(reservations.id, providerRunners.intendedReservationId), gt(reservations.expiresAt, sql`now()`))))), isNull(providerRunners.reservationReleasedAt), isNull(providerRunners.runnerSessionId));
     const idleRunners = await tx.select({
         id: providerRunners.id
-    }).from(providerRunners).where(and(eq(providerRunners.provisionerId, params.provisionerId), isNull(providerRunners.workspaceId), isNull(providerRunners.reservationId), isNull(providerRunners.intendedReservationId), isNull(providerRunners.runnerSessionId), isNotNull(providerRunners.providerRunnerId), eq(providerRunners.state, 'running'), arrayContains(providerRunners.labels, params.requiredLabels), exists(tx.select({
+    }).from(providerRunners).where(and(eq(providerRunners.provisionerId, params.provisionerId), canBindRunner(), isNotNull(providerRunners.providerRunnerId), eq(providerRunners.state, 'running'), arrayContains(providerRunners.labels, params.requiredLabels), exists(tx.select({
         id: runnerControlSessions.id
     }).from(runnerControlSessions).where(and(eq(runnerControlSessions.runnerInstanceId, providerRunners.id), eq(runnerControlSessions.provisionerId, params.provisionerId), isNull(runnerControlSessions.closedAt), gt(runnerControlSessions.expiresAt, sql`now()`)))))).orderBy(asc(providerRunners.createdAt), asc(providerRunners.id)).limit(params.count).for('update');
     if (idleRunners.length === 0) return;
+    await tx.update(runnerActivationTokens).set({
+        revokedAt: sql`now()`
+    }).where(and(inArray(runnerActivationTokens.runnerInstanceId, idleRunners.map((runner)=>runner.id)), isNull(runnerActivationTokens.consumedAt), isNull(runnerActivationTokens.revokedAt)));
     await tx.update(providerRunners).set({
         workspaceId: params.workspaceId,
         reservationId: params.reservationId,
+        intendedReservationId: null,
         assignedAt: sql`now()`,
         updatedAt: sql`now()`
-    }).where(and(inArray(providerRunners.id, idleRunners.map((runner)=>runner.id)), isNull(providerRunners.workspaceId), isNull(providerRunners.reservationId), isNull(providerRunners.runnerSessionId)));
+    }).where(and(inArray(providerRunners.id, idleRunners.map((runner)=>runner.id)), canBindRunner()));
 }
 async function listInstallationDemandWorkspaceIds(eligibleWorkspaceIds) {
     if (eligibleWorkspaceIds.size === 0) return [];
@@ -212,16 +228,20 @@ export async function deleteReservationsByIds(ids) {
 }
 async function deleteReservationsWithCleanupTx(tx, ids, params = {}) {
     if (ids.length === 0) return 0;
-    const assignedRunners = await tx.select({
+    const affectedRunners = await tx.select({
         id: providerRunners.id,
-        reservationId: providerRunners.reservationId
-    }).from(providerRunners).where(and(inArray(providerRunners.reservationId, ids), isNull(providerRunners.runnerSessionId), isNull(providerRunners.reservationReleasedAt))).for('update');
+        reservationId: providerRunners.reservationId,
+        intendedReservationId: providerRunners.intendedReservationId,
+        runnerSessionId: providerRunners.runnerSessionId,
+        reservationReleasedAt: providerRunners.reservationReleasedAt
+    }).from(providerRunners).where(or(and(inArray(providerRunners.reservationId, ids), isNull(providerRunners.runnerSessionId), isNull(providerRunners.reservationReleasedAt)), inArray(providerRunners.intendedReservationId, ids))).orderBy(asc(providerRunners.id)).for('update');
     const reservationRows = await tx.select({
         id: reservations.id
     }).from(reservations).where(and(inArray(reservations.id, ids), params.expiredOnly ? lt(reservations.expiresAt, sql`now()`) : undefined)).for('update');
     const reservationIds = reservationRows.map((reservation)=>reservation.id);
     if (reservationIds.length === 0) return 0;
-    const assignedRunnerIds = assignedRunners.filter((runner)=>runner.reservationId && reservationIds.includes(runner.reservationId)).map((runner)=>runner.id);
+    const assignedRunnerIds = affectedRunners.filter((runner)=>runner.reservationId && reservationIds.includes(runner.reservationId) && !runner.runnerSessionId && !runner.reservationReleasedAt).map((runner)=>runner.id);
+    const intendedRunnerIds = affectedRunners.filter((runner)=>runner.intendedReservationId && reservationIds.includes(runner.intendedReservationId)).map((runner)=>runner.id);
     if (assignedRunnerIds.length > 0) {
         await tx.update(runnerActivationTokens).set({
             revokedAt: sql`now()`
@@ -233,6 +253,12 @@ async function deleteReservationsWithCleanupTx(tx, ids, params = {}) {
             updatedAt: sql`now()`
         }).where(and(inArray(providerRunners.id, assignedRunnerIds), isNull(providerRunners.runnerSessionId), isNull(providerRunners.reservationReleasedAt)));
     }
+    if (intendedRunnerIds.length > 0) {
+        await tx.update(providerRunners).set({
+            intendedReservationId: null,
+            updatedAt: sql`now()`
+        }).where(and(inArray(providerRunners.id, intendedRunnerIds), inArray(providerRunners.intendedReservationId, reservationIds)));
+    }
     const deleted = await tx.delete(reservations).where(inArray(reservations.id, reservationIds)).returning({
         id: reservations.id
     });

package/dist/db/reservations.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/db/reservations.ts"],"sourcesContent":["import {canonicalizeLabels} from '@shipfox/runner-labels';\nimport {\n  and,\n  arrayContains,\n  asc,\n  eq,\n  exists,\n  gt,\n  inArray,\n  isNotNull,\n  isNull,\n  lt,\n  sql,\n} from 'drizzle-orm';\nimport type {Tx} from './db.js';\nimport {db} from './db.js';\nimport {pendingJobExecutions} from './schema/pending-job-executions.js';\nimport {provisionerCapabilitySnapshots} from './schema/provisioner-capability-snapshots.js';\nimport {provisionerTokens} from './schema/provisioner-tokens.js';\nimport {reservations} from './schema/reservations.js';\nimport {runnerActivationTokens} from './schema/runner-activation-tokens.js';\nimport {runnerControlSessions} from './schema/runner-control-sessions.js';\nimport {providerRunners} from './schema/runner-instances.js';\n\nexport interface ReservationTemplate {\n  templateKey: string;\n  labels: string[];\n  availableSlots: number;\n  starting: number;\n  running: number;\n}\n\nexport interface DemandStat {\n  workspaceId?: string;\n  labels: string[];\n  queued: number;\n  reserved: number;\n  oldestQueuedAt: Date;\n}\n\nexport interface ReservationGrant {\n  reservationId: string;\n  workspaceId?: string;\n  labels: string[];\n  count: number;\n  expiresAt: Date;\n}\n\nexport interface PollDemandAndReserveParams {\n  workspaceId: string;\n  provisionerId: string;\n  maxReservations: number;\n  ttlSeconds: number;\n  templates: ReservationTemplate[];\n  capabilityWindowSeconds?: number;\n}\n\nexport interface InstallationPollDemandAndReserveParams {\n  provisionerId: string;\n  maxReservations: number;\n  ttlSeconds: number;\n  templates: ReservationTemplate[];\n  capabilityWindowSeconds: number;\n  eligibleWorkspaceIds: ReadonlySet<string>;\n  signal?: AbortSignal;\n  onReservations?: (reservations: ReservationGrant[]) => void;\n}\n\ninterface NormalizedTemplate {\n  templateKey: string;\n  labels: string[];\n  remainingSlots: number;\n}\n\ninterface DemandRow {\n  requiredLabels: string[];\n  queued: number;\n  oldestQueuedAt: Date;\n}\n\nexport async function pollDemandAndReserve(\n  params: PollDemandAndReserveParams,\n): Promise<{stats: DemandStat[]; reservations: ReservationGrant[]}> {\n  return await db().transaction(async (tx) => {\n    return await pollDemandAndReserveTx(tx, params);\n  });\n}\n\nexport async function pollDemandAndReserveTx(\n  tx: Tx,\n  params: PollDemandAndReserveParams,\n): Promise<{stats: DemandStat[]; reservations: ReservationGrant[]}> {\n  await tx.execute(sql`select pg_advisory_xact_lock(hashtext(${params.workspaceId}))`);\n  return await pollDemandAndReserveLockedTx(tx, params);\n}\n\nexport async function pollInstallationDemandAndReserve(\n  params: InstallationPollDemandAndReserveParams,\n): Promise<{stats: DemandStat[]; reservations: ReservationGrant[]}> {\n  const candidateWorkspaceIds = await listInstallationDemandWorkspaceIds(\n    params.eligibleWorkspaceIds,\n  );\n  const results: Array<{stats: DemandStat[]; reservations: ReservationGrant[]}> = [];\n  let remainingMaxReservations = params.maxReservations;\n  const remainingTemplates = params.templates.map((template) => ({\n    ...template,\n    labels: [...canonicalizeLabels(template.labels)],\n    remainingSlots: template.availableSlots,\n  }));\n  for (const workspaceId of candidateWorkspaceIds) {\n    if (params.signal?.aborted || remainingMaxReservations === 0) break;\n    const result = await db().transaction(async (tx) => {\n      const lockResult = await tx.execute<{locked: boolean}>(\n        sql`select pg_try_advisory_xact_lock(hashtext(${workspaceId})) as locked`,\n      );\n      const locked = lockResult.rows[0];\n      if (!locked?.locked) return {stats: [], reservations: []};\n      return await pollDemandAndReserveLockedTx(tx, {\n        workspaceId,\n        provisionerId: params.provisionerId,\n        maxReservations: remainingMaxReservations,\n        ttlSeconds: params.ttlSeconds,\n        templates: remainingTemplates.map((template) => ({\n          ...template,\n          availableSlots: template.remainingSlots,\n        })),\n        capabilityWindowSeconds: params.capabilityWindowSeconds,\n      });\n    });\n    results.push(result);\n    consumeInstallationTemplateSlots(remainingTemplates, result.reservations);\n    params.onReservations?.(result.reservations);\n    remainingMaxReservations -= result.reservations.reduce(\n      (total, reservation) => total + reservation.count,\n      0,\n    );\n  }\n  return {\n    stats: results.flatMap((result) => result.stats),\n    reservations: results.flatMap((result) => result.reservations),\n  };\n}\n\nfunction consumeInstallationTemplateSlots(\n  templates: NormalizedTemplate[],\n  reservations: ReservationGrant[],\n): void {\n  for (const reservation of reservations) {\n    const satisfyingTemplates = templates\n      .filter((template) => isSubset(reservation.labels, template.labels))\n      .sort(\n        (a, b) => a.labels.length - b.labels.length || a.templateKey.localeCompare(b.templateKey),\n      );\n    drawSlots(satisfyingTemplates, reservation.count);\n  }\n}\n\nasync function pollDemandAndReserveLockedTx(\n  tx: Tx,\n  params: PollDemandAndReserveParams,\n): Promise<{stats: DemandStat[]; reservations: ReservationGrant[]}> {\n  let demandRows = (\n    await tx\n      .select({\n        requiredLabels: pendingJobExecutions.requiredLabels,\n        queued: sql<number>`count(*)::int`,\n        oldestQueuedAt: sql<Date | string>`min(${pendingJobExecutions.createdAt})`,\n      })\n      .from(pendingJobExecutions)\n      .where(eq(pendingJobExecutions.workspaceId, params.workspaceId))\n      .groupBy(pendingJobExecutions.requiredLabels)\n  ).map((row) => ({\n    ...row,\n    oldestQueuedAt: new Date(row.oldestQueuedAt),\n  }));\n  if (params.capabilityWindowSeconds !== undefined) {\n    const capabilityLabels = await listActiveWorkspaceCapabilityLabelsTx(tx, {\n      workspaceId: params.workspaceId,\n      windowSeconds: params.capabilityWindowSeconds,\n    });\n    demandRows = demandRows.filter(\n      (demand) => !capabilityLabels.some((labels) => isSubset(demand.requiredLabels, labels)),\n    );\n  }\n\n  const activeReservationRows = await tx\n    .select({\n      requiredLabels: reservations.requiredLabels,\n      reserved: sql<number>`coalesce(sum(${reservations.count}), 0)::int`,\n    })\n    .from(reservations)\n    .where(\n      and(eq(reservations.workspaceId, params.workspaceId), gt(reservations.expiresAt, sql`now()`)),\n    )\n    .groupBy(reservations.requiredLabels);\n\n  const reservedByLabels = new Map(\n    activeReservationRows.map((row) => [labelKey(row.requiredLabels), row.reserved]),\n  );\n  const activeProvisionerReservationRows = await tx\n    .select({\n      requiredLabels: reservations.requiredLabels,\n      reserved: sql<number>`coalesce(sum(${reservations.count}), 0)::int`,\n    })\n    .from(reservations)\n    .where(\n      and(\n        eq(reservations.workspaceId, params.workspaceId),\n        eq(reservations.provisionerId, params.provisionerId),\n        gt(reservations.expiresAt, sql`now()`),\n      ),\n    )\n    .groupBy(reservations.requiredLabels);\n\n  const templates = params.templates.map((template) => ({\n    templateKey: template.templateKey,\n    labels: [...canonicalizeLabels(template.labels)],\n    remainingSlots: template.availableSlots,\n  }));\n  deductProvisionerReservations(templates, activeProvisionerReservationRows);\n  const stats: DemandStat[] = [];\n  const grants: ReservationGrant[] = [];\n  let remainingMaxReservations = params.maxReservations;\n  const workspaceIdField =\n    params.capabilityWindowSeconds === undefined ? {} : {workspaceId: params.workspaceId};\n\n  for (const demand of sortDemandRows(demandRows)) {\n    const satisfyingTemplates = templates\n      .filter((template) => isSubset(demand.requiredLabels, template.labels))\n      .sort(\n        (a, b) => a.labels.length - b.labels.length || a.templateKey.localeCompare(b.templateKey),\n      );\n\n    if (satisfyingTemplates.length === 0) continue;\n\n    const reserved = reservedByLabels.get(labelKey(demand.requiredLabels)) ?? 0;\n    const unreserved = Math.max(0, demand.queued - reserved);\n    const capacity = satisfyingTemplates.reduce(\n      (total, template) => total + template.remainingSlots,\n      0,\n    );\n    const grant = Math.min(unreserved, capacity, remainingMaxReservations);\n    let reservedAfterGrant = reserved;\n\n    if (grant > 0 && params.maxReservations > 0) {\n      const [inserted] = await tx\n        .insert(reservations)\n        .values({\n          workspaceId: params.workspaceId,\n          provisionerId: params.provisionerId,\n          requiredLabels: demand.requiredLabels,\n          count: grant,\n          expiresAt: sql`now() + (${params.ttlSeconds} || ' seconds')::interval`,\n        })\n        .returning({id: reservations.id, expiresAt: reservations.expiresAt});\n\n      if (!inserted) throw new Error('Insert returned no rows');\n\n      remainingMaxReservations -= grant;\n      drawSlots(satisfyingTemplates, grant);\n      await bindIdleRunnerInstancesTx(tx, {\n        provisionerId: params.provisionerId,\n        reservationId: inserted.id,\n        requiredLabels: demand.requiredLabels,\n        count: grant,\n        workspaceId: params.workspaceId,\n      });\n      reservedAfterGrant += grant;\n      grants.push({\n        reservationId: inserted.id,\n        ...workspaceIdField,\n        labels: demand.requiredLabels,\n        count: grant,\n        expiresAt: inserted.expiresAt,\n      });\n    }\n\n    stats.push({\n      ...workspaceIdField,\n      labels: demand.requiredLabels,\n      queued: demand.queued,\n      reserved: reservedAfterGrant,\n      oldestQueuedAt: demand.oldestQueuedAt,\n    });\n  }\n\n  return {stats, reservations: grants};\n}\n\nasync function bindIdleRunnerInstancesTx(\n  tx: Tx,\n  params: {\n    provisionerId: string;\n    reservationId: string;\n    workspaceId: string;\n    requiredLabels: string[];\n    count: number;\n  },\n): Promise<void> {\n  const idleRunners = await tx\n    .select({id: providerRunners.id})\n    .from(providerRunners)\n    .where(\n      and(\n        eq(providerRunners.provisionerId, params.provisionerId),\n        isNull(providerRunners.workspaceId),\n        isNull(providerRunners.reservationId),\n        isNull(providerRunners.intendedReservationId),\n        isNull(providerRunners.runnerSessionId),\n        isNotNull(providerRunners.providerRunnerId),\n        eq(providerRunners.state, 'running'),\n        arrayContains(providerRunners.labels, params.requiredLabels),\n        exists(\n          tx\n            .select({id: runnerControlSessions.id})\n            .from(runnerControlSessions)\n            .where(\n              and(\n                eq(runnerControlSessions.runnerInstanceId, providerRunners.id),\n                eq(runnerControlSessions.provisionerId, params.provisionerId),\n                isNull(runnerControlSessions.closedAt),\n                gt(runnerControlSessions.expiresAt, sql`now()`),\n              ),\n            ),\n        ),\n      ),\n    )\n    .orderBy(asc(providerRunners.createdAt), asc(providerRunners.id))\n    .limit(params.count)\n    .for('update');\n\n  if (idleRunners.length === 0) return;\n\n  await tx\n    .update(providerRunners)\n    .set({\n      workspaceId: params.workspaceId,\n      reservationId: params.reservationId,\n      assignedAt: sql`now()`,\n      updatedAt: sql`now()`,\n    })\n    .where(\n      and(\n        inArray(\n          providerRunners.id,\n          idleRunners.map((runner) => runner.id),\n        ),\n        isNull(providerRunners.workspaceId),\n        isNull(providerRunners.reservationId),\n        isNull(providerRunners.runnerSessionId),\n      ),\n    );\n}\n\nasync function listInstallationDemandWorkspaceIds(eligibleWorkspaceIds: ReadonlySet<string>) {\n  if (eligibleWorkspaceIds.size === 0) return [];\n  const rows = await db()\n    .select({\n      workspaceId: pendingJobExecutions.workspaceId,\n      oldestQueuedAt: sql<Date>`min(${pendingJobExecutions.createdAt})`,\n    })\n    .from(pendingJobExecutions)\n    .where(inArray(pendingJobExecutions.workspaceId, [...eligibleWorkspaceIds]))\n    .groupBy(pendingJobExecutions.workspaceId)\n    .orderBy(\n      asc(sql`min(${pendingJobExecutions.createdAt})`),\n      asc(pendingJobExecutions.workspaceId),\n    );\n  return rows.map((row) => row.workspaceId);\n}\n\nexport async function listQueuedDemandWorkspaceIds(): Promise<string[]> {\n  const rows = await db()\n    .select({workspaceId: pendingJobExecutions.workspaceId})\n    .from(pendingJobExecutions)\n    .groupBy(pendingJobExecutions.workspaceId);\n  return rows.map((row) => row.workspaceId);\n}\n\nasync function listActiveWorkspaceCapabilityLabelsTx(\n  tx: Tx,\n  params: {workspaceId: string; windowSeconds: number},\n): Promise<string[][]> {\n  const rows = await tx\n    .select({labels: provisionerCapabilitySnapshots.labels})\n    .from(provisionerCapabilitySnapshots)\n    .innerJoin(\n      provisionerTokens,\n      eq(provisionerTokens.id, provisionerCapabilitySnapshots.provisionerId),\n    )\n    .where(\n      and(\n        eq(provisionerCapabilitySnapshots.workspaceId, params.workspaceId),\n        eq(provisionerTokens.workspaceId, params.workspaceId),\n        eq(provisionerTokens.scope, 'workspace'),\n        sql`${provisionerCapabilitySnapshots.advertisedAt} > now() - (${params.windowSeconds} || ' seconds')::interval`,\n        sql`${provisionerTokens.revokedAt} is null`,\n        sql`(${provisionerTokens.expiresAt} is null or ${provisionerTokens.expiresAt} > now())`,\n      ),\n    );\n  return rows.map((row) => row.labels);\n}\n\nexport async function deleteExpiredReservations(params?: {limit?: number}): Promise<number> {\n  return await db().transaction(async (tx) => {\n    const expiredRows = await tx\n      .select({id: reservations.id})\n      .from(reservations)\n      .where(lt(reservations.expiresAt, sql`now()`))\n      .orderBy(asc(reservations.expiresAt))\n      .limit(params?.limit ?? 1000);\n    return await deleteReservationsWithCleanupTx(\n      tx,\n      expiredRows.map((reservation) => reservation.id),\n      {expiredOnly: true},\n    );\n  });\n}\n\nexport async function deleteReservationsByIds(ids: string[]): Promise<number> {\n  if (ids.length === 0) return 0;\n\n  return await db().transaction(async (tx) => {\n    return await deleteReservationsWithCleanupTx(tx, ids);\n  });\n}\n\nasync function deleteReservationsWithCleanupTx(\n  tx: Tx,\n  ids: string[],\n  params: {expiredOnly?: boolean} = {},\n): Promise<number> {\n  if (ids.length === 0) return 0;\n\n  const assignedRunners = await tx\n    .select({id: providerRunners.id, reservationId: providerRunners.reservationId})\n    .from(providerRunners)\n    .where(\n      and(\n        inArray(providerRunners.reservationId, ids),\n        isNull(providerRunners.runnerSessionId),\n        isNull(providerRunners.reservationReleasedAt),\n      ),\n    )\n    .for('update');\n  const reservationRows = await tx\n    .select({id: reservations.id})\n    .from(reservations)\n    .where(\n      and(\n        inArray(reservations.id, ids),\n        params.expiredOnly ? lt(reservations.expiresAt, sql`now()`) : undefined,\n      ),\n    )\n    .for('update');\n  const reservationIds = reservationRows.map((reservation) => reservation.id);\n\n  if (reservationIds.length === 0) return 0;\n\n  const assignedRunnerIds = assignedRunners\n    .filter((runner) => runner.reservationId && reservationIds.includes(runner.reservationId))\n    .map((runner) => runner.id);\n  if (assignedRunnerIds.length > 0) {\n    await tx\n      .update(runnerActivationTokens)\n      .set({revokedAt: sql`now()`})\n      .where(\n        and(\n          inArray(runnerActivationTokens.runnerInstanceId, assignedRunnerIds),\n          isNull(runnerActivationTokens.consumedAt),\n          isNull(runnerActivationTokens.revokedAt),\n        ),\n      );\n    await tx\n      .update(providerRunners)\n      .set({\n        workspaceId: null,\n        reservationId: null,\n        assignedAt: null,\n        updatedAt: sql`now()`,\n      })\n      .where(\n        and(\n          inArray(providerRunners.id, assignedRunnerIds),\n          isNull(providerRunners.runnerSessionId),\n          isNull(providerRunners.reservationReleasedAt),\n        ),\n      );\n  }\n\n  const deleted = await tx\n    .delete(reservations)\n    .where(inArray(reservations.id, reservationIds))\n    .returning({id: reservations.id});\n\n  return deleted.length;\n}\n\nexport async function releaseReservationUnits(\n  tx: Tx,\n  params: {\n    workspaceId: string;\n    provisionerId: string;\n    releases: Array<{reservationId: string; count: number}>;\n  },\n): Promise<number> {\n  let released = 0;\n\n  for (const release of params.releases) {\n    if (release.count <= 0) continue;\n\n    const decremented = await tx\n      .update(reservations)\n      .set({count: sql`${reservations.count} - ${release.count}`})\n      .where(\n        and(\n          eq(reservations.id, release.reservationId),\n          eq(reservations.workspaceId, params.workspaceId),\n          eq(reservations.provisionerId, params.provisionerId),\n          gt(reservations.count, release.count),\n          gt(reservations.expiresAt, sql`now()`),\n        ),\n      )\n      .returning({id: reservations.id});\n\n    if (decremented.length > 0) {\n      released += release.count;\n      continue;\n    }\n\n    const deleted = await tx\n      .delete(reservations)\n      .where(\n        and(\n          eq(reservations.id, release.reservationId),\n          eq(reservations.workspaceId, params.workspaceId),\n          eq(reservations.provisionerId, params.provisionerId),\n          sql`${reservations.count} <= ${release.count}`,\n          gt(reservations.expiresAt, sql`now()`),\n        ),\n      )\n      .returning({count: reservations.count});\n\n    released += deleted.reduce((total, row) => total + row.count, 0);\n  }\n\n  return released;\n}\n\nfunction deductProvisionerReservations(\n  templates: NormalizedTemplate[],\n  activeReservations: {requiredLabels: string[]; reserved: number}[],\n): void {\n  for (const reservation of sortReservationRows(activeReservations)) {\n    const satisfyingTemplates = templates\n      .filter((template) => isSubset(reservation.requiredLabels, template.labels))\n      .sort(\n        (a, b) => a.labels.length - b.labels.length || a.templateKey.localeCompare(b.templateKey),\n      );\n    drawSlots(satisfyingTemplates, reservation.reserved);\n  }\n}\n\nfunction sortDemandRows(rows: DemandRow[]): DemandRow[] {\n  return [...rows].sort((a, b) => {\n    const specificity = b.requiredLabels.length - a.requiredLabels.length;\n    if (specificity !== 0) return specificity;\n    return a.oldestQueuedAt.getTime() - b.oldestQueuedAt.getTime();\n  });\n}\n\nfunction sortReservationRows<T extends {requiredLabels: string[]}>(rows: T[]): T[] {\n  return [...rows].sort((a, b) => {\n    const specificity = b.requiredLabels.length - a.requiredLabels.length;\n    if (specificity !== 0) return specificity;\n    return labelKey(a.requiredLabels).localeCompare(labelKey(b.requiredLabels));\n  });\n}\n\nfunction isSubset(requiredLabels: string[], availableLabels: string[]): boolean {\n  return requiredLabels.every((label) => availableLabels.includes(label));\n}\n\nfunction drawSlots(templates: NormalizedTemplate[], count: number): void {\n  let remaining = count;\n  for (const template of templates) {\n    if (remaining === 0) return;\n    const used = Math.min(template.remainingSlots, remaining);\n    template.remainingSlots -= used;\n    remaining -= used;\n  }\n}\n\nfunction labelKey(labels: string[]): string {\n  return JSON.stringify(labels);\n}\n"],"names":["canonicalizeLabels","and","arrayContains","asc","eq","exists","gt","inArray","isNotNull","isNull","lt","sql","db","pendingJobExecutions","provisionerCapabilitySnapshots","provisionerTokens","reservations","runnerActivationTokens","runnerControlSessions","providerRunners","pollDemandAndReserve","params","transaction","tx","pollDemandAndReserveTx","execute","workspaceId","pollDemandAndReserveLockedTx","pollInstallationDemandAndReserve","candidateWorkspaceIds","listInstallationDemandWorkspaceIds","eligibleWorkspaceIds","results","remainingMaxReservations","maxReservations","remainingTemplates","templates","map","template","labels","remainingSlots","availableSlots","signal","aborted","result","lockResult","locked","rows","stats","provisionerId","ttlSeconds","capabilityWindowSeconds","push","consumeInstallationTemplateSlots","onReservations","reduce","total","reservation","count","flatMap","satisfyingTemplates","filter","isSubset","sort","a","b","length","templateKey","localeCompare","drawSlots","demandRows","select","requiredLabels","queued","oldestQueuedAt","createdAt","from","where","groupBy","row","Date","undefined","capabilityLabels","listActiveWorkspaceCapabilityLabelsTx","windowSeconds","demand","some","activeReservationRows","reserved","expiresAt","reservedByLabels","Map","labelKey","activeProvisionerReservationRows","deductProvisionerReservations","grants","workspaceIdField","sortDemandRows","get","unreserved","Math","max","capacity","grant","min","reservedAfterGrant","inserted","insert","values","returning","id","Error","bindIdleRunnerInstancesTx","reservationId","idleRunners","intendedReservationId","runnerSessionId","providerRunnerId","state","runnerInstanceId","closedAt","orderBy","limit","for","update","set","assignedAt","updatedAt","runner","size","listQueuedDemandWorkspaceIds","innerJoin","scope","advertisedAt","revokedAt","deleteExpiredReservations","expiredRows","deleteReservationsWithCleanupTx","expiredOnly","deleteReservationsByIds","ids","assignedRunners","reservationReleasedAt","reservationRows","reservationIds","assignedRunnerIds","includes","consumedAt","deleted","delete","releaseReservationUnits","released","release","releases","decremented","activeReservations","sortReservationRows","specificity","getTime","availableLabels","every","label","remaining","used","JSON","stringify"],"mappings":"AAAA,SAAQA,kBAAkB,QAAO,yBAAyB;AAC1D,SACEC,GAAG,EACHC,aAAa,EACbC,GAAG,EACHC,EAAE,EACFC,MAAM,EACNC,EAAE,EACFC,OAAO,EACPC,SAAS,EACTC,MAAM,EACNC,EAAE,EACFC,GAAG,QACE,cAAc;AAErB,SAAQC,EAAE,QAAO,UAAU;AAC3B,SAAQC,oBAAoB,QAAO,qCAAqC;AACxE,SAAQC,8BAA8B,QAAO,+CAA+C;AAC5F,SAAQC,iBAAiB,QAAO,iCAAiC;AACjE,SAAQC,YAAY,QAAO,2BAA2B;AACtD,SAAQC,sBAAsB,QAAO,uCAAuC;AAC5E,SAAQC,qBAAqB,QAAO,sCAAsC;AAC1E,SAAQC,eAAe,QAAO,+BAA+B;AA0D7D,OAAO,eAAeC,qBACpBC,MAAkC;IAElC,OAAO,MAAMT,KAAKU,WAAW,CAAC,OAAOC;QACnC,OAAO,MAAMC,uBAAuBD,IAAIF;IAC1C;AACF;AAEA,OAAO,eAAeG,uBACpBD,EAAM,EACNF,MAAkC;IAElC,MAAME,GAAGE,OAAO,CAACd,GAAG,CAAC,sCAAsC,EAAEU,OAAOK,WAAW,CAAC,EAAE,CAAC;IACnF,OAAO,MAAMC,6BAA6BJ,IAAIF;AAChD;AAEA,OAAO,eAAeO,iCACpBP,MAA8C;IAE9C,MAAMQ,wBAAwB,MAAMC,mCAClCT,OAAOU,oBAAoB;IAE7B,MAAMC,UAA0E,EAAE;IAClF,IAAIC,2BAA2BZ,OAAOa,eAAe;IACrD,MAAMC,qBAAqBd,OAAOe,SAAS,CAACC,GAAG,CAAC,CAACC,WAAc,CAAA;YAC7D,GAAGA,QAAQ;YACXC,QAAQ;mBAAIvC,mBAAmBsC,SAASC,MAAM;aAAE;YAChDC,gBAAgBF,SAASG,cAAc;QACzC,CAAA;IACA,KAAK,MAAMf,eAAeG,sBAAuB;QAC/C,IAAIR,OAAOqB,MAAM,EAAEC,WAAWV,6BAA6B,GAAG;QAC9D,MAAMW,SAAS,MAAMhC,KAAKU,WAAW,CAAC,OAAOC;YAC3C,MAAMsB,aAAa,MAAMtB,GAAGE,OAAO,CACjCd,GAAG,CAAC,0CAA0C,EAAEe,YAAY,YAAY,CAAC;YAE3E,MAAMoB,SAASD,WAAWE,IAAI,CAAC,EAAE;YACjC,IAAI,CAACD,QAAQA,QAAQ,OAAO;gBAACE,OAAO,EAAE;gBAAEhC,cAAc,EAAE;YAAA;YACxD,OAAO,MAAMW,6BAA6BJ,IAAI;gBAC5CG;gBACAuB,eAAe5B,OAAO4B,aAAa;gBACnCf,iBAAiBD;gBACjBiB,YAAY7B,OAAO6B,UAAU;gBAC7Bd,WAAWD,mBAAmBE,GAAG,CAAC,CAACC,WAAc,CAAA;wBAC/C,GAAGA,QAAQ;wBACXG,gBAAgBH,SAASE,cAAc;oBACzC,CAAA;gBACAW,yBAAyB9B,OAAO8B,uBAAuB;YACzD;QACF;QACAnB,QAAQoB,IAAI,CAACR;QACbS,iCAAiClB,oBAAoBS,OAAO5B,YAAY;QACxEK,OAAOiC,cAAc,GAAGV,OAAO5B,YAAY;QAC3CiB,4BAA4BW,OAAO5B,YAAY,CAACuC,MAAM,CACpD,CAACC,OAAOC,cAAgBD,QAAQC,YAAYC,KAAK,EACjD;IAEJ;IACA,OAAO;QACLV,OAAOhB,QAAQ2B,OAAO,CAAC,CAACf,SAAWA,OAAOI,KAAK;QAC/ChC,cAAcgB,QAAQ2B,OAAO,CAAC,CAACf,SAAWA,OAAO5B,YAAY;IAC/D;AACF;AAEA,SAASqC,iCACPjB,SAA+B,EAC/BpB,YAAgC;IAEhC,KAAK,MAAMyC,eAAezC,aAAc;QACtC,MAAM4C,sBAAsBxB,UACzByB,MAAM,CAAC,CAACvB,WAAawB,SAASL,YAAYlB,MAAM,EAAED,SAASC,MAAM,GACjEwB,IAAI,CACH,CAACC,GAAGC,IAAMD,EAAEzB,MAAM,CAAC2B,MAAM,GAAGD,EAAE1B,MAAM,CAAC2B,MAAM,IAAIF,EAAEG,WAAW,CAACC,aAAa,CAACH,EAAEE,WAAW;QAE5FE,UAAUT,qBAAqBH,YAAYC,KAAK;IAClD;AACF;AAEA,eAAe/B,6BACbJ,EAAM,EACNF,MAAkC;IAElC,IAAIiD,aAAa,AACf,CAAA,MAAM/C,GACHgD,MAAM,CAAC;QACNC,gBAAgB3D,qBAAqB2D,cAAc;QACnDC,QAAQ9D,GAAW,CAAC,aAAa,CAAC;QAClC+D,gBAAgB/D,GAAkB,CAAC,IAAI,EAAEE,qBAAqB8D,SAAS,CAAC,CAAC,CAAC;IAC5E,GACCC,IAAI,CAAC/D,sBACLgE,KAAK,CAACzE,GAAGS,qBAAqBa,WAAW,EAAEL,OAAOK,WAAW,GAC7DoD,OAAO,CAACjE,qBAAqB2D,cAAc,CAAA,EAC9CnC,GAAG,CAAC,CAAC0C,MAAS,CAAA;YACd,GAAGA,GAAG;YACNL,gBAAgB,IAAIM,KAAKD,IAAIL,cAAc;QAC7C,CAAA;IACA,IAAIrD,OAAO8B,uBAAuB,KAAK8B,WAAW;QAChD,MAAMC,mBAAmB,MAAMC,sCAAsC5D,IAAI;YACvEG,aAAaL,OAAOK,WAAW;YAC/B0D,eAAe/D,OAAO8B,uBAAuB;QAC/C;QACAmB,aAAaA,WAAWT,MAAM,CAC5B,CAACwB,SAAW,CAACH,iBAAiBI,IAAI,CAAC,CAAC/C,SAAWuB,SAASuB,OAAOb,cAAc,EAAEjC;IAEnF;IAEA,MAAMgD,wBAAwB,MAAMhE,GACjCgD,MAAM,CAAC;QACNC,gBAAgBxD,aAAawD,cAAc;QAC3CgB,UAAU7E,GAAW,CAAC,aAAa,EAAEK,aAAa0C,KAAK,CAAC,UAAU,CAAC;IACrE,GACCkB,IAAI,CAAC5D,cACL6D,KAAK,CACJ5E,IAAIG,GAAGY,aAAaU,WAAW,EAAEL,OAAOK,WAAW,GAAGpB,GAAGU,aAAayE,SAAS,EAAE9E,GAAG,CAAC,KAAK,CAAC,IAE5FmE,OAAO,CAAC9D,aAAawD,cAAc;IAEtC,MAAMkB,mBAAmB,IAAIC,IAC3BJ,sBAAsBlD,GAAG,CAAC,CAAC0C,MAAQ;YAACa,SAASb,IAAIP,cAAc;YAAGO,IAAIS,QAAQ;SAAC;IAEjF,MAAMK,mCAAmC,MAAMtE,GAC5CgD,MAAM,CAAC;QACNC,gBAAgBxD,aAAawD,cAAc;QAC3CgB,UAAU7E,GAAW,CAAC,aAAa,EAAEK,aAAa0C,KAAK,CAAC,UAAU,CAAC;IACrE,GACCkB,IAAI,CAAC5D,cACL6D,KAAK,CACJ5E,IACEG,GAAGY,aAAaU,WAAW,EAAEL,OAAOK,WAAW,GAC/CtB,GAAGY,aAAaiC,aAAa,EAAE5B,OAAO4B,aAAa,GACnD3C,GAAGU,aAAayE,SAAS,EAAE9E,GAAG,CAAC,KAAK,CAAC,IAGxCmE,OAAO,CAAC9D,aAAawD,cAAc;IAEtC,MAAMpC,YAAYf,OAAOe,SAAS,CAACC,GAAG,CAAC,CAACC,WAAc,CAAA;YACpD6B,aAAa7B,SAAS6B,WAAW;YACjC5B,QAAQ;mBAAIvC,mBAAmBsC,SAASC,MAAM;aAAE;YAChDC,gBAAgBF,SAASG,cAAc;QACzC,CAAA;IACAqD,8BAA8B1D,WAAWyD;IACzC,MAAM7C,QAAsB,EAAE;IAC9B,MAAM+C,SAA6B,EAAE;IACrC,IAAI9D,2BAA2BZ,OAAOa,eAAe;IACrD,MAAM8D,mBACJ3E,OAAO8B,uBAAuB,KAAK8B,YAAY,CAAC,IAAI;QAACvD,aAAaL,OAAOK,WAAW;IAAA;IAEtF,KAAK,MAAM2D,UAAUY,eAAe3B,YAAa;QAC/C,MAAMV,sBAAsBxB,UACzByB,MAAM,CAAC,CAACvB,WAAawB,SAASuB,OAAOb,cAAc,EAAElC,SAASC,MAAM,GACpEwB,IAAI,CACH,CAACC,GAAGC,IAAMD,EAAEzB,MAAM,CAAC2B,MAAM,GAAGD,EAAE1B,MAAM,CAAC2B,MAAM,IAAIF,EAAEG,WAAW,CAACC,aAAa,CAACH,EAAEE,WAAW;QAG5F,IAAIP,oBAAoBM,MAAM,KAAK,GAAG;QAEtC,MAAMsB,WAAWE,iBAAiBQ,GAAG,CAACN,SAASP,OAAOb,cAAc,MAAM;QAC1E,MAAM2B,aAAaC,KAAKC,GAAG,CAAC,GAAGhB,OAAOZ,MAAM,GAAGe;QAC/C,MAAMc,WAAW1C,oBAAoBL,MAAM,CACzC,CAACC,OAAOlB,WAAakB,QAAQlB,SAASE,cAAc,EACpD;QAEF,MAAM+D,QAAQH,KAAKI,GAAG,CAACL,YAAYG,UAAUrE;QAC7C,IAAIwE,qBAAqBjB;QAEzB,IAAIe,QAAQ,KAAKlF,OAAOa,eAAe,GAAG,GAAG;YAC3C,MAAM,CAACwE,SAAS,GAAG,MAAMnF,GACtBoF,MAAM,CAAC3F,cACP4F,MAAM,CAAC;gBACNlF,aAAaL,OAAOK,WAAW;gBAC/BuB,eAAe5B,OAAO4B,aAAa;gBACnCuB,gBAAgBa,OAAOb,cAAc;gBACrCd,OAAO6C;gBACPd,WAAW9E,GAAG,CAAC,SAAS,EAAEU,OAAO6B,UAAU,CAAC,yBAAyB,CAAC;YACxE,GACC2D,SAAS,CAAC;gBAACC,IAAI9F,aAAa8F,EAAE;gBAAErB,WAAWzE,aAAayE,SAAS;YAAA;YAEpE,IAAI,CAACiB,UAAU,MAAM,IAAIK,MAAM;YAE/B9E,4BAA4BsE;YAC5BlC,UAAUT,qBAAqB2C;YAC/B,MAAMS,0BAA0BzF,IAAI;gBAClC0B,eAAe5B,OAAO4B,aAAa;gBACnCgE,eAAeP,SAASI,EAAE;gBAC1BtC,gBAAgBa,OAAOb,cAAc;gBACrCd,OAAO6C;gBACP7E,aAAaL,OAAOK,WAAW;YACjC;YACA+E,sBAAsBF;YACtBR,OAAO3C,IAAI,CAAC;gBACV6D,eAAeP,SAASI,EAAE;gBAC1B,GAAGd,gBAAgB;gBACnBzD,QAAQ8C,OAAOb,cAAc;gBAC7Bd,OAAO6C;gBACPd,WAAWiB,SAASjB,SAAS;YAC/B;QACF;QAEAzC,MAAMI,IAAI,CAAC;YACT,GAAG4C,gBAAgB;YACnBzD,QAAQ8C,OAAOb,cAAc;YAC7BC,QAAQY,OAAOZ,MAAM;YACrBe,UAAUiB;YACV/B,gBAAgBW,OAAOX,cAAc;QACvC;IACF;IAEA,OAAO;QAAC1B;QAAOhC,cAAc+E;IAAM;AACrC;AAEA,eAAeiB,0BACbzF,EAAM,EACNF,MAMC;IAED,MAAM6F,cAAc,MAAM3F,GACvBgD,MAAM,CAAC;QAACuC,IAAI3F,gBAAgB2F,EAAE;IAAA,GAC9BlC,IAAI,CAACzD,iBACL0D,KAAK,CACJ5E,IACEG,GAAGe,gBAAgB8B,aAAa,EAAE5B,OAAO4B,aAAa,GACtDxC,OAAOU,gBAAgBO,WAAW,GAClCjB,OAAOU,gBAAgB8F,aAAa,GACpCxG,OAAOU,gBAAgBgG,qBAAqB,GAC5C1G,OAAOU,gBAAgBiG,eAAe,GACtC5G,UAAUW,gBAAgBkG,gBAAgB,GAC1CjH,GAAGe,gBAAgBmG,KAAK,EAAE,YAC1BpH,cAAciB,gBAAgBoB,MAAM,EAAElB,OAAOmD,cAAc,GAC3DnE,OACEkB,GACGgD,MAAM,CAAC;QAACuC,IAAI5F,sBAAsB4F,EAAE;IAAA,GACpClC,IAAI,CAAC1D,uBACL2D,KAAK,CACJ5E,IACEG,GAAGc,sBAAsBqG,gBAAgB,EAAEpG,gBAAgB2F,EAAE,GAC7D1G,GAAGc,sBAAsB+B,aAAa,EAAE5B,OAAO4B,aAAa,GAC5DxC,OAAOS,sBAAsBsG,QAAQ,GACrClH,GAAGY,sBAAsBuE,SAAS,EAAE9E,GAAG,CAAC,KAAK,CAAC,OAMzD8G,OAAO,CAACtH,IAAIgB,gBAAgBwD,SAAS,GAAGxE,IAAIgB,gBAAgB2F,EAAE,GAC9DY,KAAK,CAACrG,OAAOqC,KAAK,EAClBiE,GAAG,CAAC;IAEP,IAAIT,YAAYhD,MAAM,KAAK,GAAG;IAE9B,MAAM3C,GACHqG,MAAM,CAACzG,iBACP0G,GAAG,CAAC;QACHnG,aAAaL,OAAOK,WAAW;QAC/BuF,eAAe5F,OAAO4F,aAAa;QACnCa,YAAYnH,GAAG,CAAC,KAAK,CAAC;QACtBoH,WAAWpH,GAAG,CAAC,KAAK,CAAC;IACvB,GACCkE,KAAK,CACJ5E,IACEM,QACEY,gBAAgB2F,EAAE,EAClBI,YAAY7E,GAAG,CAAC,CAAC2F,SAAWA,OAAOlB,EAAE,IAEvCrG,OAAOU,gBAAgBO,WAAW,GAClCjB,OAAOU,gBAAgB8F,aAAa,GACpCxG,OAAOU,gBAAgBiG,eAAe;AAG9C;AAEA,eAAetF,mCAAmCC,oBAAyC;IACzF,IAAIA,qBAAqBkG,IAAI,KAAK,GAAG,OAAO,EAAE;IAC9C,MAAMlF,OAAO,MAAMnC,KAChB2D,MAAM,CAAC;QACN7C,aAAab,qBAAqBa,WAAW;QAC7CgD,gBAAgB/D,GAAS,CAAC,IAAI,EAAEE,qBAAqB8D,SAAS,CAAC,CAAC,CAAC;IACnE,GACCC,IAAI,CAAC/D,sBACLgE,KAAK,CAACtE,QAAQM,qBAAqBa,WAAW,EAAE;WAAIK;KAAqB,GACzE+C,OAAO,CAACjE,qBAAqBa,WAAW,EACxC+F,OAAO,CACNtH,IAAIQ,GAAG,CAAC,IAAI,EAAEE,qBAAqB8D,SAAS,CAAC,CAAC,CAAC,GAC/CxE,IAAIU,qBAAqBa,WAAW;IAExC,OAAOqB,KAAKV,GAAG,CAAC,CAAC0C,MAAQA,IAAIrD,WAAW;AAC1C;AAEA,OAAO,eAAewG;IACpB,MAAMnF,OAAO,MAAMnC,KAChB2D,MAAM,CAAC;QAAC7C,aAAab,qBAAqBa,WAAW;IAAA,GACrDkD,IAAI,CAAC/D,sBACLiE,OAAO,CAACjE,qBAAqBa,WAAW;IAC3C,OAAOqB,KAAKV,GAAG,CAAC,CAAC0C,MAAQA,IAAIrD,WAAW;AAC1C;AAEA,eAAeyD,sCACb5D,EAAM,EACNF,MAAoD;IAEpD,MAAM0B,OAAO,MAAMxB,GAChBgD,MAAM,CAAC;QAAChC,QAAQzB,+BAA+ByB,MAAM;IAAA,GACrDqC,IAAI,CAAC9D,gCACLqH,SAAS,CACRpH,mBACAX,GAAGW,kBAAkB+F,EAAE,EAAEhG,+BAA+BmC,aAAa,GAEtE4B,KAAK,CACJ5E,IACEG,GAAGU,+BAA+BY,WAAW,EAAEL,OAAOK,WAAW,GACjEtB,GAAGW,kBAAkBW,WAAW,EAAEL,OAAOK,WAAW,GACpDtB,GAAGW,kBAAkBqH,KAAK,EAAE,cAC5BzH,GAAG,CAAC,EAAEG,+BAA+BuH,YAAY,CAAC,YAAY,EAAEhH,OAAO+D,aAAa,CAAC,yBAAyB,CAAC,EAC/GzE,GAAG,CAAC,EAAEI,kBAAkBuH,SAAS,CAAC,QAAQ,CAAC,EAC3C3H,GAAG,CAAC,CAAC,EAAEI,kBAAkB0E,SAAS,CAAC,YAAY,EAAE1E,kBAAkB0E,SAAS,CAAC,SAAS,CAAC;IAG7F,OAAO1C,KAAKV,GAAG,CAAC,CAAC0C,MAAQA,IAAIxC,MAAM;AACrC;AAEA,OAAO,eAAegG,0BAA0BlH,MAAyB;IACvE,OAAO,MAAMT,KAAKU,WAAW,CAAC,OAAOC;QACnC,MAAMiH,cAAc,MAAMjH,GACvBgD,MAAM,CAAC;YAACuC,IAAI9F,aAAa8F,EAAE;QAAA,GAC3BlC,IAAI,CAAC5D,cACL6D,KAAK,CAACnE,GAAGM,aAAayE,SAAS,EAAE9E,GAAG,CAAC,KAAK,CAAC,GAC3C8G,OAAO,CAACtH,IAAIa,aAAayE,SAAS,GAClCiC,KAAK,CAACrG,QAAQqG,SAAS;QAC1B,OAAO,MAAMe,gCACXlH,IACAiH,YAAYnG,GAAG,CAAC,CAACoB,cAAgBA,YAAYqD,EAAE,GAC/C;YAAC4B,aAAa;QAAI;IAEtB;AACF;AAEA,OAAO,eAAeC,wBAAwBC,GAAa;IACzD,IAAIA,IAAI1E,MAAM,KAAK,GAAG,OAAO;IAE7B,OAAO,MAAMtD,KAAKU,WAAW,CAAC,OAAOC;QACnC,OAAO,MAAMkH,gCAAgClH,IAAIqH;IACnD;AACF;AAEA,eAAeH,gCACblH,EAAM,EACNqH,GAAa,EACbvH,SAAkC,CAAC,CAAC;IAEpC,IAAIuH,IAAI1E,MAAM,KAAK,GAAG,OAAO;IAE7B,MAAM2E,kBAAkB,MAAMtH,GAC3BgD,MAAM,CAAC;QAACuC,IAAI3F,gBAAgB2F,EAAE;QAAEG,eAAe9F,gBAAgB8F,aAAa;IAAA,GAC5ErC,IAAI,CAACzD,iBACL0D,KAAK,CACJ5E,IACEM,QAAQY,gBAAgB8F,aAAa,EAAE2B,MACvCnI,OAAOU,gBAAgBiG,eAAe,GACtC3G,OAAOU,gBAAgB2H,qBAAqB,IAG/CnB,GAAG,CAAC;IACP,MAAMoB,kBAAkB,MAAMxH,GAC3BgD,MAAM,CAAC;QAACuC,IAAI9F,aAAa8F,EAAE;IAAA,GAC3BlC,IAAI,CAAC5D,cACL6D,KAAK,CACJ5E,IACEM,QAAQS,aAAa8F,EAAE,EAAE8B,MACzBvH,OAAOqH,WAAW,GAAGhI,GAAGM,aAAayE,SAAS,EAAE9E,GAAG,CAAC,KAAK,CAAC,IAAIsE,YAGjE0C,GAAG,CAAC;IACP,MAAMqB,iBAAiBD,gBAAgB1G,GAAG,CAAC,CAACoB,cAAgBA,YAAYqD,EAAE;IAE1E,IAAIkC,eAAe9E,MAAM,KAAK,GAAG,OAAO;IAExC,MAAM+E,oBAAoBJ,gBACvBhF,MAAM,CAAC,CAACmE,SAAWA,OAAOf,aAAa,IAAI+B,eAAeE,QAAQ,CAAClB,OAAOf,aAAa,GACvF5E,GAAG,CAAC,CAAC2F,SAAWA,OAAOlB,EAAE;IAC5B,IAAImC,kBAAkB/E,MAAM,GAAG,GAAG;QAChC,MAAM3C,GACHqG,MAAM,CAAC3G,wBACP4G,GAAG,CAAC;YAACS,WAAW3H,GAAG,CAAC,KAAK,CAAC;QAAA,GAC1BkE,KAAK,CACJ5E,IACEM,QAAQU,uBAAuBsG,gBAAgB,EAAE0B,oBACjDxI,OAAOQ,uBAAuBkI,UAAU,GACxC1I,OAAOQ,uBAAuBqH,SAAS;QAG7C,MAAM/G,GACHqG,MAAM,CAACzG,iBACP0G,GAAG,CAAC;YACHnG,aAAa;YACbuF,eAAe;YACfa,YAAY;YACZC,WAAWpH,GAAG,CAAC,KAAK,CAAC;QACvB,GACCkE,KAAK,CACJ5E,IACEM,QAAQY,gBAAgB2F,EAAE,EAAEmC,oBAC5BxI,OAAOU,gBAAgBiG,eAAe,GACtC3G,OAAOU,gBAAgB2H,qBAAqB;IAGpD;IAEA,MAAMM,UAAU,MAAM7H,GACnB8H,MAAM,CAACrI,cACP6D,KAAK,CAACtE,QAAQS,aAAa8F,EAAE,EAAEkC,iBAC/BnC,SAAS,CAAC;QAACC,IAAI9F,aAAa8F,EAAE;IAAA;IAEjC,OAAOsC,QAAQlF,MAAM;AACvB;AAEA,OAAO,eAAeoF,wBACpB/H,EAAM,EACNF,MAIC;IAED,IAAIkI,WAAW;IAEf,KAAK,MAAMC,WAAWnI,OAAOoI,QAAQ,CAAE;QACrC,IAAID,QAAQ9F,KAAK,IAAI,GAAG;QAExB,MAAMgG,cAAc,MAAMnI,GACvBqG,MAAM,CAAC5G,cACP6G,GAAG,CAAC;YAACnE,OAAO/C,GAAG,CAAC,EAAEK,aAAa0C,KAAK,CAAC,GAAG,EAAE8F,QAAQ9F,KAAK,CAAC,CAAC;QAAA,GACzDmB,KAAK,CACJ5E,IACEG,GAAGY,aAAa8F,EAAE,EAAE0C,QAAQvC,aAAa,GACzC7G,GAAGY,aAAaU,WAAW,EAAEL,OAAOK,WAAW,GAC/CtB,GAAGY,aAAaiC,aAAa,EAAE5B,OAAO4B,aAAa,GACnD3C,GAAGU,aAAa0C,KAAK,EAAE8F,QAAQ9F,KAAK,GACpCpD,GAAGU,aAAayE,SAAS,EAAE9E,GAAG,CAAC,KAAK,CAAC,IAGxCkG,SAAS,CAAC;YAACC,IAAI9F,aAAa8F,EAAE;QAAA;QAEjC,IAAI4C,YAAYxF,MAAM,GAAG,GAAG;YAC1BqF,YAAYC,QAAQ9F,KAAK;YACzB;QACF;QAEA,MAAM0F,UAAU,MAAM7H,GACnB8H,MAAM,CAACrI,cACP6D,KAAK,CACJ5E,IACEG,GAAGY,aAAa8F,EAAE,EAAE0C,QAAQvC,aAAa,GACzC7G,GAAGY,aAAaU,WAAW,EAAEL,OAAOK,WAAW,GAC/CtB,GAAGY,aAAaiC,aAAa,EAAE5B,OAAO4B,aAAa,GACnDtC,GAAG,CAAC,EAAEK,aAAa0C,KAAK,CAAC,IAAI,EAAE8F,QAAQ9F,KAAK,CAAC,CAAC,EAC9CpD,GAAGU,aAAayE,SAAS,EAAE9E,GAAG,CAAC,KAAK,CAAC,IAGxCkG,SAAS,CAAC;YAACnD,OAAO1C,aAAa0C,KAAK;QAAA;QAEvC6F,YAAYH,QAAQ7F,MAAM,CAAC,CAACC,OAAOuB,MAAQvB,QAAQuB,IAAIrB,KAAK,EAAE;IAChE;IAEA,OAAO6F;AACT;AAEA,SAASzD,8BACP1D,SAA+B,EAC/BuH,kBAAkE;IAElE,KAAK,MAAMlG,eAAemG,oBAAoBD,oBAAqB;QACjE,MAAM/F,sBAAsBxB,UACzByB,MAAM,CAAC,CAACvB,WAAawB,SAASL,YAAYe,cAAc,EAAElC,SAASC,MAAM,GACzEwB,IAAI,CACH,CAACC,GAAGC,IAAMD,EAAEzB,MAAM,CAAC2B,MAAM,GAAGD,EAAE1B,MAAM,CAAC2B,MAAM,IAAIF,EAAEG,WAAW,CAACC,aAAa,CAACH,EAAEE,WAAW;QAE5FE,UAAUT,qBAAqBH,YAAY+B,QAAQ;IACrD;AACF;AAEA,SAASS,eAAelD,IAAiB;IACvC,OAAO;WAAIA;KAAK,CAACgB,IAAI,CAAC,CAACC,GAAGC;QACxB,MAAM4F,cAAc5F,EAAEO,cAAc,CAACN,MAAM,GAAGF,EAAEQ,cAAc,CAACN,MAAM;QACrE,IAAI2F,gBAAgB,GAAG,OAAOA;QAC9B,OAAO7F,EAAEU,cAAc,CAACoF,OAAO,KAAK7F,EAAES,cAAc,CAACoF,OAAO;IAC9D;AACF;AAEA,SAASF,oBAA0D7G,IAAS;IAC1E,OAAO;WAAIA;KAAK,CAACgB,IAAI,CAAC,CAACC,GAAGC;QACxB,MAAM4F,cAAc5F,EAAEO,cAAc,CAACN,MAAM,GAAGF,EAAEQ,cAAc,CAACN,MAAM;QACrE,IAAI2F,gBAAgB,GAAG,OAAOA;QAC9B,OAAOjE,SAAS5B,EAAEQ,cAAc,EAAEJ,aAAa,CAACwB,SAAS3B,EAAEO,cAAc;IAC3E;AACF;AAEA,SAASV,SAASU,cAAwB,EAAEuF,eAAyB;IACnE,OAAOvF,eAAewF,KAAK,CAAC,CAACC,QAAUF,gBAAgBb,QAAQ,CAACe;AAClE;AAEA,SAAS5F,UAAUjC,SAA+B,EAAEsB,KAAa;IAC/D,IAAIwG,YAAYxG;IAChB,KAAK,MAAMpB,YAAYF,UAAW;QAChC,IAAI8H,cAAc,GAAG;QACrB,MAAMC,OAAO/D,KAAKI,GAAG,CAAClE,SAASE,cAAc,EAAE0H;QAC/C5H,SAASE,cAAc,IAAI2H;QAC3BD,aAAaC;IACf;AACF;AAEA,SAASvE,SAASrD,MAAgB;IAChC,OAAO6H,KAAKC,SAAS,CAAC9H;AACxB"}
+{"version":3,"sources":["../../src/db/reservations.ts"],"sourcesContent":["import {canonicalizeLabels} from '@shipfox/runner-labels';\nimport {\n  and,\n  arrayContains,\n  asc,\n  eq,\n  exists,\n  gt,\n  inArray,\n  isNotNull,\n  isNull,\n  lt,\n  notExists,\n  or,\n  sql,\n} from 'drizzle-orm';\nimport type {Tx} from './db.js';\nimport {db} from './db.js';\nimport {pendingJobExecutions} from './schema/pending-job-executions.js';\nimport {provisionerCapabilitySnapshots} from './schema/provisioner-capability-snapshots.js';\nimport {provisionerTokens} from './schema/provisioner-tokens.js';\nimport {reservations} from './schema/reservations.js';\nimport {runnerActivationTokens} from './schema/runner-activation-tokens.js';\nimport {runnerControlSessions} from './schema/runner-control-sessions.js';\nimport {providerRunners} from './schema/runner-instances.js';\n\nexport interface ReservationTemplate {\n  templateKey: string;\n  labels: string[];\n  availableSlots: number;\n  starting: number;\n  running: number;\n}\n\nexport interface DemandStat {\n  workspaceId?: string;\n  labels: string[];\n  queued: number;\n  reserved: number;\n  oldestQueuedAt: Date;\n}\n\nexport interface ReservationGrant {\n  reservationId: string;\n  workspaceId?: string;\n  labels: string[];\n  count: number;\n  expiresAt: Date;\n}\n\nexport interface PollDemandAndReserveParams {\n  workspaceId: string;\n  provisionerId: string;\n  maxReservations: number;\n  ttlSeconds: number;\n  templates: ReservationTemplate[];\n  capabilityWindowSeconds?: number;\n}\n\nexport interface InstallationPollDemandAndReserveParams {\n  provisionerId: string;\n  maxReservations: number;\n  ttlSeconds: number;\n  templates: ReservationTemplate[];\n  capabilityWindowSeconds: number;\n  eligibleWorkspaceIds: ReadonlySet<string>;\n  signal?: AbortSignal;\n  onReservations?: (reservations: ReservationGrant[]) => void;\n}\n\ntype DemandScope = 'installation' | 'workspace';\n\ntype PollDemandAndReserveLockedParams = PollDemandAndReserveParams & {\n  scope: DemandScope;\n};\n\ninterface NormalizedTemplate {\n  templateKey: string;\n  labels: string[];\n  remainingSlots: number;\n}\n\ninterface DemandRow {\n  requiredLabels: string[];\n  queued: number;\n  oldestQueuedAt: Date;\n}\n\nexport async function pollDemandAndReserve(\n  params: PollDemandAndReserveParams,\n): Promise<{stats: DemandStat[]; reservations: ReservationGrant[]}> {\n  return await db().transaction(async (tx) => {\n    return await pollDemandAndReserveTx(tx, params);\n  });\n}\n\nexport async function pollDemandAndReserveTx(\n  tx: Tx,\n  params: PollDemandAndReserveParams,\n): Promise<{stats: DemandStat[]; reservations: ReservationGrant[]}> {\n  await tx.execute(sql`select pg_advisory_xact_lock(hashtext(${params.workspaceId}))`);\n  return await pollDemandAndReserveLockedTx(tx, {...params, scope: 'workspace'});\n}\n\nexport async function pollInstallationDemandAndReserve(\n  params: InstallationPollDemandAndReserveParams,\n): Promise<{stats: DemandStat[]; reservations: ReservationGrant[]}> {\n  const candidateWorkspaceIds = await listInstallationDemandWorkspaceIds(\n    params.eligibleWorkspaceIds,\n  );\n  const results: Array<{stats: DemandStat[]; reservations: ReservationGrant[]}> = [];\n  let remainingMaxReservations = params.maxReservations;\n  const remainingTemplates = params.templates.map((template) => ({\n    ...template,\n    labels: [...canonicalizeLabels(template.labels)],\n    remainingSlots: template.availableSlots,\n  }));\n  for (const workspaceId of candidateWorkspaceIds) {\n    if (params.signal?.aborted || remainingMaxReservations === 0) break;\n    const result = await db().transaction(async (tx) => {\n      const lockResult = await tx.execute<{locked: boolean}>(\n        sql`select pg_try_advisory_xact_lock(hashtext(${workspaceId})) as locked`,\n      );\n      const locked = lockResult.rows[0];\n      if (!locked?.locked) return {stats: [], reservations: []};\n      return await pollDemandAndReserveLockedTx(tx, {\n        workspaceId,\n        provisionerId: params.provisionerId,\n        maxReservations: remainingMaxReservations,\n        ttlSeconds: params.ttlSeconds,\n        templates: remainingTemplates.map((template) => ({\n          ...template,\n          availableSlots: template.remainingSlots,\n        })),\n        capabilityWindowSeconds: params.capabilityWindowSeconds,\n        scope: 'installation',\n      });\n    });\n    results.push(result);\n    consumeInstallationTemplateSlots(remainingTemplates, result.reservations);\n    params.onReservations?.(result.reservations);\n    remainingMaxReservations -= result.reservations.reduce(\n      (total, reservation) => total + reservation.count,\n      0,\n    );\n  }\n  return {\n    stats: results.flatMap((result) => result.stats),\n    reservations: results.flatMap((result) => result.reservations),\n  };\n}\n\nfunction consumeInstallationTemplateSlots(\n  templates: NormalizedTemplate[],\n  reservations: ReservationGrant[],\n): void {\n  for (const reservation of reservations) {\n    const satisfyingTemplates = templates\n      .filter((template) => isSubset(reservation.labels, template.labels))\n      .sort(\n        (a, b) => a.labels.length - b.labels.length || a.templateKey.localeCompare(b.templateKey),\n      );\n    drawSlots(satisfyingTemplates, reservation.count);\n  }\n}\n\nasync function pollDemandAndReserveLockedTx(\n  tx: Tx,\n  params: PollDemandAndReserveLockedParams,\n): Promise<{stats: DemandStat[]; reservations: ReservationGrant[]}> {\n  let demandRows = (\n    await tx\n      .select({\n        requiredLabels: pendingJobExecutions.requiredLabels,\n        queued: sql<number>`count(*)::int`,\n        oldestQueuedAt: sql<Date | string>`min(${pendingJobExecutions.createdAt})`,\n      })\n      .from(pendingJobExecutions)\n      .where(eq(pendingJobExecutions.workspaceId, params.workspaceId))\n      .groupBy(pendingJobExecutions.requiredLabels)\n  ).map((row) => ({\n    ...row,\n    oldestQueuedAt: new Date(row.oldestQueuedAt),\n  }));\n  if (params.capabilityWindowSeconds !== undefined) {\n    const capabilityLabels = await listActiveWorkspaceCapabilityLabelsTx(tx, {\n      workspaceId: params.workspaceId,\n      windowSeconds: params.capabilityWindowSeconds,\n    });\n    demandRows = demandRows.filter(\n      (demand) => !capabilityLabels.some((labels) => isSubset(demand.requiredLabels, labels)),\n    );\n  }\n\n  const activeReservationRows = await tx\n    .select({\n      requiredLabels: reservations.requiredLabels,\n      reserved: sql<number>`coalesce(sum(${reservations.count}), 0)::int`,\n    })\n    .from(reservations)\n    .where(\n      and(eq(reservations.workspaceId, params.workspaceId), gt(reservations.expiresAt, sql`now()`)),\n    )\n    .groupBy(reservations.requiredLabels);\n\n  const reservedByLabels = new Map(\n    activeReservationRows.map((row) => [labelKey(row.requiredLabels), row.reserved]),\n  );\n  const activeProvisionerReservationRows = await tx\n    .select({\n      requiredLabels: reservations.requiredLabels,\n      reserved: sql<number>`coalesce(sum(${reservations.count}), 0)::int`,\n    })\n    .from(reservations)\n    .where(\n      and(\n        eq(reservations.workspaceId, params.workspaceId),\n        eq(reservations.provisionerId, params.provisionerId),\n        gt(reservations.expiresAt, sql`now()`),\n      ),\n    )\n    .groupBy(reservations.requiredLabels);\n\n  const templates = params.templates.map((template) => ({\n    templateKey: template.templateKey,\n    labels: [...canonicalizeLabels(template.labels)],\n    remainingSlots: template.availableSlots,\n  }));\n  deductProvisionerReservations(templates, activeProvisionerReservationRows);\n  const stats: DemandStat[] = [];\n  const grants: ReservationGrant[] = [];\n  let remainingMaxReservations = params.maxReservations;\n  const workspaceIdField =\n    params.capabilityWindowSeconds === undefined ? {} : {workspaceId: params.workspaceId};\n\n  for (const demand of sortDemandRows(demandRows)) {\n    const satisfyingTemplates = templates\n      .filter((template) => isSubset(demand.requiredLabels, template.labels))\n      .sort(\n        (a, b) => a.labels.length - b.labels.length || a.templateKey.localeCompare(b.templateKey),\n      );\n\n    if (satisfyingTemplates.length === 0) continue;\n\n    const reserved = reservedByLabels.get(labelKey(demand.requiredLabels)) ?? 0;\n    const unreserved = Math.max(0, demand.queued - reserved);\n    const capacity = satisfyingTemplates.reduce(\n      (total, template) => total + template.remainingSlots,\n      0,\n    );\n    const grant = Math.min(unreserved, capacity, remainingMaxReservations);\n    let reservedAfterGrant = reserved;\n\n    if (grant > 0 && params.maxReservations > 0) {\n      const [inserted] = await tx\n        .insert(reservations)\n        .values({\n          workspaceId: params.workspaceId,\n          provisionerId: params.provisionerId,\n          requiredLabels: demand.requiredLabels,\n          count: grant,\n          expiresAt: sql`now() + (${params.ttlSeconds} || ' seconds')::interval`,\n        })\n        .returning({id: reservations.id, expiresAt: reservations.expiresAt});\n\n      if (!inserted) throw new Error('Insert returned no rows');\n\n      remainingMaxReservations -= grant;\n      drawSlots(satisfyingTemplates, grant);\n      await bindIdleRunnerInstancesTx(tx, {\n        provisionerId: params.provisionerId,\n        reservationId: inserted.id,\n        requiredLabels: demand.requiredLabels,\n        count: grant,\n        workspaceId: params.workspaceId,\n        scope: params.scope,\n      });\n      reservedAfterGrant += grant;\n      grants.push({\n        reservationId: inserted.id,\n        ...workspaceIdField,\n        labels: demand.requiredLabels,\n        count: grant,\n        expiresAt: inserted.expiresAt,\n      });\n    }\n\n    stats.push({\n      ...workspaceIdField,\n      labels: demand.requiredLabels,\n      queued: demand.queued,\n      reserved: reservedAfterGrant,\n      oldestQueuedAt: demand.oldestQueuedAt,\n    });\n  }\n\n  return {stats, reservations: grants};\n}\n\nasync function bindIdleRunnerInstancesTx(\n  tx: Tx,\n  params: {\n    provisionerId: string;\n    reservationId: string;\n    workspaceId: string;\n    requiredLabels: string[];\n    count: number;\n    scope: DemandScope;\n  },\n): Promise<void> {\n  // Reservation expiry is the activation grace deadline. A live reservation\n  // protects a runner that is still booting; an expired or missing row is stale.\n  const canBindRunner = () =>\n    and(\n      or(\n        and(isNull(providerRunners.workspaceId), isNull(providerRunners.reservationId)),\n        and(\n          isNotNull(providerRunners.reservationId),\n          params.scope === 'installation'\n            ? undefined\n            : or(\n                isNull(providerRunners.workspaceId),\n                eq(providerRunners.workspaceId, params.workspaceId),\n              ),\n          notExists(\n            tx\n              .select({id: reservations.id})\n              .from(reservations)\n              .where(\n                and(\n                  eq(reservations.id, providerRunners.reservationId),\n                  gt(reservations.expiresAt, sql`now()`),\n                ),\n              ),\n          ),\n        ),\n      ),\n      or(\n        isNull(providerRunners.intendedReservationId),\n        notExists(\n          tx\n            .select({id: reservations.id})\n            .from(reservations)\n            .where(\n              and(\n                eq(reservations.id, providerRunners.intendedReservationId),\n                gt(reservations.expiresAt, sql`now()`),\n              ),\n            ),\n        ),\n      ),\n      isNull(providerRunners.reservationReleasedAt),\n      isNull(providerRunners.runnerSessionId),\n    );\n\n  const idleRunners = await tx\n    .select({id: providerRunners.id})\n    .from(providerRunners)\n    .where(\n      and(\n        eq(providerRunners.provisionerId, params.provisionerId),\n        canBindRunner(),\n        isNotNull(providerRunners.providerRunnerId),\n        eq(providerRunners.state, 'running'),\n        arrayContains(providerRunners.labels, params.requiredLabels),\n        exists(\n          tx\n            .select({id: runnerControlSessions.id})\n            .from(runnerControlSessions)\n            .where(\n              and(\n                eq(runnerControlSessions.runnerInstanceId, providerRunners.id),\n                eq(runnerControlSessions.provisionerId, params.provisionerId),\n                isNull(runnerControlSessions.closedAt),\n                gt(runnerControlSessions.expiresAt, sql`now()`),\n              ),\n            ),\n        ),\n      ),\n    )\n    .orderBy(asc(providerRunners.createdAt), asc(providerRunners.id))\n    .limit(params.count)\n    .for('update');\n\n  if (idleRunners.length === 0) return;\n\n  await tx\n    .update(runnerActivationTokens)\n    .set({revokedAt: sql`now()`})\n    .where(\n      and(\n        inArray(\n          runnerActivationTokens.runnerInstanceId,\n          idleRunners.map((runner) => runner.id),\n        ),\n        isNull(runnerActivationTokens.consumedAt),\n        isNull(runnerActivationTokens.revokedAt),\n      ),\n    );\n\n  await tx\n    .update(providerRunners)\n    .set({\n      workspaceId: params.workspaceId,\n      reservationId: params.reservationId,\n      intendedReservationId: null,\n      assignedAt: sql`now()`,\n      updatedAt: sql`now()`,\n    })\n    .where(\n      and(\n        inArray(\n          providerRunners.id,\n          idleRunners.map((runner) => runner.id),\n        ),\n        canBindRunner(),\n      ),\n    );\n}\n\nasync function listInstallationDemandWorkspaceIds(eligibleWorkspaceIds: ReadonlySet<string>) {\n  if (eligibleWorkspaceIds.size === 0) return [];\n  const rows = await db()\n    .select({\n      workspaceId: pendingJobExecutions.workspaceId,\n      oldestQueuedAt: sql<Date>`min(${pendingJobExecutions.createdAt})`,\n    })\n    .from(pendingJobExecutions)\n    .where(inArray(pendingJobExecutions.workspaceId, [...eligibleWorkspaceIds]))\n    .groupBy(pendingJobExecutions.workspaceId)\n    .orderBy(\n      asc(sql`min(${pendingJobExecutions.createdAt})`),\n      asc(pendingJobExecutions.workspaceId),\n    );\n  return rows.map((row) => row.workspaceId);\n}\n\nexport async function listQueuedDemandWorkspaceIds(): Promise<string[]> {\n  const rows = await db()\n    .select({workspaceId: pendingJobExecutions.workspaceId})\n    .from(pendingJobExecutions)\n    .groupBy(pendingJobExecutions.workspaceId);\n  return rows.map((row) => row.workspaceId);\n}\n\nasync function listActiveWorkspaceCapabilityLabelsTx(\n  tx: Tx,\n  params: {workspaceId: string; windowSeconds: number},\n): Promise<string[][]> {\n  const rows = await tx\n    .select({labels: provisionerCapabilitySnapshots.labels})\n    .from(provisionerCapabilitySnapshots)\n    .innerJoin(\n      provisionerTokens,\n      eq(provisionerTokens.id, provisionerCapabilitySnapshots.provisionerId),\n    )\n    .where(\n      and(\n        eq(provisionerCapabilitySnapshots.workspaceId, params.workspaceId),\n        eq(provisionerTokens.workspaceId, params.workspaceId),\n        eq(provisionerTokens.scope, 'workspace'),\n        sql`${provisionerCapabilitySnapshots.advertisedAt} > now() - (${params.windowSeconds} || ' seconds')::interval`,\n        sql`${provisionerTokens.revokedAt} is null`,\n        sql`(${provisionerTokens.expiresAt} is null or ${provisionerTokens.expiresAt} > now())`,\n      ),\n    );\n  return rows.map((row) => row.labels);\n}\n\nexport async function deleteExpiredReservations(params?: {limit?: number}): Promise<number> {\n  return await db().transaction(async (tx) => {\n    const expiredRows = await tx\n      .select({id: reservations.id})\n      .from(reservations)\n      .where(lt(reservations.expiresAt, sql`now()`))\n      .orderBy(asc(reservations.expiresAt))\n      .limit(params?.limit ?? 1000);\n    return await deleteReservationsWithCleanupTx(\n      tx,\n      expiredRows.map((reservation) => reservation.id),\n      {expiredOnly: true},\n    );\n  });\n}\n\nexport async function deleteReservationsByIds(ids: string[]): Promise<number> {\n  if (ids.length === 0) return 0;\n\n  return await db().transaction(async (tx) => {\n    return await deleteReservationsWithCleanupTx(tx, ids);\n  });\n}\n\nasync function deleteReservationsWithCleanupTx(\n  tx: Tx,\n  ids: string[],\n  params: {expiredOnly?: boolean} = {},\n): Promise<number> {\n  if (ids.length === 0) return 0;\n\n  const affectedRunners = await tx\n    .select({\n      id: providerRunners.id,\n      reservationId: providerRunners.reservationId,\n      intendedReservationId: providerRunners.intendedReservationId,\n      runnerSessionId: providerRunners.runnerSessionId,\n      reservationReleasedAt: providerRunners.reservationReleasedAt,\n    })\n    .from(providerRunners)\n    .where(\n      or(\n        and(\n          inArray(providerRunners.reservationId, ids),\n          isNull(providerRunners.runnerSessionId),\n          isNull(providerRunners.reservationReleasedAt),\n        ),\n        inArray(providerRunners.intendedReservationId, ids),\n      ),\n    )\n    .orderBy(asc(providerRunners.id))\n    .for('update');\n  const reservationRows = await tx\n    .select({id: reservations.id})\n    .from(reservations)\n    .where(\n      and(\n        inArray(reservations.id, ids),\n        params.expiredOnly ? lt(reservations.expiresAt, sql`now()`) : undefined,\n      ),\n    )\n    .for('update');\n  const reservationIds = reservationRows.map((reservation) => reservation.id);\n\n  if (reservationIds.length === 0) return 0;\n\n  const assignedRunnerIds = affectedRunners\n    .filter(\n      (runner) =>\n        runner.reservationId &&\n        reservationIds.includes(runner.reservationId) &&\n        !runner.runnerSessionId &&\n        !runner.reservationReleasedAt,\n    )\n    .map((runner) => runner.id);\n  const intendedRunnerIds = affectedRunners\n    .filter(\n      (runner) =>\n        runner.intendedReservationId && reservationIds.includes(runner.intendedReservationId),\n    )\n    .map((runner) => runner.id);\n  if (assignedRunnerIds.length > 0) {\n    await tx\n      .update(runnerActivationTokens)\n      .set({revokedAt: sql`now()`})\n      .where(\n        and(\n          inArray(runnerActivationTokens.runnerInstanceId, assignedRunnerIds),\n          isNull(runnerActivationTokens.consumedAt),\n          isNull(runnerActivationTokens.revokedAt),\n        ),\n      );\n    await tx\n      .update(providerRunners)\n      .set({\n        workspaceId: null,\n        reservationId: null,\n        assignedAt: null,\n        updatedAt: sql`now()`,\n      })\n      .where(\n        and(\n          inArray(providerRunners.id, assignedRunnerIds),\n          isNull(providerRunners.runnerSessionId),\n          isNull(providerRunners.reservationReleasedAt),\n        ),\n      );\n  }\n  if (intendedRunnerIds.length > 0) {\n    await tx\n      .update(providerRunners)\n      .set({intendedReservationId: null, updatedAt: sql`now()`})\n      .where(\n        and(\n          inArray(providerRunners.id, intendedRunnerIds),\n          inArray(providerRunners.intendedReservationId, reservationIds),\n        ),\n      );\n  }\n\n  const deleted = await tx\n    .delete(reservations)\n    .where(inArray(reservations.id, reservationIds))\n    .returning({id: reservations.id});\n\n  return deleted.length;\n}\n\nexport async function releaseReservationUnits(\n  tx: Tx,\n  params: {\n    workspaceId: string;\n    provisionerId: string;\n    releases: Array<{reservationId: string; count: number}>;\n  },\n): Promise<number> {\n  let released = 0;\n\n  for (const release of params.releases) {\n    if (release.count <= 0) continue;\n\n    const decremented = await tx\n      .update(reservations)\n      .set({count: sql`${reservations.count} - ${release.count}`})\n      .where(\n        and(\n          eq(reservations.id, release.reservationId),\n          eq(reservations.workspaceId, params.workspaceId),\n          eq(reservations.provisionerId, params.provisionerId),\n          gt(reservations.count, release.count),\n          gt(reservations.expiresAt, sql`now()`),\n        ),\n      )\n      .returning({id: reservations.id});\n\n    if (decremented.length > 0) {\n      released += release.count;\n      continue;\n    }\n\n    const deleted = await tx\n      .delete(reservations)\n      .where(\n        and(\n          eq(reservations.id, release.reservationId),\n          eq(reservations.workspaceId, params.workspaceId),\n          eq(reservations.provisionerId, params.provisionerId),\n          sql`${reservations.count} <= ${release.count}`,\n          gt(reservations.expiresAt, sql`now()`),\n        ),\n      )\n      .returning({count: reservations.count});\n\n    released += deleted.reduce((total, row) => total + row.count, 0);\n  }\n\n  return released;\n}\n\nfunction deductProvisionerReservations(\n  templates: NormalizedTemplate[],\n  activeReservations: {requiredLabels: string[]; reserved: number}[],\n): void {\n  for (const reservation of sortReservationRows(activeReservations)) {\n    const satisfyingTemplates = templates\n      .filter((template) => isSubset(reservation.requiredLabels, template.labels))\n      .sort(\n        (a, b) => a.labels.length - b.labels.length || a.templateKey.localeCompare(b.templateKey),\n      );\n    drawSlots(satisfyingTemplates, reservation.reserved);\n  }\n}\n\nfunction sortDemandRows(rows: DemandRow[]): DemandRow[] {\n  return [...rows].sort((a, b) => {\n    const specificity = b.requiredLabels.length - a.requiredLabels.length;\n    if (specificity !== 0) return specificity;\n    return a.oldestQueuedAt.getTime() - b.oldestQueuedAt.getTime();\n  });\n}\n\nfunction sortReservationRows<T extends {requiredLabels: string[]}>(rows: T[]): T[] {\n  return [...rows].sort((a, b) => {\n    const specificity = b.requiredLabels.length - a.requiredLabels.length;\n    if (specificity !== 0) return specificity;\n    return labelKey(a.requiredLabels).localeCompare(labelKey(b.requiredLabels));\n  });\n}\n\nfunction isSubset(requiredLabels: string[], availableLabels: string[]): boolean {\n  return requiredLabels.every((label) => availableLabels.includes(label));\n}\n\nfunction drawSlots(templates: NormalizedTemplate[], count: number): void {\n  let remaining = count;\n  for (const template of templates) {\n    if (remaining === 0) return;\n    const used = Math.min(template.remainingSlots, remaining);\n    template.remainingSlots -= used;\n    remaining -= used;\n  }\n}\n\nfunction labelKey(labels: string[]): string {\n  return JSON.stringify(labels);\n}\n"],"names":["canonicalizeLabels","and","arrayContains","asc","eq","exists","gt","inArray","isNotNull","isNull","lt","notExists","or","sql","db","pendingJobExecutions","provisionerCapabilitySnapshots","provisionerTokens","reservations","runnerActivationTokens","runnerControlSessions","providerRunners","pollDemandAndReserve","params","transaction","tx","pollDemandAndReserveTx","execute","workspaceId","pollDemandAndReserveLockedTx","scope","pollInstallationDemandAndReserve","candidateWorkspaceIds","listInstallationDemandWorkspaceIds","eligibleWorkspaceIds","results","remainingMaxReservations","maxReservations","remainingTemplates","templates","map","template","labels","remainingSlots","availableSlots","signal","aborted","result","lockResult","locked","rows","stats","provisionerId","ttlSeconds","capabilityWindowSeconds","push","consumeInstallationTemplateSlots","onReservations","reduce","total","reservation","count","flatMap","satisfyingTemplates","filter","isSubset","sort","a","b","length","templateKey","localeCompare","drawSlots","demandRows","select","requiredLabels","queued","oldestQueuedAt","createdAt","from","where","groupBy","row","Date","undefined","capabilityLabels","listActiveWorkspaceCapabilityLabelsTx","windowSeconds","demand","some","activeReservationRows","reserved","expiresAt","reservedByLabels","Map","labelKey","activeProvisionerReservationRows","deductProvisionerReservations","grants","workspaceIdField","sortDemandRows","get","unreserved","Math","max","capacity","grant","min","reservedAfterGrant","inserted","insert","values","returning","id","Error","bindIdleRunnerInstancesTx","reservationId","canBindRunner","intendedReservationId","reservationReleasedAt","runnerSessionId","idleRunners","providerRunnerId","state","runnerInstanceId","closedAt","orderBy","limit","for","update","set","revokedAt","runner","consumedAt","assignedAt","updatedAt","size","listQueuedDemandWorkspaceIds","innerJoin","advertisedAt","deleteExpiredReservations","expiredRows","deleteReservationsWithCleanupTx","expiredOnly","deleteReservationsByIds","ids","affectedRunners","reservationRows","reservationIds","assignedRunnerIds","includes","intendedRunnerIds","deleted","delete","releaseReservationUnits","released","release","releases","decremented","activeReservations","sortReservationRows","specificity","getTime","availableLabels","every","label","remaining","used","JSON","stringify"],"mappings":"AAAA,SAAQA,kBAAkB,QAAO,yBAAyB;AAC1D,SACEC,GAAG,EACHC,aAAa,EACbC,GAAG,EACHC,EAAE,EACFC,MAAM,EACNC,EAAE,EACFC,OAAO,EACPC,SAAS,EACTC,MAAM,EACNC,EAAE,EACFC,SAAS,EACTC,EAAE,EACFC,GAAG,QACE,cAAc;AAErB,SAAQC,EAAE,QAAO,UAAU;AAC3B,SAAQC,oBAAoB,QAAO,qCAAqC;AACxE,SAAQC,8BAA8B,QAAO,+CAA+C;AAC5F,SAAQC,iBAAiB,QAAO,iCAAiC;AACjE,SAAQC,YAAY,QAAO,2BAA2B;AACtD,SAAQC,sBAAsB,QAAO,uCAAuC;AAC5E,SAAQC,qBAAqB,QAAO,sCAAsC;AAC1E,SAAQC,eAAe,QAAO,+BAA+B;AAgE7D,OAAO,eAAeC,qBACpBC,MAAkC;IAElC,OAAO,MAAMT,KAAKU,WAAW,CAAC,OAAOC;QACnC,OAAO,MAAMC,uBAAuBD,IAAIF;IAC1C;AACF;AAEA,OAAO,eAAeG,uBACpBD,EAAM,EACNF,MAAkC;IAElC,MAAME,GAAGE,OAAO,CAACd,GAAG,CAAC,sCAAsC,EAAEU,OAAOK,WAAW,CAAC,EAAE,CAAC;IACnF,OAAO,MAAMC,6BAA6BJ,IAAI;QAAC,GAAGF,MAAM;QAAEO,OAAO;IAAW;AAC9E;AAEA,OAAO,eAAeC,iCACpBR,MAA8C;IAE9C,MAAMS,wBAAwB,MAAMC,mCAClCV,OAAOW,oBAAoB;IAE7B,MAAMC,UAA0E,EAAE;IAClF,IAAIC,2BAA2Bb,OAAOc,eAAe;IACrD,MAAMC,qBAAqBf,OAAOgB,SAAS,CAACC,GAAG,CAAC,CAACC,WAAc,CAAA;YAC7D,GAAGA,QAAQ;YACXC,QAAQ;mBAAI1C,mBAAmByC,SAASC,MAAM;aAAE;YAChDC,gBAAgBF,SAASG,cAAc;QACzC,CAAA;IACA,KAAK,MAAMhB,eAAeI,sBAAuB;QAC/C,IAAIT,OAAOsB,MAAM,EAAEC,WAAWV,6BAA6B,GAAG;QAC9D,MAAMW,SAAS,MAAMjC,KAAKU,WAAW,CAAC,OAAOC;YAC3C,MAAMuB,aAAa,MAAMvB,GAAGE,OAAO,CACjCd,GAAG,CAAC,0CAA0C,EAAEe,YAAY,YAAY,CAAC;YAE3E,MAAMqB,SAASD,WAAWE,IAAI,CAAC,EAAE;YACjC,IAAI,CAACD,QAAQA,QAAQ,OAAO;gBAACE,OAAO,EAAE;gBAAEjC,cAAc,EAAE;YAAA;YACxD,OAAO,MAAMW,6BAA6BJ,IAAI;gBAC5CG;gBACAwB,eAAe7B,OAAO6B,aAAa;gBACnCf,iBAAiBD;gBACjBiB,YAAY9B,OAAO8B,UAAU;gBAC7Bd,WAAWD,mBAAmBE,GAAG,CAAC,CAACC,WAAc,CAAA;wBAC/C,GAAGA,QAAQ;wBACXG,gBAAgBH,SAASE,cAAc;oBACzC,CAAA;gBACAW,yBAAyB/B,OAAO+B,uBAAuB;gBACvDxB,OAAO;YACT;QACF;QACAK,QAAQoB,IAAI,CAACR;QACbS,iCAAiClB,oBAAoBS,OAAO7B,YAAY;QACxEK,OAAOkC,cAAc,GAAGV,OAAO7B,YAAY;QAC3CkB,4BAA4BW,OAAO7B,YAAY,CAACwC,MAAM,CACpD,CAACC,OAAOC,cAAgBD,QAAQC,YAAYC,KAAK,EACjD;IAEJ;IACA,OAAO;QACLV,OAAOhB,QAAQ2B,OAAO,CAAC,CAACf,SAAWA,OAAOI,KAAK;QAC/CjC,cAAciB,QAAQ2B,OAAO,CAAC,CAACf,SAAWA,OAAO7B,YAAY;IAC/D;AACF;AAEA,SAASsC,iCACPjB,SAA+B,EAC/BrB,YAAgC;IAEhC,KAAK,MAAM0C,eAAe1C,aAAc;QACtC,MAAM6C,sBAAsBxB,UACzByB,MAAM,CAAC,CAACvB,WAAawB,SAASL,YAAYlB,MAAM,EAAED,SAASC,MAAM,GACjEwB,IAAI,CACH,CAACC,GAAGC,IAAMD,EAAEzB,MAAM,CAAC2B,MAAM,GAAGD,EAAE1B,MAAM,CAAC2B,MAAM,IAAIF,EAAEG,WAAW,CAACC,aAAa,CAACH,EAAEE,WAAW;QAE5FE,UAAUT,qBAAqBH,YAAYC,KAAK;IAClD;AACF;AAEA,eAAehC,6BACbJ,EAAM,EACNF,MAAwC;IAExC,IAAIkD,aAAa,AACf,CAAA,MAAMhD,GACHiD,MAAM,CAAC;QACNC,gBAAgB5D,qBAAqB4D,cAAc;QACnDC,QAAQ/D,GAAW,CAAC,aAAa,CAAC;QAClCgE,gBAAgBhE,GAAkB,CAAC,IAAI,EAAEE,qBAAqB+D,SAAS,CAAC,CAAC,CAAC;IAC5E,GACCC,IAAI,CAAChE,sBACLiE,KAAK,CAAC5E,GAAGW,qBAAqBa,WAAW,EAAEL,OAAOK,WAAW,GAC7DqD,OAAO,CAAClE,qBAAqB4D,cAAc,CAAA,EAC9CnC,GAAG,CAAC,CAAC0C,MAAS,CAAA;YACd,GAAGA,GAAG;YACNL,gBAAgB,IAAIM,KAAKD,IAAIL,cAAc;QAC7C,CAAA;IACA,IAAItD,OAAO+B,uBAAuB,KAAK8B,WAAW;QAChD,MAAMC,mBAAmB,MAAMC,sCAAsC7D,IAAI;YACvEG,aAAaL,OAAOK,WAAW;YAC/B2D,eAAehE,OAAO+B,uBAAuB;QAC/C;QACAmB,aAAaA,WAAWT,MAAM,CAC5B,CAACwB,SAAW,CAACH,iBAAiBI,IAAI,CAAC,CAAC/C,SAAWuB,SAASuB,OAAOb,cAAc,EAAEjC;IAEnF;IAEA,MAAMgD,wBAAwB,MAAMjE,GACjCiD,MAAM,CAAC;QACNC,gBAAgBzD,aAAayD,cAAc;QAC3CgB,UAAU9E,GAAW,CAAC,aAAa,EAAEK,aAAa2C,KAAK,CAAC,UAAU,CAAC;IACrE,GACCkB,IAAI,CAAC7D,cACL8D,KAAK,CACJ/E,IAAIG,GAAGc,aAAaU,WAAW,EAAEL,OAAOK,WAAW,GAAGtB,GAAGY,aAAa0E,SAAS,EAAE/E,GAAG,CAAC,KAAK,CAAC,IAE5FoE,OAAO,CAAC/D,aAAayD,cAAc;IAEtC,MAAMkB,mBAAmB,IAAIC,IAC3BJ,sBAAsBlD,GAAG,CAAC,CAAC0C,MAAQ;YAACa,SAASb,IAAIP,cAAc;YAAGO,IAAIS,QAAQ;SAAC;IAEjF,MAAMK,mCAAmC,MAAMvE,GAC5CiD,MAAM,CAAC;QACNC,gBAAgBzD,aAAayD,cAAc;QAC3CgB,UAAU9E,GAAW,CAAC,aAAa,EAAEK,aAAa2C,KAAK,CAAC,UAAU,CAAC;IACrE,GACCkB,IAAI,CAAC7D,cACL8D,KAAK,CACJ/E,IACEG,GAAGc,aAAaU,WAAW,EAAEL,OAAOK,WAAW,GAC/CxB,GAAGc,aAAakC,aAAa,EAAE7B,OAAO6B,aAAa,GACnD9C,GAAGY,aAAa0E,SAAS,EAAE/E,GAAG,CAAC,KAAK,CAAC,IAGxCoE,OAAO,CAAC/D,aAAayD,cAAc;IAEtC,MAAMpC,YAAYhB,OAAOgB,SAAS,CAACC,GAAG,CAAC,CAACC,WAAc,CAAA;YACpD6B,aAAa7B,SAAS6B,WAAW;YACjC5B,QAAQ;mBAAI1C,mBAAmByC,SAASC,MAAM;aAAE;YAChDC,gBAAgBF,SAASG,cAAc;QACzC,CAAA;IACAqD,8BAA8B1D,WAAWyD;IACzC,MAAM7C,QAAsB,EAAE;IAC9B,MAAM+C,SAA6B,EAAE;IACrC,IAAI9D,2BAA2Bb,OAAOc,eAAe;IACrD,MAAM8D,mBACJ5E,OAAO+B,uBAAuB,KAAK8B,YAAY,CAAC,IAAI;QAACxD,aAAaL,OAAOK,WAAW;IAAA;IAEtF,KAAK,MAAM4D,UAAUY,eAAe3B,YAAa;QAC/C,MAAMV,sBAAsBxB,UACzByB,MAAM,CAAC,CAACvB,WAAawB,SAASuB,OAAOb,cAAc,EAAElC,SAASC,MAAM,GACpEwB,IAAI,CACH,CAACC,GAAGC,IAAMD,EAAEzB,MAAM,CAAC2B,MAAM,GAAGD,EAAE1B,MAAM,CAAC2B,MAAM,IAAIF,EAAEG,WAAW,CAACC,aAAa,CAACH,EAAEE,WAAW;QAG5F,IAAIP,oBAAoBM,MAAM,KAAK,GAAG;QAEtC,MAAMsB,WAAWE,iBAAiBQ,GAAG,CAACN,SAASP,OAAOb,cAAc,MAAM;QAC1E,MAAM2B,aAAaC,KAAKC,GAAG,CAAC,GAAGhB,OAAOZ,MAAM,GAAGe;QAC/C,MAAMc,WAAW1C,oBAAoBL,MAAM,CACzC,CAACC,OAAOlB,WAAakB,QAAQlB,SAASE,cAAc,EACpD;QAEF,MAAM+D,QAAQH,KAAKI,GAAG,CAACL,YAAYG,UAAUrE;QAC7C,IAAIwE,qBAAqBjB;QAEzB,IAAIe,QAAQ,KAAKnF,OAAOc,eAAe,GAAG,GAAG;YAC3C,MAAM,CAACwE,SAAS,GAAG,MAAMpF,GACtBqF,MAAM,CAAC5F,cACP6F,MAAM,CAAC;gBACNnF,aAAaL,OAAOK,WAAW;gBAC/BwB,eAAe7B,OAAO6B,aAAa;gBACnCuB,gBAAgBa,OAAOb,cAAc;gBACrCd,OAAO6C;gBACPd,WAAW/E,GAAG,CAAC,SAAS,EAAEU,OAAO8B,UAAU,CAAC,yBAAyB,CAAC;YACxE,GACC2D,SAAS,CAAC;gBAACC,IAAI/F,aAAa+F,EAAE;gBAAErB,WAAW1E,aAAa0E,SAAS;YAAA;YAEpE,IAAI,CAACiB,UAAU,MAAM,IAAIK,MAAM;YAE/B9E,4BAA4BsE;YAC5BlC,UAAUT,qBAAqB2C;YAC/B,MAAMS,0BAA0B1F,IAAI;gBAClC2B,eAAe7B,OAAO6B,aAAa;gBACnCgE,eAAeP,SAASI,EAAE;gBAC1BtC,gBAAgBa,OAAOb,cAAc;gBACrCd,OAAO6C;gBACP9E,aAAaL,OAAOK,WAAW;gBAC/BE,OAAOP,OAAOO,KAAK;YACrB;YACA8E,sBAAsBF;YACtBR,OAAO3C,IAAI,CAAC;gBACV6D,eAAeP,SAASI,EAAE;gBAC1B,GAAGd,gBAAgB;gBACnBzD,QAAQ8C,OAAOb,cAAc;gBAC7Bd,OAAO6C;gBACPd,WAAWiB,SAASjB,SAAS;YAC/B;QACF;QAEAzC,MAAMI,IAAI,CAAC;YACT,GAAG4C,gBAAgB;YACnBzD,QAAQ8C,OAAOb,cAAc;YAC7BC,QAAQY,OAAOZ,MAAM;YACrBe,UAAUiB;YACV/B,gBAAgBW,OAAOX,cAAc;QACvC;IACF;IAEA,OAAO;QAAC1B;QAAOjC,cAAcgF;IAAM;AACrC;AAEA,eAAeiB,0BACb1F,EAAM,EACNF,MAOC;IAED,0EAA0E;IAC1E,+EAA+E;IAC/E,MAAM8F,gBAAgB,IACpBpH,IACEW,GACEX,IAAIQ,OAAOY,gBAAgBO,WAAW,GAAGnB,OAAOY,gBAAgB+F,aAAa,IAC7EnH,IACEO,UAAUa,gBAAgB+F,aAAa,GACvC7F,OAAOO,KAAK,KAAK,iBACbsD,YACAxE,GACEH,OAAOY,gBAAgBO,WAAW,GAClCxB,GAAGiB,gBAAgBO,WAAW,EAAEL,OAAOK,WAAW,IAExDjB,UACEc,GACGiD,MAAM,CAAC;YAACuC,IAAI/F,aAAa+F,EAAE;QAAA,GAC3BlC,IAAI,CAAC7D,cACL8D,KAAK,CACJ/E,IACEG,GAAGc,aAAa+F,EAAE,EAAE5F,gBAAgB+F,aAAa,GACjD9G,GAAGY,aAAa0E,SAAS,EAAE/E,GAAG,CAAC,KAAK,CAAC,QAMjDD,GACEH,OAAOY,gBAAgBiG,qBAAqB,GAC5C3G,UACEc,GACGiD,MAAM,CAAC;YAACuC,IAAI/F,aAAa+F,EAAE;QAAA,GAC3BlC,IAAI,CAAC7D,cACL8D,KAAK,CACJ/E,IACEG,GAAGc,aAAa+F,EAAE,EAAE5F,gBAAgBiG,qBAAqB,GACzDhH,GAAGY,aAAa0E,SAAS,EAAE/E,GAAG,CAAC,KAAK,CAAC,OAK/CJ,OAAOY,gBAAgBkG,qBAAqB,GAC5C9G,OAAOY,gBAAgBmG,eAAe;IAG1C,MAAMC,cAAc,MAAMhG,GACvBiD,MAAM,CAAC;QAACuC,IAAI5F,gBAAgB4F,EAAE;IAAA,GAC9BlC,IAAI,CAAC1D,iBACL2D,KAAK,CACJ/E,IACEG,GAAGiB,gBAAgB+B,aAAa,EAAE7B,OAAO6B,aAAa,GACtDiE,iBACA7G,UAAUa,gBAAgBqG,gBAAgB,GAC1CtH,GAAGiB,gBAAgBsG,KAAK,EAAE,YAC1BzH,cAAcmB,gBAAgBqB,MAAM,EAAEnB,OAAOoD,cAAc,GAC3DtE,OACEoB,GACGiD,MAAM,CAAC;QAACuC,IAAI7F,sBAAsB6F,EAAE;IAAA,GACpClC,IAAI,CAAC3D,uBACL4D,KAAK,CACJ/E,IACEG,GAAGgB,sBAAsBwG,gBAAgB,EAAEvG,gBAAgB4F,EAAE,GAC7D7G,GAAGgB,sBAAsBgC,aAAa,EAAE7B,OAAO6B,aAAa,GAC5D3C,OAAOW,sBAAsByG,QAAQ,GACrCvH,GAAGc,sBAAsBwE,SAAS,EAAE/E,GAAG,CAAC,KAAK,CAAC,OAMzDiH,OAAO,CAAC3H,IAAIkB,gBAAgByD,SAAS,GAAG3E,IAAIkB,gBAAgB4F,EAAE,GAC9Dc,KAAK,CAACxG,OAAOsC,KAAK,EAClBmE,GAAG,CAAC;IAEP,IAAIP,YAAYpD,MAAM,KAAK,GAAG;IAE9B,MAAM5C,GACHwG,MAAM,CAAC9G,wBACP+G,GAAG,CAAC;QAACC,WAAWtH,GAAG,CAAC,KAAK,CAAC;IAAA,GAC1BmE,KAAK,CACJ/E,IACEM,QACEY,uBAAuByG,gBAAgB,EACvCH,YAAYjF,GAAG,CAAC,CAAC4F,SAAWA,OAAOnB,EAAE,IAEvCxG,OAAOU,uBAAuBkH,UAAU,GACxC5H,OAAOU,uBAAuBgH,SAAS;IAI7C,MAAM1G,GACHwG,MAAM,CAAC5G,iBACP6G,GAAG,CAAC;QACHtG,aAAaL,OAAOK,WAAW;QAC/BwF,eAAe7F,OAAO6F,aAAa;QACnCE,uBAAuB;QACvBgB,YAAYzH,GAAG,CAAC,KAAK,CAAC;QACtB0H,WAAW1H,GAAG,CAAC,KAAK,CAAC;IACvB,GACCmE,KAAK,CACJ/E,IACEM,QACEc,gBAAgB4F,EAAE,EAClBQ,YAAYjF,GAAG,CAAC,CAAC4F,SAAWA,OAAOnB,EAAE,IAEvCI;AAGR;AAEA,eAAepF,mCAAmCC,oBAAyC;IACzF,IAAIA,qBAAqBsG,IAAI,KAAK,GAAG,OAAO,EAAE;IAC9C,MAAMtF,OAAO,MAAMpC,KAChB4D,MAAM,CAAC;QACN9C,aAAab,qBAAqBa,WAAW;QAC7CiD,gBAAgBhE,GAAS,CAAC,IAAI,EAAEE,qBAAqB+D,SAAS,CAAC,CAAC,CAAC;IACnE,GACCC,IAAI,CAAChE,sBACLiE,KAAK,CAACzE,QAAQQ,qBAAqBa,WAAW,EAAE;WAAIM;KAAqB,GACzE+C,OAAO,CAAClE,qBAAqBa,WAAW,EACxCkG,OAAO,CACN3H,IAAIU,GAAG,CAAC,IAAI,EAAEE,qBAAqB+D,SAAS,CAAC,CAAC,CAAC,GAC/C3E,IAAIY,qBAAqBa,WAAW;IAExC,OAAOsB,KAAKV,GAAG,CAAC,CAAC0C,MAAQA,IAAItD,WAAW;AAC1C;AAEA,OAAO,eAAe6G;IACpB,MAAMvF,OAAO,MAAMpC,KAChB4D,MAAM,CAAC;QAAC9C,aAAab,qBAAqBa,WAAW;IAAA,GACrDmD,IAAI,CAAChE,sBACLkE,OAAO,CAAClE,qBAAqBa,WAAW;IAC3C,OAAOsB,KAAKV,GAAG,CAAC,CAAC0C,MAAQA,IAAItD,WAAW;AAC1C;AAEA,eAAe0D,sCACb7D,EAAM,EACNF,MAAoD;IAEpD,MAAM2B,OAAO,MAAMzB,GAChBiD,MAAM,CAAC;QAAChC,QAAQ1B,+BAA+B0B,MAAM;IAAA,GACrDqC,IAAI,CAAC/D,gCACL0H,SAAS,CACRzH,mBACAb,GAAGa,kBAAkBgG,EAAE,EAAEjG,+BAA+BoC,aAAa,GAEtE4B,KAAK,CACJ/E,IACEG,GAAGY,+BAA+BY,WAAW,EAAEL,OAAOK,WAAW,GACjExB,GAAGa,kBAAkBW,WAAW,EAAEL,OAAOK,WAAW,GACpDxB,GAAGa,kBAAkBa,KAAK,EAAE,cAC5BjB,GAAG,CAAC,EAAEG,+BAA+B2H,YAAY,CAAC,YAAY,EAAEpH,OAAOgE,aAAa,CAAC,yBAAyB,CAAC,EAC/G1E,GAAG,CAAC,EAAEI,kBAAkBkH,SAAS,CAAC,QAAQ,CAAC,EAC3CtH,GAAG,CAAC,CAAC,EAAEI,kBAAkB2E,SAAS,CAAC,YAAY,EAAE3E,kBAAkB2E,SAAS,CAAC,SAAS,CAAC;IAG7F,OAAO1C,KAAKV,GAAG,CAAC,CAAC0C,MAAQA,IAAIxC,MAAM;AACrC;AAEA,OAAO,eAAekG,0BAA0BrH,MAAyB;IACvE,OAAO,MAAMT,KAAKU,WAAW,CAAC,OAAOC;QACnC,MAAMoH,cAAc,MAAMpH,GACvBiD,MAAM,CAAC;YAACuC,IAAI/F,aAAa+F,EAAE;QAAA,GAC3BlC,IAAI,CAAC7D,cACL8D,KAAK,CAACtE,GAAGQ,aAAa0E,SAAS,EAAE/E,GAAG,CAAC,KAAK,CAAC,GAC3CiH,OAAO,CAAC3H,IAAIe,aAAa0E,SAAS,GAClCmC,KAAK,CAACxG,QAAQwG,SAAS;QAC1B,OAAO,MAAMe,gCACXrH,IACAoH,YAAYrG,GAAG,CAAC,CAACoB,cAAgBA,YAAYqD,EAAE,GAC/C;YAAC8B,aAAa;QAAI;IAEtB;AACF;AAEA,OAAO,eAAeC,wBAAwBC,GAAa;IACzD,IAAIA,IAAI5E,MAAM,KAAK,GAAG,OAAO;IAE7B,OAAO,MAAMvD,KAAKU,WAAW,CAAC,OAAOC;QACnC,OAAO,MAAMqH,gCAAgCrH,IAAIwH;IACnD;AACF;AAEA,eAAeH,gCACbrH,EAAM,EACNwH,GAAa,EACb1H,SAAkC,CAAC,CAAC;IAEpC,IAAI0H,IAAI5E,MAAM,KAAK,GAAG,OAAO;IAE7B,MAAM6E,kBAAkB,MAAMzH,GAC3BiD,MAAM,CAAC;QACNuC,IAAI5F,gBAAgB4F,EAAE;QACtBG,eAAe/F,gBAAgB+F,aAAa;QAC5CE,uBAAuBjG,gBAAgBiG,qBAAqB;QAC5DE,iBAAiBnG,gBAAgBmG,eAAe;QAChDD,uBAAuBlG,gBAAgBkG,qBAAqB;IAC9D,GACCxC,IAAI,CAAC1D,iBACL2D,KAAK,CACJpE,GACEX,IACEM,QAAQc,gBAAgB+F,aAAa,EAAE6B,MACvCxI,OAAOY,gBAAgBmG,eAAe,GACtC/G,OAAOY,gBAAgBkG,qBAAqB,IAE9ChH,QAAQc,gBAAgBiG,qBAAqB,EAAE2B,OAGlDnB,OAAO,CAAC3H,IAAIkB,gBAAgB4F,EAAE,GAC9Be,GAAG,CAAC;IACP,MAAMmB,kBAAkB,MAAM1H,GAC3BiD,MAAM,CAAC;QAACuC,IAAI/F,aAAa+F,EAAE;IAAA,GAC3BlC,IAAI,CAAC7D,cACL8D,KAAK,CACJ/E,IACEM,QAAQW,aAAa+F,EAAE,EAAEgC,MACzB1H,OAAOwH,WAAW,GAAGrI,GAAGQ,aAAa0E,SAAS,EAAE/E,GAAG,CAAC,KAAK,CAAC,IAAIuE,YAGjE4C,GAAG,CAAC;IACP,MAAMoB,iBAAiBD,gBAAgB3G,GAAG,CAAC,CAACoB,cAAgBA,YAAYqD,EAAE;IAE1E,IAAImC,eAAe/E,MAAM,KAAK,GAAG,OAAO;IAExC,MAAMgF,oBAAoBH,gBACvBlF,MAAM,CACL,CAACoE,SACCA,OAAOhB,aAAa,IACpBgC,eAAeE,QAAQ,CAAClB,OAAOhB,aAAa,KAC5C,CAACgB,OAAOZ,eAAe,IACvB,CAACY,OAAOb,qBAAqB,EAEhC/E,GAAG,CAAC,CAAC4F,SAAWA,OAAOnB,EAAE;IAC5B,MAAMsC,oBAAoBL,gBACvBlF,MAAM,CACL,CAACoE,SACCA,OAAOd,qBAAqB,IAAI8B,eAAeE,QAAQ,CAAClB,OAAOd,qBAAqB,GAEvF9E,GAAG,CAAC,CAAC4F,SAAWA,OAAOnB,EAAE;IAC5B,IAAIoC,kBAAkBhF,MAAM,GAAG,GAAG;QAChC,MAAM5C,GACHwG,MAAM,CAAC9G,wBACP+G,GAAG,CAAC;YAACC,WAAWtH,GAAG,CAAC,KAAK,CAAC;QAAA,GAC1BmE,KAAK,CACJ/E,IACEM,QAAQY,uBAAuByG,gBAAgB,EAAEyB,oBACjD5I,OAAOU,uBAAuBkH,UAAU,GACxC5H,OAAOU,uBAAuBgH,SAAS;QAG7C,MAAM1G,GACHwG,MAAM,CAAC5G,iBACP6G,GAAG,CAAC;YACHtG,aAAa;YACbwF,eAAe;YACfkB,YAAY;YACZC,WAAW1H,GAAG,CAAC,KAAK,CAAC;QACvB,GACCmE,KAAK,CACJ/E,IACEM,QAAQc,gBAAgB4F,EAAE,EAAEoC,oBAC5B5I,OAAOY,gBAAgBmG,eAAe,GACtC/G,OAAOY,gBAAgBkG,qBAAqB;IAGpD;IACA,IAAIgC,kBAAkBlF,MAAM,GAAG,GAAG;QAChC,MAAM5C,GACHwG,MAAM,CAAC5G,iBACP6G,GAAG,CAAC;YAACZ,uBAAuB;YAAMiB,WAAW1H,GAAG,CAAC,KAAK,CAAC;QAAA,GACvDmE,KAAK,CACJ/E,IACEM,QAAQc,gBAAgB4F,EAAE,EAAEsC,oBAC5BhJ,QAAQc,gBAAgBiG,qBAAqB,EAAE8B;IAGvD;IAEA,MAAMI,UAAU,MAAM/H,GACnBgI,MAAM,CAACvI,cACP8D,KAAK,CAACzE,QAAQW,aAAa+F,EAAE,EAAEmC,iBAC/BpC,SAAS,CAAC;QAACC,IAAI/F,aAAa+F,EAAE;IAAA;IAEjC,OAAOuC,QAAQnF,MAAM;AACvB;AAEA,OAAO,eAAeqF,wBACpBjI,EAAM,EACNF,MAIC;IAED,IAAIoI,WAAW;IAEf,KAAK,MAAMC,WAAWrI,OAAOsI,QAAQ,CAAE;QACrC,IAAID,QAAQ/F,KAAK,IAAI,GAAG;QAExB,MAAMiG,cAAc,MAAMrI,GACvBwG,MAAM,CAAC/G,cACPgH,GAAG,CAAC;YAACrE,OAAOhD,GAAG,CAAC,EAAEK,aAAa2C,KAAK,CAAC,GAAG,EAAE+F,QAAQ/F,KAAK,CAAC,CAAC;QAAA,GACzDmB,KAAK,CACJ/E,IACEG,GAAGc,aAAa+F,EAAE,EAAE2C,QAAQxC,aAAa,GACzChH,GAAGc,aAAaU,WAAW,EAAEL,OAAOK,WAAW,GAC/CxB,GAAGc,aAAakC,aAAa,EAAE7B,OAAO6B,aAAa,GACnD9C,GAAGY,aAAa2C,KAAK,EAAE+F,QAAQ/F,KAAK,GACpCvD,GAAGY,aAAa0E,SAAS,EAAE/E,GAAG,CAAC,KAAK,CAAC,IAGxCmG,SAAS,CAAC;YAACC,IAAI/F,aAAa+F,EAAE;QAAA;QAEjC,IAAI6C,YAAYzF,MAAM,GAAG,GAAG;YAC1BsF,YAAYC,QAAQ/F,KAAK;YACzB;QACF;QAEA,MAAM2F,UAAU,MAAM/H,GACnBgI,MAAM,CAACvI,cACP8D,KAAK,CACJ/E,IACEG,GAAGc,aAAa+F,EAAE,EAAE2C,QAAQxC,aAAa,GACzChH,GAAGc,aAAaU,WAAW,EAAEL,OAAOK,WAAW,GAC/CxB,GAAGc,aAAakC,aAAa,EAAE7B,OAAO6B,aAAa,GACnDvC,GAAG,CAAC,EAAEK,aAAa2C,KAAK,CAAC,IAAI,EAAE+F,QAAQ/F,KAAK,CAAC,CAAC,EAC9CvD,GAAGY,aAAa0E,SAAS,EAAE/E,GAAG,CAAC,KAAK,CAAC,IAGxCmG,SAAS,CAAC;YAACnD,OAAO3C,aAAa2C,KAAK;QAAA;QAEvC8F,YAAYH,QAAQ9F,MAAM,CAAC,CAACC,OAAOuB,MAAQvB,QAAQuB,IAAIrB,KAAK,EAAE;IAChE;IAEA,OAAO8F;AACT;AAEA,SAAS1D,8BACP1D,SAA+B,EAC/BwH,kBAAkE;IAElE,KAAK,MAAMnG,eAAeoG,oBAAoBD,oBAAqB;QACjE,MAAMhG,sBAAsBxB,UACzByB,MAAM,CAAC,CAACvB,WAAawB,SAASL,YAAYe,cAAc,EAAElC,SAASC,MAAM,GACzEwB,IAAI,CACH,CAACC,GAAGC,IAAMD,EAAEzB,MAAM,CAAC2B,MAAM,GAAGD,EAAE1B,MAAM,CAAC2B,MAAM,IAAIF,EAAEG,WAAW,CAACC,aAAa,CAACH,EAAEE,WAAW;QAE5FE,UAAUT,qBAAqBH,YAAY+B,QAAQ;IACrD;AACF;AAEA,SAASS,eAAelD,IAAiB;IACvC,OAAO;WAAIA;KAAK,CAACgB,IAAI,CAAC,CAACC,GAAGC;QACxB,MAAM6F,cAAc7F,EAAEO,cAAc,CAACN,MAAM,GAAGF,EAAEQ,cAAc,CAACN,MAAM;QACrE,IAAI4F,gBAAgB,GAAG,OAAOA;QAC9B,OAAO9F,EAAEU,cAAc,CAACqF,OAAO,KAAK9F,EAAES,cAAc,CAACqF,OAAO;IAC9D;AACF;AAEA,SAASF,oBAA0D9G,IAAS;IAC1E,OAAO;WAAIA;KAAK,CAACgB,IAAI,CAAC,CAACC,GAAGC;QACxB,MAAM6F,cAAc7F,EAAEO,cAAc,CAACN,MAAM,GAAGF,EAAEQ,cAAc,CAACN,MAAM;QACrE,IAAI4F,gBAAgB,GAAG,OAAOA;QAC9B,OAAOlE,SAAS5B,EAAEQ,cAAc,EAAEJ,aAAa,CAACwB,SAAS3B,EAAEO,cAAc;IAC3E;AACF;AAEA,SAASV,SAASU,cAAwB,EAAEwF,eAAyB;IACnE,OAAOxF,eAAeyF,KAAK,CAAC,CAACC,QAAUF,gBAAgBb,QAAQ,CAACe;AAClE;AAEA,SAAS7F,UAAUjC,SAA+B,EAAEsB,KAAa;IAC/D,IAAIyG,YAAYzG;IAChB,KAAK,MAAMpB,YAAYF,UAAW;QAChC,IAAI+H,cAAc,GAAG;QACrB,MAAMC,OAAOhE,KAAKI,GAAG,CAAClE,SAASE,cAAc,EAAE2H;QAC/C7H,SAASE,cAAc,IAAI4H;QAC3BD,aAAaC;IACf;AACF;AAEA,SAASxE,SAASrD,MAAgB;IAChC,OAAO8H,KAAKC,SAAS,CAAC/H;AACxB"}

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

@@ -26,6 +26,7 @@ export interface RunnerInstanceReportEvent {
     reportedAt: Date;
 }
 export interface ReportRunnerInstancesParams {
+    scope: 'installation' | 'workspace';
     workspaceId: string | null;
     provisionerId: string;
     events: RunnerInstanceReportEvent[];
@@ -60,6 +61,9 @@ export declare function listActiveRunnerInstanceCountsByTemplateTx(tx: Tx, param
     workspaceId: string;
     provisionerId: string;
 }): Promise<ActiveRunnerInstanceTemplateCount[]>;
+export declare function countStaleEnrolledRunnerInstances(params: {
+    graceSeconds: number;
+}): Promise<number>;
 export declare function listActiveRunnerInstances(params: {
     workspaceId: string;
     windowSeconds: number;

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

@@ -1 +1 @@
-{"version":3,"file":"runner-instances.d.ts","sourceRoot":"","sources":["../../src/db/runner-instances.ts"],"names":[],"mappings":"AAoBA,OAAO,KAAK,EAAC,cAAc,EAAE,mBAAmB,EAAC,MAAM,mCAAmC,CAAC;AAC3F,OAAO,KAAK,EAAC,EAAE,EAAC,MAAM,SAAS,CAAC;AAEhC,OAAO,EAEL,KAAK,+BAA+B,EACrC,MAAM,qBAAqB,CAAC;AAS7B,eAAO,MAAM,cAAc,8CAIwB,CAAC;AACpD,eAAO,MAAM,YAAY,8CAI0B,CAAC;AACpD,eAAO,MAAM,qBAAqB,kCAG/B,CAAC;AAEJ,MAAM,MAAM,mCAAmC,GAAG,eAAe,CAAC;AAElE,MAAM,WAAW,6BAA6B;IAC5C,gBAAgB,EAAE,MAAM,CAAC;IACzB,MAAM,EAAE,mCAAmC,CAAC;CAC7C;AAED,MAAM,WAAW,iCAAiC;IAChD,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,CAAC,OAAO,qBAAqB,CAAC,CAAC,MAAM,CAAC,CAAC;IAC9C,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,yBAAyB;IACxC,gBAAgB,EAAE,MAAM,CAAC;IACzB,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,KAAK,EAAE,mBAAmB,CAAC;IAC3B,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,eAAe,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,UAAU,EAAE,IAAI,CAAC;CAClB;AAED,MAAM,WAAW,2BAA2B;IAC1C,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,yBAAyB,EAAE,CAAC;CACrC;AAED,MAAM,WAAW,8BAA8B;IAC7C,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,aAAa,EAAE,MAAM,CAAC;IACtB,yBAAyB,EAAE,MAAM,EAAE,CAAC;IACpC,qBAAqB,EAAE,MAAM,CAAC;CAC/B;AAED,MAAM,WAAW,gCAAgC;IAC/C,YAAY,EAAE,cAAc,EAAE,CAAC;IAC/B,oCAAoC,EAAE,GAAG,CAAC,MAAM,EAAE,+BAA+B,CAAC,CAAC;IACnF,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,oBAAoB,EAAE,MAAM,CAAC;CAC9B;AAED,MAAM,WAAW,8BAA8B;IAC7C,MAAM,EAAE,MAAM,CAAC;IACf,oBAAoB,EAAE,MAAM,CAAC;CAC9B;AAED,wBAAsB,8BAA8B,CAAC,MAAM,EAAE;IAC3D,gBAAgB,EAAE,MAAM,CAAC;IACzB,aAAa,EAAE,MAAM,CAAC;IACtB,gBAAgB,EAAE,MAAM,CAAC;CAC1B,GAAG,OAAO,CAAC,OAAO,CAAC,CAcnB;AAiBD,wBAAsB,qBAAqB,CAAC,MAAM,EAAE,2BAA2B,GAAG,OAAO,CAAC;IACxF,QAAQ,EAAE,MAAM,CAAC;IACjB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,uBAAuB,EAAE,6BAA6B,EAAE,CAAC;CAC1D,CAAC,CA2ED;AAED,wBAAsB,0CAA0C,CAC9D,EAAE,EAAE,EAAE,EACN,MAAM,EAAE;IAAC,WAAW,EAAE,MAAM,CAAC;IAAC,aAAa,EAAE,MAAM,CAAA;CAAC,GACnD,OAAO,CAAC,iCAAiC,EAAE,CAAC,CAuB9C;AAED,wBAAsB,yBAAyB,CAAC,MAAM,EAAE;IACtD,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,CAe5B;AAED,wBAAsB,+BAA+B,CAAC,MAAM,EAAE;IAC5D,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,MAAM,CAAC;CACf,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAKpB;AAED,wBAAsB,oCAAoC,CACxD,EAAE,EAAE,EAAE,EACN,MAAM,EAAE;IACN,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,MAAM,CAAC;CACf,GACA,OAAO,CAAC,6BAA6B,EAAE,CAAC,CAsB1C;AAqFD,wBAAsB,wBAAwB,CAC5C,MAAM,EAAE,8BAA8B,GACrC,OAAO,CAAC,gCAAgC,CAAC,CAqG3C;AAED,wBAAsB,wBAAwB,CAAC,MAAM,EAAE;IACrD,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,EAAE,MAAM,CAAC;CACf,GAAG,OAAO,CAAC,8BAA8B,CAAC,CA4D1C;AAkcD,wBAAgB,eAAe,CAAC,KAAK,EAAE,mBAAmB,GAAG,OAAO,CAEnE"}
+{"version":3,"file":"runner-instances.d.ts","sourceRoot":"","sources":["../../src/db/runner-instances.ts"],"names":[],"mappings":"AAmBA,OAAO,KAAK,EAAC,cAAc,EAAE,mBAAmB,EAAC,MAAM,mCAAmC,CAAC;AAE3F,OAAO,KAAK,EAAC,EAAE,EAAC,MAAM,SAAS,CAAC;AAEhC,OAAO,EAEL,KAAK,+BAA+B,EACrC,MAAM,qBAAqB,CAAC;AAU7B,eAAO,MAAM,cAAc,8CAIwB,CAAC;AACpD,eAAO,MAAM,YAAY,8CAI0B,CAAC;AACpD,eAAO,MAAM,qBAAqB,kCAG/B,CAAC;AAEJ,MAAM,MAAM,mCAAmC,GAAG,eAAe,CAAC;AAElE,MAAM,WAAW,6BAA6B;IAC5C,gBAAgB,EAAE,MAAM,CAAC;IACzB,MAAM,EAAE,mCAAmC,CAAC;CAC7C;AAED,MAAM,WAAW,iCAAiC;IAChD,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,CAAC,OAAO,qBAAqB,CAAC,CAAC,MAAM,CAAC,CAAC;IAC9C,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,yBAAyB;IACxC,gBAAgB,EAAE,MAAM,CAAC;IACzB,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,KAAK,EAAE,mBAAmB,CAAC;IAC3B,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,eAAe,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,UAAU,EAAE,IAAI,CAAC;CAClB;AAED,MAAM,WAAW,2BAA2B;IAC1C,KAAK,EAAE,cAAc,GAAG,WAAW,CAAC;IACpC,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,yBAAyB,EAAE,CAAC;CACrC;AAED,MAAM,WAAW,8BAA8B;IAC7C,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,aAAa,EAAE,MAAM,CAAC;IACtB,yBAAyB,EAAE,MAAM,EAAE,CAAC;IACpC,qBAAqB,EAAE,MAAM,CAAC;CAC/B;AAED,MAAM,WAAW,gCAAgC;IAC/C,YAAY,EAAE,cAAc,EAAE,CAAC;IAC/B,oCAAoC,EAAE,GAAG,CAAC,MAAM,EAAE,+BAA+B,CAAC,CAAC;IACnF,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,oBAAoB,EAAE,MAAM,CAAC;CAC9B;AAED,MAAM,WAAW,8BAA8B;IAC7C,MAAM,EAAE,MAAM,CAAC;IACf,oBAAoB,EAAE,MAAM,CAAC;CAC9B;AAED,wBAAsB,8BAA8B,CAAC,MAAM,EAAE;IAC3D,gBAAgB,EAAE,MAAM,CAAC;IACzB,aAAa,EAAE,MAAM,CAAC;IACtB,gBAAgB,EAAE,MAAM,CAAC;CAC1B,GAAG,OAAO,CAAC,OAAO,CAAC,CAcnB;AAiBD,wBAAsB,qBAAqB,CAAC,MAAM,EAAE,2BAA2B,GAAG,OAAO,CAAC;IACxF,QAAQ,EAAE,MAAM,CAAC;IACjB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,uBAAuB,EAAE,6BAA6B,EAAE,CAAC;CAC1D,CAAC,CA+ED;AAED,wBAAsB,0CAA0C,CAC9D,EAAE,EAAE,EAAE,EACN,MAAM,EAAE;IAAC,WAAW,EAAE,MAAM,CAAC;IAAC,aAAa,EAAE,MAAM,CAAA;CAAC,GACnD,OAAO,CAAC,iCAAiC,EAAE,CAAC,CAuB9C;AAED,wBAAsB,iCAAiC,CAAC,MAAM,EAAE;IAC9D,YAAY,EAAE,MAAM,CAAC;CACtB,GAAG,OAAO,CAAC,MAAM,CAAC,CA0BlB;AAED,wBAAsB,yBAAyB,CAAC,MAAM,EAAE;IACtD,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,CAe5B;AAED,wBAAsB,+BAA+B,CAAC,MAAM,EAAE;IAC5D,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,MAAM,CAAC;CACf,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAKpB;AAED,wBAAsB,oCAAoC,CACxD,EAAE,EAAE,EAAE,EACN,MAAM,EAAE;IACN,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,MAAM,CAAC;CACf,GACA,OAAO,CAAC,6BAA6B,EAAE,CAAC,CAsB1C;AAqFD,wBAAsB,wBAAwB,CAC5C,MAAM,EAAE,8BAA8B,GACrC,OAAO,CAAC,gCAAgC,CAAC,CAqG3C;AAED,wBAAsB,wBAAwB,CAAC,MAAM,EAAE;IACrD,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,EAAE,MAAM,CAAC;CACf,GAAG,OAAO,CAAC,8BAA8B,CAAC,CA4D1C;AAkcD,wBAAgB,eAAe,CAAC,KAAK,EAAE,mBAAmB,GAAG,OAAO,CAEnE"}

package/dist/db/runner-instances.js

@@ -1,13 +1,14 @@
 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 { sanitizeRunnerLabels } from '#core/runner-labels.js';
 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 { reservations } from './schema/reservations.js';
+import { runnerControlSessions } from './schema/runner-control-sessions.js';
 import { providerRunners, toRunnerInstance } from './schema/runner-instances.js';
 import { runnerSessions } from './schema/runner-sessions.js';
 import { runningJobExecutions } from './schema/running-job-executions.js';
@@ -57,9 +58,11 @@ export async function reportRunnerInstances(params) {
                 providerRunnerId: event.providerRunnerId,
                 reservationId: event.reservationId,
                 templateKey: event.templateKey,
-                labels: [
-                    ...canonicalizeLabels(event.labels)
-                ],
+                labels: sanitizeRunnerLabels(event.labels, {
+                    scope: params.scope,
+                    logLevel: 'debug',
+                    source: 'runner instance report'
+                }),
                 state: event.state,
                 reason: event.reason,
                 runnerSessionId: event.runnerSessionId,
@@ -78,7 +81,7 @@ export async function reportRunnerInstances(params) {
             ],
             targetWhere: isNotNull(providerRunners.providerRunnerId),
             set: {
-                reservationId: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN coalesce(excluded.reservation_id, ${providerRunners.reservationId}) ELSE ${providerRunners.reservationId} END`,
+                reservationId: sql`CASE WHEN ${providerRunnerProjectionUpdateCondition()} THEN coalesce(${providerRunners.reservationId}, excluded.reservation_id) 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`,
@@ -120,6 +123,14 @@ export async function listActiveRunnerInstanceCountsByTemplateTx(tx, params) {
             }
         ] : []);
 }
+export async function countStaleEnrolledRunnerInstances(params) {
+    const [row] = await db().select({
+        count: sql`count(*)::int`
+    }).from(providerRunners).where(and(eq(providerRunners.state, 'running'), isNull(providerRunners.workspaceId), isNull(providerRunners.runnerSessionId), lt(providerRunners.reportedAt, staleRunnerInstanceCutoff(params.graceSeconds)), exists(db().select({
+        id: runnerControlSessions.id
+    }).from(runnerControlSessions).where(and(eq(runnerControlSessions.runnerInstanceId, providerRunners.id), isNull(runnerControlSessions.closedAt), gt(runnerControlSessions.expiresAt, sql`now()`))))));
+    return row?.count ?? 0;
+}
 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);