/**
 * Copyright 2011-2015 John Ericksen
 *
 * 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.androidtransfuse.gen.variableBuilder;

import com.sun.codemodel.JExpr;
import com.sun.codemodel.JExpression;
import com.sun.codemodel.JInvocation;
import com.sun.codemodel.JType;
import org.androidtransfuse.gen.ClassGenerationUtil;
import org.androidtransfuse.gen.InjectionBuilderContext;
import org.androidtransfuse.gen.InjectionExpressionBuilder;
import org.androidtransfuse.gen.IntentFactoryStrategyGenerator;
import org.androidtransfuse.gen.variableDecorator.TypedExpressionFactory;
import org.androidtransfuse.model.InjectionNode;
import org.androidtransfuse.model.TypedExpression;
import org.androidtransfuse.util.AndroidLiterals;
import org.androidtransfuse.util.ExtraUtil;

import javax.inject.Inject;
import javax.inject.Named;

/**
 * @author John Ericksen
 */
public class ExtraVariableBuilder extends ConsistentTypeVariableBuilder {

    private final boolean parcelerWrapped;
    private final String extraId;
    private final InjectionNode activityInjectionNode;
    private final InjectionExpressionBuilder injectionExpressionBuilder;
    private final boolean nullable;
    private final ClassGenerationUtil generationUtil;
    private final GetExtraExpressionBuilder getExtraExpressionBuilder;

    @Inject
    public ExtraVariableBuilder(/*@Assisted*/ String extraId,
                                /*@Assisted*/ InjectionNode activityInjectionNode,
                                /*@Assisted("nullable")*/ @Named("nullable") boolean nullable,
                                /*@Assisted("wrapped")*/ @Named("wrapped") boolean parcelerWrapped,
                                /*@Assisted*/ GetExtraExpressionBuilder getExtraExpressionBuilder,
                                InjectionExpressionBuilder injectionExpressionBuilder,
                                ClassGenerationUtil generationUtil,
                                TypedExpressionFactory typedExpressionFactory) {
        super(Object.class, typedExpressionFactory);
        this.extraId = extraId;
        this.activityInjectionNode = activityInjectionNode;
        this.injectionExpressionBuilder = injectionExpressionBuilder;
        this.nullable = nullable;
        this.generationUtil = generationUtil;
        this.parcelerWrapped = parcelerWrapped;
        this.getExtraExpressionBuilder = getExtraExpressionBuilder;
    }

    @Override
    public JExpression buildExpression(InjectionBuilderContext injectionBuilderContext, InjectionNode injectionNode) {
        TypedExpression contextVar = injectionExpressionBuilder.buildVariable(injectionBuilderContext, activityInjectionNode);

        JInvocation getExtraInvocation = generationUtil.ref(ExtraUtil.class)
                .staticInvoke(ExtraUtil.GET_EXTRA)
                .arg(getExtraExpressionBuilder.buildGetExtraBundle(contextVar.getExpression()))
                .arg(JExpr.lit(extraId))
                .arg(JExpr.lit(nullable));

        if (parcelerWrapped) {
            JType parcelableType = generationUtil.ref(AndroidLiterals.PARCELABLE);
            getExtraInvocation = generationUtil.ref(IntentFactoryStrategyGenerator.PARCELS_NAME)
                    .staticInvoke(IntentFactoryStrategyGenerator.UNWRAP_METHOD)
                    .arg(JExpr.cast(parcelableType, getExtraInvocation));
        }

        return getExtraInvocation;
    }

    public interface GetExtraExpressionBuilder {

        JExpression buildGetExtraBundle(JExpression expression);
    }
}