antongolub/npm-registry-firewall

import http from 'node:http'
import https from 'node:https'
import {getConfig} from '../config.js'
import {isPlainObject} from '../util.js'

const agents = {
  http: new Map(),
  https: new Map()
}

const agentOpts = {
  keepAliveMsecs: 20_000,
  keepAlive: true,
  maxSockets: 1_000,
  timeout: 20_000
}

export const getAgent = (secure) => {
  const opts = getConfig()?.agent || agentOpts
  const map = secure ? agents.https : agents.http
  if (!map.has(opts)) {
    const value = isPlainObject(opts) ? new (secure ? https.Agent : http.Agent)(opts) : opts
    map.set(opts, value)
  }

  return map.get(opts)
}