115 lines
3.2 KiB
JavaScript
115 lines
3.2 KiB
JavaScript
const fetch = require('node-fetch');
|
|
const fs = require('fs');
|
|
const { URLSearchParams } = require('url');
|
|
const crypto = require('crypto');
|
|
const Response = require('./classes/response.js');
|
|
const MemoryCache = require('./classes/caching/memory_cache.js');
|
|
|
|
const CACHE_VERSION = 2;
|
|
|
|
function md5(str) {
|
|
return crypto.createHash('md5').update(str).digest('hex');
|
|
}
|
|
|
|
// Since the bounday in FormData is random,
|
|
// we ignore it for purposes of calculating
|
|
// the cache key.
|
|
function getFormDataCacheKey(formData) {
|
|
const cacheKey = { ...formData };
|
|
|
|
if (typeof formData.getBoundary === 'function') {
|
|
const boundary = formData.getBoundary();
|
|
|
|
// eslint-disable-next-line no-underscore-dangle
|
|
delete cacheKey._boundary;
|
|
|
|
// eslint-disable-next-line no-underscore-dangle
|
|
if (Array.isArray(cacheKey._streams)) {
|
|
const boundaryReplaceRegex = new RegExp(boundary, 'g');
|
|
|
|
// eslint-disable-next-line no-underscore-dangle
|
|
cacheKey._streams = cacheKey._streams.map((s) => {
|
|
if (typeof s === 'string') {
|
|
return s.replace(boundaryReplaceRegex, '');
|
|
}
|
|
|
|
return s;
|
|
});
|
|
}
|
|
}
|
|
|
|
return cacheKey;
|
|
}
|
|
|
|
function getBodyCacheKeyJson(body) {
|
|
if (!body) {
|
|
return body;
|
|
} if (typeof body === 'string') {
|
|
return body;
|
|
} if (body instanceof URLSearchParams) {
|
|
return body.toString();
|
|
} if (body instanceof fs.ReadStream) {
|
|
return body.path;
|
|
} if (body.toString && body.toString() === '[object FormData]') {
|
|
return getFormDataCacheKey(body);
|
|
}
|
|
|
|
throw new Error('Unsupported body type');
|
|
}
|
|
|
|
function getCacheKey(requestArguments) {
|
|
const resource = requestArguments[0];
|
|
const init = requestArguments[1] || {};
|
|
|
|
const resourceCacheKeyJson = typeof resource === 'string' ? { url: resource } : { ...resource };
|
|
const initCacheKeyJson = { ...init };
|
|
|
|
resourceCacheKeyJson.body = getBodyCacheKeyJson(resourceCacheKeyJson.body);
|
|
initCacheKeyJson.body = getBodyCacheKeyJson(initCacheKeyJson.body);
|
|
|
|
return md5(JSON.stringify([resourceCacheKeyJson, initCacheKeyJson, CACHE_VERSION]));
|
|
}
|
|
|
|
async function createRawResponse(fetchRes) {
|
|
const buffer = await fetchRes.buffer();
|
|
|
|
const rawHeaders = Array.from(fetchRes.headers.entries())
|
|
.reduce((aggregate, entry) => ({ ...aggregate, [entry[0]]: entry[1] }), {});
|
|
|
|
return {
|
|
status: fetchRes.status,
|
|
statusText: fetchRes.statusText,
|
|
type: fetchRes.type,
|
|
url: fetchRes.url,
|
|
ok: fetchRes.ok,
|
|
headers: rawHeaders,
|
|
redirected: fetchRes.redirected,
|
|
bodyBuffer: buffer,
|
|
};
|
|
}
|
|
|
|
async function getResponse(cache, requestArguments) {
|
|
const cacheKey = getCacheKey(requestArguments);
|
|
const cachedValue = await cache.get(cacheKey);
|
|
|
|
const ejectSelfFromCache = () => cache.remove(cacheKey);
|
|
|
|
if (cachedValue) {
|
|
return new Response(cachedValue, ejectSelfFromCache, true);
|
|
}
|
|
|
|
const fetchResponse = await fetch(...requestArguments);
|
|
const rawResponse = await createRawResponse(fetchResponse);
|
|
await cache.set(cacheKey, rawResponse);
|
|
return new Response(rawResponse, ejectSelfFromCache, false);
|
|
}
|
|
|
|
function createFetchWithCache(cache) {
|
|
const fetchCache = (...args) => getResponse(cache, args);
|
|
fetchCache.withCache = createFetchWithCache;
|
|
|
|
return fetchCache;
|
|
}
|
|
|
|
module.exports = createFetchWithCache(new MemoryCache());
|