From 4ef713fc36346bf2dd108b17fa356432a321f916 Mon Sep 17 00:00:00 2001
From: Alexis Tyler <xo@wvvw.me>
Date: Fri, 22 Jan 2021 23:11:00 +1030
Subject: [PATCH] feat: add internal graphql class

---
 app/mothership/sockets/internal-graphql.ts | 90 ++++++++++++++++++++++
 1 file changed, 90 insertions(+)
 create mode 100644 app/mothership/sockets/internal-graphql.ts

diff --git a/app/mothership/sockets/internal-graphql.ts b/app/mothership/sockets/internal-graphql.ts
new file mode 100644
index 000000000..4c3ae4214
--- /dev/null
+++ b/app/mothership/sockets/internal-graphql.ts
@@ -0,0 +1,90 @@
+import { INTERNAL_WS_LINK } from '../../consts';
+import { apiManager, relayLogger } from '../../core';
+import { sleep } from '../../core/utils';
+import { AppError } from '../../core/errors';
+import { CustomSocket, WebSocketWithHeartBeat } from '../custom-socket';
+import { MothershipSocket } from './mothership';
+
+export class InternalGraphql extends CustomSocket {
+	private mothership?: MothershipSocket;
+
+	constructor(options: CustomSocket['options'] = {}) {
+        super({
+            name: 'InternalGraphql',
+            uri: INTERNAL_WS_LINK,
+			logger: relayLogger,
+            ...options
+        });
+    }
+
+	protected async getApiKey() {
+		const key = apiManager.getKey('my_servers');
+		if (!key) {
+			throw new AppError('No API key found.');
+		}
+
+		return key.key;
+	}
+
+	onMessage() {
+		const internalGraphql = this;
+		return async function (this: WebSocketWithHeartBeat, data: string) {
+			try {
+				internalGraphql.mothership?.connection?.send(data);
+			} catch (error) {
+				// Relay socket is closed, close internal one
+				if (error.message.includes('WebSocket is not open')) {
+					this.close(4200, JSON.stringify({
+						message: error.emss
+					}));
+					return;
+				}
+			}
+		};
+	}
+
+	onError() {
+		const internalGraphql = this;
+		return async function (error: NodeJS.ErrnoException) {
+			if (error.message === 'WebSocket was closed before the connection was established') {
+				// Likely the internal relay-ws connection was started but then mothership
+				// decided the key was invalid so it killed it
+				// When this happens the relay-ws sometimes is still in the CONNECTING phase
+				// This isn't an actual error so we skip it
+				return;
+			}
+	
+			// Socket was still offline try again?
+			if (error.code && ['ENOENT', 'ECONNREFUSED'].includes(error.code)) {
+				// Wait 1s
+				await sleep(1000);
+	
+				// Re-connect to internal graphql server
+				internalGraphql.connect();
+				return;
+			}
+	
+			internalGraphql.logger.error(error);
+		};
+	}
+
+	onConnect() {
+		const internalGraphql = this;
+		return async function (this: WebSocketWithHeartBeat) {
+			// No API key, close internal connection
+			if (!internalGraphql.apiKey) {
+				this.close(4200, JSON.stringify({
+					message: 'No API key'
+				}));
+			}
+
+			// Authenticate with ourselves
+			this.send(JSON.stringify({
+				type: 'connection_init',
+				payload: {
+					'x-api-key': internalGraphql.apiKey
+				}
+			}));
+		};
+	}
+};
\ No newline at end of file