talk/client/coral-framework/services/client.js

import ApolloClient, {
  IntrospectionFragmentMatcher,
  createNetworkInterface,
} from 'apollo-client';
import {
  SubscriptionClient,
  addGraphQLSubscriptions,
} from 'subscriptions-transport-ws';
import MessageTypes from 'subscriptions-transport-ws/dist/message-types';

// Redux middleware to report any errors to the console.
export const apolloErrorReporter = () => next => action => {
  if (action.type === 'APOLLO_QUERY_ERROR') {
    console.error(action.error);
  }
  return next(action);
};

function resolveToken(token) {
  return typeof token === 'function' ? token() : token;
}

/**
 * createClient setups and returns an Apollo GraphQL Client
 * @param  {Object}          [options]          configuration
 * @param  {string|function} [options.token]    auth token
 * @param  {string}          [options.uri]      uri of the graphql server
 * @param  {string}          [options.liveUri]  uri of the graphql subscription server
 * @param  {Object}          [options.introspectionData] introspection query result data
 * @return {Object}          apollo client
 */
export function createClient(options = {}) {
  const { token, uri, liveUri, introspectionData } = options;
  const wsClient = new SubscriptionClient(liveUri, {
    reconnect: true,
    lazy: true,
    connectionParams: {
      get token() {
        return resolveToken(token);
      },
    },
  });

  const networkInterface = createNetworkInterface({
    uri,
    opts: {
      credentials: 'same-origin',
    },
  });

  networkInterface.use([
    {
      applyMiddleware(req, next) {
        if (!req.options.headers) {
          req.options.headers = {}; // Create the header object if needed.
        }

        let authToken = resolveToken(token);
        if (authToken) {
          req.options.headers['authorization'] = `Bearer ${authToken}`;
        }

        next();
      },
    },
  ]);

  const networkInterfaceWithSubscriptions = addGraphQLSubscriptions(
    networkInterface,
    wsClient
  );

  const client = new ApolloClient({
    connectToDevTools: true,
    addTypename: true,
    fragmentMatcher: new IntrospectionFragmentMatcher({
      introspectionQueryResultData: introspectionData,
    }),
    dataIdFromObject: result => {
      if (result.id && result.__typename) {
        // eslint-disable-line no-underscore-dangle
        return `${result.__typename}_${result.id}`; // eslint-disable-line no-underscore-dangle
      }
      return null;
    },
    networkInterface: networkInterfaceWithSubscriptions,
  });

  client.resetWebsocket = () => {
    // Close socket connection which will also unregister subscriptions on the server-side.
    wsClient.close();

    // Reconnect to the server.
    wsClient.connect();

    // Reregister all subscriptions (uses non public api).
    // See: https://github.com/apollographql/subscriptions-transport-ws/issues/171
    Object.keys(wsClient.operations).forEach(id => {
      wsClient.sendMessage(
        id,
        MessageTypes.GQL_START,
        wsClient.operations[id].options
      );
    });
  };

  return client;
}