/*
 * Copyright 2014 Groupon, Inc
 *
 * Groupon licenses this file to you 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.killbill.billing.payment.core.sm;

import java.util.List;

import org.joda.time.DateTime;
import org.killbill.automaton.State;
import org.killbill.automaton.State.LeavingStateCallback;
import org.killbill.billing.payment.api.PluginProperty;
import org.killbill.billing.payment.api.TransactionType;
import org.killbill.billing.payment.dao.PaymentModelDao;
import org.killbill.billing.payment.dao.PaymentAttemptModelDao;
import org.killbill.billing.payment.dao.PaymentDao;
import org.killbill.billing.payment.dao.PluginPropertyModelDao;

import com.google.common.base.Function;
import com.google.common.base.Preconditions;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Iterables;

public class RetryLeavingStateCallback implements LeavingStateCallback {

    private PluginControlledDirectPaymentAutomatonRunner retryableDirectPaymentAutomatonRunner;
    private final RetryableDirectPaymentStateContext stateContext;
    private final State initialState;
    private final State retriedState;
    private final TransactionType transactionType;
    private final PaymentDao paymentDao;

    public RetryLeavingStateCallback(final PluginControlledDirectPaymentAutomatonRunner retryableDirectPaymentAutomatonRunner, final DirectPaymentStateContext stateContext, final PaymentDao paymentDao,
                                     final State initialState, final State retriedState, final TransactionType transactionType) {
        this.retryableDirectPaymentAutomatonRunner = retryableDirectPaymentAutomatonRunner;
        this.paymentDao = paymentDao;
        this.initialState = initialState;
        this.retriedState = retriedState;
        this.stateContext = (RetryableDirectPaymentStateContext) stateContext;
        this.transactionType = transactionType;
    }

    @Override
    public void leavingState(final State state) {

        final DateTime utcNow = retryableDirectPaymentAutomatonRunner.clock.getUTCNow();

        Preconditions.checkState(stateContext.getDirectPaymentExternalKey() != null || /* AUTH, PURCHASE, CREDIT calls will provide the payment  */
                                 stateContext.getDirectPaymentId() != null);
        if (stateContext.getDirectPaymentExternalKey() == null) {
            final PaymentModelDao payment = paymentDao.getDirectPayment(stateContext.getDirectPaymentId(), stateContext.internalCallContext);
            Preconditions.checkState(payment != null);
            stateContext.setDirectPaymentExternalKey(payment.getExternalKey());
        }


        if (state.getName().equals(initialState.getName()) ||
            state.getName().equals(retriedState.getName())) {
            final List<PluginPropertyModelDao> properties = ImmutableList.copyOf(Iterables.transform(stateContext.getProperties(), new Function<PluginProperty, PluginPropertyModelDao>() {
                @Override
                public PluginPropertyModelDao apply(final PluginProperty input) {
                    // STEPH how to serialize more complex values such as item adjustments. json ?
                    final String value = (input.getValue() instanceof String) ? (String) input.getValue() : "TODO: could not serialize";
                    return new PluginPropertyModelDao(stateContext.getDirectPaymentExternalKey(), stateContext.directPaymentTransactionExternalKey, stateContext.getAccount().getId(),
                                                      stateContext.getPluginName(), input.getKey(), value, stateContext.getCallContext().getUserName(), stateContext.getCallContext().getCreatedDate());
                }
            }));
            final PaymentAttemptModelDao attempt = retryableDirectPaymentAutomatonRunner.paymentDao.insertPaymentAttemptWithProperties(new PaymentAttemptModelDao(stateContext.getAccount().getId(), stateContext.getPaymentMethodId(),
                                                                                                        utcNow, utcNow, stateContext.getDirectPaymentExternalKey(), null,
                                                                                                                           stateContext.directPaymentTransactionExternalKey, transactionType, state.getName(),
                                                                                                                            stateContext.getAmount(), stateContext.getCurrency(),
                                                                                                                           stateContext.getPluginName()),
                                                                                                properties, stateContext.internalCallContext);

            stateContext.setAttemptId(attempt.getId());
        }
    }
}
