com.commercetools.sunrise.framework.components.controllers.CartFieldsUpdaterControllerComponent Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common_2.11 Show documentation
common
The newest version!
package com.commercetools.sunrise.framework.components.controllers;

import com.commercetools.sunrise.framework.hooks.ctpactions.CartLoadedActionHook;
import com.commercetools.sunrise.framework.hooks.ctpactions.CustomerSignedInActionHook;
import com.commercetools.sunrise.framework.injection.RequestScoped;
import com.neovisionaries.i18n.CountryCode;
import io.sphere.sdk.carts.Cart;
import io.sphere.sdk.carts.commands.CartUpdateCommand;
import io.sphere.sdk.carts.commands.updateactions.SetCountry;
import io.sphere.sdk.carts.commands.updateactions.SetCustomerEmail;
import io.sphere.sdk.carts.commands.updateactions.SetShippingAddress;
import io.sphere.sdk.client.SphereClient;
import io.sphere.sdk.commands.UpdateAction;
import io.sphere.sdk.customers.Customer;
import io.sphere.sdk.customers.CustomerSignInResult;
import io.sphere.sdk.expansion.ExpansionPathContainer;
import io.sphere.sdk.models.Address;

import javax.annotation.Nullable;
import javax.inject.Inject;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import java.util.concurrent.CompletionStage;

import static java.util.concurrent.CompletableFuture.completedFuture;

@RequestScoped
public class CartFieldsUpdaterControllerComponent implements ControllerComponent, CartLoadedActionHook, CustomerSignedInActionHook {

    private final CountryCode country;
    private final SphereClient sphereClient;

    @Inject
    public CartFieldsUpdaterControllerComponent(final CountryCode country, final SphereClient sphereClient) {
        this.country = country;
        this.sphereClient = sphereClient;
    }

    @Override
    public CompletionStage onCustomerSignedInAction(final CustomerSignInResult customerSignInResult,
                                                                          @Nullable final ExpansionPathContainer expansionPathContainer) {
        return Optional.ofNullable(customerSignInResult.getCart())
                .map(cart -> buildRequest(cart, customerSignInResult.getCustomer()))
                .filter(command -> !command.getUpdateActions().isEmpty())
                .map(command -> sphereClient.execute(command)
                        // Expansion path is honored, since only possible expansion is on customer (on the CreateCustomerCommand)
                        // and customer is kept as it was received in the original response
                        .thenApply(updatedCart -> buildCustomerSignInResult(customerSignInResult.getCustomer(), updatedCart)))
                .orElseGet(() -> completedFuture(customerSignInResult));
    }

    @Override
    public CompletionStage onCartLoadedAction(final Cart cart, final ExpansionPathContainer expansionPathContainer) {
        // TODO Handle case where some line items do not exist for this country
        final CartUpdateCommand updateCommand = buildRequest(cart, null);
        if (!updateCommand.getUpdateActions().isEmpty()) {
            return sphereClient.execute(updateCommand.withExpansionPaths(expansionPathContainer));
        } else {
            return completedFuture(cart);
        }
    }

    protected CartUpdateCommand buildRequest(final Cart cart, @Nullable final Customer customer) {
        return CartUpdateCommand.of(cart, buildUpdateActions(cart, customer));
    }

    private List> buildUpdateActions(final Cart cart, @Nullable final Customer customer) {
        final List> updateActions = new ArrayList<>();
        final boolean hasDifferentCountry = !country.equals(cart.getCountry());
        if (hasDifferentCountry) {
            final Address shippingAddress = Optional.ofNullable(cart.getShippingAddress())
                    .map(address -> address.withCountry(country))
                    .orElseGet(() -> Address.of(country));
            updateActions.add(SetShippingAddress.of(shippingAddress));
            updateActions.add(SetCountry.of(country));
        }
        final boolean hasNoCustomerEmail = customer != null && cart.getCustomerEmail() == null;
        if (hasNoCustomerEmail) {
            updateActions.add(SetCustomerEmail.of(customer.getEmail()));
        }
        return updateActions;
    }

    private CustomerSignInResult buildCustomerSignInResult(final Customer customer, @Nullable final Cart cart) {
        return new CustomerSignInResult() {
            @Override
            public Customer getCustomer() {
                return customer;
            }

            @Nullable
            @Override
            public Cart getCart() {
                return cart;
            }
        };
    }
}