From 17b2e9d375c4e7bb55f5e43aafd7f8b1e92f15de Mon Sep 17 00:00:00 2001
From: Andrey Goncharov <aigoncharov@fb.com>
Date: Mon, 23 May 2022 03:38:23 -0700
Subject: [PATCH] Connect companion events to Flipper Server

Reviewed By: passy

Differential Revision: D36517762

fbshipit-source-id: c6318b1b0705d1e6cbcf73973e2d6c9196914491
---
 desktop/flipper-server/src/attachSocketServer.tsx | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/desktop/flipper-server/src/attachSocketServer.tsx b/desktop/flipper-server/src/attachSocketServer.tsx
index 63c35fd10..d0c57eb5f 100644
--- a/desktop/flipper-server/src/attachSocketServer.tsx
+++ b/desktop/flipper-server/src/attachSocketServer.tsx
@@ -17,6 +17,7 @@ import {
   UserError,
   SystemError,
   getLogger,
+  CompanionEventWebSocketMessage,
 } from 'flipper-common';
 import {FlipperServerImpl} from 'flipper-server-core';
 import {WebSocketServer} from 'ws';
@@ -104,6 +105,19 @@ export function attachSocketServer(
 
     flipperServer.onAny(onServerEvent);
 
+    async function onServerCompanionEvent(event: string, payload: any) {
+      const message = {
+        event: 'companion-event',
+        payload: {
+          event,
+          data: payload,
+        },
+      } as CompanionEventWebSocketMessage;
+      client.send(JSON.stringify(message));
+    }
+
+    flipperServerCompanion?.onAny(onServerCompanionEvent);
+
     client.on('message', (data) => {
       let [event, payload]: [event: string | null, payload: any | null] = [
         null,