mongo patch

Author: EspressoTrip-v2

README.md

@@ -63,3 +63,37 @@ but adapted to expose raw Sockets, instead of just http/https requests.
     }
 
     run().catch(console.error);
+
+
+## Usage - MongoDB
+
+    import * as mongo from 'mongodb';
+    import { useProxyForMongo } from '@journeyapps/https-proxy-socket';
+
+    const SRV_URI = 'mongodb+srv://<username>:<password>@cluster0.jzuewet.mongodb.net';
+    const PROXY = 'us-cc-proxy.journeyapps.com'; // Or za-cc-proxy.journeyapps.com
+    const PROXY_PORT = 443
+
+    // Register the proxy globally for MongoDB
+    useProxyForMongo({
+      proxy: PROXY,
+      auth: <egress_token>
+    });
+
+    async function run() {
+      const client = new mongo.MongoClient(SRV_URI, {
+        proxyPort: PROXY_PORT,
+        proxyHost: PROXY,
+      });
+      try {
+        const database = client.db('poc');
+        const data = database.collection('data');
+
+        const results = await data.find({ index: { $lt: 5 } }).toArray();
+        console.log(results);
+      } finally {
+      await client.close();
+      }
+    }
+
+    run().catch(console.error);

package.json

@@ -27,7 +27,8 @@
   },
   "main": "./lib/index.js",
   "dependencies": {
-    "agent-base": "^6.0.0"
+    "agent-base": "^6.0.0",
+    "socks": "^2.8.7"
   },
   "files": [
     "/README.md",

src/index.ts

@@ -1,2 +1,3 @@
 export * from './HttpsProxySocket';
 export * from './tediousPatch';
+export * from './mongoPatch';

src/mongoPatch.ts

@@ -0,0 +1,24 @@
+import * as socks from 'socks';
+import {HttpsProxySocket} from './HttpsProxySocket';
+
+interface Config {
+  /** The journey apps cc egress token */
+  auth: string;
+  /** The journey apps cc egress proxy domain */
+  proxy: string;
+}
+/**
+ *  The patch should be called before instantiating the MongoClient
+ *  @param config - The configuration for the proxy
+ */
+export function useProxyForMongo(config: Config) {
+  socks.SocksClient.createConnection = async (options, callback) => {
+    const proxy = new HttpsProxySocket(`https://${config.proxy}`, { auth: config.auth });
+    return new Promise(async (resolve, reject) => {
+      const socket = await proxy.connect({ host: options.destination.host, port: options.destination.port });
+      resolve({
+        socket,
+      });
+    });
+  };
+}

yarn.lock

@@ -758,6 +758,11 @@ inquirer@^7.0.0:
     strip-ansi "^5.1.0"
     through "^2.3.6"
 
+ip-address@^10.0.1:
+  version "10.0.1"
+  resolved "https://registry.yarnpkg.com/ip-address/-/ip-address-10.0.1.tgz#a8180b783ce7788777d796286d61bce4276818ed"
+  integrity sha512-NWv9YLW4PoW2B7xtzaS3NCot75m6nK7Icdv0o3lfMceJVRfSoQwqD4wEH5rLwoKJwUiZ/rfpiVBhnaF0FK4HoA==
+
 is-buffer@~2.0.3:
   version "2.0.4"
   resolved "https://registry.yarnpkg.com/is-buffer/-/is-buffer-2.0.4.tgz#3e572f23c8411a5cfd9557c849e3665e0b290623"
@@ -1264,6 +1269,19 @@ signal-exit@^3.0.2:
   resolved "https://registry.yarnpkg.com/signal-exit/-/signal-exit-3.0.2.tgz#b5fdc08f1287ea1178628e415e25132b73646c6d"
   integrity sha1-tf3AjxKH6hF4Yo5BXiUTK3NkbG0=
 
+smart-buffer@^4.2.0:
+  version "4.2.0"
+  resolved "https://registry.yarnpkg.com/smart-buffer/-/smart-buffer-4.2.0.tgz#6e1d71fa4f18c05f7d0ff216dd16a481d0e8d9ae"
+  integrity sha512-94hK0Hh8rPqQl2xXc3HsaBoOXKV20MToPkcXvwbISWLEs+64sBq5kFgn2kJDHb1Pry9yrP0dxrCI9RRci7RXKg==
+
+socks@^2.8.7:
+  version "2.8.7"
+  resolved "https://registry.yarnpkg.com/socks/-/socks-2.8.7.tgz#e2fb1d9a603add75050a2067db8c381a0b5669ea"
+  integrity sha512-HLpt+uLy/pxB+bum/9DzAgiKS8CX1EvbWxI4zlmgGCExImLdiad2iCwXT5Z4c9c3Eq8rP2318mPW2c+QbtjK8A==
+  dependencies:
+    ip-address "^10.0.1"
+    smart-buffer "^4.2.0"
+
 source-map-support@^0.5.6:
   version "0.5.16"
   resolved "https://registry.yarnpkg.com/source-map-support/-/source-map-support-0.5.16.tgz#0ae069e7fe3ba7538c64c98515e35339eac5a042"