@overlordai/server 1.0.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/database/migrations/001-init-schema.sql +226 -0
- package/database/migrations/002-add-indexes.sql +17 -0
- package/database/migrations/003-add-settings-table.sql +4 -0
- package/database/migrations/004-add-developer-id-index.sql +5 -0
- package/dist/adapters/adapter.interface.d.ts +41 -0
- package/dist/adapters/adapter.interface.d.ts.map +1 -0
- package/dist/adapters/adapter.interface.js +6 -0
- package/dist/adapters/adapter.interface.js.map +1 -0
- package/dist/adapters/adapter.module.d.ts +3 -0
- package/dist/adapters/adapter.module.d.ts.map +1 -0
- package/dist/adapters/adapter.module.js +54 -0
- package/dist/adapters/adapter.module.js.map +1 -0
- package/dist/adapters/adapter.registry.d.ts +19 -0
- package/dist/adapters/adapter.registry.d.ts.map +1 -0
- package/dist/adapters/adapter.registry.js +51 -0
- package/dist/adapters/adapter.registry.js.map +1 -0
- package/dist/adapters/lark/lark-card.builder.d.ts +48 -0
- package/dist/adapters/lark/lark-card.builder.d.ts.map +1 -0
- package/dist/adapters/lark/lark-card.builder.js +259 -0
- package/dist/adapters/lark/lark-card.builder.js.map +1 -0
- package/dist/adapters/lark/lark-message.parser.d.ts +51 -0
- package/dist/adapters/lark/lark-message.parser.d.ts.map +1 -0
- package/dist/adapters/lark/lark-message.parser.js +189 -0
- package/dist/adapters/lark/lark-message.parser.js.map +1 -0
- package/dist/adapters/lark/lark-signature.d.ts +13 -0
- package/dist/adapters/lark/lark-signature.d.ts.map +1 -0
- package/dist/adapters/lark/lark-signature.js +58 -0
- package/dist/adapters/lark/lark-signature.js.map +1 -0
- package/dist/adapters/lark/lark.adapter.d.ts +65 -0
- package/dist/adapters/lark/lark.adapter.d.ts.map +1 -0
- package/dist/adapters/lark/lark.adapter.js +565 -0
- package/dist/adapters/lark/lark.adapter.js.map +1 -0
- package/dist/adapters/lark/lark.controller.d.ts +21 -0
- package/dist/adapters/lark/lark.controller.d.ts.map +1 -0
- package/dist/adapters/lark/lark.controller.js +120 -0
- package/dist/adapters/lark/lark.controller.js.map +1 -0
- package/dist/adapters/slack/slack.adapter.d.ts +19 -0
- package/dist/adapters/slack/slack.adapter.d.ts.map +1 -0
- package/dist/adapters/slack/slack.adapter.js +42 -0
- package/dist/adapters/slack/slack.adapter.js.map +1 -0
- package/dist/app.module.d.ts +5 -0
- package/dist/app.module.d.ts.map +1 -0
- package/dist/app.module.js +48 -0
- package/dist/app.module.js.map +1 -0
- package/dist/auth/auth.controller.d.ts +15 -0
- package/dist/auth/auth.controller.d.ts.map +1 -0
- package/dist/auth/auth.controller.js +67 -0
- package/dist/auth/auth.controller.js.map +1 -0
- package/dist/auth/auth.module.d.ts +3 -0
- package/dist/auth/auth.module.d.ts.map +1 -0
- package/dist/auth/auth.module.js +46 -0
- package/dist/auth/auth.module.js.map +1 -0
- package/dist/auth/auth.service.d.ts +62 -0
- package/dist/auth/auth.service.d.ts.map +1 -0
- package/dist/auth/auth.service.js +307 -0
- package/dist/auth/auth.service.js.map +1 -0
- package/dist/auth/decorators/allow-totp-setup.decorator.d.ts +3 -0
- package/dist/auth/decorators/allow-totp-setup.decorator.d.ts.map +1 -0
- package/dist/auth/decorators/allow-totp-setup.decorator.js +8 -0
- package/dist/auth/decorators/allow-totp-setup.decorator.js.map +1 -0
- package/dist/auth/decorators/project-roles.decorator.d.ts +4 -0
- package/dist/auth/decorators/project-roles.decorator.d.ts.map +1 -0
- package/dist/auth/decorators/project-roles.decorator.js +8 -0
- package/dist/auth/decorators/project-roles.decorator.js.map +1 -0
- package/dist/auth/decorators/roles.decorator.d.ts +4 -0
- package/dist/auth/decorators/roles.decorator.d.ts.map +1 -0
- package/dist/auth/decorators/roles.decorator.js +8 -0
- package/dist/auth/decorators/roles.decorator.js.map +1 -0
- package/dist/auth/extract-user.middleware.d.ts +21 -0
- package/dist/auth/extract-user.middleware.d.ts.map +1 -0
- package/dist/auth/extract-user.middleware.js +57 -0
- package/dist/auth/extract-user.middleware.js.map +1 -0
- package/dist/auth/guards/jwt-auth.guard.d.ts +14 -0
- package/dist/auth/guards/jwt-auth.guard.d.ts.map +1 -0
- package/dist/auth/guards/jwt-auth.guard.js +139 -0
- package/dist/auth/guards/jwt-auth.guard.js.map +1 -0
- package/dist/auth/guards/project-role.guard.d.ts +10 -0
- package/dist/auth/guards/project-role.guard.d.ts.map +1 -0
- package/dist/auth/guards/project-role.guard.js +72 -0
- package/dist/auth/guards/project-role.guard.js.map +1 -0
- package/dist/auth/guards/roles.guard.d.ts +8 -0
- package/dist/auth/guards/roles.guard.d.ts.map +1 -0
- package/dist/auth/guards/roles.guard.js +56 -0
- package/dist/auth/guards/roles.guard.js.map +1 -0
- package/dist/auth/jwt.strategy.d.ts +23 -0
- package/dist/auth/jwt.strategy.d.ts.map +1 -0
- package/dist/auth/jwt.strategy.js +49 -0
- package/dist/auth/jwt.strategy.js.map +1 -0
- package/dist/common/crypto.service.d.ts +31 -0
- package/dist/common/crypto.service.d.ts.map +1 -0
- package/dist/common/crypto.service.js +120 -0
- package/dist/common/crypto.service.js.map +1 -0
- package/dist/common/error-filter.d.ts +6 -0
- package/dist/common/error-filter.d.ts.map +1 -0
- package/dist/common/error-filter.js +78 -0
- package/dist/common/error-filter.js.map +1 -0
- package/dist/common/health.controller.d.ts +13 -0
- package/dist/common/health.controller.d.ts.map +1 -0
- package/dist/common/health.controller.js +75 -0
- package/dist/common/health.controller.js.map +1 -0
- package/dist/common/logger.service.d.ts +11 -0
- package/dist/common/logger.service.d.ts.map +1 -0
- package/dist/common/logger.service.js +48 -0
- package/dist/common/logger.service.js.map +1 -0
- package/dist/common/pagination.d.ts +18 -0
- package/dist/common/pagination.d.ts.map +1 -0
- package/dist/common/pagination.js +39 -0
- package/dist/common/pagination.js.map +1 -0
- package/dist/common/rate-limit.guard.d.ts +48 -0
- package/dist/common/rate-limit.guard.d.ts.map +1 -0
- package/dist/common/rate-limit.guard.js +129 -0
- package/dist/common/rate-limit.guard.js.map +1 -0
- package/dist/common/sensitive-filter.d.ts +7 -0
- package/dist/common/sensitive-filter.d.ts.map +1 -0
- package/dist/common/sensitive-filter.js +20 -0
- package/dist/common/sensitive-filter.js.map +1 -0
- package/dist/database/database.module.d.ts +3 -0
- package/dist/database/database.module.d.ts.map +1 -0
- package/dist/database/database.module.js +22 -0
- package/dist/database/database.module.js.map +1 -0
- package/dist/database/database.service.d.ts +13 -0
- package/dist/database/database.service.d.ts.map +1 -0
- package/dist/database/database.service.js +107 -0
- package/dist/database/database.service.js.map +1 -0
- package/dist/database/migration-runner.d.ts +5 -0
- package/dist/database/migration-runner.d.ts.map +1 -0
- package/dist/database/migration-runner.js +86 -0
- package/dist/database/migration-runner.js.map +1 -0
- package/dist/database/repositories/audit-log.repository.d.ts +29 -0
- package/dist/database/repositories/audit-log.repository.d.ts.map +1 -0
- package/dist/database/repositories/audit-log.repository.js +80 -0
- package/dist/database/repositories/audit-log.repository.js.map +1 -0
- package/dist/database/repositories/bot.repository.d.ts +67 -0
- package/dist/database/repositories/bot.repository.d.ts.map +1 -0
- package/dist/database/repositories/bot.repository.js +133 -0
- package/dist/database/repositories/bot.repository.js.map +1 -0
- package/dist/database/repositories/developer-token.repository.d.ts +40 -0
- package/dist/database/repositories/developer-token.repository.d.ts.map +1 -0
- package/dist/database/repositories/developer-token.repository.js +84 -0
- package/dist/database/repositories/developer-token.repository.js.map +1 -0
- package/dist/database/repositories/developer.repository.d.ts +25 -0
- package/dist/database/repositories/developer.repository.d.ts.map +1 -0
- package/dist/database/repositories/developer.repository.js +139 -0
- package/dist/database/repositories/developer.repository.js.map +1 -0
- package/dist/database/repositories/machine.repository.d.ts +39 -0
- package/dist/database/repositories/machine.repository.d.ts.map +1 -0
- package/dist/database/repositories/machine.repository.js +176 -0
- package/dist/database/repositories/machine.repository.js.map +1 -0
- package/dist/database/repositories/notification.repository.d.ts +19 -0
- package/dist/database/repositories/notification.repository.d.ts.map +1 -0
- package/dist/database/repositories/notification.repository.js +94 -0
- package/dist/database/repositories/notification.repository.js.map +1 -0
- package/dist/database/repositories/project-member.repository.d.ts +30 -0
- package/dist/database/repositories/project-member.repository.d.ts.map +1 -0
- package/dist/database/repositories/project-member.repository.js +75 -0
- package/dist/database/repositories/project-member.repository.js.map +1 -0
- package/dist/database/repositories/project.repository.d.ts +24 -0
- package/dist/database/repositories/project.repository.d.ts.map +1 -0
- package/dist/database/repositories/project.repository.js +154 -0
- package/dist/database/repositories/project.repository.js.map +1 -0
- package/dist/database/repositories/session.repository.d.ts +19 -0
- package/dist/database/repositories/session.repository.d.ts.map +1 -0
- package/dist/database/repositories/session.repository.js +117 -0
- package/dist/database/repositories/session.repository.js.map +1 -0
- package/dist/database/repositories/task.repository.d.ts +37 -0
- package/dist/database/repositories/task.repository.d.ts.map +1 -0
- package/dist/database/repositories/task.repository.js +229 -0
- package/dist/database/repositories/task.repository.js.map +1 -0
- package/dist/database/repositories/worker-token.repository.d.ts +20 -0
- package/dist/database/repositories/worker-token.repository.d.ts.map +1 -0
- package/dist/database/repositories/worker-token.repository.js +94 -0
- package/dist/database/repositories/worker-token.repository.js.map +1 -0
- package/dist/database/repositories/workspace.repository.d.ts +19 -0
- package/dist/database/repositories/workspace.repository.d.ts.map +1 -0
- package/dist/database/repositories/workspace.repository.js +82 -0
- package/dist/database/repositories/workspace.repository.js.map +1 -0
- package/dist/dispatcher/capability.service.d.ts +50 -0
- package/dist/dispatcher/capability.service.d.ts.map +1 -0
- package/dist/dispatcher/capability.service.js +159 -0
- package/dist/dispatcher/capability.service.js.map +1 -0
- package/dist/dispatcher/cleanup.service.d.ts +23 -0
- package/dist/dispatcher/cleanup.service.d.ts.map +1 -0
- package/dist/dispatcher/cleanup.service.js +107 -0
- package/dist/dispatcher/cleanup.service.js.map +1 -0
- package/dist/dispatcher/dedup.service.d.ts +48 -0
- package/dist/dispatcher/dedup.service.d.ts.map +1 -0
- package/dist/dispatcher/dedup.service.js +189 -0
- package/dist/dispatcher/dedup.service.js.map +1 -0
- package/dist/dispatcher/dispatcher.module.d.ts +3 -0
- package/dist/dispatcher/dispatcher.module.d.ts.map +1 -0
- package/dist/dispatcher/dispatcher.module.js +76 -0
- package/dist/dispatcher/dispatcher.module.js.map +1 -0
- package/dist/dispatcher/dispatcher.service.d.ts +134 -0
- package/dist/dispatcher/dispatcher.service.d.ts.map +1 -0
- package/dist/dispatcher/dispatcher.service.js +1034 -0
- package/dist/dispatcher/dispatcher.service.js.map +1 -0
- package/dist/dispatcher/heartbeat.service.d.ts +50 -0
- package/dist/dispatcher/heartbeat.service.d.ts.map +1 -0
- package/dist/dispatcher/heartbeat.service.js +154 -0
- package/dist/dispatcher/heartbeat.service.js.map +1 -0
- package/dist/dispatcher/machine-selector.d.ts +18 -0
- package/dist/dispatcher/machine-selector.d.ts.map +1 -0
- package/dist/dispatcher/machine-selector.js +144 -0
- package/dist/dispatcher/machine-selector.js.map +1 -0
- package/dist/dispatcher/pty-relay.service.d.ts +75 -0
- package/dist/dispatcher/pty-relay.service.d.ts.map +1 -0
- package/dist/dispatcher/pty-relay.service.js +404 -0
- package/dist/dispatcher/pty-relay.service.js.map +1 -0
- package/dist/dispatcher/reconciler.d.ts +39 -0
- package/dist/dispatcher/reconciler.d.ts.map +1 -0
- package/dist/dispatcher/reconciler.js +556 -0
- package/dist/dispatcher/reconciler.js.map +1 -0
- package/dist/dispatcher/scheduler.service.d.ts +50 -0
- package/dist/dispatcher/scheduler.service.d.ts.map +1 -0
- package/dist/dispatcher/scheduler.service.js +287 -0
- package/dist/dispatcher/scheduler.service.js.map +1 -0
- package/dist/dispatcher/state-machine.d.ts +16 -0
- package/dist/dispatcher/state-machine.d.ts.map +1 -0
- package/dist/dispatcher/state-machine.js +77 -0
- package/dist/dispatcher/state-machine.js.map +1 -0
- package/dist/dispatcher/task-log-batcher.d.ts +50 -0
- package/dist/dispatcher/task-log-batcher.d.ts.map +1 -0
- package/dist/dispatcher/task-log-batcher.js +184 -0
- package/dist/dispatcher/task-log-batcher.js.map +1 -0
- package/dist/dispatcher/worker-connection.manager.d.ts +49 -0
- package/dist/dispatcher/worker-connection.manager.d.ts.map +1 -0
- package/dist/dispatcher/worker-connection.manager.js +128 -0
- package/dist/dispatcher/worker-connection.manager.js.map +1 -0
- package/dist/main.d.ts +2 -0
- package/dist/main.d.ts.map +1 -0
- package/dist/main.js +85 -0
- package/dist/main.js.map +1 -0
- package/dist/notifier/debouncer.d.ts +39 -0
- package/dist/notifier/debouncer.d.ts.map +1 -0
- package/dist/notifier/debouncer.js +123 -0
- package/dist/notifier/debouncer.js.map +1 -0
- package/dist/notifier/notification-consumer.d.ts +88 -0
- package/dist/notifier/notification-consumer.d.ts.map +1 -0
- package/dist/notifier/notification-consumer.js +186 -0
- package/dist/notifier/notification-consumer.js.map +1 -0
- package/dist/notifier/notifier.module.d.ts +9 -0
- package/dist/notifier/notifier.module.d.ts.map +1 -0
- package/dist/notifier/notifier.module.js +58 -0
- package/dist/notifier/notifier.module.js.map +1 -0
- package/dist/notifier/notifier.service.d.ts +40 -0
- package/dist/notifier/notifier.service.d.ts.map +1 -0
- package/dist/notifier/notifier.service.js +191 -0
- package/dist/notifier/notifier.service.js.map +1 -0
- package/dist/notifier/template.service.d.ts +42 -0
- package/dist/notifier/template.service.d.ts.map +1 -0
- package/dist/notifier/template.service.js +201 -0
- package/dist/notifier/template.service.js.map +1 -0
- package/dist/redis/redis.module.d.ts +3 -0
- package/dist/redis/redis.module.d.ts.map +1 -0
- package/dist/redis/redis.module.js +22 -0
- package/dist/redis/redis.module.js.map +1 -0
- package/dist/redis/redis.service.d.ts +19 -0
- package/dist/redis/redis.service.d.ts.map +1 -0
- package/dist/redis/redis.service.js +69 -0
- package/dist/redis/redis.service.js.map +1 -0
- package/dist/web/admin/admin-audit.controller.d.ts +7 -0
- package/dist/web/admin/admin-audit.controller.d.ts.map +1 -0
- package/dist/web/admin/admin-audit.controller.js +53 -0
- package/dist/web/admin/admin-audit.controller.js.map +1 -0
- package/dist/web/admin/admin-bot.controller.d.ts +79 -0
- package/dist/web/admin/admin-bot.controller.d.ts.map +1 -0
- package/dist/web/admin/admin-bot.controller.js +193 -0
- package/dist/web/admin/admin-bot.controller.js.map +1 -0
- package/dist/web/admin/admin-developer.controller.d.ts +52 -0
- package/dist/web/admin/admin-developer.controller.d.ts.map +1 -0
- package/dist/web/admin/admin-developer.controller.js +160 -0
- package/dist/web/admin/admin-developer.controller.js.map +1 -0
- package/dist/web/admin/admin-machine.controller.d.ts +64 -0
- package/dist/web/admin/admin-machine.controller.d.ts.map +1 -0
- package/dist/web/admin/admin-machine.controller.js +111 -0
- package/dist/web/admin/admin-machine.controller.js.map +1 -0
- package/dist/web/admin/admin-project.controller.d.ts +45 -0
- package/dist/web/admin/admin-project.controller.d.ts.map +1 -0
- package/dist/web/admin/admin-project.controller.js +207 -0
- package/dist/web/admin/admin-project.controller.js.map +1 -0
- package/dist/web/admin/admin-settings.controller.d.ts +18 -0
- package/dist/web/admin/admin-settings.controller.d.ts.map +1 -0
- package/dist/web/admin/admin-settings.controller.js +93 -0
- package/dist/web/admin/admin-settings.controller.js.map +1 -0
- package/dist/web/admin/admin-token.controller.d.ts +45 -0
- package/dist/web/admin/admin-token.controller.d.ts.map +1 -0
- package/dist/web/admin/admin-token.controller.js +182 -0
- package/dist/web/admin/admin-token.controller.js.map +1 -0
- package/dist/web/dashboard.controller.d.ts +16 -0
- package/dist/web/dashboard.controller.d.ts.map +1 -0
- package/dist/web/dashboard.controller.js +78 -0
- package/dist/web/dashboard.controller.js.map +1 -0
- package/dist/web/dashboard.service.d.ts +39 -0
- package/dist/web/dashboard.service.d.ts.map +1 -0
- package/dist/web/dashboard.service.js +234 -0
- package/dist/web/dashboard.service.js.map +1 -0
- package/dist/web/interaction.service.d.ts +42 -0
- package/dist/web/interaction.service.d.ts.map +1 -0
- package/dist/web/interaction.service.js +102 -0
- package/dist/web/interaction.service.js.map +1 -0
- package/dist/web/machine.controller.d.ts +102 -0
- package/dist/web/machine.controller.d.ts.map +1 -0
- package/dist/web/machine.controller.js +121 -0
- package/dist/web/machine.controller.js.map +1 -0
- package/dist/web/notification.controller.d.ts +22 -0
- package/dist/web/notification.controller.d.ts.map +1 -0
- package/dist/web/notification.controller.js +70 -0
- package/dist/web/notification.controller.js.map +1 -0
- package/dist/web/profile.controller.d.ts +70 -0
- package/dist/web/profile.controller.d.ts.map +1 -0
- package/dist/web/profile.controller.js +262 -0
- package/dist/web/profile.controller.js.map +1 -0
- package/dist/web/project.controller.d.ts +8 -0
- package/dist/web/project.controller.d.ts.map +1 -0
- package/dist/web/project.controller.js +54 -0
- package/dist/web/project.controller.js.map +1 -0
- package/dist/web/pty.gateway.d.ts +32 -0
- package/dist/web/pty.gateway.d.ts.map +1 -0
- package/dist/web/pty.gateway.js +358 -0
- package/dist/web/pty.gateway.js.map +1 -0
- package/dist/web/search.service.d.ts +34 -0
- package/dist/web/search.service.d.ts.map +1 -0
- package/dist/web/search.service.js +106 -0
- package/dist/web/search.service.js.map +1 -0
- package/dist/web/task.controller.d.ts +54 -0
- package/dist/web/task.controller.d.ts.map +1 -0
- package/dist/web/task.controller.js +266 -0
- package/dist/web/task.controller.js.map +1 -0
- package/dist/web/web.module.d.ts +3 -0
- package/dist/web/web.module.d.ts.map +1 -0
- package/dist/web/web.module.js +97 -0
- package/dist/web/web.module.js.map +1 -0
- package/dist/web/worker-channel.gateway.d.ts +45 -0
- package/dist/web/worker-channel.gateway.d.ts.map +1 -0
- package/dist/web/worker-channel.gateway.js +283 -0
- package/dist/web/worker-channel.gateway.js.map +1 -0
- package/dist/web/worker.controller.d.ts +14 -0
- package/dist/web/worker.controller.d.ts.map +1 -0
- package/dist/web/worker.controller.js +73 -0
- package/dist/web/worker.controller.js.map +1 -0
- package/dist/web/workspace.controller.d.ts +109 -0
- package/dist/web/workspace.controller.d.ts.map +1 -0
- package/dist/web/workspace.controller.js +386 -0
- package/dist/web/workspace.controller.js.map +1 -0
- package/package.json +61 -0
|
@@ -0,0 +1,283 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
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;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
var __metadata = (this && this.__metadata) || function (k, v) {
|
|
9
|
+
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
|
10
|
+
};
|
|
11
|
+
var WorkerChannelGateway_1;
|
|
12
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
13
|
+
exports.WorkerChannelGateway = void 0;
|
|
14
|
+
const websockets_1 = require("@nestjs/websockets");
|
|
15
|
+
const common_1 = require("@nestjs/common");
|
|
16
|
+
const ws_1 = require("ws");
|
|
17
|
+
const protocol_1 = require("@overlordai/protocol");
|
|
18
|
+
const auth_service_1 = require("../auth/auth.service");
|
|
19
|
+
const worker_connection_manager_1 = require("../dispatcher/worker-connection.manager");
|
|
20
|
+
const heartbeat_service_1 = require("../dispatcher/heartbeat.service");
|
|
21
|
+
const reconciler_1 = require("../dispatcher/reconciler");
|
|
22
|
+
const task_log_batcher_1 = require("../dispatcher/task-log-batcher");
|
|
23
|
+
const dedup_service_1 = require("../dispatcher/dedup.service");
|
|
24
|
+
const machine_repository_1 = require("../database/repositories/machine.repository");
|
|
25
|
+
const task_repository_1 = require("../database/repositories/task.repository");
|
|
26
|
+
const worker_token_repository_1 = require("../database/repositories/worker-token.repository");
|
|
27
|
+
const notifier_service_1 = require("../notifier/notifier.service");
|
|
28
|
+
const protocol_2 = require("@overlordai/protocol");
|
|
29
|
+
let WorkerChannelGateway = WorkerChannelGateway_1 = class WorkerChannelGateway {
|
|
30
|
+
authService;
|
|
31
|
+
workerConnectionManager;
|
|
32
|
+
heartbeatService;
|
|
33
|
+
reconciler;
|
|
34
|
+
taskLogBatcher;
|
|
35
|
+
dedupService;
|
|
36
|
+
machineRepo;
|
|
37
|
+
taskRepo;
|
|
38
|
+
workerTokenRepo;
|
|
39
|
+
notifierService;
|
|
40
|
+
logger = new common_1.Logger(WorkerChannelGateway_1.name);
|
|
41
|
+
authedSockets = new Map();
|
|
42
|
+
AUTH_TIMEOUT_MS = 10_000;
|
|
43
|
+
constructor(authService, workerConnectionManager, heartbeatService, reconciler, taskLogBatcher, dedupService, machineRepo, taskRepo, workerTokenRepo, notifierService) {
|
|
44
|
+
this.authService = authService;
|
|
45
|
+
this.workerConnectionManager = workerConnectionManager;
|
|
46
|
+
this.heartbeatService = heartbeatService;
|
|
47
|
+
this.reconciler = reconciler;
|
|
48
|
+
this.taskLogBatcher = taskLogBatcher;
|
|
49
|
+
this.dedupService = dedupService;
|
|
50
|
+
this.machineRepo = machineRepo;
|
|
51
|
+
this.taskRepo = taskRepo;
|
|
52
|
+
this.workerTokenRepo = workerTokenRepo;
|
|
53
|
+
this.notifierService = notifierService;
|
|
54
|
+
}
|
|
55
|
+
handleConnection(client, _req) {
|
|
56
|
+
this.logger.debug('New worker channel connection, awaiting auth frame');
|
|
57
|
+
// Set auth timeout - disconnect if no auth frame within 10s
|
|
58
|
+
const authTimeout = setTimeout(() => {
|
|
59
|
+
if (!this.authedSockets.has(client)) {
|
|
60
|
+
this.logger.warn('Worker channel auth timeout, closing connection');
|
|
61
|
+
client.close(4001, 'Auth timeout');
|
|
62
|
+
}
|
|
63
|
+
}, this.AUTH_TIMEOUT_MS);
|
|
64
|
+
client.on('message', (rawData) => {
|
|
65
|
+
let frame;
|
|
66
|
+
try {
|
|
67
|
+
const str = typeof rawData === 'string' ? rawData : rawData.toString('utf8');
|
|
68
|
+
frame = JSON.parse(str);
|
|
69
|
+
}
|
|
70
|
+
catch {
|
|
71
|
+
this.logger.warn('Invalid JSON from worker channel');
|
|
72
|
+
client.close(4002, 'Invalid JSON');
|
|
73
|
+
return;
|
|
74
|
+
}
|
|
75
|
+
const authed = this.authedSockets.get(client);
|
|
76
|
+
if (!authed) {
|
|
77
|
+
// First message must be auth
|
|
78
|
+
if (frame.type !== 'auth') {
|
|
79
|
+
this.logger.warn(`Expected auth frame, got ${frame.type}; closing`);
|
|
80
|
+
client.close(4003, 'Expected auth frame');
|
|
81
|
+
return;
|
|
82
|
+
}
|
|
83
|
+
clearTimeout(authTimeout);
|
|
84
|
+
this.handleAuth(client, frame);
|
|
85
|
+
return;
|
|
86
|
+
}
|
|
87
|
+
// Dispatch authenticated frames
|
|
88
|
+
switch (frame.type) {
|
|
89
|
+
case 'heartbeat':
|
|
90
|
+
this.handleHeartbeat(authed, frame);
|
|
91
|
+
break;
|
|
92
|
+
case 'ack':
|
|
93
|
+
this.handleAck(frame);
|
|
94
|
+
break;
|
|
95
|
+
case 'reconnect':
|
|
96
|
+
void this.handleReconnect(authed, frame);
|
|
97
|
+
break;
|
|
98
|
+
case 'progress':
|
|
99
|
+
this.handleProgress(frame);
|
|
100
|
+
break;
|
|
101
|
+
case 'recovery_secret_ack':
|
|
102
|
+
this.handleRecoverySecretAck(authed, frame);
|
|
103
|
+
break;
|
|
104
|
+
case 'stage_confirm_request':
|
|
105
|
+
void this.handleStageConfirmRequest(frame);
|
|
106
|
+
break;
|
|
107
|
+
default:
|
|
108
|
+
this.logger.debug(`Unknown frame type: ${frame.type}`);
|
|
109
|
+
}
|
|
110
|
+
});
|
|
111
|
+
}
|
|
112
|
+
handleDisconnect(client) {
|
|
113
|
+
const authed = this.authedSockets.get(client);
|
|
114
|
+
if (authed) {
|
|
115
|
+
clearInterval(authed.revalidationTimer);
|
|
116
|
+
this.workerConnectionManager.unregister(authed.machineId);
|
|
117
|
+
this.authedSockets.delete(client);
|
|
118
|
+
this.logger.log(`Worker ${authed.machineId} disconnected`);
|
|
119
|
+
}
|
|
120
|
+
}
|
|
121
|
+
handleAuth(client, frame) {
|
|
122
|
+
// Validate JWT
|
|
123
|
+
let payload;
|
|
124
|
+
try {
|
|
125
|
+
payload = this.authService.verifyWorkerJwt(frame.token);
|
|
126
|
+
}
|
|
127
|
+
catch {
|
|
128
|
+
this.logger.warn('Worker auth failed: invalid JWT');
|
|
129
|
+
client.close(4004, 'Invalid JWT');
|
|
130
|
+
return;
|
|
131
|
+
}
|
|
132
|
+
// Validate protocol version (major must match)
|
|
133
|
+
const clientMajor = frame.protocolVersion.split('.')[0];
|
|
134
|
+
const serverMajor = protocol_1.PROTOCOL_VERSION.split('.')[0];
|
|
135
|
+
if (clientMajor !== serverMajor) {
|
|
136
|
+
this.logger.warn(`Protocol version mismatch: client=${frame.protocolVersion}, server=${protocol_1.PROTOCOL_VERSION}`);
|
|
137
|
+
client.close(4005, 'Protocol version mismatch');
|
|
138
|
+
return;
|
|
139
|
+
}
|
|
140
|
+
// Validate token against machine record
|
|
141
|
+
const machine = this.machineRepo.findById(payload.sub);
|
|
142
|
+
if (!machine) {
|
|
143
|
+
this.logger.warn(`Auth failed: machine ${payload.sub} not found`);
|
|
144
|
+
client.close(4006, 'Machine not found');
|
|
145
|
+
return;
|
|
146
|
+
}
|
|
147
|
+
if (machine.tokenId !== payload.tokenId) {
|
|
148
|
+
this.logger.warn(`Auth failed: tokenId mismatch for machine ${payload.sub}`);
|
|
149
|
+
client.close(4007, 'Token ID mismatch');
|
|
150
|
+
return;
|
|
151
|
+
}
|
|
152
|
+
// Validate worker token is active
|
|
153
|
+
const workerToken = this.workerTokenRepo.findById(payload.tokenId);
|
|
154
|
+
if (!workerToken || workerToken.status !== 'active') {
|
|
155
|
+
this.logger.warn(`Auth failed: worker token ${payload.tokenId} not active`);
|
|
156
|
+
client.close(4008, 'Worker token not active');
|
|
157
|
+
return;
|
|
158
|
+
}
|
|
159
|
+
// Register connection
|
|
160
|
+
this.workerConnectionManager.register(payload.sub, client);
|
|
161
|
+
// Mark machine online
|
|
162
|
+
this.machineRepo.updateStatus(payload.sub, protocol_1.MachineStatus.ONLINE);
|
|
163
|
+
// Set up periodic revalidation
|
|
164
|
+
const revalidationTimer = setInterval(() => {
|
|
165
|
+
this.revalidateWorkerJwt(client, payload.sub, payload.tokenId);
|
|
166
|
+
}, protocol_1.TOKEN_REVALIDATION_INTERVAL_MS);
|
|
167
|
+
const authedSocket = {
|
|
168
|
+
ws: client,
|
|
169
|
+
machineId: payload.sub,
|
|
170
|
+
tokenId: payload.tokenId,
|
|
171
|
+
authedAt: Date.now(),
|
|
172
|
+
revalidationTimer,
|
|
173
|
+
};
|
|
174
|
+
this.authedSockets.set(client, authedSocket);
|
|
175
|
+
this.logger.log(`Worker ${payload.sub} authenticated`);
|
|
176
|
+
}
|
|
177
|
+
handleHeartbeat(authed, frame) {
|
|
178
|
+
this.heartbeatService.processHeartbeat(authed.machineId, {
|
|
179
|
+
cpuUsage: frame.metrics.cpuUsage,
|
|
180
|
+
memoryUsage: frame.metrics.memoryUsage,
|
|
181
|
+
diskUsage: frame.metrics.diskUsage,
|
|
182
|
+
activeSlots: frame.metrics.activeSlots,
|
|
183
|
+
}, frame.capabilities);
|
|
184
|
+
// Send heartbeat_ack
|
|
185
|
+
const ack = {
|
|
186
|
+
type: 'heartbeat_ack',
|
|
187
|
+
timestamp: frame.timestamp,
|
|
188
|
+
serverTime: new Date().toISOString(),
|
|
189
|
+
};
|
|
190
|
+
if (authed.ws.readyState === ws_1.WebSocket.OPEN) {
|
|
191
|
+
authed.ws.send(JSON.stringify(ack));
|
|
192
|
+
}
|
|
193
|
+
}
|
|
194
|
+
handleAck(frame) {
|
|
195
|
+
this.workerConnectionManager.handleAck(frame.msgId, frame.payload);
|
|
196
|
+
}
|
|
197
|
+
async handleReconnect(authed, frame) {
|
|
198
|
+
this.logger.log(`Worker ${authed.machineId} reconnecting with ${frame.tasks.length} task(s)`);
|
|
199
|
+
// Update machine status to online
|
|
200
|
+
this.machineRepo.updateStatus(authed.machineId, protocol_1.MachineStatus.ONLINE);
|
|
201
|
+
// Forward to reconciler
|
|
202
|
+
await this.reconciler.handleReconnect(authed.machineId, frame.tasks);
|
|
203
|
+
}
|
|
204
|
+
handleProgress(frame) {
|
|
205
|
+
this.taskLogBatcher.add(frame.taskId, protocol_2.TaskLogType.STAGE_CHANGE, `Stage: ${frame.stage} - ${frame.status}${frame.message ? `: ${frame.message}` : ''}`);
|
|
206
|
+
}
|
|
207
|
+
handleRecoverySecretAck(authed, _frame) {
|
|
208
|
+
this.logger.log(`Worker ${authed.machineId} acknowledged recovery secret bootstrap`);
|
|
209
|
+
}
|
|
210
|
+
async handleStageConfirmRequest(frame) {
|
|
211
|
+
const { taskId, stageIndex, mode, prompt, choices, timeout } = frame;
|
|
212
|
+
this.logger.log(`Stage confirm request for task #${taskId}, stage ${stageIndex}, mode=${mode}`);
|
|
213
|
+
// Store pending confirm in Redis via DedupService
|
|
214
|
+
const confirmKey = `stage:${taskId}:${stageIndex}`;
|
|
215
|
+
await this.dedupService.storePendingConfirm(confirmKey, {
|
|
216
|
+
taskId,
|
|
217
|
+
stageIndex,
|
|
218
|
+
mode,
|
|
219
|
+
prompt,
|
|
220
|
+
choices,
|
|
221
|
+
timeout,
|
|
222
|
+
createdAt: new Date().toISOString(),
|
|
223
|
+
});
|
|
224
|
+
// Notify web clients via the notification channel
|
|
225
|
+
const task = this.taskRepo.findById(taskId);
|
|
226
|
+
if (task) {
|
|
227
|
+
await this.notifierService.notifyStageConfirmRequired(task, `stage_${stageIndex}`);
|
|
228
|
+
}
|
|
229
|
+
else {
|
|
230
|
+
this.logger.warn(`Task #${taskId} not found when processing stage confirm request`);
|
|
231
|
+
}
|
|
232
|
+
}
|
|
233
|
+
/**
|
|
234
|
+
* Periodically revalidate Worker JWT. Disconnect if token is invalid,
|
|
235
|
+
* machine tokenId changed, or worker token is revoked/expired.
|
|
236
|
+
* Honors a 2-minute grace period for old JWTs during token refresh.
|
|
237
|
+
*/
|
|
238
|
+
revalidateWorkerJwt(client, machineId, tokenId) {
|
|
239
|
+
// Check machine still exists and tokenId matches
|
|
240
|
+
const machine = this.machineRepo.findById(machineId);
|
|
241
|
+
if (!machine) {
|
|
242
|
+
this.logger.warn(`Revalidation failed: machine ${machineId} not found, disconnecting`);
|
|
243
|
+
client.close(4010, 'Machine not found');
|
|
244
|
+
return;
|
|
245
|
+
}
|
|
246
|
+
if (machine.tokenId !== tokenId) {
|
|
247
|
+
// Check grace period
|
|
248
|
+
const authed = this.authedSockets.get(client);
|
|
249
|
+
if (authed) {
|
|
250
|
+
const elapsed = Date.now() - authed.authedAt;
|
|
251
|
+
if (elapsed < protocol_1.WORKER_JWT_GRACE_PERIOD_MS) {
|
|
252
|
+
this.logger.debug(`TokenId mismatch for ${machineId} but within grace period (${elapsed}ms)`);
|
|
253
|
+
return;
|
|
254
|
+
}
|
|
255
|
+
}
|
|
256
|
+
this.logger.warn(`Revalidation failed: tokenId mismatch for machine ${machineId}, disconnecting`);
|
|
257
|
+
client.close(4011, 'Token ID mismatch');
|
|
258
|
+
return;
|
|
259
|
+
}
|
|
260
|
+
// Check worker token is still active
|
|
261
|
+
const workerToken = this.workerTokenRepo.findById(tokenId);
|
|
262
|
+
if (!workerToken || workerToken.status !== 'active') {
|
|
263
|
+
this.logger.warn(`Revalidation failed: worker token ${tokenId} not active, disconnecting`);
|
|
264
|
+
client.close(4012, 'Worker token not active');
|
|
265
|
+
return;
|
|
266
|
+
}
|
|
267
|
+
}
|
|
268
|
+
};
|
|
269
|
+
exports.WorkerChannelGateway = WorkerChannelGateway;
|
|
270
|
+
exports.WorkerChannelGateway = WorkerChannelGateway = WorkerChannelGateway_1 = __decorate([
|
|
271
|
+
(0, websockets_1.WebSocketGateway)({ path: '/api/workers/channel' }),
|
|
272
|
+
__metadata("design:paramtypes", [auth_service_1.AuthService,
|
|
273
|
+
worker_connection_manager_1.WorkerConnectionManager,
|
|
274
|
+
heartbeat_service_1.HeartbeatService,
|
|
275
|
+
reconciler_1.Reconciler,
|
|
276
|
+
task_log_batcher_1.TaskLogBatcher,
|
|
277
|
+
dedup_service_1.DedupService,
|
|
278
|
+
machine_repository_1.MachineRepository,
|
|
279
|
+
task_repository_1.TaskRepository,
|
|
280
|
+
worker_token_repository_1.WorkerTokenRepository,
|
|
281
|
+
notifier_service_1.NotifierService])
|
|
282
|
+
], WorkerChannelGateway);
|
|
283
|
+
//# sourceMappingURL=worker-channel.gateway.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"worker-channel.gateway.js","sourceRoot":"","sources":["../../src/web/worker-channel.gateway.ts"],"names":[],"mappings":";;;;;;;;;;;;;AAAA,mDAI4B;AAC5B,2CAAwC;AAExC,2BAA+B;AAE/B,mDAK8B;AAa9B,uDAAmD;AACnD,uFAAkF;AAClF,uEAAmE;AACnE,yDAAsD;AACtD,qEAAgE;AAChE,+DAA2D;AAC3D,oFAAgF;AAChF,8EAA0E;AAC1E,8FAAyF;AACzF,mEAA+D;AAC/D,mDAAmD;AAW5C,IAAM,oBAAoB,4BAA1B,MAAM,oBAAoB;IAQZ;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IAdF,MAAM,GAAG,IAAI,eAAM,CAAC,sBAAoB,CAAC,IAAI,CAAC,CAAC;IAC/C,aAAa,GAAG,IAAI,GAAG,EAAkC,CAAC;IAC1D,eAAe,GAAG,MAAM,CAAC;IAE1C,YACmB,WAAwB,EACxB,uBAAgD,EAChD,gBAAkC,EAClC,UAAsB,EACtB,cAA8B,EAC9B,YAA0B,EAC1B,WAA8B,EAC9B,QAAwB,EACxB,eAAsC,EACtC,eAAgC;QAThC,gBAAW,GAAX,WAAW,CAAa;QACxB,4BAAuB,GAAvB,uBAAuB,CAAyB;QAChD,qBAAgB,GAAhB,gBAAgB,CAAkB;QAClC,eAAU,GAAV,UAAU,CAAY;QACtB,mBAAc,GAAd,cAAc,CAAgB;QAC9B,iBAAY,GAAZ,YAAY,CAAc;QAC1B,gBAAW,GAAX,WAAW,CAAmB;QAC9B,aAAQ,GAAR,QAAQ,CAAgB;QACxB,oBAAe,GAAf,eAAe,CAAuB;QACtC,oBAAe,GAAf,eAAe,CAAiB;IAChD,CAAC;IAEJ,gBAAgB,CAAC,MAAiB,EAAE,IAAqB;QACvD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,oDAAoD,CAAC,CAAC;QAExE,4DAA4D;QAC5D,MAAM,WAAW,GAAG,UAAU,CAAC,GAAG,EAAE;YAClC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBACpC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iDAAiD,CAAC,CAAC;gBACpE,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;YACrC,CAAC;QACH,CAAC,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;QAEzB,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,OAAwB,EAAE,EAAE;YAChD,IAAI,KAAoB,CAAC;YACzB,IAAI,CAAC;gBACH,MAAM,GAAG,GACP,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;gBACnE,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAkB,CAAC;YAC3C,CAAC;YAAC,MAAM,CAAC;gBACP,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;gBACrD,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;gBACnC,OAAO;YACT,CAAC;YAED,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAE9C,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,6BAA6B;gBAC7B,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAC1B,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,4BAA4B,KAAK,CAAC,IAAI,WAAW,CAClD,CAAC;oBACF,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,qBAAqB,CAAC,CAAC;oBAC1C,OAAO;gBACT,CAAC;gBACD,YAAY,CAAC,WAAW,CAAC,CAAC;gBAC1B,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,KAAwB,CAAC,CAAC;gBAClD,OAAO;YACT,CAAC;YAED,gCAAgC;YAChC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gBACnB,KAAK,WAAW;oBACd,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,KAA6B,CAAC,CAAC;oBAC5D,MAAM;gBACR,KAAK,KAAK;oBACR,IAAI,CAAC,SAAS,CAAC,KAAuB,CAAC,CAAC;oBACxC,MAAM;gBACR,KAAK,WAAW;oBACd,KAAK,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,KAA6B,CAAC,CAAC;oBACjE,MAAM;gBACR,KAAK,UAAU;oBACb,IAAI,CAAC,cAAc,CAAC,KAA4B,CAAC,CAAC;oBAClD,MAAM;gBACR,KAAK,qBAAqB;oBACxB,IAAI,CAAC,uBAAuB,CAC1B,MAAM,EACN,KAAqC,CACtC,CAAC;oBACF,MAAM;gBACR,KAAK,uBAAuB;oBAC1B,KAAK,IAAI,CAAC,yBAAyB,CACjC,KAAiC,CAClC,CAAC;oBACF,MAAM;gBACR;oBACE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uBAAwB,KAA0B,CAAC,IAAI,EAAE,CAAC,CAAC;YACjF,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,gBAAgB,CAAC,MAAiB;QAChC,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC9C,IAAI,MAAM,EAAE,CAAC;YACX,aAAa,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;YACxC,IAAI,CAAC,uBAAuB,CAAC,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAC1D,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAClC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,MAAM,CAAC,SAAS,eAAe,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAEO,UAAU,CAAC,MAAiB,EAAE,KAAsB;QAC1D,eAAe;QACf,IAAI,OAAyC,CAAC;QAC9C,IAAI,CAAC;YACH,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC1D,CAAC;QAAC,MAAM,CAAC;YACP,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;YACpD,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;YAClC,OAAO;QACT,CAAC;QAED,+CAA+C;QAC/C,MAAM,WAAW,GAAG,KAAK,CAAC,eAAe,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACxD,MAAM,WAAW,GAAG,2BAAgB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACnD,IAAI,WAAW,KAAK,WAAW,EAAE,CAAC;YAChC,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,qCAAqC,KAAK,CAAC,eAAe,YAAY,2BAAgB,EAAE,CACzF,CAAC;YACF,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,2BAA2B,CAAC,CAAC;YAChD,OAAO;QACT,CAAC;QAED,wCAAwC;QACxC,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACvD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,wBAAwB,OAAO,CAAC,GAAG,YAAY,CAAC,CAAC;YAClE,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAC;YACxC,OAAO;QACT,CAAC;QAED,IAAI,OAAO,CAAC,OAAO,KAAK,OAAO,CAAC,OAAO,EAAE,CAAC;YACxC,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,6CAA6C,OAAO,CAAC,GAAG,EAAE,CAC3D,CAAC;YACF,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAC;YACxC,OAAO;QACT,CAAC;QAED,kCAAkC;QAClC,MAAM,WAAW,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACnE,IAAI,CAAC,WAAW,IAAI,WAAW,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YACpD,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,6BAA6B,OAAO,CAAC,OAAO,aAAa,CAC1D,CAAC;YACF,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,yBAAyB,CAAC,CAAC;YAC9C,OAAO;QACT,CAAC;QAED,sBAAsB;QACtB,IAAI,CAAC,uBAAuB,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAE3D,sBAAsB;QACtB,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,EAAE,wBAAa,CAAC,MAAM,CAAC,CAAC;QAEjE,+BAA+B;QAC/B,MAAM,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE;YACzC,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QACjE,CAAC,EAAE,yCAA8B,CAAC,CAAC;QAEnC,MAAM,YAAY,GAAwB;YACxC,EAAE,EAAE,MAAM;YACV,SAAS,EAAE,OAAO,CAAC,GAAG;YACtB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE;YACpB,iBAAiB;SAClB,CAAC;QAEF,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAC7C,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,OAAO,CAAC,GAAG,gBAAgB,CAAC,CAAC;IACzD,CAAC;IAEO,eAAe,CACrB,MAA2B,EAC3B,KAA2B;QAE3B,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,CACpC,MAAM,CAAC,SAAS,EAChB;YACE,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,QAAQ;YAChC,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,WAAW;YACtC,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,SAAS;YAClC,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,WAAW;SACvC,EACD,KAAK,CAAC,YAAY,CACnB,CAAC;QAEF,qBAAqB;QACrB,MAAM,GAAG,GAAsB;YAC7B,IAAI,EAAE,eAAe;YACrB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACrC,CAAC;QAEF,IAAI,MAAM,CAAC,EAAE,CAAC,UAAU,KAAK,cAAS,CAAC,IAAI,EAAE,CAAC;YAC5C,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,KAAqB;QACrC,IAAI,CAAC,uBAAuB,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;IACrE,CAAC;IAEO,KAAK,CAAC,eAAe,CAC3B,MAA2B,EAC3B,KAA2B;QAE3B,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,UAAU,MAAM,CAAC,SAAS,sBAAsB,KAAK,CAAC,KAAK,CAAC,MAAM,UAAU,CAC7E,CAAC;QAEF,kCAAkC;QAClC,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,MAAM,CAAC,SAAS,EAAE,wBAAa,CAAC,MAAM,CAAC,CAAC;QAEtE,wBAAwB;QACxB,MAAM,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC,MAAM,CAAC,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;IACvE,CAAC;IAEO,cAAc,CAAC,KAA0B;QAC/C,IAAI,CAAC,cAAc,CAAC,GAAG,CACrB,KAAK,CAAC,MAAM,EACZ,sBAAW,CAAC,YAAY,EACxB,UAAU,KAAK,CAAC,KAAK,MAAM,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CACtF,CAAC;IACJ,CAAC;IAEO,uBAAuB,CAC7B,MAA2B,EAC3B,MAAoC;QAEpC,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,UAAU,MAAM,CAAC,SAAS,yCAAyC,CACpE,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,yBAAyB,CACrC,KAA+B;QAE/B,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;QAErE,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,mCAAmC,MAAM,WAAW,UAAU,UAAU,IAAI,EAAE,CAC/E,CAAC;QAEF,kDAAkD;QAClD,MAAM,UAAU,GAAG,SAAS,MAAM,IAAI,UAAU,EAAE,CAAC;QACnD,MAAM,IAAI,CAAC,YAAY,CAAC,mBAAmB,CAAC,UAAU,EAAE;YACtD,MAAM;YACN,UAAU;YACV,IAAI;YACJ,MAAM;YACN,OAAO;YACP,OAAO;YACP,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC,CAAC;QAEH,kDAAkD;QAClD,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QAC5C,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,IAAI,CAAC,eAAe,CAAC,0BAA0B,CACnD,IAAI,EACJ,SAAS,UAAU,EAAE,CACtB,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,SAAS,MAAM,kDAAkD,CAClE,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;;OAIG;IACK,mBAAmB,CACzB,MAAiB,EACjB,SAAiB,EACjB,OAAe;QAEf,iDAAiD;QACjD,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QACrD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,gCAAgC,SAAS,2BAA2B,CACrE,CAAC;YACF,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAC;YACxC,OAAO;QACT,CAAC;QAED,IAAI,OAAO,CAAC,OAAO,KAAK,OAAO,EAAE,CAAC;YAChC,qBAAqB;YACrB,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC9C,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC;gBAC7C,IAAI,OAAO,GAAG,qCAA0B,EAAE,CAAC;oBACzC,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,wBAAwB,SAAS,6BAA6B,OAAO,KAAK,CAC3E,CAAC;oBACF,OAAO;gBACT,CAAC;YACH,CAAC;YAED,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,qDAAqD,SAAS,iBAAiB,CAChF,CAAC;YACF,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAC;YACxC,OAAO;QACT,CAAC;QAED,qCAAqC;QACrC,MAAM,WAAW,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC3D,IAAI,CAAC,WAAW,IAAI,WAAW,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YACpD,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,qCAAqC,OAAO,4BAA4B,CACzE,CAAC;YACF,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,yBAAyB,CAAC,CAAC;YAC9C,OAAO;QACT,CAAC;IACH,CAAC;CACF,CAAA;AA/TY,oDAAoB;+BAApB,oBAAoB;IADhC,IAAA,6BAAgB,EAAC,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC;qCASjB,0BAAW;QACC,mDAAuB;QAC9B,oCAAgB;QACtB,uBAAU;QACN,iCAAc;QAChB,4BAAY;QACb,sCAAiB;QACpB,gCAAc;QACP,+CAAqB;QACrB,kCAAe;GAjBxC,oBAAoB,CA+ThC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { WorkerRegisterRequest, WorkerRegisterResponse, WorkerRefreshRequest, WorkerRefreshResponse, WorkerRecoveryRequest, WorkerRecoveryResponse } from '@overlordai/protocol';
|
|
2
|
+
import { DispatcherService } from '../dispatcher/dispatcher.service';
|
|
3
|
+
import { AuthService } from '../auth/auth.service';
|
|
4
|
+
export declare class WorkerController {
|
|
5
|
+
private readonly dispatcher;
|
|
6
|
+
private readonly authService;
|
|
7
|
+
constructor(dispatcher: DispatcherService, authService: AuthService);
|
|
8
|
+
register(body: WorkerRegisterRequest): Promise<WorkerRegisterResponse>;
|
|
9
|
+
refreshToken(body: WorkerRefreshRequest, req: {
|
|
10
|
+
headers?: Record<string, string>;
|
|
11
|
+
}): Promise<WorkerRefreshResponse>;
|
|
12
|
+
recover(body: WorkerRecoveryRequest): Promise<WorkerRecoveryResponse>;
|
|
13
|
+
}
|
|
14
|
+
//# sourceMappingURL=worker.controller.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"worker.controller.d.ts","sourceRoot":"","sources":["../../src/web/worker.controller.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EACV,qBAAqB,EACrB,sBAAsB,EACtB,oBAAoB,EACpB,qBAAqB,EACrB,qBAAqB,EACrB,sBAAsB,EACvB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,iBAAiB,EAAE,MAAM,kCAAkC,CAAC;AACrE,OAAO,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAGnD,qBACa,gBAAgB;IAEzB,OAAO,CAAC,QAAQ,CAAC,UAAU;IAC3B,OAAO,CAAC,QAAQ,CAAC,WAAW;gBADX,UAAU,EAAE,iBAAiB,EAC7B,WAAW,EAAE,WAAW;IAKrC,QAAQ,CACJ,IAAI,EAAE,qBAAqB,GAClC,OAAO,CAAC,sBAAsB,CAAC;IAK5B,YAAY,CACR,IAAI,EAAE,oBAAoB,EAC3B,GAAG,EAAE;QAAE,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;KAAE,GAC/C,OAAO,CAAC,qBAAqB,CAAC;IAY3B,OAAO,CACH,IAAI,EAAE,qBAAqB,GAClC,OAAO,CAAC,sBAAsB,CAAC;CAGnC"}
|
|
@@ -0,0 +1,73 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
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;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
var __metadata = (this && this.__metadata) || function (k, v) {
|
|
9
|
+
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
|
10
|
+
};
|
|
11
|
+
var __param = (this && this.__param) || function (paramIndex, decorator) {
|
|
12
|
+
return function (target, key) { decorator(target, key, paramIndex); }
|
|
13
|
+
};
|
|
14
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
15
|
+
exports.WorkerController = void 0;
|
|
16
|
+
const common_1 = require("@nestjs/common");
|
|
17
|
+
const dispatcher_service_1 = require("../dispatcher/dispatcher.service");
|
|
18
|
+
const auth_service_1 = require("../auth/auth.service");
|
|
19
|
+
const rate_limit_guard_1 = require("../common/rate-limit.guard");
|
|
20
|
+
let WorkerController = class WorkerController {
|
|
21
|
+
dispatcher;
|
|
22
|
+
authService;
|
|
23
|
+
constructor(dispatcher, authService) {
|
|
24
|
+
this.dispatcher = dispatcher;
|
|
25
|
+
this.authService = authService;
|
|
26
|
+
}
|
|
27
|
+
async register(body) {
|
|
28
|
+
return this.dispatcher.registerWorker(body);
|
|
29
|
+
}
|
|
30
|
+
async refreshToken(body, req) {
|
|
31
|
+
const authHeader = req.headers?.authorization;
|
|
32
|
+
if (!authHeader?.startsWith('Bearer ')) {
|
|
33
|
+
throw new common_1.UnauthorizedException('Missing or invalid Authorization header');
|
|
34
|
+
}
|
|
35
|
+
const token = authHeader.slice(7);
|
|
36
|
+
const payload = this.authService.verifyWorkerJwt(token);
|
|
37
|
+
return this.dispatcher.refreshWorkerToken(body, { sub: payload.sub, tokenId: payload.tokenId });
|
|
38
|
+
}
|
|
39
|
+
async recover(body) {
|
|
40
|
+
return this.dispatcher.recoverWorker(body);
|
|
41
|
+
}
|
|
42
|
+
};
|
|
43
|
+
exports.WorkerController = WorkerController;
|
|
44
|
+
__decorate([
|
|
45
|
+
(0, common_1.Post)('register'),
|
|
46
|
+
(0, rate_limit_guard_1.RateLimit)({ limit: 5, windowMs: 60_000, byIp: true }),
|
|
47
|
+
__param(0, (0, common_1.Body)()),
|
|
48
|
+
__metadata("design:type", Function),
|
|
49
|
+
__metadata("design:paramtypes", [Object]),
|
|
50
|
+
__metadata("design:returntype", Promise)
|
|
51
|
+
], WorkerController.prototype, "register", null);
|
|
52
|
+
__decorate([
|
|
53
|
+
(0, common_1.Post)('refresh-token'),
|
|
54
|
+
__param(0, (0, common_1.Body)()),
|
|
55
|
+
__param(1, (0, common_1.Req)()),
|
|
56
|
+
__metadata("design:type", Function),
|
|
57
|
+
__metadata("design:paramtypes", [Object, Object]),
|
|
58
|
+
__metadata("design:returntype", Promise)
|
|
59
|
+
], WorkerController.prototype, "refreshToken", null);
|
|
60
|
+
__decorate([
|
|
61
|
+
(0, common_1.Post)('recover'),
|
|
62
|
+
(0, rate_limit_guard_1.RateLimit)({ limit: 5, windowMs: 60_000, byIp: true }),
|
|
63
|
+
__param(0, (0, common_1.Body)()),
|
|
64
|
+
__metadata("design:type", Function),
|
|
65
|
+
__metadata("design:paramtypes", [Object]),
|
|
66
|
+
__metadata("design:returntype", Promise)
|
|
67
|
+
], WorkerController.prototype, "recover", null);
|
|
68
|
+
exports.WorkerController = WorkerController = __decorate([
|
|
69
|
+
(0, common_1.Controller)('api/workers'),
|
|
70
|
+
__metadata("design:paramtypes", [dispatcher_service_1.DispatcherService,
|
|
71
|
+
auth_service_1.AuthService])
|
|
72
|
+
], WorkerController);
|
|
73
|
+
//# sourceMappingURL=worker.controller.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"worker.controller.js","sourceRoot":"","sources":["../../src/web/worker.controller.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,2CAMwB;AASxB,yEAAqE;AACrE,uDAAmD;AACnD,iEAAuD;AAGhD,IAAM,gBAAgB,GAAtB,MAAM,gBAAgB;IAER;IACA;IAFnB,YACmB,UAA6B,EAC7B,WAAwB;QADxB,eAAU,GAAV,UAAU,CAAmB;QAC7B,gBAAW,GAAX,WAAW,CAAa;IACxC,CAAC;IAIE,AAAN,KAAK,CAAC,QAAQ,CACJ,IAA2B;QAEnC,OAAO,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IAC9C,CAAC;IAGK,AAAN,KAAK,CAAC,YAAY,CACR,IAA0B,EAC3B,GAAyC;QAEhD,MAAM,UAAU,GAAG,GAAG,CAAC,OAAO,EAAE,aAAa,CAAC;QAC9C,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YACvC,MAAM,IAAI,8BAAqB,CAAC,yCAAyC,CAAC,CAAC;QAC7E,CAAC;QACD,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QACxD,OAAO,IAAI,CAAC,UAAU,CAAC,kBAAkB,CAAC,IAAI,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IAClG,CAAC;IAIK,AAAN,KAAK,CAAC,OAAO,CACH,IAA2B;QAEnC,OAAO,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;IAC7C,CAAC;CACF,CAAA;AAnCY,4CAAgB;AAQrB;IAFL,IAAA,aAAI,EAAC,UAAU,CAAC;IAChB,IAAA,4BAAS,EAAC,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IAEnD,WAAA,IAAA,aAAI,GAAE,CAAA;;;;gDAGR;AAGK;IADL,IAAA,aAAI,EAAC,eAAe,CAAC;IAEnB,WAAA,IAAA,aAAI,GAAE,CAAA;IACN,WAAA,IAAA,YAAG,GAAE,CAAA;;;;oDASP;AAIK;IAFL,IAAA,aAAI,EAAC,SAAS,CAAC;IACf,IAAA,4BAAS,EAAC,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IAEnD,WAAA,IAAA,aAAI,GAAE,CAAA;;;;+CAGR;2BAlCU,gBAAgB;IAD5B,IAAA,mBAAU,EAAC,aAAa,CAAC;qCAGO,sCAAiB;QAChB,0BAAW;GAHhC,gBAAgB,CAmC5B"}
|
|
@@ -0,0 +1,109 @@
|
|
|
1
|
+
import { WorkspaceRepository } from '../database/repositories/workspace.repository';
|
|
2
|
+
import { TaskRepository } from '../database/repositories/task.repository';
|
|
3
|
+
import { WorkerConnectionManager } from '../dispatcher/worker-connection.manager';
|
|
4
|
+
/**
|
|
5
|
+
* Tunnel lifecycle states:
|
|
6
|
+
* IDLE -> STARTING -> CONNECTED -> EXPIRED -> CLOSING -> CLOSED
|
|
7
|
+
* | ^
|
|
8
|
+
* +--- (manual stop) -----+
|
|
9
|
+
*/
|
|
10
|
+
type TunnelStatus = 'IDLE' | 'STARTING' | 'CONNECTED' | 'EXPIRED' | 'CLOSING' | 'CLOSED';
|
|
11
|
+
export declare class WorkspaceController {
|
|
12
|
+
private readonly workspaceRepo;
|
|
13
|
+
private readonly taskRepo;
|
|
14
|
+
private readonly workerConnectionManager;
|
|
15
|
+
private readonly logger;
|
|
16
|
+
/**
|
|
17
|
+
* In-memory tunnel state per taskId.
|
|
18
|
+
* NOTE: This state is lost on server restart. This is acceptable because
|
|
19
|
+
* tunnels have a TTL and self-expire, and Workers will detect stale tunnels
|
|
20
|
+
* via heartbeat. After a restart, getTunnelStatus returns 'IDLE' which
|
|
21
|
+
* allows clients to start a new tunnel.
|
|
22
|
+
*/
|
|
23
|
+
private readonly tunnels;
|
|
24
|
+
constructor(workspaceRepo: WorkspaceRepository, taskRepo: TaskRepository, workerConnectionManager: WorkerConnectionManager);
|
|
25
|
+
getWorkspace(taskId: number, req: {
|
|
26
|
+
user: {
|
|
27
|
+
sub: number;
|
|
28
|
+
role: string;
|
|
29
|
+
};
|
|
30
|
+
}): import("@overlordai/protocol").Workspace;
|
|
31
|
+
/**
|
|
32
|
+
* Start a Cursor Remote Tunnel for the given task's workspace.
|
|
33
|
+
* Sends a start_tunnel command to the Worker via the WebSocket control channel.
|
|
34
|
+
* Only the task owner or an admin can start a tunnel.
|
|
35
|
+
*/
|
|
36
|
+
startTunnel(taskId: number, req: {
|
|
37
|
+
user: {
|
|
38
|
+
sub: number;
|
|
39
|
+
role: string;
|
|
40
|
+
};
|
|
41
|
+
}): Promise<{
|
|
42
|
+
status: TunnelStatus;
|
|
43
|
+
taskId: number;
|
|
44
|
+
tunnelUrl: string | undefined;
|
|
45
|
+
expiresAt: string | undefined;
|
|
46
|
+
}>;
|
|
47
|
+
/**
|
|
48
|
+
* Stop a running Cursor Remote Tunnel for the given task.
|
|
49
|
+
* Sends a stop_tunnel command to the Worker.
|
|
50
|
+
* Only the task owner or an admin can stop a tunnel.
|
|
51
|
+
*/
|
|
52
|
+
stopTunnel(taskId: number, req: {
|
|
53
|
+
user: {
|
|
54
|
+
sub: number;
|
|
55
|
+
role: string;
|
|
56
|
+
};
|
|
57
|
+
}): Promise<{
|
|
58
|
+
status: string;
|
|
59
|
+
taskId: number;
|
|
60
|
+
}>;
|
|
61
|
+
/**
|
|
62
|
+
* Get the current tunnel status for a task.
|
|
63
|
+
* tunnelUrl is only returned when the tunnel is in CONNECTED state.
|
|
64
|
+
*/
|
|
65
|
+
getTunnelStatus(taskId: number, req: {
|
|
66
|
+
user: {
|
|
67
|
+
sub: number;
|
|
68
|
+
role: string;
|
|
69
|
+
};
|
|
70
|
+
}): {
|
|
71
|
+
status: TunnelStatus;
|
|
72
|
+
taskId: number;
|
|
73
|
+
tunnelUrl?: undefined;
|
|
74
|
+
expiresAt?: undefined;
|
|
75
|
+
} | {
|
|
76
|
+
status: TunnelStatus;
|
|
77
|
+
taskId: number;
|
|
78
|
+
tunnelUrl: string | undefined;
|
|
79
|
+
expiresAt: string | undefined;
|
|
80
|
+
};
|
|
81
|
+
/**
|
|
82
|
+
* Handle a tunnel_started event from the Worker (called by WorkerChannelGateway).
|
|
83
|
+
* Transitions the tunnel state to CONNECTED and starts the expiry timer.
|
|
84
|
+
*/
|
|
85
|
+
handleTunnelStarted(taskId: number, tunnelUrl: string): void;
|
|
86
|
+
/**
|
|
87
|
+
* Handle a tunnel_closed event from the Worker (called by WorkerChannelGateway).
|
|
88
|
+
* Worker-side tunnel process exited unexpectedly.
|
|
89
|
+
*/
|
|
90
|
+
handleTunnelClosed(taskId: number, _reason: string): void;
|
|
91
|
+
/**
|
|
92
|
+
* Force-close all tunnels for a task (called when task reaches terminal state).
|
|
93
|
+
*/
|
|
94
|
+
cleanupTunnelForTask(taskId: number, machineId?: string | null): Promise<void>;
|
|
95
|
+
/**
|
|
96
|
+
* Transition a tunnel to CONNECTED state with a URL and start the expiry timer.
|
|
97
|
+
*/
|
|
98
|
+
private transitionToConnected;
|
|
99
|
+
/**
|
|
100
|
+
* Handle tunnel expiry: transition to EXPIRED, then send stop_tunnel to Worker.
|
|
101
|
+
*/
|
|
102
|
+
private handleTunnelExpired;
|
|
103
|
+
/**
|
|
104
|
+
* Assert that the requesting user is either the task owner or an admin.
|
|
105
|
+
*/
|
|
106
|
+
private assertOwnerOrAdmin;
|
|
107
|
+
}
|
|
108
|
+
export {};
|
|
109
|
+
//# sourceMappingURL=workspace.controller.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"workspace.controller.d.ts","sourceRoot":"","sources":["../../src/web/workspace.controller.ts"],"names":[],"mappings":"AAgBA,OAAO,EAAE,mBAAmB,EAAE,MAAM,+CAA+C,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,MAAM,0CAA0C,CAAC;AAC1E,OAAO,EAAE,uBAAuB,EAAE,MAAM,yCAAyC,CAAC;AASlF;;;;;GAKG;AACH,KAAK,YAAY,GAAG,MAAM,GAAG,UAAU,GAAG,WAAW,GAAG,SAAS,GAAG,SAAS,GAAG,QAAQ,CAAC;AAUzF,qBAEa,mBAAmB;IAY5B,OAAO,CAAC,QAAQ,CAAC,aAAa;IAC9B,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,uBAAuB;IAb1C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAwC;IAC/D;;;;;;OAMG;IACH,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAkC;gBAGvC,aAAa,EAAE,mBAAmB,EAClC,QAAQ,EAAE,cAAc,EACxB,uBAAuB,EAAE,uBAAuB;IAInE,YAAY,CACqB,MAAM,EAAE,MAAM,EACtC,GAAG,EAAE;QAAE,IAAI,EAAE;YAAE,GAAG,EAAE,MAAM,CAAC;YAAC,IAAI,EAAE,MAAM,CAAA;SAAE,CAAA;KAAE;IAkBrD;;;;OAIG;IAEG,WAAW,CACgB,MAAM,EAAE,MAAM,EACtC,GAAG,EAAE;QAAE,IAAI,EAAE;YAAE,GAAG,EAAE,MAAM,CAAC;YAAC,IAAI,EAAE,MAAM,CAAA;SAAE,CAAA;KAAE;;;;;;IAqFrD;;;;OAIG;IAEG,UAAU,CACiB,MAAM,EAAE,MAAM,EACtC,GAAG,EAAE;QAAE,IAAI,EAAE;YAAE,GAAG,EAAE,MAAM,CAAC;YAAC,IAAI,EAAE,MAAM,CAAA;SAAE,CAAA;KAAE;;;;IA4DrD;;;OAGG;IAEH,eAAe,CACkB,MAAM,EAAE,MAAM,EACtC,GAAG,EAAE;QAAE,IAAI,EAAE;YAAE,GAAG,EAAE,MAAM,CAAC;YAAC,IAAI,EAAE,MAAM,CAAA;SAAE,CAAA;KAAE;gBAetB,YAAY;;;;;;;;;;IAW3C;;;OAGG;IACH,mBAAmB,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,IAAI;IAI5D;;;OAGG;IACH,kBAAkB,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAezD;;OAEG;IACG,oBAAoB,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAiCpF;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAyB7B;;OAEG;YACW,mBAAmB;IAkCjC;;OAEG;IACH,OAAO,CAAC,kBAAkB;CAW3B"}
|