api/app/mothership/subscribe-to-servers-endpoint.ts

import * as Sentry from '@sentry/node';
import { pubsub } from '@unraid/core';
import { SubscriptionClient } from 'graphql-subscriptions-client';
import { MOTHERSHIP_GRAPHQL_LINK } from '../consts';
import { userCache, CachedServers } from '../cache';

const client = new SubscriptionClient(MOTHERSHIP_GRAPHQL_LINK, {
    reconnect: true,
    lazy: true, // only connect when there is a query
    connectionCallback: (error) => {
        if (error) {
            Sentry.captureException(error);
        }
    }
});

export const subscribeToServersEndpoint = (apiKey: string) => {
    return client.request({
        query: `subscription servers ($apiKey: String!) {
            servers @auth(apiKey: $apiKey)
        }`,
        variables: {
            apiKey
        }
    })
    .subscribe({
        next({ data, errors }) {
            if (errors) {
                // Send all errors to Sentry
                errors.forEach((error: Error) => {
                    Sentry.captureException(error);
                });
            }
            if (data) {
                const obj = {
                    servers: data.servers
                };

                // Update internal cache
                userCache.set<CachedServers>('mine', obj);
                // Update subscribers
                pubsub.publish('servers', obj);
            }
        }
    });
};