@overlordai/server 1.0.1

package/dist/dispatcher/dispatcher.module.js

@@ -0,0 +1,76 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DispatcherModule = void 0;
+const common_1 = require("@nestjs/common");
+const database_module_1 = require("../database/database.module");
+const auth_module_1 = require("../auth/auth.module");
+const notifier_module_1 = require("../notifier/notifier.module");
+const task_repository_1 = require("../database/repositories/task.repository");
+const machine_repository_1 = require("../database/repositories/machine.repository");
+const project_repository_1 = require("../database/repositories/project.repository");
+const project_member_repository_1 = require("../database/repositories/project-member.repository");
+const worker_token_repository_1 = require("../database/repositories/worker-token.repository");
+const audit_log_repository_1 = require("../database/repositories/audit-log.repository");
+const developer_repository_1 = require("../database/repositories/developer.repository");
+const workspace_repository_1 = require("../database/repositories/workspace.repository");
+const crypto_service_1 = require("../common/crypto.service");
+const dedup_service_1 = require("./dedup.service");
+const machine_selector_1 = require("./machine-selector");
+const scheduler_service_1 = require("./scheduler.service");
+const dispatcher_service_1 = require("./dispatcher.service");
+const worker_connection_manager_1 = require("./worker-connection.manager");
+const heartbeat_service_1 = require("./heartbeat.service");
+const capability_service_1 = require("./capability.service");
+const pty_relay_service_1 = require("./pty-relay.service");
+const cleanup_service_1 = require("./cleanup.service");
+const task_log_batcher_1 = require("./task-log-batcher");
+const reconciler_1 = require("./reconciler");
+let DispatcherModule = class DispatcherModule {
+};
+exports.DispatcherModule = DispatcherModule;
+exports.DispatcherModule = DispatcherModule = __decorate([
+    (0, common_1.Module)({
+        imports: [database_module_1.DatabaseModule, auth_module_1.AuthModule, notifier_module_1.NotifierModule],
+        providers: [
+            task_repository_1.TaskRepository,
+            machine_repository_1.MachineRepository,
+            project_repository_1.ProjectRepository,
+            project_member_repository_1.ProjectMemberRepository,
+            worker_token_repository_1.WorkerTokenRepository,
+            audit_log_repository_1.AuditLogRepository,
+            developer_repository_1.DeveloperRepository,
+            workspace_repository_1.WorkspaceRepository,
+            crypto_service_1.CryptoService,
+            dedup_service_1.DedupService,
+            machine_selector_1.MachineSelector,
+            scheduler_service_1.SchedulerService,
+            dispatcher_service_1.DispatcherService,
+            worker_connection_manager_1.WorkerConnectionManager,
+            heartbeat_service_1.HeartbeatService,
+            capability_service_1.CapabilityService,
+            pty_relay_service_1.PtyRelayService,
+            cleanup_service_1.CleanupService,
+            task_log_batcher_1.TaskLogBatcher,
+            reconciler_1.Reconciler,
+        ],
+        exports: [
+            notifier_module_1.NotifierModule,
+            dedup_service_1.DedupService,
+            dispatcher_service_1.DispatcherService,
+            worker_connection_manager_1.WorkerConnectionManager,
+            heartbeat_service_1.HeartbeatService,
+            capability_service_1.CapabilityService,
+            pty_relay_service_1.PtyRelayService,
+            cleanup_service_1.CleanupService,
+            task_log_batcher_1.TaskLogBatcher,
+            reconciler_1.Reconciler,
+        ],
+    })
+], DispatcherModule);
+//# sourceMappingURL=dispatcher.module.js.map

package/dist/dispatcher/dispatcher.module.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"dispatcher.module.js","sourceRoot":"","sources":["../../src/dispatcher/dispatcher.module.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAAwC;AACxC,iEAA6D;AAC7D,qDAAiD;AACjD,iEAA6D;AAC7D,8EAA0E;AAC1E,oFAAgF;AAChF,oFAAgF;AAChF,kGAA6F;AAC7F,8FAAyF;AACzF,wFAAmF;AACnF,wFAAoF;AACpF,wFAAoF;AACpF,6DAAyD;AACzD,mDAA+C;AAC/C,yDAAqD;AACrD,2DAAuD;AACvD,6DAAyD;AACzD,2EAAsE;AACtE,2DAAuD;AACvD,6DAAyD;AACzD,2DAAsD;AACtD,uDAAmD;AACnD,yDAAoD;AACpD,6CAA0C;AAuCnC,IAAM,gBAAgB,GAAtB,MAAM,gBAAgB;CAAG,CAAA;AAAnB,4CAAgB;2BAAhB,gBAAgB;IArC5B,IAAA,eAAM,EAAC;QACN,OAAO,EAAE,CAAC,gCAAc,EAAE,wBAAU,EAAE,gCAAc,CAAC;QACrD,SAAS,EAAE;YACT,gCAAc;YACd,sCAAiB;YACjB,sCAAiB;YACjB,mDAAuB;YACvB,+CAAqB;YACrB,yCAAkB;YAClB,0CAAmB;YACnB,0CAAmB;YACnB,8BAAa;YACb,4BAAY;YACZ,kCAAe;YACf,oCAAgB;YAChB,sCAAiB;YACjB,mDAAuB;YACvB,oCAAgB;YAChB,sCAAiB;YACjB,mCAAe;YACf,gCAAc;YACd,iCAAc;YACd,uBAAU;SACX;QACD,OAAO,EAAE;YACP,gCAAc;YACd,4BAAY;YACZ,sCAAiB;YACjB,mDAAuB;YACvB,oCAAgB;YAChB,sCAAiB;YACjB,mCAAe;YACf,gCAAc;YACd,iCAAc;YACd,uBAAU;SACX;KACF,CAAC;GACW,gBAAgB,CAAG"}

package/dist/dispatcher/dispatcher.service.d.ts

@@ -0,0 +1,134 @@
+import type { Command, Task, PaginatedResult, CreateTaskRequest, WorkerRegisterRequest, WorkerRegisterResponse, WorkerRefreshRequest, WorkerRefreshResponse, WorkerRecoveryRequest, WorkerRecoveryResponse, TaskListQuery } from '@overlordai/protocol';
+import { TaskRepository } from '../database/repositories/task.repository';
+import { MachineRepository } from '../database/repositories/machine.repository';
+import { ProjectRepository } from '../database/repositories/project.repository';
+import { ProjectMemberRepository } from '../database/repositories/project-member.repository';
+import { WorkerTokenRepository } from '../database/repositories/worker-token.repository';
+import { AuditLogRepository } from '../database/repositories/audit-log.repository';
+import { DeveloperRepository } from '../database/repositories/developer.repository';
+import { DatabaseService } from '../database/database.service';
+import { AuthService } from '../auth/auth.service';
+import { CryptoService } from '../common/crypto.service';
+import { RedisService } from '../redis/redis.service';
+import { DedupService } from './dedup.service';
+import { SchedulerService } from './scheduler.service';
+import { WorkerConnectionManager } from './worker-connection.manager';
+import { CleanupService } from './cleanup.service';
+import { WorkspaceRepository } from '../database/repositories/workspace.repository';
+/** Result returned from handleCommand to the adapter/web layer. */
+export interface CommandResult {
+    success: boolean;
+    message?: string;
+    task?: Task;
+    tasks?: Task[];
+    confirmRequired?: {
+        reason: string;
+        message: string;
+        pendingAction: string;
+        taskId?: number;
+        mrUrl?: string;
+        confirmKey: string;
+    };
+}
+export declare class DispatcherService {
+    private readonly taskRepo;
+    private readonly machineRepo;
+    private readonly projectRepo;
+    private readonly projectMemberRepo;
+    private readonly workerTokenRepo;
+    private readonly auditLogRepo;
+    private readonly developerRepo;
+    private readonly databaseService;
+    private readonly authService;
+    private readonly cryptoService;
+    private readonly redis;
+    private readonly dedupService;
+    private readonly schedulerService;
+    private readonly workerConnectionManager;
+    private readonly cleanupService;
+    private readonly workspaceRepo;
+    private readonly logger;
+    constructor(taskRepo: TaskRepository, machineRepo: MachineRepository, projectRepo: ProjectRepository, projectMemberRepo: ProjectMemberRepository, workerTokenRepo: WorkerTokenRepository, auditLogRepo: AuditLogRepository, developerRepo: DeveloperRepository, databaseService: DatabaseService, authService: AuthService, cryptoService: CryptoService, redis: RedisService, dedupService: DedupService, schedulerService: SchedulerService, workerConnectionManager: WorkerConnectionManager, cleanupService: CleanupService, workspaceRepo: WorkspaceRepository);
+    handleCommand(command: Command): Promise<CommandResult>;
+    private handleDevelopCommand;
+    private handleCancelCommand;
+    private handleRetryCommand;
+    private handleConfirmCommand;
+    private handleListCommand;
+    private handleDrainCommand;
+    createTask(request: CreateTaskRequest, createdBy: number, source?: TaskSource): Promise<Task>;
+    private createTaskInternal;
+    cancelTask(taskId: number, cancelledBy: number): Promise<void>;
+    retryTask(taskId: number, retriedBy: number): Promise<Task>;
+    private executeRetry;
+    confirmStage(taskId: number, stageIndex: number, result: string): Promise<void>;
+    getTask(taskId: number): Task | null;
+    listTasks(query: TaskListQuery): PaginatedResult<Task>;
+    registerWorker(request: WorkerRegisterRequest): Promise<WorkerRegisterResponse>;
+    refreshWorkerToken(request: WorkerRefreshRequest, currentJwt: {
+        sub: string;
+        tokenId: number;
+    }): Promise<WorkerRefreshResponse>;
+    recoverWorker(request: WorkerRecoveryRequest): Promise<WorkerRecoveryResponse>;
+    issuePtyToken(taskId: number, aud: 'worker' | 'web'): string;
+    private resolveProject;
+    private parseWorkerToken;
+    private validateProtocolVersion;
+    /**
+     * Resolve a pending confirmation from Redis.
+     * Tries reply-to msgId lookup first, then taskId-based index.
+     * Validates that the confirming user matches the original creator.
+     * Consumes (deletes) the pending confirm on success.
+     */
+    private resolvePendingConfirm;
+    /**
+     * Update task status with CAS retry (up to CAS_MAX_RETRIES).
+     * Returns true if successful, throws ConflictException on exhaustion.
+     */
+    private updateTaskStatusWithRetry;
+    /**
+     * Send cancel frame to Worker and wait for ack within CANCEL_ACK_TIMEOUT_MS.
+     */
+    private sendCancelToWorkerWithAck;
+    /**
+     * Send cancel frame to Worker (fire and forget).
+     */
+    private sendCancelToWorker;
+    /**
+     * Check if the task's branch has an existing open MR/PR on the Git platform.
+     * Returns the MR/PR URL if found, null otherwise.
+     * On timeout or error, returns null (allows retry to proceed).
+     *
+     * Supports GitLab and GitHub APIs. Extracts the project path from the repo URL
+     * and queries the platform API for open merge/pull requests on the branch.
+     */
+    private checkExistingMergeRequest;
+    /**
+     * Query GitLab API for open merge requests on a branch.
+     * GET /api/v4/projects/:id/merge_requests?source_branch=...&state=opened
+     */
+    private checkGitLabMergeRequest;
+    /**
+     * Query GitHub API for open pull requests on a branch.
+     * GET /repos/:owner/:repo/pulls?head=owner:branch&state=open
+     */
+    private checkGitHubPullRequest;
+    /**
+     * Extract the project path (e.g. "group/project") from a git repo URL.
+     * Supports HTTPS and SSH formats.
+     */
+    private extractProjectPath;
+    /**
+     * Extract the base URL (scheme + host) from a git repo URL.
+     */
+    private extractGitBaseUrl;
+}
+/** Source metadata from adapter for task creation. */
+interface TaskSource {
+    platform?: string;
+    msgId?: string;
+    chatId?: string;
+    appId?: string;
+}
+export {};
+//# sourceMappingURL=dispatcher.service.d.ts.map

package/dist/dispatcher/dispatcher.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"dispatcher.service.d.ts","sourceRoot":"","sources":["../../src/dispatcher/dispatcher.service.ts"],"names":[],"mappings":"AAUA,OAAO,KAAK,EACV,OAAO,EACP,IAAI,EAEJ,eAAe,EACf,iBAAiB,EACjB,qBAAqB,EACrB,sBAAsB,EACtB,oBAAoB,EACpB,qBAAqB,EACrB,qBAAqB,EACrB,sBAAsB,EACtB,aAAa,EAEd,MAAM,sBAAsB,CAAC;AAY9B,OAAO,EAAE,cAAc,EAAE,MAAM,0CAA0C,CAAC;AAC1E,OAAO,EAAE,iBAAiB,EAAE,MAAM,6CAA6C,CAAC;AAChF,OAAO,EAAE,iBAAiB,EAAE,MAAM,6CAA6C,CAAC;AAChF,OAAO,EAAE,uBAAuB,EAAE,MAAM,oDAAoD,CAAC;AAC7F,OAAO,EAAE,qBAAqB,EAAE,MAAM,kDAAkD,CAAC;AACzF,OAAO,EAAE,kBAAkB,EAAE,MAAM,+CAA+C,CAAC;AACnF,OAAO,EAAE,mBAAmB,EAAE,MAAM,+CAA+C,CAAC;AACpF,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AACzD,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AAGtD,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC/C,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,EAAE,uBAAuB,EAAE,MAAM,6BAA6B,CAAC;AACtE,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EAAE,mBAAmB,EAAE,MAAM,+CAA+C,CAAC;AAEpF,mEAAmE;AACnE,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,IAAI,CAAC;IACZ,KAAK,CAAC,EAAE,IAAI,EAAE,CAAC;IACf,eAAe,CAAC,EAAE;QAChB,MAAM,EAAE,MAAM,CAAC;QACf,OAAO,EAAE,MAAM,CAAC;QAChB,aAAa,EAAE,MAAM,CAAC;QACtB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,UAAU,EAAE,MAAM,CAAC;KACpB,CAAC;CACH;AAED,qBACa,iBAAiB;IAI1B,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,WAAW;IAC5B,OAAO,CAAC,QAAQ,CAAC,WAAW;IAC5B,OAAO,CAAC,QAAQ,CAAC,iBAAiB;IAClC,OAAO,CAAC,QAAQ,CAAC,eAAe;IAChC,OAAO,CAAC,QAAQ,CAAC,YAAY;IAC7B,OAAO,CAAC,QAAQ,CAAC,aAAa;IAC9B,OAAO,CAAC,QAAQ,CAAC,eAAe;IAChC,OAAO,CAAC,QAAQ,CAAC,WAAW;IAC5B,OAAO,CAAC,QAAQ,CAAC,aAAa;IAC9B,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,YAAY;IAC7B,OAAO,CAAC,QAAQ,CAAC,gBAAgB;IACjC,OAAO,CAAC,QAAQ,CAAC,uBAAuB;IACxC,OAAO,CAAC,QAAQ,CAAC,cAAc;IAC/B,OAAO,CAAC,QAAQ,CAAC,aAAa;IAlBhC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAsC;gBAG1C,QAAQ,EAAE,cAAc,EACxB,WAAW,EAAE,iBAAiB,EAC9B,WAAW,EAAE,iBAAiB,EAC9B,iBAAiB,EAAE,uBAAuB,EAC1C,eAAe,EAAE,qBAAqB,EACtC,YAAY,EAAE,kBAAkB,EAChC,aAAa,EAAE,mBAAmB,EAClC,eAAe,EAAE,eAAe,EAChC,WAAW,EAAE,WAAW,EACxB,aAAa,EAAE,aAAa,EAC5B,KAAK,EAAE,YAAY,EACnB,YAAY,EAAE,YAAY,EAC1B,gBAAgB,EAAE,gBAAgB,EAClC,uBAAuB,EAAE,uBAAuB,EAChD,cAAc,EAAE,cAAc,EAC9B,aAAa,EAAE,mBAAmB;IAO/C,aAAa,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,aAAa,CAAC;YA4E/C,oBAAoB;YAoCpB,mBAAmB;YAUnB,kBAAkB;YAuBlB,oBAAoB;IAsClC,OAAO,CAAC,iBAAiB;IAgBzB,OAAO,CAAC,kBAAkB;IAiCpB,UAAU,CACd,OAAO,EAAE,iBAAiB,EAC1B,SAAS,EAAE,MAAM,EACjB,MAAM,CAAC,EAAE,UAAU,GAClB,OAAO,CAAC,IAAI,CAAC;YAIF,kBAAkB;IA0I1B,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA+G9D,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;YA8DnD,YAAY;IAgDpB,YAAY,CAChB,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,IAAI,CAAC;IAoDhB,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,GAAG,IAAI;IAIpC,SAAS,CAAC,KAAK,EAAE,aAAa,GAAG,eAAe,CAAC,IAAI,CAAC;IAQhD,cAAc,CAClB,OAAO,EAAE,qBAAqB,GAC7B,OAAO,CAAC,sBAAsB,CAAC;IAwF5B,kBAAkB,CACtB,OAAO,EAAE,oBAAoB,EAC7B,UAAU,EAAE;QAAE,GAAG,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,GAC3C,OAAO,CAAC,qBAAqB,CAAC;IAmC3B,aAAa,CACjB,OAAO,EAAE,qBAAqB,GAC7B,OAAO,CAAC,sBAAsB,CAAC;IAqFlC,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,QAAQ,GAAG,KAAK,GAAG,MAAM;IAa5D,OAAO,CAAC,cAAc;IActB,OAAO,CAAC,gBAAgB;IAexB,OAAO,CAAC,uBAAuB;IAiB/B;;;;;OAKG;YACW,qBAAqB;IAsCnC;;;OAGG;IACH,OAAO,CAAC,yBAAyB;IAkCjC;;OAEG;YACW,yBAAyB;IAevC;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAS1B;;;;;;;OAOG;YACW,yBAAyB;IA+EvC;;;OAGG;YACW,uBAAuB;IAoCrC;;;OAGG;YACW,sBAAsB;IAyCpC;;;OAGG;IACH,OAAO,CAAC,kBAAkB;IAgB1B;;OAEG;IACH,OAAO,CAAC,iBAAiB;CAa1B;AAED,sDAAsD;AACtD,UAAU,UAAU;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB"}