index.js 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169
  1. 'use strict'
  2. const Client = require('./lib/dispatcher/client')
  3. const Dispatcher = require('./lib/dispatcher/dispatcher')
  4. const Pool = require('./lib/dispatcher/pool')
  5. const BalancedPool = require('./lib/dispatcher/balanced-pool')
  6. const Agent = require('./lib/dispatcher/agent')
  7. const ProxyAgent = require('./lib/dispatcher/proxy-agent')
  8. const EnvHttpProxyAgent = require('./lib/dispatcher/env-http-proxy-agent')
  9. const RetryAgent = require('./lib/dispatcher/retry-agent')
  10. const errors = require('./lib/core/errors')
  11. const util = require('./lib/core/util')
  12. const { InvalidArgumentError } = errors
  13. const api = require('./lib/api')
  14. const buildConnector = require('./lib/core/connect')
  15. const MockClient = require('./lib/mock/mock-client')
  16. const MockAgent = require('./lib/mock/mock-agent')
  17. const MockPool = require('./lib/mock/mock-pool')
  18. const mockErrors = require('./lib/mock/mock-errors')
  19. const RetryHandler = require('./lib/handler/retry-handler')
  20. const { getGlobalDispatcher, setGlobalDispatcher } = require('./lib/global')
  21. const DecoratorHandler = require('./lib/handler/decorator-handler')
  22. const RedirectHandler = require('./lib/handler/redirect-handler')
  23. const createRedirectInterceptor = require('./lib/interceptor/redirect-interceptor')
  24. Object.assign(Dispatcher.prototype, api)
  25. module.exports.Dispatcher = Dispatcher
  26. module.exports.Client = Client
  27. module.exports.Pool = Pool
  28. module.exports.BalancedPool = BalancedPool
  29. module.exports.Agent = Agent
  30. module.exports.ProxyAgent = ProxyAgent
  31. module.exports.EnvHttpProxyAgent = EnvHttpProxyAgent
  32. module.exports.RetryAgent = RetryAgent
  33. module.exports.RetryHandler = RetryHandler
  34. module.exports.DecoratorHandler = DecoratorHandler
  35. module.exports.RedirectHandler = RedirectHandler
  36. module.exports.createRedirectInterceptor = createRedirectInterceptor
  37. module.exports.interceptors = {
  38. redirect: require('./lib/interceptor/redirect'),
  39. retry: require('./lib/interceptor/retry'),
  40. dump: require('./lib/interceptor/dump'),
  41. dns: require('./lib/interceptor/dns')
  42. }
  43. module.exports.buildConnector = buildConnector
  44. module.exports.errors = errors
  45. module.exports.util = {
  46. parseHeaders: util.parseHeaders,
  47. headerNameToString: util.headerNameToString
  48. }
  49. function makeDispatcher (fn) {
  50. return (url, opts, handler) => {
  51. if (typeof opts === 'function') {
  52. handler = opts
  53. opts = null
  54. }
  55. if (!url || (typeof url !== 'string' && typeof url !== 'object' && !(url instanceof URL))) {
  56. throw new InvalidArgumentError('invalid url')
  57. }
  58. if (opts != null && typeof opts !== 'object') {
  59. throw new InvalidArgumentError('invalid opts')
  60. }
  61. if (opts && opts.path != null) {
  62. if (typeof opts.path !== 'string') {
  63. throw new InvalidArgumentError('invalid opts.path')
  64. }
  65. let path = opts.path
  66. if (!opts.path.startsWith('/')) {
  67. path = `/${path}`
  68. }
  69. url = new URL(util.parseOrigin(url).origin + path)
  70. } else {
  71. if (!opts) {
  72. opts = typeof url === 'object' ? url : {}
  73. }
  74. url = util.parseURL(url)
  75. }
  76. const { agent, dispatcher = getGlobalDispatcher() } = opts
  77. if (agent) {
  78. throw new InvalidArgumentError('unsupported opts.agent. Did you mean opts.client?')
  79. }
  80. return fn.call(dispatcher, {
  81. ...opts,
  82. origin: url.origin,
  83. path: url.search ? `${url.pathname}${url.search}` : url.pathname,
  84. method: opts.method || (opts.body ? 'PUT' : 'GET')
  85. }, handler)
  86. }
  87. }
  88. module.exports.setGlobalDispatcher = setGlobalDispatcher
  89. module.exports.getGlobalDispatcher = getGlobalDispatcher
  90. const fetchImpl = require('./lib/web/fetch').fetch
  91. module.exports.fetch = async function fetch (init, options = undefined) {
  92. try {
  93. return await fetchImpl(init, options)
  94. } catch (err) {
  95. if (err && typeof err === 'object') {
  96. Error.captureStackTrace(err)
  97. }
  98. throw err
  99. }
  100. }
  101. module.exports.Headers = require('./lib/web/fetch/headers').Headers
  102. module.exports.Response = require('./lib/web/fetch/response').Response
  103. module.exports.Request = require('./lib/web/fetch/request').Request
  104. module.exports.FormData = require('./lib/web/fetch/formdata').FormData
  105. module.exports.File = globalThis.File ?? require('node:buffer').File
  106. module.exports.FileReader = require('./lib/web/fileapi/filereader').FileReader
  107. const { setGlobalOrigin, getGlobalOrigin } = require('./lib/web/fetch/global')
  108. module.exports.setGlobalOrigin = setGlobalOrigin
  109. module.exports.getGlobalOrigin = getGlobalOrigin
  110. const { CacheStorage } = require('./lib/web/cache/cachestorage')
  111. const { kConstruct } = require('./lib/web/cache/symbols')
  112. // Cache & CacheStorage are tightly coupled with fetch. Even if it may run
  113. // in an older version of Node, it doesn't have any use without fetch.
  114. module.exports.caches = new CacheStorage(kConstruct)
  115. const { deleteCookie, getCookies, getSetCookies, setCookie } = require('./lib/web/cookies')
  116. module.exports.deleteCookie = deleteCookie
  117. module.exports.getCookies = getCookies
  118. module.exports.getSetCookies = getSetCookies
  119. module.exports.setCookie = setCookie
  120. const { parseMIMEType, serializeAMimeType } = require('./lib/web/fetch/data-url')
  121. module.exports.parseMIMEType = parseMIMEType
  122. module.exports.serializeAMimeType = serializeAMimeType
  123. const { CloseEvent, ErrorEvent, MessageEvent } = require('./lib/web/websocket/events')
  124. module.exports.WebSocket = require('./lib/web/websocket/websocket').WebSocket
  125. module.exports.CloseEvent = CloseEvent
  126. module.exports.ErrorEvent = ErrorEvent
  127. module.exports.MessageEvent = MessageEvent
  128. module.exports.request = makeDispatcher(api.request)
  129. module.exports.stream = makeDispatcher(api.stream)
  130. module.exports.pipeline = makeDispatcher(api.pipeline)
  131. module.exports.connect = makeDispatcher(api.connect)
  132. module.exports.upgrade = makeDispatcher(api.upgrade)
  133. module.exports.MockClient = MockClient
  134. module.exports.MockPool = MockPool
  135. module.exports.MockAgent = MockAgent
  136. module.exports.mockErrors = mockErrors
  137. const { EventSource } = require('./lib/web/eventsource/eventsource')
  138. module.exports.EventSource = EventSource