File size: 4,038 Bytes
68d7816 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 | import { ScopeActivation } from '#/_base/di/instantiation';
import type { ServiceIdentifier } from '#/_base/di/instantiation';
import type {
AgentToolCtor,
AnyAgentTool,
} from '#/agent/toolRegistry/toolContribution';
import { IFlagService } from '#/app/flag/flag';
import { LifecycleScope } from '#/app/scopes';
import { Feature } from '#/features/feature';
import { registerFeature } from '#/features/featureRegistry';
import { TOWER_FLAG_ID } from './tower';
import { ITowerRateLimitService } from './towerRateLimit';
import { TowerRateLimitService } from './towerRateLimitService';
import { ITowerFindingTool } from './tools/finding/finding';
import { TowerFindingTool } from './tools/finding/findingTool';
import { ITowerInboxTool } from './tools/inbox/inbox';
import { TowerInboxTool } from './tools/inbox/inboxTool';
import { ITowerInitTool } from './tools/init/init';
import { TowerInitTool } from './tools/init/initTool';
import { ITowerMergeTool } from './tools/merge/merge';
import { TowerMergeTool } from './tools/merge/mergeTool';
import { ITowerMissionTool } from './tools/mission/mission';
import { TowerMissionTool } from './tools/mission/missionTool';
import { ITowerPlanTool } from './tools/plan/plan';
import { TowerPlanTool } from './tools/plan/planTool';
import { ITowerReviewTool } from './tools/review/review';
import { TowerReviewTool } from './tools/review/reviewTool';
import { ITowerSendTool } from './tools/send/send';
import { TowerSendTool } from './tools/send/sendTool';
import { ITowerSpawnTool } from './tools/spawn/spawn';
import { TowerSpawnTool } from './tools/spawn/spawnTool';
import { ITowerStatusTool } from './tools/status/status';
import { TowerStatusTool } from './tools/status/statusTool';
import { ITowerTeardownTool } from './tools/teardown/teardown';
import { TowerTeardownTool } from './tools/teardown/teardownTool';
import { TOWER_WORKER_PROFILE_DEF } from './workerProfile';
interface TowerToolContribution {
readonly id: ServiceIdentifier<AnyAgentTool>;
readonly ctor: AgentToolCtor;
readonly name: string;
}
export const TOWER_TOOL_CONTRIBUTIONS: readonly TowerToolContribution[] = [
{ id: ITowerInitTool, ctor: TowerInitTool, name: 'TowerInit' },
{ id: ITowerPlanTool, ctor: TowerPlanTool, name: 'TowerPlan' },
{ id: ITowerSpawnTool, ctor: TowerSpawnTool, name: 'TowerSpawn' },
{ id: ITowerMergeTool, ctor: TowerMergeTool, name: 'TowerMerge' },
{ id: ITowerTeardownTool, ctor: TowerTeardownTool, name: 'TowerTeardown' },
{ id: ITowerSendTool, ctor: TowerSendTool, name: 'TowerSend' },
{ id: ITowerInboxTool, ctor: TowerInboxTool, name: 'TowerInbox' },
{ id: ITowerFindingTool, ctor: TowerFindingTool, name: 'TowerFinding' },
{ id: ITowerReviewTool, ctor: TowerReviewTool, name: 'TowerReview' },
{ id: ITowerMissionTool, ctor: TowerMissionTool, name: 'TowerMission' },
{ id: ITowerStatusTool, ctor: TowerStatusTool, name: 'TowerStatus' },
];
export class TowerFeature extends Feature {
static override readonly name = 'tower';
constructor(@IFlagService flags: IFlagService) {
super();
if (!flags.enabled(TOWER_FLAG_ID)) return;
assembledFlagServices.add(flags);
this.onDispose(() => {
assembledFlagServices.delete(flags);
});
this.contributeService(LifecycleScope.App, ITowerRateLimitService, TowerRateLimitService, {
activation: ScopeActivation.OnDemand,
});
for (const tool of TOWER_TOOL_CONTRIBUTIONS) {
this.contributeTool(tool.id, tool.ctor, {
name: tool.name,
domain: 'tower',
});
}
this.contributeProfiles([TOWER_WORKER_PROFILE_DEF]);
}
}
const assembledFlagServices = new WeakSet<IFlagService>();
let assembledOverrideForTests: boolean | undefined;
export function isTowerFeatureAssembled(flags: IFlagService): boolean {
return assembledOverrideForTests ?? assembledFlagServices.has(flags);
}
export function _setTowerFeatureAssembledForTests(value: boolean | undefined): void {
assembledOverrideForTests = value;
}
registerFeature(TowerFeature);
|