refactor: colocate relay assignment admission owners (#32142)

cloud/apps/relay/src/app.ts

@@ -37,7 +37,7 @@ import {
   type RelayAssignmentStore,
   type ResolvedRelayAssignment
 } from './assignment-store.js'
-import { ASSIGNMENT_LEASE_AUDIENCE, assignmentLeaseKeyId } from './assignment-lease.js'
+import { ASSIGNMENT_LEASE_AUDIENCE, assignmentLeaseKeyId } from './assignment/assignment-lease.js'
 import { AssignmentRejectionLogWindow } from './assignment/assignment-rejection-log-window.js'
 import { CELL_ADMISSION_STATES } from './cell-admission-selector.js'
 import { registerCellSeatFeedRoute } from './cell-seat-feed-route.js'

cloud/apps/relay/src/assignment-store.ts

@@ -58,7 +58,7 @@ import {
 import {
   ASSIGNMENT_CONNECTION_HEADROOM_QUERY
 } from './assignment-connection-headroom-query.js'
-import { AssignmentIdentityQueue } from './assignment-identity-queue.js'
+import { AssignmentIdentityQueue } from './assignment/assignment-identity-queue.js'
 import {
   CONTROL_RENEWAL_BATCH_SQL,
   CONTROL_RENEWAL_STATEMENT_OUTCOMES,

cloud/apps/relay/src/assignment/assignment-lease-shadow.ts

@@ -1,4 +1,4 @@
-import { percentile } from './relay-observability.js'
+import { percentile } from '../relay-observability.js'
 import { classifyAssignmentLease, type AssignmentLeaseClass } from './assignment-lease.js'
 
 // What `ticketCheck=enforce` would have said, beside the database answer that still decides.

cloud/apps/relay/src/assignment/assignment-lease.test.ts

@@ -11,12 +11,12 @@ import {
   classifyAssignmentLease
 } from './assignment-lease.js'
 import { ASSIGNMENT_LEASE_SHADOW_EVENT, AssignmentLeaseShadow } from './assignment-lease-shadow.js'
-import type { RelayAssignmentStore } from './assignment-store.js'
-import type { RelayConfig } from './config.js'
-import type { RelayCredentialStore } from './credential-store.js'
-import { HostSessionRegistry } from './host-session-registry.js'
-import type { RelayTokenClaims } from './relay-token-verifier.js'
-import { ProcessQueuedByteBudget } from './splice-forwarder.js'
+import type { RelayAssignmentStore } from '../assignment-store.js'
+import type { RelayConfig } from '../config.js'
+import type { RelayCredentialStore } from '../credential-store.js'
+import { HostSessionRegistry } from '../host-session-registry.js'
+import type { RelayTokenClaims } from '../relay-token-verifier.js'
+import { ProcessQueuedByteBudget } from '../splice-forwarder.js'
 
 const KEY = new TextEncoder().encode('0123456789abcdef0123456789abcdef')
 const OTHER_KEY = new TextEncoder().encode('fedcba9876543210fedcba9876543210')

cloud/apps/relay/src/control-renewal-statement.ts

@@ -1,4 +1,4 @@
-import type { AssignmentIdentity } from './assignment-identity-queue.js'
+import type { AssignmentIdentity } from './assignment/assignment-identity-queue.js'
 import type { SqlRow } from './database.js'
 
 export type ControlRenewalOutcome =

cloud/apps/relay/src/host-session-registry.ts

@@ -30,8 +30,8 @@ import nacl from 'tweetnacl'
 import type WebSocket from 'ws'
 import type { RawData } from 'ws'
 import type { RelayConfig } from './config.js'
-import type { AssignmentLeaseClass } from './assignment-lease.js'
-import type { AssignmentLeaseShadow } from './assignment-lease-shadow.js'
+import type { AssignmentLeaseClass } from './assignment/assignment-lease.js'
+import type { AssignmentLeaseShadow } from './assignment/assignment-lease-shadow.js'
 import type { RelayAssignmentStore } from './assignment-store.js'
 import type { CellAdmitMode, CellReserveBook } from './cell-reserve-book.js'
 import type { DemoteRequest, ReserveOutcome, ReserveRequest } from './cell-reserve-contract.js'

cloud/apps/relay/src/relay-server.ts

@@ -21,11 +21,11 @@ import type { RawData } from 'ws'
 import { CellAdmitEffectiveWriter } from './cell-admit-effective-writer.js'
 import { ReserveDeadMan, reserveDeadManWindowMs } from './cell-reserve-dead-man.js'
 import { createRelayApp } from './app.js'
-import { classifyAssignmentLease } from './assignment-lease.js'
+import { classifyAssignmentLease } from './assignment/assignment-lease.js'
 import { defaultIntakePerSec, type CellFlags } from './cell-flags.js'
 import { CellReserveBook } from './cell-reserve-book.js'
 import type { DemoteRequest, ReserveRequest } from './cell-reserve-contract.js'
-import { AssignmentLeaseShadow } from './assignment-lease-shadow.js'
+import { AssignmentLeaseShadow } from './assignment/assignment-lease-shadow.js'
 import { RelayAssignmentStore } from './assignment-store.js'
 import type { RelayConfig } from './config.js'
 import { RelayCredentialStore } from './credential-store.js'

cloud/apps/relay/src/relay.blackbox.test.ts

@@ -17,7 +17,7 @@ import nacl from 'tweetnacl'
 import { afterAll, beforeAll, describe, expect, it } from 'vitest'
 import WebSocket from 'ws'
 import type { RawData } from 'ws'
-import { assignmentLeaseKeyId } from './assignment-lease.js'
+import { assignmentLeaseKeyId } from './assignment/assignment-lease.js'
 import { RelayAssignmentStore } from './assignment-store.js'
 import {
   encodeMembership,