@atlaskit/editor-common/dist/es2019/utils/traversor.js

A package that contains common classes and components for editor and renderer

import _defineProperty from "@babel/runtime/helpers/defineProperty";
class ADFTraversor {
  constructor(doc) {
    _defineProperty(this, "subscribers", new Map());
    this.doc = doc;
  }
  subscribe(type, callback) {
    const callbacks = this.subscribers.get(type);
    if (!callbacks) {
      this.subscribers.set(type, [callback]);
    } else {
      callbacks.push(callback);
    }
  }
  exec() {
    if (!this.doc || !Array.isArray(this.doc.content)) {
      return;
    }
    let candidates = this.doc.content.slice(0);
    while (candidates.length) {
      const node = candidates.shift();
      if (Array.isArray(node.content)) {
        candidates = candidates.concat(node.content);
      }
      const callbacks = this.subscribers.get(node.type);
      if (!callbacks) {
        continue;
      }
      for (let i = 0; i < callbacks.length; i++) {
        callbacks[i](node);
      }
    }
  }
}
export default ADFTraversor;