FunctionExpressionTreeImpl.java revision 1786:80120e9b3273 
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
26package jdk.nashorn.api.tree;
27
28import java.util.List;
29import jdk.nashorn.internal.ir.FunctionNode;
30
31final 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}
102