/*
* Copyright 2014 The Billing Project, Inc.
*
* Ning 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.
*/
package org.killbill.billing.invoice.usage;
import java.math.BigDecimal;
import java.util.Collections;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.UUID;
import org.joda.time.DateTime;
import org.joda.time.DateTimeZone;
import org.joda.time.LocalDate;
import org.killbill.billing.catalog.api.CatalogApiException;
import org.killbill.billing.catalog.api.Currency;
import org.killbill.billing.catalog.api.TieredBlock;
import org.killbill.billing.catalog.api.Usage;
import org.killbill.billing.invoice.api.InvoiceItem;
import org.killbill.billing.invoice.api.InvoiceItemType;
import org.killbill.billing.invoice.generator.BillingIntervalDetail;
import org.killbill.billing.invoice.model.UsageInvoiceItem;
import org.killbill.billing.junction.BillingEvent;
import org.killbill.billing.usage.api.RolledUpUsage;
import org.killbill.billing.usage.api.UsageUserApi;
import org.killbill.billing.util.callcontext.TenantContext;
import com.google.common.base.Function;
import com.google.common.base.Preconditions;
import com.google.common.base.Predicate;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Iterables;
import com.google.common.collect.Lists;
import static org.killbill.billing.invoice.usage.SubscriptionConsumableInArrear.getTieredBlocks;
import static org.killbill.billing.invoice.usage.SubscriptionConsumableInArrear.getUnitTypes;
import static org.killbill.billing.invoice.usage.SubscriptionConsumableInArrear.localDateToEndOfDayInAccountTimezone;
/**
* There is one such class per subscriptionId, matching a given in arrear/consumable usage section and
* referenced through a contiguous list of billing events.
*/
public class ContiguousInArrearUsageInterval {
private final List<LocalDate> transitionTimes;
private final List<BillingEvent> billingEvents;
private final Usage usage;
private final Set<String> unitTypes;
private final UsageUserApi usageApi;
private final LocalDate targetDate;
private final UUID invoiceId;
private final TenantContext context;
public ContiguousInArrearUsageInterval(final Usage usage, final UUID invoiceId, final UsageUserApi usageApi, final LocalDate targetDate, final TenantContext context) {
this.usage = usage;
this.invoiceId = invoiceId;
this.unitTypes = getUnitTypes(usage);
this.usageApi = usageApi;
this.targetDate = targetDate;
this.context = context;
this.billingEvents = Lists.newLinkedList();
this.transitionTimes = Lists.newLinkedList();
}
/**
* Builds the transitionTimes associated to that usage section. Those are determined based on billing events for when to start and when to stop,
* the per usage billingPeriod and finally the targetDate.
* <p/>
* Those transition dates define the well defined billing granularity periods that should be billed for that specific usage section.
*
* @param closedInterval whether there was a last billing event referencing the usage section or whether this is ongoing and
* then targetDate will define the endDate.
* @return
*/
public ContiguousInArrearUsageInterval build(final boolean closedInterval) {
Preconditions.checkState((!closedInterval && billingEvents.size() >= 1) ||
(closedInterval && billingEvents.size() >= 2));
final LocalDate startDate = new LocalDate(billingEvents.get(0).getEffectiveDate(), getAccountTimeZone());
if (targetDate.isBefore(startDate)) {
return this;
}
final LocalDate endDate = closedInterval ? new LocalDate(billingEvents.get(billingEvents.size() - 1), getAccountTimeZone()) : targetDate;
final BillingIntervalDetail bid = new BillingIntervalDetail(startDate, endDate, targetDate, getBCD(), usage.getBillingPeriod());
int numberOfPeriod = 0;
// First billingCycleDate prior startDate
LocalDate nextBillCycleDate = bid.getFutureBillingDateFor(numberOfPeriod);
transitionTimes.add(startDate);
while (!nextBillCycleDate.isAfter(endDate)) {
if (nextBillCycleDate.isAfter(startDate)) {
transitionTimes.add(nextBillCycleDate);
}
numberOfPeriod++;
nextBillCycleDate = bid.getFutureBillingDateFor(numberOfPeriod);
}
return this;
}
/**
* Compute the missing usage invoice items based on what should be billed and what has been billed ($ amount comparison).
*
* @param existingUsage existing on disk usage items for the subscription
* @return
* @throws CatalogApiException
*/
public List<InvoiceItem> computeMissingItems(final List<InvoiceItem> existingUsage) throws CatalogApiException {
final List<InvoiceItem> result = Lists.newLinkedList();
final RolledUpUsageForUnitTypesFactory factory = new RolledUpUsageForUnitTypesFactory(getRolledUpUsage(), unitTypes, getAccountTimeZone());
for (RolledUpUsageForUnitTypes ru : factory.getOrderedRolledUpUsageForUnitTypes()) {
final BigDecimal billedUsage = computeBilledUsage(ru.getStartDate(), ru.getEndDate(), existingUsage);
final BigDecimal toBeBilledUsage = BigDecimal.ZERO;
for (final String unitType : unitTypes) {
final BigDecimal usageAmountForUnitType = ru.getUsageAmountForUnitType(unitType);
final BigDecimal toBeBilledForUnit = computeToBeBilledUsage(usageAmountForUnitType, unitType);
toBeBilledUsage.add(toBeBilledForUnit);
if (billedUsage.compareTo(toBeBilledUsage) < 0) {
InvoiceItem item = new UsageInvoiceItem(invoiceId, getAccountId(), getBundleId(), getSubscriptionId(), getPlanName(),
getPhaseName(), usage.getName(), ru.getStartDate(), ru.getEndDate(), toBeBilledUsage.subtract(billedUsage), getCurrency());
result.add(item);
}
}
}
return result;
}
List<RolledUpUsage> getRolledUpUsage() {
final Iterable<DateTime> transitions = Iterables.transform(transitionTimes, new Function<LocalDate, DateTime>() {
@Override
public DateTime apply(final LocalDate input) {
return localDateToEndOfDayInAccountTimezone(input, getAccountTimeZone());
}
});
// STEPH_USAGE optimized api takes set of unitTypes -- for usage section-- and list of transitions date. Should we use dateTime or LocalDate?
return usageApi.getAllUsageForSubscription(getSubscriptionId(), unitTypes, ImmutableList.copyOf(transitions), context);
}
BigDecimal computeToBeBilledUsage(final BigDecimal nbUnits, final String unitType) throws CatalogApiException {
BigDecimal result = BigDecimal.ZERO;
final List<TieredBlock> tieredBlocks = getTieredBlocks(usage, unitType);
int remainingUnits = nbUnits.intValue();
for (TieredBlock tieredBlock : tieredBlocks) {
final int blockTierSize = tieredBlock.getSize().intValue();
final int tmp = remainingUnits / blockTierSize + (remainingUnits % blockTierSize == 0 ? 0 : 1);
final int nbUsedTierBlocks;
if (tmp > tieredBlock.getMax()) {
nbUsedTierBlocks = tieredBlock.getMax().intValue();
remainingUnits -= tieredBlock.getMax() * blockTierSize;
} else {
nbUsedTierBlocks = tmp;
remainingUnits = 0;
}
result.add(tieredBlock.getPrice().getPrice(getCurrency()).multiply(new BigDecimal(nbUsedTierBlocks)));
}
return result;
}
BigDecimal computeBilledUsage(final LocalDate startDate, final LocalDate endDate, final List<InvoiceItem> existingUsage) {
final Iterable<InvoiceItem> filteredUsageForInterval = Iterables.filter(existingUsage, new Predicate<InvoiceItem>() {
@Override
public boolean apply(final InvoiceItem input) {
if (input.getInvoiceItemType() != InvoiceItemType.USAGE) {
return false;
}
// STEPH_USAGE what happens if we discover usage period that overlap (one side or both side) the [startDate, endDate] interval
final UsageInvoiceItem usageInput = (UsageInvoiceItem) input;
return usageInput.getUsageName().equals(usage.getName()) &&
usageInput.getStartDate().compareTo(startDate) >= 0 &&
usageInput.getEndDate().compareTo(endDate) <= 0;
}
});
final BigDecimal billedAmount = BigDecimal.ZERO;
for (InvoiceItem ii : filteredUsageForInterval) {
billedAmount.add(ii.getAmount());
}
// Return the billed $ amount (not the # of units)
return billedAmount;
}
private static class RolledUpUsageForUnitTypesFactory {
private final Map<String, RolledUpUsageForUnitTypes> map;
public RolledUpUsageForUnitTypesFactory(final List<RolledUpUsage> rolledUpUsages, final Set<String> unitTypes, final DateTimeZone accountTimeZone) {
map = new HashMap<String, RolledUpUsageForUnitTypes>();
for (RolledUpUsage ru : rolledUpUsages) {
final LocalDate startRolledUpDate = new LocalDate(ru.getStartTime(), accountTimeZone);
final LocalDate endRolledUpDate = new LocalDate(ru.getEndTime(), accountTimeZone);
final String key = startRolledUpDate + "-" + endRolledUpDate;
RolledUpUsageForUnitTypes usageForUnitTypes = map.get(key);
if (usageForUnitTypes == null) {
usageForUnitTypes = new RolledUpUsageForUnitTypes(startRolledUpDate, endRolledUpDate, unitTypes);
map.put(key, usageForUnitTypes);
}
usageForUnitTypes.addUsageForUnit(ru.getUnitType(), ru.getAmount());
}
}
public List<RolledUpUsageForUnitTypes> getOrderedRolledUpUsageForUnitTypes() {
final LinkedList<RolledUpUsageForUnitTypes> result = new LinkedList<RolledUpUsageForUnitTypes>(map.values());
Collections.sort(result);
return result;
}
}
/**
* Internal classes to transform RolledUpUsage into a map of usage (types, amount) across each billable interval.
*/
private static class RolledUpUsageForUnitTypes implements Comparable {
private final LocalDate startDate;
private final LocalDate endDate;
private final Map<String, BigDecimal> unitAmounts;
private RolledUpUsageForUnitTypes(final LocalDate endDate, final LocalDate startDate, final Set<String> unitTypes) {
this.endDate = endDate;
this.startDate = startDate;
this.unitAmounts = new HashMap<String, BigDecimal>();
for (final String type : unitTypes) {
unitAmounts.put(type, BigDecimal.ZERO);
}
}
public void addUsageForUnit(final String unitType, BigDecimal amount) {
final BigDecimal currentAmount = unitAmounts.get(unitType);
unitAmounts.put(unitType, currentAmount.add(amount));
}
public LocalDate getStartDate() {
return startDate;
}
public LocalDate getEndDate() {
return endDate;
}
public BigDecimal getUsageAmountForUnitType(final String unitType) {
return unitAmounts.get(unitType);
}
@Override
public int hashCode() {
int result = startDate != null ? startDate.hashCode() : 0;
result = 31 * result + (endDate != null ? endDate.hashCode() : 0);
result = 31 * result + (unitAmounts != null ? unitAmounts.hashCode() : 0);
return result;
}
@Override
public int compareTo(final Object o) {
Preconditions.checkArgument(o instanceof RolledUpUsageForUnitTypes);
final RolledUpUsageForUnitTypes other = (RolledUpUsageForUnitTypes) o;
// We will check later intervals don't overlap.
return getEndDate().compareTo(other.getStartDate());
}
}
public void addBillingEvent(final BillingEvent event) {
billingEvents.add(event);
}
public Usage getUsage() {
return usage;
}
public int getBCD() {
return billingEvents.get(0).getBillCycleDayLocal();
}
public UUID getAccountId() {
return billingEvents.get(0).getAccount().getId();
}
public UUID getBundleId() {
return billingEvents.get(0).getSubscription().getBundleId();
}
public UUID getSubscriptionId() {
return billingEvents.get(0).getSubscription().getId();
}
// STEPH_USAGE planName/phaseName,BCD,... might not be correct if we changed plan but Usage section was exactly similar
public String getPlanName() {
return billingEvents.get(0).getPlan().getName();
}
public String getPhaseName() {
return billingEvents.get(0).getPlanPhase().getName();
}
public Currency getCurrency() {
return billingEvents.get(0).getCurrency();
}
public DateTimeZone getAccountTimeZone() {
return billingEvents.get(0).getTimeZone();
}
}