1 /*
   2  * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.  Oracle designates this
   8  * particular file as subject to the "Classpath" exception as provided
   9  * by Oracle in the LICENSE file that accompanied this code.
  10  *
  11  * This code is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14  * version 2 for more details (a copy is included in the LICENSE file that
  15  * accompanied this code).
  16  *
  17  * You should have received a copy of the GNU General Public License version
  18  * 2 along with this work; if not, write to the Free Software Foundation,
  19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  22  * or visit www.oracle.com if you need additional information or have any
  23  * questions.
  24  */
  25 
  26 package jdk.nashorn.api.tree;
  27 
  28 import java.util.List;
  29 import jdk.nashorn.internal.ir.FunctionNode;
  30 
  31 final class FunctionExpressionTreeImpl extends ExpressionTreeImpl
  32     implements FunctionExpressionTree {
  33     private final FunctionNode funcNode;
  34     private final IdentifierTree funcName;
  35     private final List<? extends ExpressionTree> params;
  36     private final Tree body;
  37 
  38     FunctionExpressionTreeImpl(final FunctionNode node,
  39             final List<? extends ExpressionTree> params,
  40             final BlockTree body) {
  41         super(node);
  42         funcNode = node;
  43         assert !funcNode.isDeclared() || funcNode.isAnonymous() : "function expression expected";
  44 
  45         final FunctionNode.Kind kind = node.getKind();
  46         if (node.isAnonymous() || kind == FunctionNode.Kind.GETTER || kind == FunctionNode.Kind.SETTER) {
  47             funcName = null;
  48         } else {
  49             funcName = new IdentifierTreeImpl(node.getIdent());
  50         }
  51 
  52         this.params = params;
  53         if (node.getFlag(FunctionNode.HAS_EXPRESSION_BODY)) {
  54             final StatementTree first = body.getStatements().get(0);
  55             assert first instanceof ReturnTree : "consise func. expression should have a return statement";
  56             this.body = ((ReturnTree)first).getExpression();
  57         } else {
  58             this.body = body;
  59         }
  60     }
  61 
  62     @Override
  63     public Tree.Kind getKind() {
  64         return Tree.Kind.FUNCTION_EXPRESSION;
  65     }
  66 
  67     @Override
  68     public IdentifierTree getName() {
  69         return funcName;
  70     }
  71 
  72     @Override
  73     public List<? extends ExpressionTree> getParameters() {
  74         return params;
  75     }
  76 
  77     @Override
  78     public Tree getBody() {
  79         return body;
  80     }
  81 
  82     @Override
  83     public boolean isStrict() {
  84         return funcNode.isStrict();
  85     }
  86 
  87     @Override
  88     public boolean isArrow() {
  89         return funcNode.getKind() == FunctionNode.Kind.ARROW;
  90     }
  91 
  92     @Override
  93     public boolean isGenerator() {
  94         return funcNode.getKind() == FunctionNode.Kind.GENERATOR;
  95     }
  96 
  97     @Override
  98     public <R,D> R accept(final TreeVisitor<R,D> visitor, final D data) {
  99         return visitor.visitFunctionExpression(this, data);
 100     }
 101 }