Files
RSSHub/lib/middleware/header.js

39 lines
1.1 KiB
JavaScript

const etagCalculate = require('etag');
const logger = require('../utils/logger');
const config = require('../config');
const headers = {
'Access-Control-Allow-Origin': '*',
'Access-Control-Allow-Headers': 'Content-Type, Content-Length, Authorization, Accept, X-Requested-With',
'Access-Control-Allow-Methods': 'GET',
'Content-Type': 'application/xml; charset=utf-8',
'Cache-Control': `public, max-age=${config.cacheExpire}`,
};
module.exports = async (ctx, next) => {
logger.info(`${ctx.url}, user IP: ${ctx.ips[0] || ctx.ip}`);
ctx.set(headers);
await next();
let body = ctx.body;
if (!body || typeof body !== 'string' || ctx.response.get('ETag')) {
return;
}
const status = (ctx.status / 100) | 0;
if (2 !== status) {
return;
}
const match = body.match(/<lastBuildDate>(.*)<\/lastBuildDate>/);
if (match) {
ctx.set({
'Last-Modified': body.match(/<lastBuildDate>(.*)<\/lastBuildDate>/)[1],
});
}
body = body.replace(/<lastBuildDate>(.*)<\/lastBuildDate>/, '');
ctx.response.etag = etagCalculate(body);
};