mirror of
https://github.com/stablyai/orca.git
synced 2026-11-14 09:57:48 +00:00
Extract automation API operations into runtime layer (#37196)
* Extract automation API operations into a separate runtime layer * Fix Source Control test fixtures for remote operation state
This commit is contained in:
Jinjing
@@ -0,0 +1,134 @@
|
||||
import type { AutomationsChangedPayload } from '../../../../shared/runtime-client-events'
|
||||
import type {
|
||||
AutomationOwnerPrecondition,
|
||||
AutomationDestination,
|
||||
AutomationUpdatePreconditions
|
||||
} from '../../../../shared/automation-owner-precondition'
|
||||
import type { Automation, AutomationRun } from '../../../../shared/automations-types'
|
||||
import type {
|
||||
RuntimeAutomationCreateInput,
|
||||
RuntimeAutomationUpdateInput
|
||||
} from '../../runtime-automation-controller'
|
||||
import {
|
||||
automationChangePublications,
|
||||
type AutomationChangeSelector,
|
||||
type AutomationListResult
|
||||
} from '../../../../shared/automation-list-scope'
|
||||
import { OrcaRuntimeWithCrewOperations } from './crew-operations'
|
||||
|
||||
export abstract class OrcaRuntimeWithAutomationApiOperations extends OrcaRuntimeWithCrewOperations {
|
||||
abstract notifyAutomationsChanged(payload?: AutomationsChangedPayload): void
|
||||
|
||||
private readAutomationRuns<T>(
|
||||
automationId: string | undefined,
|
||||
expectedOwner: AutomationOwnerPrecondition | undefined,
|
||||
read: () => T
|
||||
): T {
|
||||
if (expectedOwner && !automationId) {
|
||||
throw new Error('An expected owner requires an automation id.')
|
||||
}
|
||||
return this.automation.withExternalProbePriority(() => {
|
||||
if (automationId) {
|
||||
this.automation.assertOwner(automationId, expectedOwner, 'read')
|
||||
}
|
||||
return read()
|
||||
})
|
||||
}
|
||||
|
||||
listAutomationRuns(
|
||||
automationId?: string,
|
||||
expectedOwner?: AutomationOwnerPrecondition
|
||||
): AutomationRun[] {
|
||||
return this.readAutomationRuns(automationId, expectedOwner, () =>
|
||||
this.automation.listRuns(automationId)
|
||||
)
|
||||
}
|
||||
|
||||
listAutomationRunsPage(
|
||||
automationId?: string,
|
||||
expectedOwner?: AutomationOwnerPrecondition,
|
||||
limit?: number,
|
||||
cursor?: string
|
||||
) {
|
||||
return this.readAutomationRuns(automationId, expectedOwner, () =>
|
||||
this.automation.listRunsPage(automationId, limit, cursor)
|
||||
)
|
||||
}
|
||||
|
||||
showAutomation(id: string, expectedOwner?: AutomationOwnerPrecondition): Automation {
|
||||
const automation = this.automation.show(id)
|
||||
this.automation.assertOwner(id, expectedOwner, 'read')
|
||||
return automation
|
||||
}
|
||||
|
||||
protected automationChangeSelector(id: string): AutomationChangeSelector | null {
|
||||
return this.store?.automationChangeSelector?.(id) ?? null
|
||||
}
|
||||
|
||||
/** A store that cannot name the affected host degrades to one unscoped authority event. */
|
||||
protected publishAutomationDefinitionChange(
|
||||
before: AutomationChangeSelector | null,
|
||||
after: AutomationChangeSelector | null
|
||||
): void {
|
||||
for (const selector of automationChangePublications(before, after)) {
|
||||
this.notifyAutomationsChanged({ reason: 'definition', ...(selector ? { selector } : {}) })
|
||||
}
|
||||
}
|
||||
|
||||
createAutomation(
|
||||
input: RuntimeAutomationCreateInput,
|
||||
destination?: AutomationDestination
|
||||
): Promise<Automation> {
|
||||
return this.automation.withExternalProbePriority(() =>
|
||||
this.automation.create(input, destination).then((automation) => {
|
||||
const selector = this.automationChangeSelector(automation.id)
|
||||
this.publishAutomationDefinitionChange(selector, selector)
|
||||
return automation
|
||||
})
|
||||
)
|
||||
}
|
||||
|
||||
updateAutomation(
|
||||
id: string,
|
||||
updates: RuntimeAutomationUpdateInput,
|
||||
options?: AutomationUpdatePreconditions
|
||||
): Promise<Automation> {
|
||||
return this.automation.withExternalProbePriority(() => {
|
||||
const source = this.automationChangeSelector(id)
|
||||
return this.automation.update(id, updates, options).then((automation) => {
|
||||
this.publishAutomationDefinitionChange(source, this.automationChangeSelector(automation.id))
|
||||
return automation
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
deleteAutomation(
|
||||
id: string,
|
||||
expectedOwner?: AutomationOwnerPrecondition
|
||||
): Promise<{ removed: boolean; id: string }> {
|
||||
return this.automation.withExternalProbePriority(() => {
|
||||
const selector = this.automationChangeSelector(id)
|
||||
return this.automation.delete(id, expectedOwner).then((result) => {
|
||||
this.publishAutomationDefinitionChange(selector, selector)
|
||||
return result
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
runAutomationNow(
|
||||
id: string,
|
||||
expectedOwner?: AutomationOwnerPrecondition
|
||||
): Promise<AutomationRun> {
|
||||
return this.automation.withExternalProbePriority(() =>
|
||||
this.automation.runNow(id, expectedOwner)
|
||||
)
|
||||
}
|
||||
|
||||
listAutomationsForScope(params = {}): AutomationListResult {
|
||||
return this.automation.withExternalProbePriority(() => this.automation.listForScope(params))
|
||||
}
|
||||
|
||||
automationOwnerPrecondition(id: string): AutomationOwnerPrecondition | null {
|
||||
return this.automation.ownerPrecondition(id)
|
||||
}
|
||||
}
|
||||
@@ -1,22 +1,6 @@
|
||||
import type { AutomationsChangedPayload } from '../../../../shared/runtime-client-events'
|
||||
import { getWorkerTerminalAutoReleaseScheduler } from '../../orchestration/worker-terminal-auto-release-scheduler'
|
||||
import type { ResolvedWorktree } from '../../runtime-worktree-path-identity'
|
||||
import { OrcaRuntimeWithCrewOperations } from './crew-operations'
|
||||
import type {
|
||||
AutomationOwnerPrecondition,
|
||||
AutomationDestination,
|
||||
AutomationUpdatePreconditions
|
||||
} from '../../../../shared/automation-owner-precondition'
|
||||
import type { Automation, AutomationRun } from '../../../../shared/automations-types'
|
||||
import type {
|
||||
RuntimeAutomationCreateInput,
|
||||
RuntimeAutomationUpdateInput
|
||||
} from '../../runtime-automation-controller'
|
||||
import {
|
||||
automationChangePublications,
|
||||
type AutomationChangeSelector,
|
||||
type AutomationListResult
|
||||
} from '../../../../shared/automation-list-scope'
|
||||
import { OrcaRuntimeWithAutomationApiOperations } from './automation-api-operations'
|
||||
import { OrchestrationDb } from '../../orchestration/db'
|
||||
import {
|
||||
committedHandBackTypes,
|
||||
@@ -26,8 +10,7 @@ import { existsSync } from 'node:fs'
|
||||
import type { LegacyWorkerTerminalRecoveryPlan } from '../../orchestration/orchestration-legacy-worker-terminal-recovery'
|
||||
import type { LegacyWorkerTerminalRecoveryResult } from '../../runtime-legacy-worker-terminal-recovery-types'
|
||||
|
||||
export abstract class OrcaRuntimeWithAutomationOperations extends OrcaRuntimeWithCrewOperations {
|
||||
abstract notifyAutomationsChanged(payload?: AutomationsChangedPayload): void
|
||||
export abstract class OrcaRuntimeWithAutomationOperations extends OrcaRuntimeWithAutomationApiOperations {
|
||||
abstract ensureOrchestrationFederationRelay(runId?: string): void
|
||||
protected abstract scheduleRestoredMessageRepoints(): void
|
||||
/** Mail a worker left unread when it reported before a restart still continues it. */
|
||||
@@ -35,119 +18,6 @@ export abstract class OrcaRuntimeWithAutomationOperations extends OrcaRuntimeWit
|
||||
protected abstract getResolvedWorktreeMap(): Promise<Map<string, ResolvedWorktree>>
|
||||
protected abstract getLeafKey(tabId: string, leafId: string): string
|
||||
|
||||
private readAutomationRuns<T>(
|
||||
automationId: string | undefined,
|
||||
expectedOwner: AutomationOwnerPrecondition | undefined,
|
||||
read: () => T
|
||||
): T {
|
||||
if (expectedOwner && !automationId) {
|
||||
throw new Error('An expected owner requires an automation id.')
|
||||
}
|
||||
return this.automation.withExternalProbePriority(() => {
|
||||
if (automationId) {
|
||||
this.automation.assertOwner(automationId, expectedOwner, 'read')
|
||||
}
|
||||
return read()
|
||||
})
|
||||
}
|
||||
|
||||
listAutomationRuns(
|
||||
automationId?: string,
|
||||
expectedOwner?: AutomationOwnerPrecondition
|
||||
): AutomationRun[] {
|
||||
return this.readAutomationRuns(automationId, expectedOwner, () =>
|
||||
this.automation.listRuns(automationId)
|
||||
)
|
||||
}
|
||||
|
||||
listAutomationRunsPage(
|
||||
automationId?: string,
|
||||
expectedOwner?: AutomationOwnerPrecondition,
|
||||
limit?: number,
|
||||
cursor?: string
|
||||
) {
|
||||
return this.readAutomationRuns(automationId, expectedOwner, () =>
|
||||
this.automation.listRunsPage(automationId, limit, cursor)
|
||||
)
|
||||
}
|
||||
|
||||
showAutomation(id: string, expectedOwner?: AutomationOwnerPrecondition): Automation {
|
||||
const automation = this.automation.show(id)
|
||||
this.automation.assertOwner(id, expectedOwner, 'read')
|
||||
return automation
|
||||
}
|
||||
|
||||
protected automationChangeSelector(id: string): AutomationChangeSelector | null {
|
||||
return this.store?.automationChangeSelector?.(id) ?? null
|
||||
}
|
||||
|
||||
/** A store that cannot name the affected host degrades to one unscoped authority event. */
|
||||
protected publishAutomationDefinitionChange(
|
||||
before: AutomationChangeSelector | null,
|
||||
after: AutomationChangeSelector | null
|
||||
): void {
|
||||
for (const selector of automationChangePublications(before, after)) {
|
||||
this.notifyAutomationsChanged({ reason: 'definition', ...(selector ? { selector } : {}) })
|
||||
}
|
||||
}
|
||||
|
||||
createAutomation(
|
||||
input: RuntimeAutomationCreateInput,
|
||||
destination?: AutomationDestination
|
||||
): Promise<Automation> {
|
||||
return this.automation.withExternalProbePriority(() =>
|
||||
this.automation.create(input, destination).then((automation) => {
|
||||
const selector = this.automationChangeSelector(automation.id)
|
||||
this.publishAutomationDefinitionChange(selector, selector)
|
||||
return automation
|
||||
})
|
||||
)
|
||||
}
|
||||
|
||||
updateAutomation(
|
||||
id: string,
|
||||
updates: RuntimeAutomationUpdateInput,
|
||||
options?: AutomationUpdatePreconditions
|
||||
): Promise<Automation> {
|
||||
return this.automation.withExternalProbePriority(() => {
|
||||
const source = this.automationChangeSelector(id)
|
||||
return this.automation.update(id, updates, options).then((automation) => {
|
||||
this.publishAutomationDefinitionChange(source, this.automationChangeSelector(automation.id))
|
||||
return automation
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
deleteAutomation(
|
||||
id: string,
|
||||
expectedOwner?: AutomationOwnerPrecondition
|
||||
): Promise<{ removed: boolean; id: string }> {
|
||||
return this.automation.withExternalProbePriority(() => {
|
||||
const selector = this.automationChangeSelector(id)
|
||||
return this.automation.delete(id, expectedOwner).then((result) => {
|
||||
this.publishAutomationDefinitionChange(selector, selector)
|
||||
return result
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
runAutomationNow(
|
||||
id: string,
|
||||
expectedOwner?: AutomationOwnerPrecondition
|
||||
): Promise<AutomationRun> {
|
||||
return this.automation.withExternalProbePriority(() =>
|
||||
this.automation.runNow(id, expectedOwner)
|
||||
)
|
||||
}
|
||||
|
||||
listAutomationsForScope(params = {}): AutomationListResult {
|
||||
return this.automation.withExternalProbePriority(() => this.automation.listForScope(params))
|
||||
}
|
||||
|
||||
automationOwnerPrecondition(id: string): AutomationOwnerPrecondition | null {
|
||||
return this.automation.ownerPrecondition(id)
|
||||
}
|
||||
|
||||
// Defer until userData is final; tests can inject an in-memory DB.
|
||||
getOrchestrationDb(): OrchestrationDb {
|
||||
if (!this._orchestrationDb) {
|
||||
|
||||
Reference in New Issue
Block a user