@graphql-yoga/plugin-graphql-sse/cjs/index.js

GraphQL over Server-Sent Events Protocol plugin for GraphQL Yoga.

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.useGraphQLSSE = useGraphQLSSE;
const graphql_1 = require("graphql");
const fetch_1 = require("graphql-sse/lib/use/fetch");
const promise_helpers_1 = require("@whatwg-node/promise-helpers");
/**
 * Get [GraphQL over Server-Sent Events Protocol](https://github.com/enisdenjo/graphql-sse/blob/master/PROTOCOL.md) integration with GraphQL Yoga by simply installing this plugin!
 *
 * Note that the endpoint defaults to `/graphql/stream`, this is where your [graphql-sse](https://github.com/enisdenjo/graphql-sse) client should connect.
 */
function useGraphQLSSE(options = {}) {
    const { endpoint = '/graphql/stream', ...handlerOptions } = options;
    // eslint-disable-next-line @typescript-eslint/no-explicit-any
    const ctxForReq = new WeakMap();
    let handler;
    return {
        onYogaInit({ yoga }) {
            handler = (0, fetch_1.createHandler)({
                ...handlerOptions,
                onSubscribe(req, params) {
                    const enveloped = yoga.getEnveloped({
                        ...ctxForReq.get(req.raw),
                        request: req.raw,
                        params,
                    });
                    const document = enveloped.parse(params.query);
                    const errors = enveloped.validate(enveloped.schema, document);
                    if (errors.length > 0) {
                        return { errors };
                    }
                    return (0, promise_helpers_1.handleMaybePromise)(() => enveloped.contextFactory(), contextValue => {
                        const executionArgs = {
                            schema: enveloped.schema,
                            document,
                            contextValue,
                            variableValues: params.variables,
                            operationName: params.operationName,
                        };
                        const operation = (0, graphql_1.getOperationAST)(document, params.operationName);
                        const executeFn = operation?.operation === 'subscription' ? enveloped.subscribe : enveloped.execute;
                        return executeFn(executionArgs);
                    });
                },
            }, yoga.fetchAPI);
        },
        onRequest({ request, endResponse, serverContext }) {
            const [path, _search] = request.url.split('?');
            if (path?.endsWith(endpoint)) {
                ctxForReq.set(request, serverContext);
                return (0, promise_helpers_1.handleMaybePromise)(() => handler(request), endResponse);
            }
        },
    };
}