media.ts

// @ts-nocheck
import got from '@/utils/got';
import { config } from '@/config';
const { getDomain } = require('tldts');
const { refererMap } = require('./referer-map');

export default async (ctx) => {
    if (!config.feature.mediaProxyKey) {
        throw new Error('Internal media proxy is disabled.');
    }

    const key = ctx.req.param('key');
    if (key !== config.feature.mediaProxyKey) {
        throw new Error('Invalid media proxy key.');
    }

    const url = decodeURIComponent(ctx.req.param('url'));
    const requestUrl = new URL(url);
    const { hostname, origin } = requestUrl;

    const domain = getDomain(hostname);

    let referer = refererMap.get(domain);
    referer ||= origin;

    const { headers } = await got.head(url, {
        headers: {
            referer,
        },
    });

    const cacheControl = headers['cache-control'];
    const contentType = headers['content-type'];
    const contentLength = headers['content-length'];

    if (!contentType.startsWith('image/') || headers.server === 'RSSHub') {
        return ctx.redirect(url);
    }

    ctx.set({
        'cache-control': cacheControl || `public, max-age=${config.cache.contentExpire}`,
        'content-length': contentLength,
        'content-type': contentType,
        server: 'RSSHub',
    });

    ctx.body = await got.stream(url, {
        headers: {
            referer,
        },
    });
};