/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF 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.
 */

import org.apache.ofbiz.order.shoppingcart.ShoppingCartEvents

shoppingCart = ShoppingCartEvents.getCartObject(request)

if (userLogin) {
    context.enableShippingAddress = true
}

shippingContactMechId = shoppingCart.getShippingContactMechId()
if (shippingContactMechId) {
    context.enableShipmentMethod = true
}

shipmentMethodTypeId  = shoppingCart.getShipmentMethodTypeId()
if (shipmentMethodTypeId) {
    context.enablePaymentOptions = true
}

paymentMethodIds  = shoppingCart.getPaymentMethodIds()
paymentMethodTypeId = parameters.paymentMethodTypeId
if (paymentMethodIds || paymentMethodTypeId) {
    context.enablePaymentInformation = true
}

if (paymentMethodIds) {
    context.enableReviewOrder = true
}
