after.ts

export function after(
  $pos: ResolvedPos,
  predicate: (node: Node, pos: number) => boolean
): { $pos: ResolvedPos; node: Node; parent: Node } | null {
  let result: { $pos: ResolvedPos; node: Node; parent: Node } | null = null;
  $pos.doc.nodesBetween($pos.pos, $pos.doc.content.size, (node, pos, parent) => {
    if (pos < $pos.pos) {
      // Don't recurse into nodes that are completely before the position we're
      // interested in.
      if (pos + node.nodeSize < $pos.pos) {
        return false;
      }
      // Skip nodes that are before the position we're interested in.
      return;
    }

    if (predicate(node, pos)) {
      result = {
        $pos: $pos.doc.resolve(pos),
        node: node,
        parent: parent
      };
    }

    // `false` needs to be returned repeatedly until `nodesBetween` bubbles up
    // to the root and exits.
    if (result !== null) {
      return false;
    }

    // Continue
    return;
  });
  return result;
}