blob: a0275c6a57377ebd80f4922a55b542e3c25d753a [file] [log] [blame]
// Copyright (c) 2017, the Dart project authors. Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.
import '../js_ast/js_ast.dart';
/// Simplify `(args) => (() => { ... })()` to `(args) => { ... }`.
// TODO(jmesserly): find a better home for this function
Fun simplifyPassThroughArrowFunCallBody(Fun fn) {
if (fn.body is Block && fn.body.statements.length == 1) {
var stat = fn.body.statements.single;
if (stat is Return && stat.value is Call) {
Call call = stat.value;
var innerFun = call.target;
if (innerFun is ArrowFun &&
call.arguments.isEmpty &&
innerFun.params.isEmpty) {
var body = innerFun.body;
if (body is Block) {
return Fun(fn.params, body);
}
}
}
}
return fn;
}
Set<Identifier> findMutatedVariables(Node scope) {
var v = MutationVisitor();
scope.accept(v);
return v.mutated;
}
class MutationVisitor extends BaseVisitor {
final mutated = Set<Identifier>();
@override
visitAssignment(node) {
var id = node.leftHandSide;
if (id is Identifier) mutated.add(id);
super.visitAssignment(node);
}
}