Press n or j to go to the next uncovered block, b, p or k for the previous block.
| 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 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 | 2296x 2296x 2296x 2296x 2296x 2296x 230x 2296x 585x 2296x 746x 2296x 422x 2296x 125x 2296x 15x 2296x 8x 2296x 14x 2296x 2276x 20x 20x 16x 20x 20x 1x 19x 16x 20x 16x | /**
* StatementExpressionCollector
*
* Collects all expressions from a statement context that need to be walked
* for function calls and parameter modifications. This centralizes expression
* extraction to prevent missing cases (like issue #565).
*
* Issue #566: Extracted from CodeGenerator for improved testability.
*/
import * as Parser from "../../parser/grammar/CNextParser.js";
class StatementExpressionCollector {
/**
* Collect all expressions from a statement that need to be walked for
* function calls and modifications.
*
* Handles:
* - Simple statements (expression, assignment, variable declaration, return)
* - Control flow conditions (if, while, do-while, switch)
* - For statement parts (init, condition, update)
*/
static collectAll(stmt: Parser.StatementContext): Parser.ExpressionContext[] {
const expressions: Parser.ExpressionContext[] = [];
// Simple statements with expressions
StatementExpressionCollector._collectSimpleStatementExprs(
stmt,
expressions,
);
// Control flow conditions
StatementExpressionCollector._collectControlFlowExprs(stmt, expressions);
// For statement has multiple expression contexts
StatementExpressionCollector._collectForStatementExprs(stmt, expressions);
return expressions;
}
/**
* Collect expressions from simple statements (expression, assignment,
* variable declaration, return).
*/
private static _collectSimpleStatementExprs(
stmt: Parser.StatementContext,
expressions: Parser.ExpressionContext[],
): void {
if (stmt.expressionStatement()) {
expressions.push(stmt.expressionStatement()!.expression());
}
if (stmt.assignmentStatement()) {
expressions.push(stmt.assignmentStatement()!.expression());
}
if (stmt.variableDeclaration()?.expression()) {
expressions.push(stmt.variableDeclaration()!.expression()!);
}
if (stmt.returnStatement()?.expression()) {
expressions.push(stmt.returnStatement()!.expression()!);
}
}
/**
* Collect expressions from control flow conditions (if, while, do-while, switch).
*/
private static _collectControlFlowExprs(
stmt: Parser.StatementContext,
expressions: Parser.ExpressionContext[],
): void {
if (stmt.ifStatement()) {
expressions.push(stmt.ifStatement()!.expression());
}
if (stmt.whileStatement()) {
expressions.push(stmt.whileStatement()!.expression());
}
if (stmt.doWhileStatement()) {
expressions.push(stmt.doWhileStatement()!.expression());
}
if (stmt.switchStatement()) {
expressions.push(stmt.switchStatement()!.expression());
}
}
/**
* Collect expressions from for statement parts (init, condition, update).
*/
private static _collectForStatementExprs(
stmt: Parser.StatementContext,
expressions: Parser.ExpressionContext[],
): void {
if (!stmt.forStatement()) {
return;
}
const forStmt = stmt.forStatement()!;
// Condition (optional)
if (forStmt.expression()) {
expressions.push(forStmt.expression()!);
}
// forInit expressions
const forInit = forStmt.forInit();
if (forInit?.forAssignment()) {
expressions.push(forInit.forAssignment()!.expression());
} else if (forInit?.forVarDecl()?.expression()) {
expressions.push(forInit.forVarDecl()!.expression()!);
}
// forUpdate expression
if (forStmt.forUpdate()) {
expressions.push(forStmt.forUpdate()!.expression());
}
}
}
export default StatementExpressionCollector;
|