@graphql-hive/core/esm/client/persisted-documents.js

import LRU from 'tiny-lru';
import { http } from './http-client.js';
export function createPersistedDocuments(config) {
    var _a;
    const persistedDocumentsCache = LRU((_a = config.cache) !== null && _a !== void 0 ? _a : 10000);
    let allowArbitraryDocuments;
    if (typeof config.allowArbitraryDocuments === 'boolean') {
        let value = config.allowArbitraryDocuments;
        allowArbitraryDocuments = () => value;
    }
    else if (typeof config.allowArbitraryDocuments === 'function') {
        allowArbitraryDocuments = config.allowArbitraryDocuments;
    }
    else {
        allowArbitraryDocuments = () => false;
    }
    /** if there is already a in-flight request for a document, we re-use it. */
    const fetchCache = new Map();
    /** Batch load a persisted documents */
    function loadPersistedDocument(documentId) {
        const document = persistedDocumentsCache.get(documentId);
        if (document) {
            return document;
        }
        const cdnDocumentId = documentId.replaceAll('~', '/');
        const url = config.cdn.endpoint + '/apps/' + cdnDocumentId;
        let promise = fetchCache.get(url);
        if (!promise) {
            promise = http
                .get(url, {
                headers: {
                    'X-Hive-CDN-Key': config.cdn.accessToken,
                },
                logger: config.logger,
                isRequestOk: response => response.status === 200 || response.status === 404,
            })
                .then(async (response) => {
                if (response.status !== 200) {
                    return null;
                }
                const text = await response.text();
                persistedDocumentsCache.set(documentId, text);
                return text;
            })
                .finally(() => {
                fetchCache.delete(url);
            });
            fetchCache.set(url, promise);
        }
        return promise;
    }
    return {
        allowArbitraryDocuments,
        resolve: loadPersistedDocument,
    };
}