index.js 7.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367
  1. const errors = require('./lib/errors')
  2. class EventListener {
  3. constructor() {
  4. this.list = []
  5. this.count = 0
  6. }
  7. append(ctx, name, fn, once) {
  8. this.count++
  9. ctx.emit('newListener', name, fn) // Emit BEFORE adding
  10. this.list.push([fn, once])
  11. }
  12. prepend(ctx, name, fn, once) {
  13. this.count++
  14. ctx.emit('newListener', name, fn) // Emit BEFORE adding
  15. this.list.unshift([fn, once])
  16. }
  17. remove(ctx, name, fn) {
  18. for (let i = 0, n = this.list.length; i < n; i++) {
  19. const l = this.list[i]
  20. if (l[0] === fn) {
  21. this.list.splice(i, 1)
  22. if (this.count === 1) delete ctx._events[name]
  23. ctx.emit('removeListener', name, fn) // Emit AFTER removing
  24. this.count--
  25. return
  26. }
  27. }
  28. }
  29. removeAll(ctx, name) {
  30. const list = [...this.list]
  31. this.list = []
  32. if (this.count === list.length) delete ctx._events[name]
  33. for (let i = list.length - 1; i >= 0; i--) {
  34. ctx.emit('removeListener', name, list[i][0]) // Emit AFTER removing
  35. }
  36. this.count -= list.length
  37. }
  38. emit(ctx, name, ...args) {
  39. const list = [...this.list]
  40. for (let i = 0, n = list.length; i < n; i++) {
  41. const l = list[i]
  42. if (l[1] === true) this.remove(ctx, name, l[0])
  43. Reflect.apply(l[0], ctx, args)
  44. }
  45. return list.length > 0
  46. }
  47. }
  48. function appendListener(ctx, name, fn, once) {
  49. if (ctx._events === undefined) ctx._events = Object.create(null)
  50. const e = ctx._events[name] || (ctx._events[name] = new EventListener())
  51. e.append(ctx, name, fn, once)
  52. return ctx
  53. }
  54. function prependListener(ctx, name, fn, once) {
  55. if (ctx._events === undefined) ctx._events = Object.create(null)
  56. const e = ctx._events[name] || (ctx._events[name] = new EventListener())
  57. e.prepend(ctx, name, fn, once)
  58. return ctx
  59. }
  60. function removeListener(ctx, name, fn) {
  61. if (ctx._events === undefined) return ctx
  62. const e = ctx._events[name]
  63. if (e !== undefined) e.remove(ctx, name, fn)
  64. return ctx
  65. }
  66. function throwUnhandledError(...args) {
  67. let err
  68. if (args.length > 0) err = args[0]
  69. if (err instanceof Error === false) err = errors.UNHANDLED_ERROR(err)
  70. if (Error.captureStackTrace) {
  71. Error.captureStackTrace(err, exports.prototype.emit)
  72. }
  73. queueMicrotask(() => {
  74. throw err
  75. })
  76. }
  77. module.exports = exports = class EventEmitter {
  78. constructor() {
  79. this._events = Object.create(null)
  80. }
  81. addListener(name, fn) {
  82. return appendListener(this, name, fn, false)
  83. }
  84. addOnceListener(name, fn) {
  85. return appendListener(this, name, fn, true)
  86. }
  87. prependListener(name, fn) {
  88. return prependListener(this, name, fn, false)
  89. }
  90. prependOnceListener(name, fn) {
  91. return prependListener(this, name, fn, true)
  92. }
  93. removeListener(name, fn) {
  94. return removeListener(this, name, fn)
  95. }
  96. on(name, fn) {
  97. return appendListener(this, name, fn, false)
  98. }
  99. once(name, fn) {
  100. return appendListener(this, name, fn, true)
  101. }
  102. off(name, fn) {
  103. return removeListener(this, name, fn)
  104. }
  105. emit(name, ...args) {
  106. if (name === 'error' && this._events !== undefined && this._events.error === undefined) {
  107. throwUnhandledError(...args)
  108. }
  109. if (this._events === undefined) return false
  110. const e = this._events[name]
  111. return e === undefined ? false : e.emit(this, name, ...args)
  112. }
  113. listeners(name) {
  114. if (this._events === undefined) return []
  115. const e = this._events[name]
  116. return e === undefined ? [] : [...e.list]
  117. }
  118. listenerCount(name) {
  119. if (this._events === undefined) return 0
  120. const e = this._events[name]
  121. return e === undefined ? 0 : e.list.length
  122. }
  123. getMaxListeners() {
  124. return EventEmitter.defaultMaxListeners
  125. }
  126. setMaxListeners(n) {}
  127. removeAllListeners(name) {
  128. if (arguments.length === 0) {
  129. for (const key of Reflect.ownKeys(this._events)) {
  130. if (key === 'removeListener') continue
  131. this.removeAllListeners(key)
  132. }
  133. this.removeAllListeners('removeListener')
  134. } else {
  135. const e = this._events[name]
  136. if (e !== undefined) e.removeAll(this, name)
  137. }
  138. return this
  139. }
  140. }
  141. exports.EventEmitter = exports
  142. exports.errors = errors
  143. exports.defaultMaxListeners = 10
  144. exports.on = function on(emitter, name, opts = {}) {
  145. const { signal } = opts
  146. if (signal && signal.aborted) {
  147. throw errors.OPERATION_ABORTED(signal.reason)
  148. }
  149. let error = null
  150. let done = false
  151. const events = []
  152. const promises = []
  153. if (name !== 'error') emitter.on('error', onerror)
  154. if (signal) signal.addEventListener('abort', onabort)
  155. emitter.on(name, onevent)
  156. return {
  157. next() {
  158. if (events.length) {
  159. return Promise.resolve({ value: events.shift(), done: false })
  160. }
  161. if (error) {
  162. const err = error
  163. error = null
  164. return Promise.reject(err)
  165. }
  166. if (done) return onclose()
  167. return new Promise((resolve, reject) => promises.push({ resolve, reject }))
  168. },
  169. return() {
  170. return onclose()
  171. },
  172. throw(err) {
  173. return onerror(err)
  174. },
  175. [Symbol.asyncIterator]() {
  176. return this
  177. }
  178. }
  179. function onevent(...args) {
  180. if (promises.length) {
  181. promises.shift().resolve({ value: args, done: false })
  182. } else {
  183. events.push(args)
  184. }
  185. }
  186. function onerror(err) {
  187. emitter.off(name, onevent).off('error', onerror)
  188. if (promises.length) {
  189. promises.shift().reject(err)
  190. } else {
  191. error = err
  192. }
  193. return Promise.resolve({ done: true })
  194. }
  195. function onabort() {
  196. signal.removeEventListener('abort', onabort)
  197. onerror(errors.OPERATION_ABORTED(signal.reason))
  198. }
  199. function onclose() {
  200. emitter.off(name, onevent)
  201. if (name !== 'error') emitter.off('error', onerror)
  202. if (signal) signal.removeEventListener('abort', onabort)
  203. done = true
  204. if (promises.length) promises.shift().resolve({ done: true })
  205. return Promise.resolve({ done: true })
  206. }
  207. }
  208. exports.once = function once(emitter, name, opts = {}) {
  209. const { signal } = opts
  210. if (signal && signal.aborted) {
  211. return Promise.reject(errors.OPERATION_ABORTED(signal.reason))
  212. }
  213. return new Promise((resolve, reject) => {
  214. if (name !== 'error') emitter.on('error', onerror)
  215. if (signal) signal.addEventListener('abort', onabort)
  216. emitter.once(name, onevent)
  217. function onevent(...args) {
  218. if (name !== 'error') emitter.off('error', onerror)
  219. if (signal) signal.removeEventListener('abort', onabort)
  220. resolve(args)
  221. }
  222. function onerror(err) {
  223. emitter.off(name, onevent)
  224. if (name !== 'error') emitter.off('error', onerror)
  225. reject(err)
  226. }
  227. function onabort() {
  228. signal.removeEventListener('abort', onabort)
  229. onerror(errors.OPERATION_ABORTED(signal.reason))
  230. }
  231. })
  232. }
  233. exports.forward = function forward(from, to, names, opts = {}) {
  234. if (typeof names === 'string') names = [names]
  235. const { emit = to.emit.bind(to) } = opts
  236. const listeners = names.map(
  237. (name) =>
  238. function onevent(...args) {
  239. emit(name, ...args)
  240. }
  241. )
  242. to.on('newListener', (name) => {
  243. const i = names.indexOf(name)
  244. if (i !== -1 && to.listenerCount(name) === 0) {
  245. from.on(name, listeners[i])
  246. }
  247. }).on('removeListener', (name) => {
  248. const i = names.indexOf(name)
  249. if (i !== -1 && to.listenerCount(name) === 0) {
  250. from.off(name, listeners[i])
  251. }
  252. })
  253. }
  254. exports.listenerCount = function listenerCount(emitter, name) {
  255. return emitter.listenerCount(name)
  256. }
  257. exports.getMaxListeners = function getMaxListeners(emitter) {
  258. if (typeof emitter.getMaxListeners === 'function') {
  259. return emitter.getMaxListeners()
  260. }
  261. return exports.defaultMaxListeners
  262. }
  263. exports.setMaxListeners = function setMaxListeners(n, ...emitters) {
  264. if (emitters.length === 0) exports.defaultMaxListeners = n
  265. else {
  266. for (const emitter of emitters) {
  267. if (typeof emitter.setMaxListeners === 'function') {
  268. emitter.setMaxListeners(n)
  269. }
  270. }
  271. }
  272. }