feat: add internal graphql class

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
+				}
+			}));
+		};
+	}
+};