|
| 1 | +import type { |
| 2 | + ClientOptions, |
| 3 | + createClient as createClientType, |
| 4 | + ExecutionResult, |
| 5 | +} from 'graphql-sse'; |
| 6 | +import { Fetcher, FetcherParams } from './types'; |
| 7 | + |
| 8 | +export async function createSseFetcher(opts: ClientOptions): Promise<Fetcher> { |
| 9 | + const { createClient } = |
| 10 | + process.env.USE_IMPORT === 'false' |
| 11 | + ? (require('graphql-sse') as { createClient: typeof createClientType }) |
| 12 | + : await import('graphql-sse'); |
| 13 | + |
| 14 | + const sseClient = createClient({ |
| 15 | + retryAttempts: 0, |
| 16 | + // @ts-expect-error |
| 17 | + singleConnection: true, // or use false if you have an HTTP/2 server |
| 18 | + // @ts-expect-error |
| 19 | + lazy: false, // connect as soon as the page opens |
| 20 | + ...opts, |
| 21 | + }); |
| 22 | + |
| 23 | + function subscribe(payload: FetcherParams) { |
| 24 | + let deferred: { |
| 25 | + resolve: (arg: boolean) => void; |
| 26 | + reject: (arg: unknown) => void; |
| 27 | + }; |
| 28 | + |
| 29 | + const pending: ExecutionResult<Record<string, unknown>, unknown>[] = []; |
| 30 | + let throwMe: unknown; |
| 31 | + let done = false; |
| 32 | + |
| 33 | + const dispose = sseClient.subscribe( |
| 34 | + { |
| 35 | + ...payload, |
| 36 | + // types are different with FetcherParams |
| 37 | + operationName: payload.operationName ?? undefined, |
| 38 | + }, |
| 39 | + { |
| 40 | + next(data) { |
| 41 | + pending.push(data); |
| 42 | + deferred?.resolve(false); |
| 43 | + }, |
| 44 | + error(err) { |
| 45 | + throwMe = err; |
| 46 | + deferred?.reject(throwMe); |
| 47 | + }, |
| 48 | + complete() { |
| 49 | + done = true; |
| 50 | + deferred?.resolve(true); |
| 51 | + }, |
| 52 | + }, |
| 53 | + ); |
| 54 | + |
| 55 | + return { |
| 56 | + [Symbol.asyncIterator]() { |
| 57 | + return this; |
| 58 | + }, |
| 59 | + async next() { |
| 60 | + if (done) { |
| 61 | + return { done: true, value: undefined }; |
| 62 | + } |
| 63 | + if (throwMe) { |
| 64 | + throw throwMe; |
| 65 | + } |
| 66 | + if (pending.length) { |
| 67 | + return { value: pending.shift() }; |
| 68 | + } |
| 69 | + return (await new Promise((resolve, reject) => { |
| 70 | + deferred = { resolve, reject }; |
| 71 | + })) |
| 72 | + ? { done: true, value: undefined } |
| 73 | + : { value: pending.shift() }; |
| 74 | + }, |
| 75 | + async return() { |
| 76 | + dispose(); |
| 77 | + return { done: true, value: undefined }; |
| 78 | + }, |
| 79 | + }; |
| 80 | + } |
| 81 | + |
| 82 | + // @ts-expect-error todo: fix type |
| 83 | + return subscribe; |
| 84 | +} |
0 commit comments