@junctionpanel/server 0.1.100 → 0.1.101

package/dist/server/server/session.js

@@ -1322,10 +1322,10 @@ export class Session {
                     await this.handleCancelAgentRequest(msg.agentId);
                     break;
                 case 'restart_server_request':
-                    await this.handleRestartServerRequest(msg.requestId, msg.reason);
+                    await this.handleRestartServerRequest(msg.requestId, msg.reason, msg.sourceCommand);
                     break;
                 case 'shutdown_server_request':
-                    await this.handleShutdownServerRequest(msg.requestId);
+                    await this.handleShutdownServerRequest(msg.requestId, msg.sourceCommand);
                     break;
                 case 'fetch_agent_timeline_request':
                     await this.handleFetchAgentTimelineRequest(msg);
@@ -1595,11 +1595,12 @@ export class Session {
         }
         this.sessionLogger.warn({ streamId: frame.streamId, messageType: frame.messageType }, 'Unhandled terminal binary frame');
     }
-    async handleRestartServerRequest(requestId, reason) {
+    async handleRestartServerRequest(requestId, reason, sourceCommand) {
         this.assertLifecycleControlAllowed('restart', requestId, reason);
         this.requestServerRestart(requestId, reason, {
             logLevel: 'warn',
             logMessage: 'Restart requested via websocket',
+            sourceCommand,
         });
     }
     requestServerRestart(requestId, reason, options) {
@@ -1610,10 +1611,13 @@ export class Session {
         if (reason && reason.trim().length > 0) {
             payload.reason = reason;
         }
+        if (options?.sourceCommand) {
+            payload.sourceCommand = options.sourceCommand;
+        }
         payload.requestId = requestId;
         const logLevel = options?.logLevel ?? 'warn';
         const logMessage = options?.logMessage ?? 'Restart requested';
-        this.sessionLogger[logLevel]({ reason, requestId }, logMessage);
+        this.sessionLogger[logLevel]({ reason, requestId, sourceCommand: options?.sourceCommand ?? null }, logMessage);
         this.emit({
             type: 'status',
             payload,
@@ -1626,17 +1630,19 @@ export class Session {
             initiatorTransport: this.connectionContext.transport,
             initiatorRemoteAddress: this.connectionContext.remoteAddress,
             initiatorOrigin: this.connectionContext.origin,
+            ...(options?.sourceCommand ? { sourceCommand: options.sourceCommand } : {}),
             ...(reason ? { reason } : {}),
         });
     }
-    async handleShutdownServerRequest(requestId) {
+    async handleShutdownServerRequest(requestId, sourceCommand) {
         this.assertLifecycleControlAllowed('shutdown', requestId);
-        this.sessionLogger.warn('Shutdown requested via websocket');
+        this.sessionLogger.warn({ requestId, sourceCommand: sourceCommand ?? null }, 'Shutdown requested via websocket');
         this.emit({
             type: 'status',
             payload: {
                 status: 'shutdown_requested',
                 clientId: this.clientId,
+                ...(sourceCommand ? { sourceCommand } : {}),
                 requestId,
             },
         });
@@ -1648,6 +1654,7 @@ export class Session {
             initiatorTransport: this.connectionContext.transport,
             initiatorRemoteAddress: this.connectionContext.remoteAddress,
             initiatorOrigin: this.connectionContext.origin,
+            ...(sourceCommand ? { sourceCommand } : {}),
         });
     }
     assertLifecycleControlAllowed(action, requestId, reason) {