/*
* Copyright (C) 2011 Red Hat, Inc. and/or its affiliates.
*
* 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 org.jboss.errai.codegen.builder.callstack;
import org.jboss.errai.codegen.AssignmentOperator;
import org.jboss.errai.codegen.Context;
import org.jboss.errai.codegen.Statement;
import org.jboss.errai.codegen.VariableReference;
import org.jboss.errai.codegen.builder.impl.AssignmentBuilder;
import org.jboss.errai.codegen.exception.GenerationException;
import org.jboss.errai.codegen.meta.MetaClass;
import org.jboss.errai.codegen.util.GenUtil;
/**
* {@link CallElement} to assign a value to a variable.
*
* @author Christian Sadilek <csadilek@redhat.com>
*/
public class AssignVariable extends AbstractCallElement {
private final AssignmentOperator operator;
private final Object value;
public AssignVariable(final AssignmentOperator operator, final Object value) {
this.operator = operator;
this.value = value;
}
@Override
public void handleCall(final CallWriter writer, final Context context, final Statement statement) {
writer.reset();
final Statement stmt = new AssignmentBuilder(false, operator, (VariableReference) statement, GenUtil.generate(context,
value));
final Statement wrapperStmt;
try {
if (next == null) {
wrapperStmt = new Statement() {
@Override
public String generate(final Context context) {
return stmt.generate(context).concat(";");
}
@Override
public MetaClass getType() {
return stmt.getType();
}
};
nextOrReturn(writer, context, wrapperStmt);
}
else {
nextOrReturn(writer, context, stmt);
}
}
catch (GenerationException e) {
blameAndRethrow(e);
}
}
@Override
public String toString() {
return "[[Assignment<" + operator.getCanonicalString() + ":" + value + ">]" + next + "]";
}
}