blob: 0c78eaf1e52a2de1e75711d8050d005126b0542e [file] [log] [blame]
// Copyright 2014 The Bazel Authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package com.google.devtools.build.lib.syntax;
import com.google.common.collect.ImmutableList;
import java.io.IOException;
/** Syntax node for a function definition. */
public final class FunctionDefStatement extends Statement {
private final Identifier identifier;
private final FunctionSignature.WithValues<Expression, Expression> signature;
private final ImmutableList<Statement> statements;
private final ImmutableList<Parameter<Expression, Expression>> parameters;
public FunctionDefStatement(Identifier identifier,
Iterable<Parameter<Expression, Expression>> parameters,
FunctionSignature.WithValues<Expression, Expression> signature,
Iterable<Statement> statements) {
this.identifier = identifier;
this.parameters = ImmutableList.copyOf(parameters);
this.signature = signature;
this.statements = ImmutableList.copyOf(statements);
}
@Override
public void prettyPrint(Appendable buffer, int indentLevel) throws IOException {
printIndent(buffer, indentLevel);
buffer.append("def ");
identifier.prettyPrint(buffer);
buffer.append('(');
String sep = "";
for (Parameter<?, ?> param : parameters) {
buffer.append(sep);
param.prettyPrint(buffer);
sep = ", ";
}
buffer.append("):\n");
printSuite(buffer, statements, indentLevel);
}
@Override
public String toString() {
return "def " + identifier + "(" + signature + "): ...\n";
}
public Identifier getIdentifier() {
return identifier;
}
public ImmutableList<Statement> getStatements() {
return statements;
}
public ImmutableList<Parameter<Expression, Expression>> getParameters() {
return parameters;
}
public FunctionSignature.WithValues<Expression, Expression> getSignature() {
return signature;
}
@Override
public void accept(SyntaxTreeVisitor visitor) {
visitor.visit(this);
}
@Override
public Kind kind() {
return Kind.FUNCTION_DEF;
}
}