All files estimate.ts

100% Statements 135/135
100% Branches 44/44
100% Functions 17/17
100% Lines 94/94

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193        1x 1x 1x     22x 22x 22x 1x   22x             9x 9x 1x   9x             9x 9x   9x 2x 2x   1x 1x 1x     7x   10x       14x 11x 4x       7x                         14x 14x 5x   10x 9x 4x 4x 4x 4x 4x 4x   5x 5x 5x 4x 4x 5x   4x   5x           1x       22x 1x     21x 21x 21x                   21x 21x 8x 8x 5x 5x 5x   8x 8x 8x 8x       8x 2x 2x   6x 6x 5x 4x 4x   6x 6x 6x     5x 7x     5x 2x 1x 2x 1x     16x 16x 16x 16x 16x 10x 9x 9x                 21x 21x 21x 21x 21x                   8x        
import {
  PricingRules, PricingField, PricingOption, Selections, FieldSelection,
  QuoteResult, UnsupportedResult, DiscountGroup,
} from './types.js';
import { applyDiscount } from './discount.js';
import { roundHalfEven } from './round.js';
import { resolveVisibleFields } from './visibility.js';
 
function unitPriceAt(rules: PricingRules, qty: number): number {
  let unitPrice = applyDiscount(rules.product.unitPrice, qty, rules.product.discountGroup);
  const mop = rules.product.minimumPrice;
  if (qty > 0 && mop && unitPrice * qty < mop) {
    unitPrice = mop / qty;
  }
  return unitPrice;
}
 
function variationSetupCost(
  source: { variationCost: number; variationCostDiscountGroup: DiscountGroup | null },
  totalQty: number
): number {
  let cost = source.variationCost;
  if (source.variationCostDiscountGroup) {
    cost = applyDiscount(cost, totalQty, source.variationCostDiscountGroup);
  }
  return roundHalfEven(cost, 3);
}
 
function variationUnitCosts(
  source: { variationUnitCost: number; variationUnitCostDiscountGroup: DiscountGroup | null },
  groupQuantities: number[]
): number[] {
  const uc = source.variationUnitCost;
  const group = source.variationUnitCostDiscountGroup;
  let list: number[];
  if (group) {
    if (group.groupRestricted) {
      list = groupQuantities.map((q) => applyDiscount(uc, q, group));
    } else {
      const total = groupQuantities.reduce((a, b) => a + b, 0);
      const discounted = applyDiscount(uc, total, group);
      list = groupQuantities.map(() => discounted);
    }
  } else {
    list = groupQuantities.map(() => uc);
  }
  return list.map((c) => roundHalfEven(c, 3));
}
 
function isEmpty(field: PricingField, sel: FieldSelection | undefined): boolean {
  if (!sel) return true;
  if (field.isSelectable) {
    return !sel.selectedOptionIds || sel.selectedOptionIds.length === 0;
  }
  // Server `is_empty` non-selectable branch is `return not self.value`, so
  // numeric 0 is empty (but string '0' stays non-empty, since `not '0'` is False).
  return (
    sel.value === undefined ||
    sel.value === null ||
    sel.value === '' ||
    sel.value === 0
  );
}
 
function costFactor(
  field: PricingField,
  sel: FieldSelection | undefined,
  groupQuantities: number[]
): { setup: number; unitList: number[] } {
  const n = groupQuantities.length;
  if (isEmpty(field, sel)) {
    return { setup: 0, unitList: new Array(n).fill(0) };
  }
  const totalQty = groupQuantities.reduce((a, b) => a + b, 0);
  if (field.isSelectable) {
    let setup = 0;
    const unitList = new Array(n).fill(0);
    const optById = new Map<number, PricingOption>();
    for (const o of field.options) {
      optById.set(o.id, o);
      if (o.originalId != null) optById.set(o.originalId, o);
    }
    for (const optId of sel!.selectedOptionIds!) {
      const opt = optById.get(optId);
      if (!opt) continue;
      setup += variationSetupCost(opt, totalQty);
      const ucs = variationUnitCosts(opt, groupQuantities);
      for (let i = 0; i < n; i++) unitList[i] += ucs[i];
    }
    return { setup, unitList };
  }
  return {
    setup: variationSetupCost(field, totalQty),
    unitList: variationUnitCosts(field, groupQuantities),
  };
}
 
export function estimateQuote(
  rules: PricingRules,
  selections: Selections
): QuoteResult | UnsupportedResult {
  if (rules.unsupported) {
    return { unsupported: rules.unsupported };
  }
 
  const visible = resolveVisibleFields(rules, selections);
  const groupCosts: number[] = [];
  let cost = 0;
  let costPerUnit: number;
  let groupQuantities: number[];
 
  // Mirror the server's `update_group_variations_cost_and_job_cost`
  // (jobs.py ~3321): the group-vs-single branch keys off whether the *job*
  // actually has variation groups (`if self.variations_groups:`), NOT the
  // product's `hasGroups` capability. A group-capable product with no groups
  // submitted is priced as a single non-group job (the server's `else` branch
  // sets `group_quantities = [self.quantity or 0]`).
  const groups = selections.groups || [];
  if (groups.length > 0) {
    groupQuantities = groups.map((g) => g.quantity || 0);
    const totalQty = groupQuantities.reduce((a, b) => a + b, 0);
    const restricted = Boolean(rules.product.discountGroup?.groupRestricted);
    const baseUnitPrice = unitPriceAt(rules, totalQty);
    const perGroupCpu: number[] = [];
 
    for (const group of groups) {
      const gQty = group.quantity || 0;
      const cpu = restricted ? unitPriceAt(rules, gQty) : baseUnitPrice;
      perGroupCpu.push(cpu);
      // Server `VariationsGroups.update_cost` (variations_groups.py ~238):
      // `if not self.quantity: self.group_cost = 0; return` — a zero-qty group
      // contributes exactly 0 and adds NO field/variation costs.
      if (!gQty) {
        groupCosts.push(0);
        continue;
      }
      let groupVariationCost = 0;
      for (const field of rules.groupFields) {
        if (!visible.has(field.id)) continue;
        const { setup, unitList } = costFactor(field, group.fieldValues[field.id], [gQty]);
        groupVariationCost += setup + unitList[0] * gQty;
      }
      const groupCost = gQty * cpu + groupVariationCost;
      groupCosts.push(groupCost);
      cost += groupCost;
    }
 
    costPerUnit = totalQty > 0
      ? perGroupCpu.reduce((acc, cpu, i) => acc + cpu * groupQuantities[i], 0) / totalQty
      : baseUnitPrice;
 
    for (const field of rules.fields) {
      if (!visible.has(field.id)) continue;
      const { setup, unitList } = costFactor(field, selections.fieldValues[field.id], groupQuantities);
      const unitTotal = unitList.reduce((acc, uc, i) => acc + uc * groupQuantities[i], 0);
      cost += setup + unitTotal;
    }
  } else {
    const qty = selections.quantity || 0;
    groupQuantities = [qty];
    costPerUnit = unitPriceAt(rules, qty);
    cost = costPerUnit * qty;
    for (const field of rules.fields) {
      if (!visible.has(field.id)) continue;
      const { setup, unitList } = costFactor(field, selections.fieldValues[field.id], groupQuantities);
      cost += setup + unitList[0] * qty;
    }
  }
 
  // Server `Jobs.update_cost` (jobs.py ~3452-3454) accumulates `self.cost` at
  // full precision and computes `self.tax_amount = tax(self.cost, ...)` on the
  // UNROUNDED cost (money_protocol.py `tax` does NOT round). `cost` and
  // `taxAmount` are then rounded INDEPENDENTLY at serialization to 2dp
  // (jobs.py ~2169-2171). So tax must be derived from the unrounded cost.
  const unroundedCost = cost;
  const roundedCost = roundHalfEven(unroundedCost, 2);
  const unroundedTax = (unroundedCost * rules.taxPercent) / 100;
  const taxAmount = roundHalfEven(unroundedTax, 2);
  return {
    costPerUnit: roundHalfEven(costPerUnit, 3),
    cost: roundedCost,
    taxAmount,
    // `totalCost` mirrors `Jobs.all_total_cost` = `self.cost + self.tax_amount`
    // (jobs.py ~1724), serialized UNROUNDED at jobs.py ~1899. We round to 3dp to
    // strip float noise while matching the server's effective Numeric scale.
    totalCost: roundHalfEven(unroundedCost + unroundedTax, 3),
    // `result["groupCost"] = self.group_cost` is serialized UNROUNDED
    // (variations_groups.py ~154); round to 3dp to match the Numeric scale.
    groupCosts: groupCosts.map((c) => roundHalfEven(c, 3)),
    currency: rules.currency,
  };
}