55 lines
1.5 KiB
TypeScript
55 lines
1.5 KiB
TypeScript
import etagCalculate from 'etag';
|
|
import type { MiddlewareHandler } from 'hono';
|
|
import { routePath } from 'hono/route';
|
|
|
|
import { config } from '@/config';
|
|
import type { Data } from '@/types';
|
|
|
|
const headers: Record<string, string> = {
|
|
'Access-Control-Allow-Methods': 'GET',
|
|
'Content-Type': 'application/xml; charset=utf-8',
|
|
'Cache-Control': `public, max-age=${config.cache.routeExpire}`,
|
|
'X-Content-Type-Options': 'nosniff',
|
|
};
|
|
if (config.nodeName) {
|
|
headers['RSSHub-Node'] = config.nodeName;
|
|
}
|
|
|
|
function etagMatches(etag: string, ifNoneMatch: string | null) {
|
|
return ifNoneMatch !== null && ifNoneMatch.split(/,\s*/).includes(etag);
|
|
}
|
|
|
|
const middleware: MiddlewareHandler = async (ctx, next) => {
|
|
for (const key in headers) {
|
|
ctx.header(key, headers[key]);
|
|
}
|
|
ctx.header('Access-Control-Allow-Origin', config.allowOrigin || new URL(ctx.req.url).host);
|
|
|
|
await next();
|
|
const rPath = routePath(ctx);
|
|
|
|
if (rPath !== '/*') {
|
|
ctx.header('X-RSSHub-Route', rPath);
|
|
}
|
|
|
|
const data: Data = ctx.get('data');
|
|
if (!data || ctx.res.headers.get('ETag')) {
|
|
return;
|
|
}
|
|
|
|
const { lastBuildDate, ...etagData } = data;
|
|
const etag = etagCalculate(JSON.stringify(etagData));
|
|
|
|
ctx.header('ETag', etag);
|
|
|
|
const ifNoneMatch = ctx.req.header('If-None-Match') ?? null;
|
|
if (etagMatches(etag, ifNoneMatch)) {
|
|
ctx.status(304);
|
|
ctx.set('no-content', true);
|
|
} else {
|
|
ctx.header('Last-Modified', lastBuildDate);
|
|
}
|
|
};
|
|
|
|
export default middleware;
|