@directus/api
Version:
Directus is a real-time API and App dashboard for managing SQL database content
93 lines (92 loc) • 4.16 kB
JavaScript
import { useEnv } from '@directus/env';
import { parse as parseBytesConfiguration } from 'bytes';
import { getCache, setCacheValue } from '../cache.js';
import getDatabase from '../database/index.js';
import { useLogger } from '../logger/index.js';
import { ExportService } from '../services/import-export.js';
import asyncHandler from '../utils/async-handler.js';
import { getCacheControlHeader } from '../utils/get-cache-headers.js';
import { getCacheKey } from '../utils/get-cache-key.js';
import { getDateFormatted } from '../utils/get-date-formatted.js';
import { getMilliseconds } from '../utils/get-milliseconds.js';
import { stringByteSize } from '../utils/get-string-byte-size.js';
import { permissionsCachable } from '../utils/permissions-cachable.js';
export const respond = asyncHandler(async (req, res) => {
const env = useEnv();
const logger = useLogger();
const { cache } = getCache();
let exceedsMaxSize = false;
if (env['CACHE_VALUE_MAX_SIZE'] !== false) {
const valueSize = res.locals['payload'] ? stringByteSize(JSON.stringify(res.locals['payload'])) : 0;
const maxSize = parseBytesConfiguration(env['CACHE_VALUE_MAX_SIZE']);
if (maxSize !== null)
exceedsMaxSize = valueSize > maxSize;
}
if ((req.method.toLowerCase() === 'get' || req.originalUrl?.startsWith('/graphql')) &&
req.originalUrl?.startsWith('/auth') === false &&
env['CACHE_ENABLED'] === true &&
cache &&
!req.sanitizedQuery.export &&
res.locals['cache'] !== false &&
exceedsMaxSize === false &&
(await permissionsCachable(req.collection, {
knex: getDatabase(),
schema: req.schema,
}, req.accountability))) {
const key = await getCacheKey(req);
try {
await setCacheValue(cache, key, res.locals['payload'], getMilliseconds(env['CACHE_TTL']));
await setCacheValue(cache, `${key}__expires_at`, { exp: Date.now() + getMilliseconds(env['CACHE_TTL'], 0) });
}
catch (err) {
logger.warn(err, `[cache] Couldn't set key ${key}. ${err}`);
}
res.setHeader('Cache-Control', getCacheControlHeader(req, getMilliseconds(env['CACHE_TTL']), true, true));
res.setHeader('Vary', 'Origin, Cache-Control');
}
else {
// Don't cache anything by default
res.setHeader('Cache-Control', 'no-cache');
res.setHeader('Vary', 'Origin, Cache-Control');
}
if (req.sanitizedQuery.export) {
const exportService = new ExportService({ accountability: req.accountability ?? null, schema: req.schema });
let filename = '';
if (req.collection) {
filename += req.collection;
}
else {
filename += 'Export';
}
filename += ' ' + getDateFormatted();
if (req.sanitizedQuery.export === 'json') {
res.attachment(`${filename}.json`);
res.set('Content-Type', 'application/json');
return res.status(200).send(exportService.transform(res.locals['payload']?.data, 'json'));
}
if (req.sanitizedQuery.export === 'xml') {
res.attachment(`${filename}.xml`);
res.set('Content-Type', 'text/xml');
return res.status(200).send(exportService.transform(res.locals['payload']?.data, 'xml'));
}
if (req.sanitizedQuery.export === 'csv') {
res.attachment(`${filename}.csv`);
res.set('Content-Type', 'text/csv');
return res.status(200).send(exportService.transform(res.locals['payload']?.data, 'csv'));
}
if (req.sanitizedQuery.export === 'yaml') {
res.attachment(`${filename}.yaml`);
res.set('Content-Type', 'text/yaml');
return res.status(200).send(exportService.transform(res.locals['payload']?.data, 'yaml'));
}
}
if (Buffer.isBuffer(res.locals['payload'])) {
return res.end(res.locals['payload']);
}
else if (res.locals['payload']) {
return res.json(res.locals['payload']);
}
else {
return res.status(204).end();
}
});