com.commercetools.sunrise.shoppingcart.LineItemExtendedBeanFactory 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

There is a newer version: 1.0.0-M10

package com.commercetools.sunrise.shoppingcart;

import com.commercetools.sunrise.common.ctp.ProductDataConfig;
import com.commercetools.sunrise.common.models.ProductAttributeBean;
import com.commercetools.sunrise.common.models.ProductAttributeBeanFactory;
import io.sphere.sdk.carts.LineItem;

import javax.inject.Inject;
import java.util.List;

import static java.util.stream.Collectors.toList;

public class LineItemExtendedBeanFactory extends LineItemBeanFactory {

    @Inject
    private ProductDataConfig productDataConfig;
    @Inject
    private ProductAttributeBeanFactory productAttributeBeanFactory;

    @Override
    public LineItemExtendedBean create(final LineItem lineItem) {
        final LineItemExtendedBean bean = new LineItemExtendedBean();
        initialize(bean, lineItem);
        return bean;
    }

    protected final void initialize(final LineItemExtendedBean bean, LineItem lineItem) {
        super.initialize(bean, lineItem);
        fillAttributes(bean, lineItem);
    }

    protected void fillAttributes(final LineItemExtendedBean bean, final LineItem lineItem) {
        final List attributes = lineItem.getVariant().getAttributes().stream()
                .filter(attr -> productDataConfig.getSelectableAttributes().contains(attr.getName()))
                .map(attr -> productAttributeBeanFactory.create(attr))
                .collect(toList());
        bean.setAttributes(attributes);
    }
}