@docker-digital/dockernet-agent 0.0.1

package/dist/agent/services/event.service.js

@@ -0,0 +1,242 @@
+"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;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var EventService_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.EventService = void 0;
+const common_1 = require("@nestjs/common");
+const schedule_1 = require("@nestjs/schedule");
+const api_service_1 = require("./api.service");
+const config_service_1 = require("./config.service");
+const profile_service_1 = require("./profile.service");
+const websocket_client_service_1 = require("./websocket-client.service");
+const event_websocket_client_service_1 = require("./event-websocket-client.service");
+let EventService = EventService_1 = class EventService {
+    constructor(apiService, configService, profileService, websocketClientService, eventWebSocketClientService) {
+        this.apiService = apiService;
+        this.configService = configService;
+        this.profileService = profileService;
+        this.websocketClientService = websocketClientService;
+        this.eventWebSocketClientService = eventWebSocketClientService;
+        this.logger = new common_1.Logger(EventService_1.name);
+        this.isPolling = false;
+        this.useEventDrivenUpdates = false;
+        this.isWebSocketListening = false;
+        this.useEventDrivenUpdates =
+            this.websocketClientService.isWebSocketConnected();
+    }
+    async startPolling() {
+        if (this.isPolling) {
+            return;
+        }
+        this.isPolling = true;
+        this.logger.log('Starting event listening...');
+        try {
+            const machineKey = await this.configService.getMachineKey();
+            if (machineKey) {
+                await this.startWebSocketListening(machineKey);
+            }
+            else {
+                this.logger.warn('Machine key not found, falling back to polling');
+                await this.startPollingFallback();
+            }
+        }
+        catch (error) {
+            this.logger.error(`Error starting WebSocket listening: ${error.message}, falling back to polling`);
+            await this.startPollingFallback();
+        }
+    }
+    async startWebSocketListening(machineKey) {
+        if (this.isWebSocketListening) {
+            return;
+        }
+        this.logger.log(`Connecting to event WebSocket for machine: ${machineKey}`);
+        try {
+            await this.eventWebSocketClientService.connect(machineKey);
+            this.eventWebSocketClientService.onEvent((event) => {
+                this.handleWebSocketEvent(event);
+            });
+            await new Promise((resolve) => setTimeout(resolve, 1000));
+            if (this.eventWebSocketClientService.isSubscribedToEvents()) {
+                const machineId = this.eventWebSocketClientService.getCurrentMachineId();
+                this.logger.log(`Successfully connected and subscribed to event WebSocket for machine: ${machineId}`);
+                this.isWebSocketListening = true;
+            }
+            else {
+                this.logger.warn('Connected to event WebSocket but subscription not confirmed yet');
+                this.isWebSocketListening = true;
+            }
+        }
+        catch (error) {
+            this.logger.error(`Failed to connect to event WebSocket: ${error.message}`);
+            throw error;
+        }
+    }
+    async startPollingFallback() {
+        this.logger.log('Starting event polling as fallback...');
+        while (this.isPolling) {
+            try {
+                await this.pollEvents();
+            }
+            catch (error) {
+                this.logger.error(`Error polling events: ${error.message}`);
+                try {
+                }
+                catch (updateError) {
+                    this.logger.error(`Error updating profiles after polling error: ${updateError.message}`);
+                }
+            }
+            const interval = this.configService.pollIntervalSeconds * 1000;
+            await this.sleep(interval);
+        }
+    }
+    handleWebSocketEvent(event) {
+        this.logger.log(`Received event via WebSocket: ${event.key} (type: ${event.type}, profile: ${event.profile_id})`);
+        try {
+            if (!event.key || !event.type || !event.profile_id) {
+                this.logger.warn('Invalid event (missing key, type, or profile_id)');
+                return;
+            }
+            if (event.status !== 'scheduled') {
+                this.logger.warn(`Skipping event ${event.key} with status=${event.status} (only processing scheduled)`);
+                return;
+            }
+            let workspaceId = event.workspace_id;
+            if (workspaceId === 'null' ||
+                workspaceId === 'undefined' ||
+                !workspaceId) {
+                workspaceId = '';
+            }
+            this.handleEvent(event.key, event.type, event.profile_id, workspaceId);
+        }
+        catch (error) {
+            this.logger.error(`Error handling WebSocket event ${event.key}: ${error.message}`);
+        }
+    }
+    stopPolling() {
+        this.isPolling = false;
+        this.logger.log('Stopped event listening');
+        if (this.isWebSocketListening) {
+            this.eventWebSocketClientService.disconnect();
+            this.isWebSocketListening = false;
+            this.logger.log('Disconnected from event WebSocket');
+        }
+    }
+    checkSubscriptionStatus() {
+        const isConnected = this.eventWebSocketClientService.isWebSocketConnected();
+        const isSubscribed = this.eventWebSocketClientService.isSubscribedToEvents();
+        const machineId = this.eventWebSocketClientService.getCurrentMachineId();
+        this.logger.log(`Event subscription status: connected=${isConnected}, subscribed=${isSubscribed}, machineId=${machineId || 'none'}`);
+    }
+    async pollEvents() {
+        const response = await this.apiService.callApi('GET', '/events?status=scheduled', undefined, -1, 5000);
+        if (!response) {
+            this.logger.log('No response from server, checking saved profiles status...');
+            try {
+            }
+            catch (error) {
+                this.logger.error(`Error updating saved profiles: ${error.message}`);
+            }
+            return;
+        }
+        const rows = response.data?.rows || response.rows || [];
+        const eventsCount = rows?.length || 0;
+        if (eventsCount === 0) {
+            this.logger.log('No new events, checking saved profiles status...');
+            try {
+            }
+            catch (error) {
+                this.logger.error(`Error updating saved profiles: ${error.message}`);
+            }
+            return;
+        }
+        this.logger.log(`Received ${eventsCount} event(s) with status=scheduled`);
+        for (const event of rows) {
+            const eventKey = event.key;
+            const eventType = event.type;
+            const profileId = event.profile_id;
+            let workspaceId = event.workspace_id;
+            if (!eventKey || !eventType || !profileId) {
+                this.logger.warn('Invalid event (missing key, type, or profile_id)');
+                continue;
+            }
+            if (event.status !== 'scheduled') {
+                this.logger.warn(`Skipping event ${eventKey} with status=${event.status} (only processing scheduled)`);
+                continue;
+            }
+            if (workspaceId === 'null' ||
+                workspaceId === 'undefined' ||
+                !workspaceId) {
+                workspaceId = '';
+            }
+            this.logger.log(`Processing event: ${eventKey} (type: ${eventType}, profile_id: ${profileId})`);
+            try {
+                await this.handleEvent(eventKey, eventType, profileId, workspaceId);
+            }
+            catch (error) {
+                this.logger.error(`Error handling event ${eventKey}: ${error.message}`);
+            }
+        }
+    }
+    async handleEvent(eventKey, eventType, profileId, workspaceId) {
+        if (eventType === 'start_profile') {
+            await this.profileService.startProfile(eventKey, profileId, workspaceId);
+        }
+        else if (eventType === 'stop_profile') {
+            await this.profileService.stopProfile(eventKey, profileId, workspaceId);
+        }
+        else if (eventType.startsWith('restart_service:')) {
+            const serviceName = eventType.replace('restart_service:', '');
+            await this.profileService.restartService(eventKey, profileId, workspaceId, serviceName);
+        }
+        else {
+            this.logger.error(`Unsupported event type: ${eventType}`);
+            await this.apiService.callApi('PUT', `/events/${eventKey}`, {
+                status: 'failed',
+                is_executed: true,
+                error: `Unsupported event type: ${eventType}`,
+            });
+        }
+    }
+    async updateSavedProfiles() {
+        if (this.useEventDrivenUpdates) {
+            const now = Date.now();
+            const seconds = Math.floor(now / 1000);
+            if (seconds % 60 !== 0) {
+                return;
+            }
+            this.logger.debug('Running backup status poll (event-driven updates enabled)');
+        }
+        try {
+        }
+        catch (error) {
+            this.logger.error(`Error updating saved profiles: ${error.message}`);
+        }
+    }
+    sleep(ms) {
+        return new Promise((resolve) => setTimeout(resolve, ms));
+    }
+};
+exports.EventService = EventService;
+__decorate([
+    (0, schedule_1.Cron)(schedule_1.CronExpression.EVERY_5_SECONDS),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], EventService.prototype, "updateSavedProfiles", null);
+exports.EventService = EventService = EventService_1 = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [api_service_1.ApiService,
+        config_service_1.ConfigService,
+        profile_service_1.ProfileService,
+        websocket_client_service_1.WebSocketClientService,
+        event_websocket_client_service_1.EventWebSocketClientService])
+], EventService);
+//# sourceMappingURL=event.service.js.map

package/dist/agent/services/event.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"event.service.js","sourceRoot":"","sources":["../../../src/agent/services/event.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;AAAA,2CAAoD;AACpD,+CAAwD;AACxD,+CAA2C;AAC3C,qDAAiD;AACjD,uDAAmD;AACnD,yEAAoE;AACpE,qFAA+E;AAGxE,IAAM,YAAY,oBAAlB,MAAM,YAAY;IAMvB,YACU,UAAsB,EACtB,aAA4B,EAC5B,cAA8B,EAC9B,sBAA8C,EAC9C,2BAAwD;QAJxD,eAAU,GAAV,UAAU,CAAY;QACtB,kBAAa,GAAb,aAAa,CAAe;QAC5B,mBAAc,GAAd,cAAc,CAAgB;QAC9B,2BAAsB,GAAtB,sBAAsB,CAAwB;QAC9C,gCAA2B,GAA3B,2BAA2B,CAA6B;QAVjD,WAAM,GAAG,IAAI,eAAM,CAAC,cAAY,CAAC,IAAI,CAAC,CAAC;QAChD,cAAS,GAAG,KAAK,CAAC;QAClB,0BAAqB,GAAG,KAAK,CAAC;QAC9B,yBAAoB,GAAG,KAAK,CAAC;QAUnC,IAAI,CAAC,qBAAqB;YACxB,IAAI,CAAC,sBAAsB,CAAC,oBAAoB,EAAE,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;QAG/C,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,aAAa,EAAE,CAAC;YAC5D,IAAI,UAAU,EAAE,CAAC;gBACf,MAAM,IAAI,CAAC,uBAAuB,CAAC,UAAU,CAAC,CAAC;YACjD,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gDAAgD,CAAC,CAAC;gBACnE,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;YACpC,CAAC;QACH,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,uCAAuC,KAAK,CAAC,OAAO,2BAA2B,CAChF,CAAC;YACF,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;QACpC,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,uBAAuB,CAAC,UAAkB;QACtD,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC9B,OAAO;QACT,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,8CAA8C,UAAU,EAAE,CAAC,CAAC;QAE5E,IAAI,CAAC;YAEH,MAAM,IAAI,CAAC,2BAA2B,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAG3D,IAAI,CAAC,2BAA2B,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;gBACjD,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;YACnC,CAAC,CAAC,CAAC;YAGH,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;YAG1D,IAAI,IAAI,CAAC,2BAA2B,CAAC,oBAAoB,EAAE,EAAE,CAAC;gBAC5D,MAAM,SAAS,GACb,IAAI,CAAC,2BAA2B,CAAC,mBAAmB,EAAE,CAAC;gBACzD,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,yEAAyE,SAAS,EAAE,CACrF,CAAC;gBACF,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;YACnC,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,iEAAiE,CAClE,CAAC;gBACF,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;YACnC,CAAC;QACH,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,yCAAyC,KAAK,CAAC,OAAO,EAAE,CACzD,CAAC;YACF,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,oBAAoB;QAChC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC;QAEzD,OAAO,IAAI,CAAC,SAAS,EAAE,CAAC;YACtB,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;YAC1B,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,yBAAyB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;gBAE5D,IAAI,CAAC;gBAEL,CAAC;gBAAC,OAAO,WAAgB,EAAE,CAAC;oBAC1B,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,gDAAgD,WAAW,CAAC,OAAO,EAAE,CACtE,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,mBAAmB,GAAG,IAAI,CAAC;YAC/D,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IAEO,oBAAoB,CAAC,KAAU;QACrC,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,iCAAiC,KAAK,CAAC,GAAG,WAAW,KAAK,CAAC,IAAI,cAAc,KAAK,CAAC,UAAU,GAAG,CACjG,CAAC;QAEF,IAAI,CAAC;YAEH,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC;gBACnD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC;gBACrE,OAAO;YACT,CAAC;YAED,IAAI,KAAK,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBACjC,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,kBAAkB,KAAK,CAAC,GAAG,gBAAgB,KAAK,CAAC,MAAM,8BAA8B,CACtF,CAAC;gBACF,OAAO;YACT,CAAC;YAED,IAAI,WAAW,GAAG,KAAK,CAAC,YAAY,CAAC;YACrC,IACE,WAAW,KAAK,MAAM;gBACtB,WAAW,KAAK,WAAW;gBAC3B,CAAC,WAAW,EACZ,CAAC;gBACD,WAAW,GAAG,EAAE,CAAC;YACnB,CAAC;YAGD,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QACzE,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,kCAAkC,KAAK,CAAC,GAAG,KAAK,KAAK,CAAC,OAAO,EAAE,CAChE,CAAC;QACJ,CAAC;IACH,CAAC;IAED,WAAW;QACT,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;QAG3C,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC9B,IAAI,CAAC,2BAA2B,CAAC,UAAU,EAAE,CAAC;YAC9C,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC;YAClC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAMD,uBAAuB;QACrB,MAAM,WAAW,GAAG,IAAI,CAAC,2BAA2B,CAAC,oBAAoB,EAAE,CAAC;QAC5E,MAAM,YAAY,GAChB,IAAI,CAAC,2BAA2B,CAAC,oBAAoB,EAAE,CAAC;QAC1D,MAAM,SAAS,GAAG,IAAI,CAAC,2BAA2B,CAAC,mBAAmB,EAAE,CAAC;QAEzE,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,wCAAwC,WAAW,gBAAgB,YAAY,eAC7E,SAAS,IAAI,MACf,EAAE,CACH,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,UAAU;QAEtB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAC5C,KAAK,EACL,0BAA0B,EAC1B,SAAS,EACT,CAAC,CAAC,EACF,IAAI,CACL,CAAC;QAEF,IAAI,CAAC,QAAQ,EAAE,CAAC;YAEd,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,4DAA4D,CAC7D,CAAC;YACF,IAAI,CAAC;YAEL,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,kCAAkC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YACvE,CAAC;YACD,OAAO;QACT,CAAC;QAED,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,EAAE,IAAI,IAAI,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC;QACxD,MAAM,WAAW,GAAG,IAAI,EAAE,MAAM,IAAI,CAAC,CAAC;QAEtC,IAAI,WAAW,KAAK,CAAC,EAAE,CAAC;YAEtB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC;YACpE,IAAI,CAAC;YAEL,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,kCAAkC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YACvE,CAAC;YACD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,YAAY,WAAW,iCAAiC,CAAC,CAAC;QAE1E,KAAK,MAAM,KAAK,IAAI,IAAI,EAAE,CAAC;YACzB,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC;YAC3B,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC;YAC7B,MAAM,SAAS,GAAG,KAAK,CAAC,UAAU,CAAC;YACnC,IAAI,WAAW,GAAG,KAAK,CAAC,YAAY,CAAC;YAErC,IAAI,CAAC,QAAQ,IAAI,CAAC,SAAS,IAAI,CAAC,SAAS,EAAE,CAAC;gBAC1C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC;gBACrE,SAAS;YACX,CAAC;YAED,IAAI,KAAK,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBACjC,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,kBAAkB,QAAQ,gBAAgB,KAAK,CAAC,MAAM,8BAA8B,CACrF,CAAC;gBACF,SAAS;YACX,CAAC;YAED,IACE,WAAW,KAAK,MAAM;gBACtB,WAAW,KAAK,WAAW;gBAC3B,CAAC,WAAW,EACZ,CAAC;gBACD,WAAW,GAAG,EAAE,CAAC;YACnB,CAAC;YAED,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,qBAAqB,QAAQ,WAAW,SAAS,iBAAiB,SAAS,GAAG,CAC/E,CAAC;YAEF,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;YACtE,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,wBAAwB,QAAQ,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YAC1E,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,WAAW,CACvB,QAAgB,EAChB,SAAiB,EACjB,SAAiB,EACjB,WAAmB;QAEnB,IAAI,SAAS,KAAK,eAAe,EAAE,CAAC;YAClC,MAAM,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,QAAQ,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;QAC3E,CAAC;aAAM,IAAI,SAAS,KAAK,cAAc,EAAE,CAAC;YACxC,MAAM,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,QAAQ,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;QAC1E,CAAC;aAAM,IAAI,SAAS,CAAC,UAAU,CAAC,kBAAkB,CAAC,EAAE,CAAC;YACpD,MAAM,WAAW,GAAG,SAAS,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;YAC9D,MAAM,IAAI,CAAC,cAAc,CAAC,cAAc,CACtC,QAAQ,EACR,SAAS,EACT,WAAW,EACX,WAAW,CACZ,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2BAA2B,SAAS,EAAE,CAAC,CAAC;YAC1D,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,KAAK,EAAE,WAAW,QAAQ,EAAE,EAAE;gBAC1D,MAAM,EAAE,QAAQ;gBAChB,WAAW,EAAE,IAAI;gBACjB,KAAK,EAAE,2BAA2B,SAAS,EAAE;aAC9C,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAQK,AAAN,KAAK,CAAC,mBAAmB;QAGvB,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAC/B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACvB,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,GAAG,IAAI,CAAC,CAAC;YAEvC,IAAI,OAAO,GAAG,EAAE,KAAK,CAAC,EAAE,CAAC;gBACvB,OAAO;YACT,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,2DAA2D,CAC5D,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;QAEL,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,kCAAkC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,EAAU;QACtB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;IAC3D,CAAC;CACF,CAAA;AArTY,oCAAY;AA4RjB;IADL,IAAA,eAAI,EAAC,yBAAc,CAAC,eAAe,CAAC;;;;uDAqBpC;uBAhTU,YAAY;IADxB,IAAA,mBAAU,GAAE;qCAQW,wBAAU;QACP,8BAAa;QACZ,gCAAc;QACN,iDAAsB;QACjB,4DAA2B;GAXvD,YAAY,CAqTxB"}

package/dist/agent/services/log.service.d.ts

@@ -0,0 +1,19 @@
+import { EventEmitter } from 'events';
+import { ApiService } from './api.service';
+import { ConfigService } from './config.service';
+import { DockerService } from './docker.service';
+export declare class LogService extends EventEmitter {
+    private apiService;
+    private configService;
+    private dockerService;
+    private readonly logger;
+    private readonly logPidDir;
+    private readonly streamedContainers;
+    constructor(apiService: ApiService, configService: ConfigService, dockerService: DockerService);
+    startLogStreamsForProfile(profileId: string, workspaceId: string): Promise<void>;
+    stopLogStreamsForProfile(profileId: string): Promise<void>;
+    streamLogForContainer(containerName: string, profileId: string, serviceName: string, workspaceId: string): Promise<void>;
+    autoStreamLogsForAllContainers(): Promise<void>;
+    private processLogStream;
+    private sendLogBatch;
+}

package/dist/agent/services/log.service.js

@@ -0,0 +1,176 @@
+"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;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var LogService_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LogService = void 0;
+const common_1 = require("@nestjs/common");
+const events_1 = require("events");
+const fs = require("fs-extra");
+const path = require("path");
+const api_service_1 = require("./api.service");
+const config_service_1 = require("./config.service");
+const docker_service_1 = require("./docker.service");
+let LogService = LogService_1 = class LogService extends events_1.EventEmitter {
+    constructor(apiService, configService, dockerService) {
+        super();
+        this.apiService = apiService;
+        this.configService = configService;
+        this.dockerService = dockerService;
+        this.logger = new common_1.Logger(LogService_1.name);
+        this.streamedContainers = new Set();
+        this.logPidDir = path.join(this.configService.workDirPath, '.log_pids');
+        fs.ensureDirSync(this.logPidDir);
+    }
+    async startLogStreamsForProfile(profileId, workspaceId) {
+        const profileDir = path.join(this.configService.workDirPath, profileId);
+        const composeFile = path.join(profileDir, `docker-compose_${profileId}.yaml`);
+        if (!fs.existsSync(composeFile)) {
+            this.logger.warn(`Compose file not found: ${composeFile}`);
+            return;
+        }
+        this.logger.log(`Starting log streams for profile: ${profileId}`);
+        const containers = await this.dockerService.listContainers({
+            label: [`com.docker.compose.project=${profileId}`],
+        });
+        for (const container of containers) {
+            const containerName = container.Names[0]?.replace(/^\//, '') || '';
+            const serviceName = container.Labels['com.docker.compose.service'] || containerName;
+            if (containerName && container.State === 'running') {
+                await this.streamLogForContainer(containerName, profileId, serviceName, workspaceId);
+            }
+        }
+    }
+    async stopLogStreamsForProfile(profileId) {
+        const pidFile = path.join(this.logPidDir, `${profileId}.pids`);
+        if (!fs.existsSync(pidFile)) {
+            return;
+        }
+        const pids = await fs.readFile(pidFile, 'utf-8');
+        const pidList = pids
+            .split('\n')
+            .map((p) => p.trim())
+            .filter((p) => p);
+        for (const pidStr of pidList) {
+            const pid = parseInt(pidStr, 10);
+            if (pid && !isNaN(pid)) {
+                try {
+                    process.kill(pid, 'SIGTERM');
+                }
+                catch (error) {
+                }
+            }
+        }
+        await fs.remove(pidFile);
+        this.logger.log(`Stopped log streams for profile: ${profileId}`);
+    }
+    async streamLogForContainer(containerName, profileId, serviceName, workspaceId) {
+        if (this.streamedContainers.has(containerName)) {
+            return;
+        }
+        this.streamedContainers.add(containerName);
+        this.logger.log(`Starting log stream for container: ${containerName} (profile: ${profileId}, service: ${serviceName})`);
+        try {
+            const logStream = await this.dockerService.getContainerLogs(containerName, 50);
+            this.processLogStream(logStream, containerName, profileId, serviceName, workspaceId).catch((error) => {
+                this.logger.error(`Error processing log stream for ${containerName}: ${error.message}`);
+                this.streamedContainers.delete(containerName);
+            });
+        }
+        catch (error) {
+            this.logger.error(`Error starting log stream for ${containerName}: ${error.message}`);
+            this.streamedContainers.delete(containerName);
+        }
+    }
+    async autoStreamLogsForAllContainers() {
+        const containers = await this.dockerService.listContainers({
+            status: ['running'],
+        });
+        for (const container of containers) {
+            const containerName = container.Names[0]?.replace(/^\//, '') || '';
+            if (!containerName)
+                continue;
+            const labels = container.Labels || {};
+            const serviceName = labels['com.docker.compose.service'] || '';
+            const projectName = labels['com.docker.compose.project'] || '';
+            let profileId = projectName;
+            if (!profileId && containerName.includes('-')) {
+                const parts = containerName.split('-');
+                profileId = parts[0];
+            }
+            if (profileId && !this.streamedContainers.has(containerName)) {
+                const profileDir = path.join(this.configService.workDirPath, profileId);
+                const metadataFile = path.join(profileDir, '.profile_metadata');
+                let workspaceId = '';
+                if (fs.existsSync(metadataFile)) {
+                    const metadata = await fs.readFile(metadataFile, 'utf-8');
+                    const match = metadata.match(/^workspace_id=(.+)$/m);
+                    if (match) {
+                        workspaceId = match[1].trim();
+                    }
+                }
+                await this.streamLogForContainer(containerName, profileId, serviceName || containerName, workspaceId);
+            }
+        }
+    }
+    async processLogStream(stream, containerName, profileId, serviceName, workspaceId) {
+        const lines = [];
+        let buffer = '';
+        stream.on('data', async (chunk) => {
+            buffer += chunk.toString();
+            const newLines = buffer.split('\n');
+            buffer = newLines.pop() || '';
+            for (const line of newLines) {
+                if (line.trim()) {
+                    lines.push(line);
+                }
+            }
+            if (lines.length >= 10) {
+                await this.sendLogBatch(lines.splice(0), containerName, profileId, serviceName, workspaceId);
+            }
+        });
+        stream.on('end', async () => {
+            if (lines.length > 0 || buffer) {
+                if (buffer)
+                    lines.push(buffer);
+                await this.sendLogBatch(lines, containerName, profileId, serviceName, workspaceId);
+            }
+            this.streamedContainers.delete(containerName);
+        });
+        stream.on('error', (error) => {
+            this.logger.error(`Log stream error for ${containerName}: ${error.message}`);
+            this.streamedContainers.delete(containerName);
+        });
+    }
+    async sendLogBatch(lines, containerName, profileId, serviceName, workspaceId) {
+        for (const line of lines) {
+            if (!line.trim())
+                continue;
+            const payload = {
+                profile_id: profileId,
+                workspace_id: workspaceId,
+                service_name: serviceName,
+                container_name: containerName,
+                line: line,
+                stream: 'stdout',
+                timestamp: new Date().toISOString(),
+            };
+            await this.apiService.callApi('POST', '/logs', payload, 3, 1000);
+        }
+    }
+};
+exports.LogService = LogService;
+exports.LogService = LogService = LogService_1 = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [api_service_1.ApiService,
+        config_service_1.ConfigService,
+        docker_service_1.DockerService])
+], LogService);
+//# sourceMappingURL=log.service.js.map

package/dist/agent/services/log.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"log.service.js","sourceRoot":"","sources":["../../../src/agent/services/log.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;AAAA,2CAAoD;AACpD,mCAAsC;AACtC,+BAA+B;AAC/B,6BAA6B;AAE7B,+CAA2C;AAC3C,qDAAiD;AACjD,qDAAiD;AAG1C,IAAM,UAAU,kBAAhB,MAAM,UAAW,SAAQ,qBAAY;IAK1C,YACU,UAAsB,EACtB,aAA4B,EAC5B,aAA4B;QAEpC,KAAK,EAAE,CAAC;QAJA,eAAU,GAAV,UAAU,CAAY;QACtB,kBAAa,GAAb,aAAa,CAAe;QAC5B,kBAAa,GAAb,aAAa,CAAe;QAPrB,WAAM,GAAG,IAAI,eAAM,CAAC,YAAU,CAAC,IAAI,CAAC,CAAC;QAErC,uBAAkB,GAAG,IAAI,GAAG,EAAU,CAAC;QAQtD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;QACxE,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,yBAAyB,CAC7B,SAAiB,EACjB,WAAmB;QAEnB,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;QACxE,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAC3B,UAAU,EACV,kBAAkB,SAAS,OAAO,CACnC,CAAC;QAEF,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;YAChC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,2BAA2B,WAAW,EAAE,CAAC,CAAC;YAC3D,OAAO;QACT,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,qCAAqC,SAAS,EAAE,CAAC,CAAC;QAGlE,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC;YACzD,KAAK,EAAE,CAAC,8BAA8B,SAAS,EAAE,CAAC;SACnD,CAAC,CAAC;QAEH,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,MAAM,aAAa,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;YACnE,MAAM,WAAW,GACf,SAAS,CAAC,MAAM,CAAC,4BAA4B,CAAC,IAAI,aAAa,CAAC;YAElE,IAAI,aAAa,IAAI,SAAS,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;gBACnD,MAAM,IAAI,CAAC,qBAAqB,CAC9B,aAAa,EACb,SAAS,EACT,WAAW,EACX,WAAW,CACZ,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,wBAAwB,CAAC,SAAiB;QAC9C,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,SAAS,OAAO,CAAC,CAAC;QAE/D,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACjD,MAAM,OAAO,GAAG,IAAI;aACjB,KAAK,CAAC,IAAI,CAAC;aACX,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;aACpB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QAEpB,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACjC,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;gBACvB,IAAI,CAAC;oBACH,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;gBAC/B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;gBAEjB,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACzB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,oCAAoC,SAAS,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,KAAK,CAAC,qBAAqB,CACzB,aAAqB,EACrB,SAAiB,EACjB,WAAmB,EACnB,WAAmB;QAEnB,IAAI,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;YAC/C,OAAO;QACT,CAAC;QAED,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QAC3C,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,sCAAsC,aAAa,cAAc,SAAS,cAAc,WAAW,GAAG,CACvG,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,gBAAgB,CACzD,aAAa,EACb,EAAE,CACH,CAAC;YAGF,IAAI,CAAC,gBAAgB,CACnB,SAAS,EACT,aAAa,EACb,SAAS,EACT,WAAW,EACX,WAAW,CACZ,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBAChB,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,mCAAmC,aAAa,KAAK,KAAK,CAAC,OAAO,EAAE,CACrE,CAAC;gBACF,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAChD,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,iCAAiC,aAAa,KAAK,KAAK,CAAC,OAAO,EAAE,CACnE,CAAC;YACF,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,8BAA8B;QAClC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC;YACzD,MAAM,EAAE,CAAC,SAAS,CAAC;SACpB,CAAC,CAAC;QAEH,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,MAAM,aAAa,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;YACnE,IAAI,CAAC,aAAa;gBAAE,SAAS;YAG7B,MAAM,MAAM,GAAG,SAAS,CAAC,MAAM,IAAI,EAAE,CAAC;YACtC,MAAM,WAAW,GAAG,MAAM,CAAC,4BAA4B,CAAC,IAAI,EAAE,CAAC;YAC/D,MAAM,WAAW,GAAG,MAAM,CAAC,4BAA4B,CAAC,IAAI,EAAE,CAAC;YAG/D,IAAI,SAAS,GAAG,WAAW,CAAC;YAC5B,IAAI,CAAC,SAAS,IAAI,aAAa,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC9C,MAAM,KAAK,GAAG,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;gBACvC,SAAS,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YACvB,CAAC;YAED,IAAI,SAAS,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;gBAE7D,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;gBACxE,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,mBAAmB,CAAC,CAAC;gBAChE,IAAI,WAAW,GAAG,EAAE,CAAC;gBAErB,IAAI,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;oBAChC,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;oBAC1D,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC;oBACrD,IAAI,KAAK,EAAE,CAAC;wBACV,WAAW,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBAChC,CAAC;gBACH,CAAC;gBAED,MAAM,IAAI,CAAC,qBAAqB,CAC9B,aAAa,EACb,SAAS,EACT,WAAW,IAAI,aAAa,EAC5B,WAAW,CACZ,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC5B,MAAgB,EAChB,aAAqB,EACrB,SAAiB,EACjB,WAAmB,EACnB,WAAmB;QAEnB,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,KAAK,EAAE,KAAa,EAAE,EAAE;YACxC,MAAM,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YAC3B,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,GAAG,QAAQ,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;YAE9B,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;gBAC5B,IAAI,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;oBAChB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnB,CAAC;YACH,CAAC;YAGD,IAAI,KAAK,CAAC,MAAM,IAAI,EAAE,EAAE,CAAC;gBACvB,MAAM,IAAI,CAAC,YAAY,CACrB,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EACf,aAAa,EACb,SAAS,EACT,WAAW,EACX,WAAW,CACZ,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,IAAI,EAAE;YAE1B,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,MAAM,EAAE,CAAC;gBAC/B,IAAI,MAAM;oBAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC/B,MAAM,IAAI,CAAC,YAAY,CACrB,KAAK,EACL,aAAa,EACb,SAAS,EACT,WAAW,EACX,WAAW,CACZ,CAAC;YACJ,CAAC;YACD,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YAC3B,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,wBAAwB,aAAa,KAAK,KAAK,CAAC,OAAO,EAAE,CAC1D,CAAC;YACF,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,KAAe,EACf,aAAqB,EACrB,SAAiB,EACjB,WAAmB,EACnB,WAAmB;QAEnB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;gBAAE,SAAS;YAE3B,MAAM,OAAO,GAAG;gBACd,UAAU,EAAE,SAAS;gBACrB,YAAY,EAAE,WAAW;gBACzB,YAAY,EAAE,WAAW;gBACzB,cAAc,EAAE,aAAa;gBAC7B,IAAI,EAAE,IAAI;gBACV,MAAM,EAAE,QAAQ;gBAChB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC,CAAC;YAEF,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;CACF,CAAA;AAvPY,gCAAU;qBAAV,UAAU;IADtB,IAAA,mBAAU,GAAE;qCAOW,wBAAU;QACP,8BAAa;QACb,8BAAa;GAR3B,UAAU,CAuPtB"}

package/dist/agent/services/machine.service.d.ts

@@ -0,0 +1,15 @@
+import { ApiService } from './api.service';
+import { ConfigService } from './config.service';
+export declare class MachineService {
+    private apiService;
+    private configService;
+    private readonly logger;
+    constructor(apiService: ApiService, configService: ConfigService);
+    generateMachineKey(): string;
+    getOsInfo(): string;
+    getCurrentUser(): string;
+    getTailscaleIp(): Promise<string | null>;
+    private isValidIp;
+    registerMachine(): Promise<boolean>;
+    updateMachineStatus(status: 'live' | 'warning' | 'exit'): Promise<void>;
+}

package/dist/agent/services/machine.service.js

@@ -0,0 +1,148 @@
+"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;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var MachineService_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MachineService = void 0;
+const common_1 = require("@nestjs/common");
+const uuid_1 = require("uuid");
+const os = require("os");
+const child_process_1 = require("child_process");
+const util_1 = require("util");
+const api_service_1 = require("./api.service");
+const config_service_1 = require("./config.service");
+const execAsync = (0, util_1.promisify)(child_process_1.exec);
+let MachineService = MachineService_1 = class MachineService {
+    constructor(apiService, configService) {
+        this.apiService = apiService;
+        this.configService = configService;
+        this.logger = new common_1.Logger(MachineService_1.name);
+    }
+    generateMachineKey() {
+        try {
+            return (0, uuid_1.v4)();
+        }
+        catch (error) {
+            const hostname = os.hostname();
+            const timestamp = Date.now();
+            return `machine-${hostname}-${timestamp}`;
+        }
+    }
+    getOsInfo() {
+        const osName = os.type();
+        if (osName === 'Darwin') {
+            return 'macOS';
+        }
+        return osName;
+    }
+    getCurrentUser() {
+        return os.userInfo().username;
+    }
+    async getTailscaleIp() {
+        try {
+            try {
+                const { stdout } = await execAsync('tailscale ip -4');
+                const ip = stdout.trim();
+                if (ip && this.isValidIp(ip)) {
+                    this.logger.debug(`Found Tailscale IP via CLI: ${ip}`);
+                    return ip;
+                }
+            }
+            catch (error) {
+                this.logger.debug('Tailscale CLI not available or failed, trying network interface');
+            }
+            const networkInterfaces = os.networkInterfaces();
+            const tailscaleInterface = networkInterfaces['tailscale0'];
+            if (tailscaleInterface) {
+                for (const addr of tailscaleInterface) {
+                    const addrInfo = addr;
+                    if (addrInfo && addrInfo.family && addrInfo.address) {
+                        const isIPv4 = addrInfo.family === 'IPv4' ||
+                            (typeof addrInfo.family === 'number' && addrInfo.family === 4);
+                        if (isIPv4 && typeof addrInfo.address === 'string') {
+                            this.logger.debug(`Found Tailscale IP via network interface: ${addrInfo.address}`);
+                            return addrInfo.address;
+                        }
+                    }
+                }
+            }
+            try {
+                const { stdout } = await execAsync('tailscale status --json');
+                const status = JSON.parse(stdout);
+                if (status.Self &&
+                    status.Self.TailscaleIPs &&
+                    status.Self.TailscaleIPs.length > 0) {
+                    const ipv4 = status.Self.TailscaleIPs.find((ip) => !ip.includes(':'));
+                    if (ipv4) {
+                        this.logger.debug(`Found Tailscale IP via status: ${ipv4}`);
+                        return ipv4;
+                    }
+                }
+            }
+            catch (error) {
+                this.logger.debug('Tailscale status command failed');
+            }
+            this.logger.warn('Could not determine Tailscale IP address');
+            return null;
+        }
+        catch (error) {
+            this.logger.error(`Error getting Tailscale IP: ${error.message}`);
+            return null;
+        }
+    }
+    isValidIp(ip) {
+        const ipv4Regex = /^(\d{1,3}\.){3}\d{1,3}$/;
+        const ipv6Regex = /^([0-9a-fA-F]{1,4}:){7}[0-9a-fA-F]{1,4}$/;
+        return ipv4Regex.test(ip) || ipv6Regex.test(ip);
+    }
+    async registerMachine() {
+        const machineKey = await this.configService.getMachineKey();
+        if (!machineKey) {
+            this.logger.error('Machine key not found in config');
+            return false;
+        }
+        const osInfo = this.getOsInfo();
+        const userName = this.getCurrentUser();
+        const tailscaleIp = await this.getTailscaleIp();
+        const payload = {
+            key: machineKey,
+            os: osInfo,
+            user: userName,
+        };
+        if (tailscaleIp) {
+            payload.tailscale_ip = tailscaleIp;
+        }
+        this.logger.log(`Registering machine with key: ${machineKey}`);
+        const response = await this.apiService.callApi('POST', '/machines', payload, -1, 5000);
+        if (response) {
+            this.logger.log('Machine registered successfully');
+            return true;
+        }
+        else {
+            this.logger.warn('Failed to register machine, will retry later');
+            return false;
+        }
+    }
+    async updateMachineStatus(status) {
+        const machineKey = await this.configService.getMachineKey();
+        if (!machineKey) {
+            return;
+        }
+        const payload = { status };
+        await this.apiService.callApi('POST', `/machines/${machineKey}/status`, payload, -1, 5000);
+    }
+};
+exports.MachineService = MachineService;
+exports.MachineService = MachineService = MachineService_1 = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [api_service_1.ApiService,
+        config_service_1.ConfigService])
+], MachineService);
+//# sourceMappingURL=machine.service.js.map

package/dist/agent/services/machine.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"machine.service.js","sourceRoot":"","sources":["../../../src/agent/services/machine.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;AAAA,2CAAoD;AACpD,+BAAoC;AACpC,yBAAyB;AACzB,iDAAqC;AACrC,+BAAiC;AACjC,+CAA2C;AAC3C,qDAAiD;AAEjD,MAAM,SAAS,GAAG,IAAA,gBAAS,EAAC,oBAAI,CAAC,CAAC;AAG3B,IAAM,cAAc,sBAApB,MAAM,cAAc;IAGzB,YACU,UAAsB,EACtB,aAA4B;QAD5B,eAAU,GAAV,UAAU,CAAY;QACtB,kBAAa,GAAb,aAAa,CAAe;QAJrB,WAAM,GAAG,IAAI,eAAM,CAAC,gBAAc,CAAC,IAAI,CAAC,CAAC;IAKvD,CAAC;IAEJ,kBAAkB;QAChB,IAAI,CAAC;YACH,OAAO,IAAA,SAAM,GAAE,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAEf,MAAM,QAAQ,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;YAC/B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,OAAO,WAAW,QAAQ,IAAI,SAAS,EAAE,CAAC;QAC5C,CAAC;IACH,CAAC;IAED,SAAS;QACP,MAAM,MAAM,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC;QACzB,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;YACxB,OAAO,OAAO,CAAC;QACjB,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,cAAc;QACZ,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,cAAc;QAClB,IAAI,CAAC;YAEH,IAAI,CAAC;gBACH,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,SAAS,CAAC,iBAAiB,CAAC,CAAC;gBACtD,MAAM,EAAE,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;gBACzB,IAAI,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC;oBAC7B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,+BAA+B,EAAE,EAAE,CAAC,CAAC;oBACvD,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,iEAAiE,CAClE,CAAC;YACJ,CAAC;YAGD,MAAM,iBAAiB,GAAG,EAAE,CAAC,iBAAiB,EAAE,CAAC;YACjD,MAAM,kBAAkB,GAAG,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAE3D,IAAI,kBAAkB,EAAE,CAAC;gBAGvB,KAAK,MAAM,IAAI,IAAI,kBAAkB,EAAE,CAAC;oBAEtC,MAAM,QAAQ,GAAG,IAAW,CAAC;oBAC7B,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;wBACpD,MAAM,MAAM,GACV,QAAQ,CAAC,MAAM,KAAK,MAAM;4BAC1B,CAAC,OAAO,QAAQ,CAAC,MAAM,KAAK,QAAQ,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC;wBACjE,IAAI,MAAM,IAAI,OAAO,QAAQ,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;4BACnD,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,6CAA6C,QAAQ,CAAC,OAAO,EAAE,CAChE,CAAC;4BACF,OAAO,QAAQ,CAAC,OAAO,CAAC;wBAC1B,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAGD,IAAI,CAAC;gBACH,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,SAAS,CAAC,yBAAyB,CAAC,CAAC;gBAC9D,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAClC,IACE,MAAM,CAAC,IAAI;oBACX,MAAM,CAAC,IAAI,CAAC,YAAY;oBACxB,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EACnC,CAAC;oBAED,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CACxC,CAAC,EAAU,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,CAClC,CAAC;oBACF,IAAI,IAAI,EAAE,CAAC;wBACT,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,kCAAkC,IAAI,EAAE,CAAC,CAAC;wBAC5D,OAAO,IAAI,CAAC;oBACd,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC;YACvD,CAAC;YAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAC;YAC7D,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,+BAA+B,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YAClE,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,EAAU;QAE1B,MAAM,SAAS,GAAG,yBAAyB,CAAC;QAE5C,MAAM,SAAS,GAAG,0CAA0C,CAAC;QAC7D,OAAO,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,eAAe;QACnB,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,aAAa,EAAE,CAAC;QAC5D,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC;YACrD,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QACvC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAEhD,MAAM,OAAO,GAAQ;YACnB,GAAG,EAAE,UAAU;YACf,EAAE,EAAE,MAAM;YACV,IAAI,EAAE,QAAQ;SACf,CAAC;QAEF,IAAI,WAAW,EAAE,CAAC;YAChB,OAAO,CAAC,YAAY,GAAG,WAAW,CAAC;QACrC,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,iCAAiC,UAAU,EAAE,CAAC,CAAC;QAE/D,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAC5C,MAAM,EACN,WAAW,EACX,OAAO,EACP,CAAC,CAAC,EACF,IAAI,CACL,CAAC;QAEF,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;YACnD,OAAO,IAAI,CAAC;QACd,CAAC;aAAM,CAAC;YAEN,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC;YACjE,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,KAAK,CAAC,mBAAmB,CACvB,MAAmC;QAEnC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,aAAa,EAAE,CAAC;QAC5D,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,EAAE,MAAM,EAAE,CAAC;QAE3B,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAC3B,MAAM,EACN,aAAa,UAAU,SAAS,EAChC,OAAO,EACP,CAAC,CAAC,EACF,IAAI,CACL,CAAC;IACJ,CAAC;CACF,CAAA;AAxKY,wCAAc;yBAAd,cAAc;IAD1B,IAAA,mBAAU,GAAE;qCAKW,wBAAU;QACP,8BAAa;GAL3B,cAAc,CAwK1B"}