Spaces:
Running
Running
/** | |
* @fileoverview A class of the code path. | |
* @author Toru Nagashima | |
*/ | |
; | |
//------------------------------------------------------------------------------ | |
// Requirements | |
//------------------------------------------------------------------------------ | |
const CodePathState = require("./code-path-state"); | |
const IdGenerator = require("./id-generator"); | |
//------------------------------------------------------------------------------ | |
// Public Interface | |
//------------------------------------------------------------------------------ | |
/** | |
* A code path. | |
*/ | |
class CodePath { | |
/** | |
* Creates a new instance. | |
* @param {Object} options Options for the function (see below). | |
* @param {string} options.id An identifier. | |
* @param {string} options.origin The type of code path origin. | |
* @param {CodePath|null} options.upper The code path of the upper function scope. | |
* @param {Function} options.onLooped A callback function to notify looping. | |
*/ | |
constructor({ id, origin, upper, onLooped }) { | |
/** | |
* The identifier of this code path. | |
* Rules use it to store additional information of each rule. | |
* @type {string} | |
*/ | |
this.id = id; | |
/** | |
* The reason that this code path was started. May be "program", | |
* "function", "class-field-initializer", or "class-static-block". | |
* @type {string} | |
*/ | |
this.origin = origin; | |
/** | |
* The code path of the upper function scope. | |
* @type {CodePath|null} | |
*/ | |
this.upper = upper; | |
/** | |
* The code paths of nested function scopes. | |
* @type {CodePath[]} | |
*/ | |
this.childCodePaths = []; | |
// Initializes internal state. | |
Object.defineProperty( | |
this, | |
"internal", | |
{ value: new CodePathState(new IdGenerator(`${id}_`), onLooped) } | |
); | |
// Adds this into `childCodePaths` of `upper`. | |
if (upper) { | |
upper.childCodePaths.push(this); | |
} | |
} | |
/** | |
* Gets the state of a given code path. | |
* @param {CodePath} codePath A code path to get. | |
* @returns {CodePathState} The state of the code path. | |
*/ | |
static getState(codePath) { | |
return codePath.internal; | |
} | |
/** | |
* The initial code path segment. This is the segment that is at the head | |
* of the code path. | |
* This is a passthrough to the underlying `CodePathState`. | |
* @type {CodePathSegment} | |
*/ | |
get initialSegment() { | |
return this.internal.initialSegment; | |
} | |
/** | |
* Final code path segments. These are the terminal (tail) segments in the | |
* code path, which is the combination of `returnedSegments` and `thrownSegments`. | |
* All segments in this array are reachable. | |
* This is a passthrough to the underlying `CodePathState`. | |
* @type {CodePathSegment[]} | |
*/ | |
get finalSegments() { | |
return this.internal.finalSegments; | |
} | |
/** | |
* Final code path segments that represent normal completion of the code path. | |
* For functions, this means both explicit `return` statements and implicit returns, | |
* such as the last reachable segment in a function that does not have an | |
* explicit `return` as this implicitly returns `undefined`. For scripts, | |
* modules, class field initializers, and class static blocks, this means | |
* all lines of code have been executed. | |
* These segments are also present in `finalSegments`. | |
* This is a passthrough to the underlying `CodePathState`. | |
* @type {CodePathSegment[]} | |
*/ | |
get returnedSegments() { | |
return this.internal.returnedForkContext; | |
} | |
/** | |
* Final code path segments that represent `throw` statements. | |
* This is a passthrough to the underlying `CodePathState`. | |
* These segments are also present in `finalSegments`. | |
* @type {CodePathSegment[]} | |
*/ | |
get thrownSegments() { | |
return this.internal.thrownForkContext; | |
} | |
/** | |
* Tracks the traversal of the code path through each segment. This array | |
* starts empty and segments are added or removed as the code path is | |
* traversed. This array always ends up empty at the end of a code path | |
* traversal. The `CodePathState` uses this to track its progress through | |
* the code path. | |
* This is a passthrough to the underlying `CodePathState`. | |
* @type {CodePathSegment[]} | |
* @deprecated | |
*/ | |
get currentSegments() { | |
return this.internal.currentSegments; | |
} | |
/** | |
* Traverses all segments in this code path. | |
* | |
* codePath.traverseSegments((segment, controller) => { | |
* // do something. | |
* }); | |
* | |
* This method enumerates segments in order from the head. | |
* | |
* The `controller` argument has two methods: | |
* | |
* - `skip()` - skips the following segments in this branch | |
* - `break()` - skips all following segments in the traversal | |
* | |
* A note on the parameters: the `options` argument is optional. This means | |
* the first argument might be an options object or the callback function. | |
* @param {Object} [optionsOrCallback] Optional first and last segments to traverse. | |
* @param {CodePathSegment} [optionsOrCallback.first] The first segment to traverse. | |
* @param {CodePathSegment} [optionsOrCallback.last] The last segment to traverse. | |
* @param {Function} callback A callback function. | |
* @returns {void} | |
*/ | |
traverseSegments(optionsOrCallback, callback) { | |
// normalize the arguments into a callback and options | |
let resolvedOptions; | |
let resolvedCallback; | |
if (typeof optionsOrCallback === "function") { | |
resolvedCallback = optionsOrCallback; | |
resolvedOptions = {}; | |
} else { | |
resolvedOptions = optionsOrCallback || {}; | |
resolvedCallback = callback; | |
} | |
// determine where to start traversing from based on the options | |
const startSegment = resolvedOptions.first || this.internal.initialSegment; | |
const lastSegment = resolvedOptions.last; | |
// set up initial location information | |
let record = null; | |
let index = 0; | |
let end = 0; | |
let segment = null; | |
// segments that have already been visited during traversal | |
const visited = new Set(); | |
// tracks the traversal steps | |
const stack = [[startSegment, 0]]; | |
// tracks the last skipped segment during traversal | |
let skippedSegment = null; | |
// indicates if we exited early from the traversal | |
let broken = false; | |
/** | |
* Maintains traversal state. | |
*/ | |
const controller = { | |
/** | |
* Skip the following segments in this branch. | |
* @returns {void} | |
*/ | |
skip() { | |
if (stack.length <= 1) { | |
broken = true; | |
} else { | |
skippedSegment = stack[stack.length - 2][0]; | |
} | |
}, | |
/** | |
* Stop traversal completely - do not traverse to any | |
* other segments. | |
* @returns {void} | |
*/ | |
break() { | |
broken = true; | |
} | |
}; | |
/** | |
* Checks if a given previous segment has been visited. | |
* @param {CodePathSegment} prevSegment A previous segment to check. | |
* @returns {boolean} `true` if the segment has been visited. | |
*/ | |
function isVisited(prevSegment) { | |
return ( | |
visited.has(prevSegment) || | |
segment.isLoopedPrevSegment(prevSegment) | |
); | |
} | |
// the traversal | |
while (stack.length > 0) { | |
/* | |
* This isn't a pure stack. We use the top record all the time | |
* but don't always pop it off. The record is popped only if | |
* one of the following is true: | |
* | |
* 1) We have already visited the segment. | |
* 2) We have not visited *all* of the previous segments. | |
* 3) We have traversed past the available next segments. | |
* | |
* Otherwise, we just read the value and sometimes modify the | |
* record as we traverse. | |
*/ | |
record = stack[stack.length - 1]; | |
segment = record[0]; | |
index = record[1]; | |
if (index === 0) { | |
// Skip if this segment has been visited already. | |
if (visited.has(segment)) { | |
stack.pop(); | |
continue; | |
} | |
// Skip if all previous segments have not been visited. | |
if (segment !== startSegment && | |
segment.prevSegments.length > 0 && | |
!segment.prevSegments.every(isVisited) | |
) { | |
stack.pop(); | |
continue; | |
} | |
// Reset the skipping flag if all branches have been skipped. | |
if (skippedSegment && segment.prevSegments.includes(skippedSegment)) { | |
skippedSegment = null; | |
} | |
visited.add(segment); | |
/* | |
* If the most recent segment hasn't been skipped, then we call | |
* the callback, passing in the segment and the controller. | |
*/ | |
if (!skippedSegment) { | |
resolvedCallback.call(this, segment, controller); | |
// exit if we're at the last segment | |
if (segment === lastSegment) { | |
controller.skip(); | |
} | |
/* | |
* If the previous statement was executed, or if the callback | |
* called a method on the controller, we might need to exit the | |
* loop, so check for that and break accordingly. | |
*/ | |
if (broken) { | |
break; | |
} | |
} | |
} | |
// Update the stack. | |
end = segment.nextSegments.length - 1; | |
if (index < end) { | |
/* | |
* If we haven't yet visited all of the next segments, update | |
* the current top record on the stack to the next index to visit | |
* and then push a record for the current segment on top. | |
* | |
* Setting the current top record's index lets us know how many | |
* times we've been here and ensures that the segment won't be | |
* reprocessed (because we only process segments with an index | |
* of 0). | |
*/ | |
record[1] += 1; | |
stack.push([segment.nextSegments[index], 0]); | |
} else if (index === end) { | |
/* | |
* If we are at the last next segment, then reset the top record | |
* in the stack to next segment and set its index to 0 so it will | |
* be processed next. | |
*/ | |
record[0] = segment.nextSegments[index]; | |
record[1] = 0; | |
} else { | |
/* | |
* If index > end, that means we have no more segments that need | |
* processing. So, we pop that record off of the stack in order to | |
* continue traversing at the next level up. | |
*/ | |
stack.pop(); | |
} | |
} | |
} | |
} | |
module.exports = CodePath; | |