/*
* Copyright 2010-2013 Ning, 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 com.ning.billing.jaxrs.json;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.UUID;
import javax.annotation.Nullable;
import com.ning.billing.account.api.Account;
import com.ning.billing.subscription.api.timeline.BundleTimeline;
import com.ning.billing.invoice.api.Invoice;
import com.ning.billing.invoice.api.InvoiceItem;
import com.ning.billing.invoice.api.InvoiceItemType;
import com.ning.billing.invoice.api.InvoicePayment;
import com.ning.billing.payment.api.Payment;
import com.ning.billing.payment.api.Refund;
import com.ning.billing.util.audit.AuditLog;
import com.ning.billing.util.audit.AuditLogsForBundles;
import com.ning.billing.util.audit.AuditLogsForInvoicePayments;
import com.ning.billing.util.audit.AuditLogsForInvoices;
import com.ning.billing.util.audit.AuditLogsForPayments;
import com.ning.billing.util.audit.AuditLogsForRefunds;
import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Multimap;
public class AccountTimelineJson {
private final AccountJsonSimple account;
private final List<BundleJsonWithSubscriptions> bundles;
private final List<InvoiceJsonWithBundleKeys> invoices;
private final List<PaymentJsonWithBundleKeys> payments;
@JsonCreator
public AccountTimelineJson(@JsonProperty("account") final AccountJsonSimple account,
@JsonProperty("bundles") final List<BundleJsonWithSubscriptions> bundles,
@JsonProperty("invoices") final List<InvoiceJsonWithBundleKeys> invoices,
@JsonProperty("payments") final List<PaymentJsonWithBundleKeys> payments) {
this.account = account;
this.bundles = bundles;
this.invoices = invoices;
this.payments = payments;
}
private String getBundleExternalKey(final UUID invoiceId, final List<Invoice> invoices, final List<BundleTimeline> bundles) {
for (final Invoice cur : invoices) {
if (cur.getId().equals(invoiceId)) {
return getBundleExternalKey(cur, bundles);
}
}
return null;
}
private String getBundleExternalKey(final Invoice invoice, final List<BundleTimeline> bundles) {
final Set<UUID> b = new HashSet<UUID>();
for (final InvoiceItem cur : invoice.getInvoiceItems()) {
b.add(cur.getBundleId());
}
boolean first = true;
final StringBuilder tmp = new StringBuilder();
for (final UUID cur : b) {
for (final BundleTimeline bt : bundles) {
if (bt.getId().equals(cur)) {
if (!first) {
tmp.append(",");
}
tmp.append(bt.getExternalKey());
first = false;
break;
}
}
}
return tmp.toString();
}
public AccountTimelineJson(final Account account, final List<Invoice> invoices, final List<Payment> payments,
final List<BundleTimeline> bundlesTimeline, final Multimap<UUID, Refund> refundsByPayment,
final Multimap<UUID, InvoicePayment> chargebacksByPayment, @Nullable final AuditLogsForInvoices invoicesAuditLogs,
@Nullable final AuditLogsForPayments paymentsAuditLogs, @Nullable final AuditLogsForRefunds refundsAuditLogs,
@Nullable final AuditLogsForInvoicePayments chargebacksAuditLogs, @Nullable final AuditLogsForBundles bundlesAuditLogs) {
this(account, invoices, payments, bundlesTimeline, refundsByPayment, chargebacksByPayment,
invoicesAuditLogs == null ? ImmutableMap.<UUID, List<AuditLog>>of() : invoicesAuditLogs.getInvoiceAuditLogs(),
invoicesAuditLogs == null ? ImmutableMap.<UUID, List<AuditLog>>of() : invoicesAuditLogs.getInvoiceItemsAuditLogs(),
paymentsAuditLogs == null ? ImmutableMap.<UUID, List<AuditLog>>of() : paymentsAuditLogs.getPaymentsAuditLogs(),
refundsAuditLogs == null ? ImmutableMap.<UUID, List<AuditLog>>of() : refundsAuditLogs.getRefundsAuditLogs(),
chargebacksAuditLogs == null ? ImmutableMap.<UUID, List<AuditLog>>of() : chargebacksAuditLogs.getInvoicePaymentsAuditLogs(),
bundlesAuditLogs == null ? ImmutableMap.<UUID, List<AuditLog>>of() : bundlesAuditLogs.getBundlesAuditLogs(),
bundlesAuditLogs == null ? ImmutableMap.<UUID, List<AuditLog>>of() : bundlesAuditLogs.getSubscriptionsAuditLogs(),
bundlesAuditLogs == null ? ImmutableMap.<UUID, List<AuditLog>>of() : bundlesAuditLogs.getSubscriptionEventsAuditLogs());
}
public AccountTimelineJson(final Account account, final List<Invoice> invoices, final List<Payment> payments, final List<BundleTimeline> bundles,
final Multimap<UUID, Refund> refundsByPayment, final Multimap<UUID, InvoicePayment> chargebacksByPayment,
final Map<UUID, List<AuditLog>> invoiceAuditLogs, final Map<UUID, List<AuditLog>> invoiceItemsAuditLogs,
final Map<UUID, List<AuditLog>> paymentsAuditLogs, final Map<UUID, List<AuditLog>> refundsAuditLogs,
final Map<UUID, List<AuditLog>> chargebacksAuditLogs, final Map<UUID, List<AuditLog>> bundlesAuditLogs,
final Map<UUID, List<AuditLog>> subscriptionsAuditLogs, final Map<UUID, List<AuditLog>> subscriptionEventsAuditLogs) {
this.account = new AccountJsonSimple(account.getId().toString(), account.getExternalKey());
this.bundles = new LinkedList<BundleJsonWithSubscriptions>();
for (final BundleTimeline bundle : bundles) {
final List<AuditLog> bundleAuditLogs = bundlesAuditLogs.get(bundle.getId());
final BundleJsonWithSubscriptions jsonWithSubscriptions = new BundleJsonWithSubscriptions(bundle, bundleAuditLogs,
subscriptionsAuditLogs, subscriptionEventsAuditLogs);
this.bundles.add(jsonWithSubscriptions);
}
this.invoices = new LinkedList<InvoiceJsonWithBundleKeys>();
// Extract the credits from the invoices first
final List<CreditJson> credits = new ArrayList<CreditJson>();
for (final Invoice invoice : invoices) {
for (final InvoiceItem invoiceItem : invoice.getInvoiceItems()) {
if (InvoiceItemType.CREDIT_ADJ.equals(invoiceItem.getInvoiceItemType())) {
final List<AuditLog> auditLogs = invoiceItemsAuditLogs.get(invoiceItem.getId());
credits.add(new CreditJson(invoiceItem, account.getTimeZone(), auditLogs));
}
}
}
// Create now the invoice json objects
for (final Invoice invoice : invoices) {
final List<AuditLog> auditLogs = invoiceAuditLogs.get(invoice.getId());
this.invoices.add(new InvoiceJsonWithBundleKeys(invoice,
getBundleExternalKey(invoice, bundles),
credits,
auditLogs));
}
this.payments = new LinkedList<PaymentJsonWithBundleKeys>();
for (final Payment payment : payments) {
final List<RefundJson> refunds = new ArrayList<RefundJson>();
for (final Refund refund : refundsByPayment.get(payment.getId())) {
final List<AuditLog> auditLogs = refundsAuditLogs.get(refund.getId());
// TODO add adjusted invoice items?
refunds.add(new RefundJson(refund, null, auditLogs));
}
final List<ChargebackJson> chargebacks = new ArrayList<ChargebackJson>();
for (final InvoicePayment chargeback : chargebacksByPayment.get(payment.getId())) {
final List<AuditLog> auditLogs = chargebacksAuditLogs.get(chargeback.getId());
chargebacks.add(new ChargebackJson(chargeback, auditLogs));
}
final int nbOfPaymentAttempts = payment.getAttempts().size();
final String status = payment.getPaymentStatus().toString();
final List<AuditLog> auditLogs = paymentsAuditLogs.get(payment.getId());
this.payments.add(new PaymentJsonWithBundleKeys(payment,
status,
nbOfPaymentAttempts,
getBundleExternalKey(payment.getInvoiceId(), invoices, bundles),
account.getId(),
refunds,
chargebacks,
auditLogs));
}
}
public AccountJsonSimple getAccount() {
return account;
}
public List<BundleJsonWithSubscriptions> getBundles() {
return bundles;
}
public List<InvoiceJsonWithBundleKeys> getInvoices() {
return invoices;
}
public List<PaymentJsonWithBundleKeys> getPayments() {
return payments;
}
@Override
public String toString() {
final StringBuilder sb = new StringBuilder();
sb.append("AccountTimelineJson");
sb.append("{account=").append(account);
sb.append(", bundles=").append(bundles);
sb.append(", invoices=").append(invoices);
sb.append(", payments=").append(payments);
sb.append('}');
return sb.toString();
}
@Override
public boolean equals(final Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
final AccountTimelineJson that = (AccountTimelineJson) o;
if (account != null ? !account.equals(that.account) : that.account != null) {
return false;
}
if (bundles != null ? !bundles.equals(that.bundles) : that.bundles != null) {
return false;
}
if (invoices != null ? !invoices.equals(that.invoices) : that.invoices != null) {
return false;
}
if (payments != null ? !payments.equals(that.payments) : that.payments != null) {
return false;
}
return true;
}
@Override
public int hashCode() {
int result = account != null ? account.hashCode() : 0;
result = 31 * result + (bundles != null ? bundles.hashCode() : 0);
result = 31 * result + (invoices != null ? invoices.hashCode() : 0);
result = 31 * result + (payments != null ? payments.hashCode() : 0);
return result;
}
}