google-nomulus/java/google/registry/pricing/TldSpecificLogicProxy.java

// Copyright 2016 The Domain Registry Authors. All Rights Reserved.
//
// Licensed 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 google.registry.pricing;

import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.base.Preconditions.checkState;
import static google.registry.model.ofy.ObjectifyService.ofy;
import static google.registry.pricing.PricingEngineProxy.getPricesForDomainName;
import static google.registry.util.PreconditionsUtils.checkArgumentNotNull;

import com.google.common.base.Optional;
import com.google.common.collect.ImmutableList;
import com.googlecode.objectify.Key;
import google.registry.model.ImmutableObject;
import google.registry.model.domain.DomainCommand.Create;
import google.registry.model.domain.LrpToken;
import google.registry.model.domain.fee.BaseFee.FeeType;
import google.registry.model.domain.fee.EapFee;
import google.registry.model.domain.fee.Fee;
import google.registry.model.pricing.PremiumPricingEngine.DomainPrices;
import google.registry.model.registry.Registry;
import org.joda.money.CurrencyUnit;
import org.joda.money.Money;
import org.joda.time.DateTime;

/**
 * Provides pricing, billing, and update logic, with call-outs that can be customized by providing
 * implementations on a per-TLD basis.
 */
public final class TldSpecificLogicProxy {
  /**
   * A collection of fees for a specific event.
   */
  public static final class EppCommandOperations extends ImmutableObject {
    private final CurrencyUnit currency;
    private final ImmutableList<Fee> fees;

    EppCommandOperations(CurrencyUnit currency, ImmutableList<Fee> fees) {
      this.currency = checkArgumentNotNull(
          currency, "Currency may not be null in EppCommandOperations.");
      checkArgument(!fees.isEmpty(), "You must specify one or more fees.");
      this.fees = checkArgumentNotNull(fees, "Fees may not be null in EppCommandOperations.");
    }

    private Money getTotalCostForType(FeeType type) {
      Money result = Money.zero(currency);
      checkArgumentNotNull(type);
      for (Fee fee : fees) {
        if (fee.getType() == type) {
          result = result.plus(fee.getCost());
        }
      }
      return result;
    }

    /** Returns the total cost of all fees for the event. */
    public Money getTotalCost() {
      Money result = Money.zero(currency);
      for (Fee fee : fees) {
        result = result.plus(fee.getCost());
      }
      return result;
    }

    /** Returns the create cost for the event. */
    public Money getCreateCost() {
      return getTotalCostForType(FeeType.CREATE);
    }

    /** Returns the EAP cost for the event. */
    public Money getEapCost() {
      return getTotalCostForType(FeeType.EAP);
    }

    /**
     * Returns all costs for the event as a list of fees.
     */
    public ImmutableList<Fee> getFees() {
      return fees;
    }

    /**
     * Returns the currency for all fees in the event.
     */
    public final CurrencyUnit getCurrency() {
      return currency;
    }
  }

  private TldSpecificLogicProxy() {}

  /**
   * Returns a new "create" price for the Pricer.
   */
  public static EppCommandOperations getCreatePrice(
      Registry registry, String domainName, DateTime date, int years) {
    DomainPrices prices = getPricesForDomainName(domainName, date);
    CurrencyUnit currency = registry.getCurrency();
    ImmutableList.Builder<Fee> feeBuilder = new ImmutableList.Builder<>();

    // Add Create cost.
    feeBuilder.add(
        Fee.create(prices.getCreateCost().multipliedBy(years).getAmount(), FeeType.CREATE));

    // Add EAP Fee.
    EapFee eapFee = registry.getEapFeeFor(date);
    Money eapFeeCost = eapFee.getCost();
    checkState(eapFeeCost.getCurrencyUnit().equals(currency));
    if (!eapFeeCost.getAmount().equals(Money.zero(currency).getAmount())) {
      feeBuilder.add(
          Fee.create(
              eapFeeCost.getAmount(), FeeType.EAP, eapFee.getPeriod().upperEndpoint()));
    }

    return new EppCommandOperations(currency, feeBuilder.build());
  }

  /**
   * Returns a new renew price for the pricer.
   */
  public static EppCommandOperations getRenewPrice(
      Registry registry, String domainName, DateTime date, int years) {
    DomainPrices prices = getPricesForDomainName(domainName, date);
    return new EppCommandOperations(
        registry.getCurrency(),
        ImmutableList.of(
            Fee.create(
                prices.getRenewCost().multipliedBy(years).getAmount(), FeeType.RENEW)));
  }

  /**
   * Returns a new restore price for the pricer.
   */
  public static EppCommandOperations getRestorePrice(
      Registry registry, String domainName, DateTime date, int years) {
    DomainPrices prices = getPricesForDomainName(domainName, date);
    return new EppCommandOperations(
        registry.getCurrency(),
        ImmutableList.of(
            Fee.create(
                prices.getRenewCost().multipliedBy(years).getAmount(), FeeType.RENEW),
            Fee.create(registry.getStandardRestoreCost().getAmount(), FeeType.RESTORE)));
  }

  /**
   * Returns a new transfer price for the pricer.
   */
  public static EppCommandOperations getTransferPrice(
      Registry registry, String domainName, DateTime transferDate, int additionalYears) {
    // Currently, all transfer prices = renew prices, so just pass through.
    return getRenewPrice(registry, domainName, transferDate, additionalYears);
  }

  /**
   * Returns the fee class for a given domain and date.
   */
  public static Optional<String> getFeeClass(String domainName, DateTime date) {
    return getPricesForDomainName(domainName, date).getFeeClass();
  }

  /**
   * Checks whether a {@link Create} command has a valid {@link LrpToken} for a particular TLD, and
   * return that token (wrapped in an {@link Optional}) if one exists.
   * 
   * <p>This method has no knowledge of whether or not an auth code (interpreted here as an LRP
   * token) has already been checked against the reserved list for QLP (anchor tenant), as auth
   * codes are used for both types of registrations.
   */
  public static Optional<LrpToken> getMatchingLrpToken(Create createCommand, String tld) {
    // Note that until the actual per-TLD logic is built out, what's being done here is a basic
    // domain-name-to-assignee match.
    String lrpToken = createCommand.getAuthInfo().getPw().getValue();
    LrpToken token = ofy().load().key(Key.create(LrpToken.class, lrpToken)).now();
    if (token != null) {
      if (token.getAssignee().equalsIgnoreCase(createCommand.getFullyQualifiedDomainName())
          && token.getRedemptionHistoryEntry() == null
          && token.getValidTlds().contains(tld)) {
        return Optional.of(token);
      }
    }
    return Optional.<LrpToken>absent();
  }
}