hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2026-01-08 12:10:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

JavaScript: Add support for parsing postfix generator comprehensions.

Browse Source
This commit is contained in:
Max Schaefer
2019-01-29 15:53:11 +00:00
parent bb93cef20a
commit d3ae2954ff
3 changed files with 667 additions and 507 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
javascript/extractor/src/com/semmle/jcorn/CustomParser.java
View File

@@ -314,7 +314,29 @@ public class CustomParser extends FlowParser {
}
}
return super.parseParenAndDistinguishExpression(canBeArrow);
Expression res = super.parseParenAndDistinguishExpression(canBeArrow);
if (res instanceof ParenthesizedExpression) {
ParenthesizedExpression p = (ParenthesizedExpression) res;
if (p.getExpression() instanceof ComprehensionExpression) {
ComprehensionExpression c = (ComprehensionExpression) p.getExpression();
if (c.isGenerator())
return new ComprehensionExpression(p.getLoc(), c.getBody(), c.getBlocks(), c.getFilter(), c.isGenerator());
}
}
return res;
}
@Override
protected boolean parseParenthesisedExpression(DestructuringErrors refDestructuringErrors,
boolean allowTrailingComma, ParenthesisedExpressions parenExprs, boolean first) {
boolean cont = super.parseParenthesisedExpression(refDestructuringErrors, allowTrailingComma, parenExprs, first);
if (options.mozExtensions() && parenExprs.exprList.size() == 1 && this.type == TokenType._for) {
Expression body = parenExprs.exprList.remove(0);
ComprehensionExpression c = parseComprehension(body.getLoc().getStart(), true, body);
parenExprs.exprList.add(this.finishNode(c));
return false;
}
return cont;
}
// add parsing of for-each loops