summaryrefslogtreecommitdiffstats
path: root/devtools/client/debugger/src/workers/parser/steps.js
blob: 4847294460a3d62b268a6001c6d32e4ac706bce2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at <http://mozilla.org/MPL/2.0/>. */

// @flow

import * as t from "@babel/types";
import type { SimplePath } from "./utils/simple-path";
import type { SourceLocation, SourceId } from "../../types";
import type { AstPosition } from "./types";
import { getClosestPath } from "./utils/closest";
import { isAwaitExpression, isYieldExpression } from "./utils/helpers";

export function getNextStep(
  sourceId: SourceId,
  pausedPosition: AstPosition
): ?SourceLocation {
  const currentExpression = getSteppableExpression(sourceId, pausedPosition);
  if (!currentExpression) {
    return null;
  }

  const currentStatement = currentExpression.find(p => {
    return p.inList && t.isStatement(p.node);
  });

  if (!currentStatement) {
    throw new Error(
      "Assertion failure - this should always find at least Program"
    );
  }

  return _getNextStep(currentStatement, sourceId, pausedPosition);
}

function getSteppableExpression(
  sourceId: SourceId,
  pausedPosition: AstPosition
) {
  const closestPath = getClosestPath(sourceId, pausedPosition);

  if (!closestPath) {
    return null;
  }

  if (isAwaitExpression(closestPath) || isYieldExpression(closestPath)) {
    return closestPath;
  }

  return closestPath.find(
    p => t.isAwaitExpression(p.node) || t.isYieldExpression(p.node)
  );
}

function _getNextStep(
  statement: SimplePath,
  sourceId: SourceId,
  position: AstPosition
): ?SourceLocation {
  const nextStatement = statement.getSibling(1);
  if (nextStatement) {
    return {
      ...nextStatement.node.loc.start,
      sourceId,
    };
  }

  return null;
}