[6a3a178] | 1 | "use strict";
|
---|
| 2 |
|
---|
| 3 | Object.defineProperty(exports, "__esModule", {
|
---|
| 4 | value: true
|
---|
| 5 | });
|
---|
| 6 | exports.default = traverse;
|
---|
| 7 |
|
---|
| 8 | var _definitions = require("../definitions");
|
---|
| 9 |
|
---|
| 10 | function traverse(node, handlers, state) {
|
---|
| 11 | if (typeof handlers === "function") {
|
---|
| 12 | handlers = {
|
---|
| 13 | enter: handlers
|
---|
| 14 | };
|
---|
| 15 | }
|
---|
| 16 |
|
---|
| 17 | const {
|
---|
| 18 | enter,
|
---|
| 19 | exit
|
---|
| 20 | } = handlers;
|
---|
| 21 | traverseSimpleImpl(node, enter, exit, state, []);
|
---|
| 22 | }
|
---|
| 23 |
|
---|
| 24 | function traverseSimpleImpl(node, enter, exit, state, ancestors) {
|
---|
| 25 | const keys = _definitions.VISITOR_KEYS[node.type];
|
---|
| 26 | if (!keys) return;
|
---|
| 27 | if (enter) enter(node, ancestors, state);
|
---|
| 28 |
|
---|
| 29 | for (const key of keys) {
|
---|
| 30 | const subNode = node[key];
|
---|
| 31 |
|
---|
| 32 | if (Array.isArray(subNode)) {
|
---|
| 33 | for (let i = 0; i < subNode.length; i++) {
|
---|
| 34 | const child = subNode[i];
|
---|
| 35 | if (!child) continue;
|
---|
| 36 | ancestors.push({
|
---|
| 37 | node,
|
---|
| 38 | key,
|
---|
| 39 | index: i
|
---|
| 40 | });
|
---|
| 41 | traverseSimpleImpl(child, enter, exit, state, ancestors);
|
---|
| 42 | ancestors.pop();
|
---|
| 43 | }
|
---|
| 44 | } else if (subNode) {
|
---|
| 45 | ancestors.push({
|
---|
| 46 | node,
|
---|
| 47 | key
|
---|
| 48 | });
|
---|
| 49 | traverseSimpleImpl(subNode, enter, exit, state, ancestors);
|
---|
| 50 | ancestors.pop();
|
---|
| 51 | }
|
---|
| 52 | }
|
---|
| 53 |
|
---|
| 54 | if (exit) exit(node, ancestors, state);
|
---|
| 55 | } |
---|