RocketChat/Rocket.Chat

View on GitHub
packages/server-fetch/src/index.ts

Summary

Maintainability
A
55 mins
Test Coverage
import http from 'http';
import https from 'https';

import { AbortController } from 'abort-controller';
import type { HttpProxyAgent } from 'http-proxy-agent';
import { default as createHttpProxyAgent } from 'http-proxy-agent';
import type { HttpsProxyAgent } from 'https-proxy-agent';
import { default as createHttpsProxyAgent } from 'https-proxy-agent';
import fetch from 'node-fetch';
import { getProxyForUrl } from 'proxy-from-env';

import { parseRequestOptions } from './parsers';
import type { ExtendedFetchOptions } from './types';

function getFetchAgent(url: string, allowSelfSignedCerts?: boolean): http.Agent | https.Agent | null | HttpsProxyAgent | HttpProxyAgent {
    const isHttps = /^https/.test(url);

    const proxy = getProxyForUrl(url);
    if (proxy) {
        const agentFn = isHttps ? createHttpsProxyAgent : createHttpProxyAgent;
        return agentFn(proxy);
    }

    if (!allowSelfSignedCerts) {
        return null;
    }

    if (!isHttps) {
        return new http.Agent();
    }

    if (isHttps) {
        return new https.Agent({
            rejectUnauthorized: false,
        });
    }
    return null;
}

function getTimeout(timeout?: number) {
    const controller = new AbortController();
    const timeoutId = setTimeout(() => controller.abort(), timeout ?? 20000);

    return { controller, timeoutId };
}

export function serverFetch(input: string, options?: ExtendedFetchOptions, allowSelfSignedCerts?: boolean): ReturnType<typeof fetch> {
    const agent = getFetchAgent(input, allowSelfSignedCerts);
    const { controller, timeoutId } = getTimeout(options?.timeout);

    // Keeping the URLSearchParams since it handles other cases and type conversions
    const params = new URLSearchParams(options?.params);
    const url = new URL(input);
    if (params.toString()) {
        params.forEach((value, key) => {
            if (value) {
                url.searchParams.append(key, value);
            }
        });
    }

    return fetch(url.toString(), {
        // @ts-expect-error - This complained when types were moved to file :/
        signal: controller.signal,
        ...parseRequestOptions(options),
        ...(agent ? { agent } : {}),
    }).finally(() => {
        if (timeoutId) {
            clearTimeout(timeoutId);
        }
    });
}

export { Response } from 'node-fetch';