Enable pricing guidance and data in quotes

@Grab(group='org.codehaus.groovy', module='groovy-json', version='2.4.7')

// Import necessary classes and libraries
import com.imc.context.BaseContext;
import com.imc.datamodel.BusinessObject
import com.imc.iss.groovy.GroovyCtxUtil
import com.imc.iss.groovy.priceiq.GroovyPriceIQHelper
import com.imc.iss.price.model.input.document.SalesItem;
import com.imc.iss.price.services.pricing.PriceAmount;
import com.imc.model.Amount;
import com.imc.model.Currency;
import com.imc.vocabulary.Schema;

import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.util.EntityUtils;

import groovy.json.JsonSlurper;

SalesItem salesItemDTOBean = prizableDTOBean; // Retrieve the sales item DTO bean
GroovyCtxUtil groovyCtxUtil = groovyCtxUtil;// Retrieve the utility class for ad-hoc data access
BaseContext ctx = groovyCtxUtil.getCtx(); // Retrieve the base context from the utility

// Retrieve business objects related to the sales item
final BusinessObject salesItemBO = salesItemDTOBean.getInstance();
final BusinessObject quoteBO = salesItemDTOBean.getQuoteBO();
final BusinessObject ihpiBO = ctx.commandFactory().getRelAttrValue().businessObject(salesItemBO).businessAttribute(Schema.includesItemHeaderPriceItem).execute().get();
final BusinessObject productBO = ctx.commandFactory().getRelAttrValue().businessObject(salesItemBO).businessAttribute(Schema.isProduct).execute().get();
// If the system uses an opportunity, additional verification is needed here
final BusinessObject accountBO = ctx.commandFactory().getRelAttrValue().businessObject(quoteBO).businessAttribute(Schema.quoteContainedBy).execute().get();

// Retrieve ERP IDs, quantity, and other information needed for discount calculation
final String productErpId = (String) ctx.commandFactory().getDataAttrValue().businessObject(productBO).businessAttribute(Schema.objectERPId).execute().orElse("");
final String accountErpId = (String) ctx.commandFactory().getDataAttrValue().businessObject(accountBO).businessAttribute(Schema.objectERPId).execute().orElse("");
final BigDecimal qty = (BigDecimal) ctx.commandFactory().getDataAttrValue().businessObject(ihpiBO).businessAttribute(Schema.itemHeaderQuantity).execute().get();

// Retrieve discount guidance from the Zilliant database based on product ERP ID, account ERP ID, and quantity
Guidance guidance = getGuidance(productErpId, accountErpId, qty);

// Set discount guidance to the line item table
if(guidance == null)
{
	// If no guidance is available, set default values
	ctx.commandFactory().setDataAttrValues().businessObject(salesItemBO).businessAttribute(Schema.salesItemAttribute18).values(0).allowSetReadOnlyValue().execute();
	ctx.commandFactory().setDataAttrValues().businessObject(salesItemBO).businessAttribute(Schema.salesItemAttribute19).values(0).allowSetReadOnlyValue().execute();
	ctx.commandFactory().setDataAttrValues().businessObject(salesItemBO).businessAttribute(Schema.salesItemAttribute20).values(0).allowSetReadOnlyValue().execute();
	ctx.commandFactory().setDataAttrValues().businessObject(salesItemBO).businessAttribute(Schema.salesItemAttribute21).values("").allowSetReadOnlyValue().execute();
}
else
{
	// If guidance is available, set the calculated values
    ctx.commandFactory().setDataAttrValues().businessObject(salesItemBO).businessAttribute(Schema.salesItemAttribute18).values(guidance.startDiscount).allowSetReadOnlyValue().execute();
	ctx.commandFactory().setDataAttrValues().businessObject(salesItemBO).businessAttribute(Schema.salesItemAttribute19).values(guidance.targetDiscount).allowSetReadOnlyValue().execute();
	ctx.commandFactory().setDataAttrValues().businessObject(salesItemBO).businessAttribute(Schema.salesItemAttribute20).values(guidance.floorDiscount).allowSetReadOnlyValue().execute();
	
	// Update health status based on the calculated discount
	BigDecimal discount = (BigDecimal) ctx.commandFactory().getDataAttrValue().businessObject(ihpiBO).businessAttribute(Schema.itemHeaderDiscount).execute().orElse(BigDecimal.ZERO);
	if(discount.compareTo(guidance.targetDiscount) <= 0)
	{
		ctx.commandFactory().setDataAttrValues().businessObject(salesItemBO).businessAttribute(Schema.salesItemAttribute21).values("🟢 GOOD").allowSetReadOnlyValue().execute();
	}
	else if(discount.compareTo(guidance.floorDiscount) <= 0)
	{
		ctx.commandFactory().setDataAttrValues().businessObject(salesItemBO).businessAttribute(Schema.salesItemAttribute21).values("🟡 MEDIUM").allowSetReadOnlyValue().execute();
	}
	else
	{
		ctx.commandFactory().setDataAttrValues().businessObject(salesItemBO).businessAttribute(Schema.salesItemAttribute21).values("🔴 BAD").allowSetReadOnlyValue().execute();
	}
}

// Return a PriceAmount object with a base amount set to zero
return getPriceAmountZERO();

// Functions to retrieve guidance based on product ERP ID, account ERP ID, and quantity
Guidance getGuidance(String productErpId, String accountErpId, BigDecimal qty)
{
	if(productErpId.isEmpty() || accountErpId.isEmpty())
		return null;

	def jsonPayload = """
		{
			"formulas": [
				"piq_fx_StartDiscount",
				"piq_fx_TargetDiscount",
				"piq_fx_FloorDiscount"
			],
			"responseDetails": [
				"none"
			],
			"lines": [
				{
					"p_CustomerId": "${accountErpId}",
					"p_LineLevel": "Product",
					"p_LevelValue": "${productErpId}",
					"p_Country": "United States",
					"p_CurrCode": "USD",
					"piq_fx_ExpectedDefaultUnits": ${qty.toPlainString()},
					"p_Uom": "each"
				}
			]
		}
	"""

	// Make request to the Formula Evaluation service
	def responseBody = groovyPriceIQHelper.formulaEvaluation(jsonPayload);
	def jsonResponse = new JsonSlurper().parseText(responseBody);
	
	// Retrieve guidance values from the response
	def startOutput = jsonResponse["piq_fx_StartDiscount"]["outputResults"][0];
	def targetOutput = jsonResponse["piq_fx_TargetDiscount"]["outputResults"][0];
	def floorOutput = jsonResponse["piq_fx_FloorDiscount"]["outputResults"][0];
	if(startOutput == null || targetOutput == null || floorOutput == null)
	{
		return null;
	}
	
	def start = startOutput["piq_fx_StartDiscount"][0];
	def target = targetOutput["piq_fx_TargetDiscount"][0];
	def floor = floorOutput["piq_fx_FloorDiscount"][0];
	if(start == "null" && target == "null" && floor == "null")
	{
		return null;
	}
	
	return new Guidance(start*100, target*100, floor*100);
}

// Function to return PriceAmount object with base amount set to zero
PriceAmount getPriceAmountZERO()
{
	// Set base amount 
	SalesItem salesItemDTOBean = prizableDTOBean;
	
	final Currency salesItemCurrency = salesItemDTOBean.getCurrency(); //sales item currency
	BigDecimal targetTotalAmountValue = BigDecimal.ZERO;
	Amount amount = new Amount();
	amount.setAmountCurrency(salesItemCurrency);
	amount.setAmountValue(targetTotalAmountValue);
	
	PriceAmount pa = new PriceAmount();
	pa.setBaseAmount(amount);
	pa.setPricingStep(pricingStep);
	return pa;
}

// Class to hold guidance values
class Guidance
{
	public BigDecimal startDiscount;
	public BigDecimal targetDiscount;
	public BigDecimal floorDiscount;
	
	public Guidance(BigDecimal startDiscount, BigDecimal targetDiscount, BigDecimal floorDiscount)
	{
		this.startDiscount = startDiscount;
		this.targetDiscount = targetDiscount;
		this.floorDiscount = floorDiscount;
	}
}