feat(daemon): Add TCP connect to Node powers

endojs · Feb 13, 2024 · 82c45af · 82c45af
1 parent d0a9168
commit 82c45af
Showing 1 changed file with 21 additions and 2 deletions.
diff --git a/packages/daemon/src/daemon-node-powers.js b/packages/daemon/src/daemon-node-powers.js
@@ -238,6 +238,24 @@ export const makeSocketPowers = ({ net }) => {
       cancelled,
     );
 
+  const connectPort = ({ port, host, cancelled }) =>
+    new Promise((resolve, reject) => {
+      const conn = net.connect(port, host, err => {
+        if (err) {
+          reject(err);
+          return;
+        }
+        const reader = makeNodeReader(conn);
+        const writer = makeNodeWriter(conn);
+        const closed = new Promise(close => conn.on('close', close));
+        resolve({
+          reader,
+          writer,
+          closed,
+        });
+      });
+    });
+
   /**
    * @param {object} args
    * @param {string} args.path
@@ -263,7 +281,7 @@ export const makeSocketPowers = ({ net }) => {
     return connections;
   };
 
-  return { servePort, servePath };
+  return { servePort, servePath, connectPort };
 };
 
 /**
@@ -275,7 +293,7 @@ export const makeSocketPowers = ({ net }) => {
  */
 export const makeNetworkPowers = ({ http, ws, net }) => {
   const { servePortHttp } = makeHttpPowers({ http, ws });
-  const { servePort, servePath } = makeSocketPowers({ net });
+  const { servePort, servePath, connectPort } = makeSocketPowers({ net });
 
   const connectionNumbers = (function* generateNumbers() {
     let n = 0;
@@ -338,6 +356,7 @@ export const makeNetworkPowers = ({ http, ws, net }) => {
     servePortHttp,
     servePort,
     servePath,
+    connectPort,
     makePrivatePathService,
     makePrivateHttpService,
   });