google-nomulus/java/google/registry/rde/imports/RdeImportUtils.java

// Copyright 2017 The Nomulus 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.rde.imports;

import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.base.Preconditions.checkState;
import static google.registry.util.DateTimeUtils.END_OF_TIME;
import static google.registry.util.DateTimeUtils.START_OF_TIME;
import static java.nio.charset.StandardCharsets.US_ASCII;
import static java.nio.charset.StandardCharsets.UTF_8;

import com.google.appengine.tools.cloudstorage.GcsFilename;
import com.google.common.collect.ImmutableSet;
import com.google.common.io.BaseEncoding;
import com.googlecode.objectify.Key;
import google.registry.config.RegistryConfig.Config;
import google.registry.flows.ServerTridProvider;
import google.registry.flows.ServerTridProviderImpl;
import google.registry.gcs.GcsUtils;
import google.registry.model.EppResource;
import google.registry.model.EppResource.ForeignKeyedEppResource;
import google.registry.model.billing.BillingEvent;
import google.registry.model.billing.BillingEvent.Flag;
import google.registry.model.billing.BillingEvent.Reason;
import google.registry.model.domain.DomainResource;
import google.registry.model.eppcommon.Trid;
import google.registry.model.index.EppResourceIndex;
import google.registry.model.index.ForeignKeyIndex;
import google.registry.model.ofy.Ofy;
import google.registry.model.poll.PollMessage;
import google.registry.model.registrar.Registrar;
import google.registry.model.registry.Registry;
import google.registry.model.registry.Registry.RegistryNotFoundException;
import google.registry.model.registry.Registry.TldState;
import google.registry.model.reporting.HistoryEntry;
import google.registry.util.Clock;
import google.registry.util.FormattingLogger;
import google.registry.xjc.XjcXmlTransformer;
import google.registry.xjc.rdedomain.XjcRdeDomain;
import google.registry.xjc.rdedomain.XjcRdeDomainElement;
import google.registry.xjc.rderegistrar.XjcRdeRegistrar;
import google.registry.xml.XmlException;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.UUID;
import javax.inject.Inject;
import javax.xml.bind.JAXBException;
import javax.xml.stream.XMLStreamException;
import org.joda.time.DateTime;

/**
 * Utility functions for escrow file import.
 */
public class RdeImportUtils {

  private static final FormattingLogger logger = FormattingLogger.getLoggerForCallerClass();

  private final Ofy ofy;
  private final Clock clock;
  private final String escrowBucketName;
  private final GcsUtils gcsUtils;

  private static final ServerTridProvider serverTridProvider = new ServerTridProviderImpl();

  @Inject
  public RdeImportUtils(
      Ofy ofy, Clock clock, @Config("rdeImportBucket") String escrowBucketName, GcsUtils gcsUtils) {
    this.ofy = ofy;
    this.clock = clock;
    this.gcsUtils = gcsUtils;
    this.escrowBucketName = escrowBucketName;
  }

  public <T extends EppResource & ForeignKeyedEppResource> ImmutableSet<Object>
      createIndexesForEppResource(T resource) {
    @SuppressWarnings("unchecked")
    Class<T> resourceClass = (Class<T>) resource.getClass();
    Object existing = ofy.load().key(Key.create(resource)).now();
    if (existing != null) {
      // This will roll back the transaction and prevent duplicate history entries from being saved.
      throw new ResourceExistsException();
    }
    ForeignKeyIndex<T> existingForeignKeyIndex =
        ForeignKeyIndex.load(resourceClass, resource.getForeignKey(), START_OF_TIME);
    // ForeignKeyIndex should never have existed, since existing resource was not found.
    checkState(
        existingForeignKeyIndex == null,
        "New %s resource has existing foreign key index; foreignKey=%s, repoId=%s",
        resource.getClass().getCanonicalName(),
        resource.getForeignKey(),
        resource.getRepoId());
    return ImmutableSet.<Object>of(ForeignKeyIndex.create(resource, resource.getDeletionTime()),
        EppResourceIndex.create(Key.create(resource)));
  }

  /**
   * Imports a resource from an escrow file.
   *
   * <p>The host will only be imported if it has not been previously imported.
   *
   * <p>If the host is imported, {@link ForeignKeyIndex} and {@link EppResourceIndex} are also
   * created.
   */
  public <T extends EppResource & ForeignKeyedEppResource> void
      importEppResource(final T resource) {
    ofy.save().entities(new ImmutableSet.Builder<>()
        .add(resource)
        .addAll(createIndexesForEppResource(resource))
        .build());
    logger.infofmt(
        "Imported %s resource - ROID=%s, id=%s",
        resource.getClass().getCanonicalName(), resource.getRepoId(), resource.getForeignKey());
  }

  /**
   * Validates an escrow file for import.
   *
   * <p>Before an escrow file is imported into the registry, the following conditions must be met:
   *
   * <ul>
   * <li>The TLD must already exist in the registry
   * <li>The TLD must be in the PREDELEGATION state
   * <li>Each registrar must already exist in the registry
   * <li>Each IDN table referenced must already exist in the registry
   * </ul>
   *
   * <p>If any of the above conditions is not true, an {@link IllegalStateException} will be thrown.
   *
   * @param escrowFilePath Path to the escrow file to validate
   * @throws IOException If the escrow file cannot be read
   * @throws IllegalArgumentException if the escrow file cannot be imported
   */
  public void validateEscrowFileForImport(String escrowFilePath) throws IOException {
    // TODO (wolfgang@donuts.co): Add validation method for IDN tables
    try (InputStream input =
        gcsUtils.openInputStream(new GcsFilename(escrowBucketName, escrowFilePath))) {
      try (RdeParser parser = new RdeParser(input)) {
        // validate that tld exists and is in PREDELEGATION state
        String tld = parser.getHeader().getTld();
        try {
          Registry registry = Registry.get(tld);
          TldState currentState = registry.getTldState(clock.nowUtc());
          checkArgument(
              currentState == TldState.PREDELEGATION,
              String.format("Tld '%s' is in state %s and cannot be imported", tld, currentState));
        } catch (RegistryNotFoundException e) {
          throw new IllegalArgumentException(
              String.format("Tld '%s' not found in the registry", tld));
        }
        // validate that all registrars exist
        while (parser.nextRegistrar()) {
          XjcRdeRegistrar registrar = parser.getRegistrar();
          if (Registrar.loadByClientIdCached(registrar.getId()) == null) {
            throw new IllegalArgumentException(
                String.format("Registrar '%s' not found in the registry", registrar.getId()));
          }
        }
      } catch (XMLStreamException | JAXBException e) {
        throw new IllegalArgumentException(
            String.format("Invalid XML file: '%s'", escrowFilePath), e);
      }
    }
  }

  /** Generates a random {@link Trid} for rde import. */
  public static Trid generateTridForImport() {
    // Client trids must be a token between 3 and 64 characters long
    // Base64 encoded UUID string meets this requirement
    return Trid.create(
        "Import_" + BaseEncoding.base64().encode(UUID.randomUUID().toString().getBytes(US_ASCII)),
        serverTridProvider.createServerTrid());
  }

  public static BillingEvent.Recurring createAutoRenewBillingEventForDomainImport(
      XjcRdeDomain domain, HistoryEntry historyEntry) {
    final BillingEvent.Recurring billingEvent =
        new BillingEvent.Recurring.Builder()
            .setReason(Reason.RENEW)
            .setFlags(ImmutableSet.of(Flag.AUTO_RENEW))
            .setTargetId(domain.getRoid())
            .setClientId(domain.getClID())
            .setEventTime(domain.getExDate())
            .setRecurrenceEndTime(END_OF_TIME)
            .setParent(historyEntry)
            .build();
    return billingEvent;
  }

  public static PollMessage.Autorenew createAutoRenewPollMessageForDomainImport(
      XjcRdeDomain domain, HistoryEntry historyEntry) {
    final PollMessage.Autorenew pollMessage =
        new PollMessage.Autorenew.Builder()
            .setTargetId(domain.getRoid())
            .setClientId(domain.getClID())
            .setEventTime(domain.getExDate())
            .setMsg("Domain was auto-renewed.")
            .setParent(historyEntry)
            .build();
    return pollMessage;
  }

  public static HistoryEntry createHistoryEntryForDomainImport(XjcRdeDomain domain) {
    XjcRdeDomainElement element = new XjcRdeDomainElement(domain);
    final HistoryEntry historyEntry =
        new HistoryEntry.Builder()
            .setType(HistoryEntry.Type.RDE_IMPORT)
            .setClientId(domain.getClID())
            .setTrid(generateTridForImport())
            .setModificationTime(DateTime.now())
            .setXmlBytes(getObjectXml(element))
            .setBySuperuser(true)
            .setReason("RDE Import")
            .setRequestedByRegistrar(false)
            .setParent(Key.create(null, DomainResource.class, domain.getRoid()))
            .build();
    return historyEntry;
  }

  public static byte[] getObjectXml(Object jaxbElement) {
    try {
      ByteArrayOutputStream bout = new ByteArrayOutputStream();
      XjcXmlTransformer.marshalLenient(jaxbElement, bout, UTF_8);
      return bout.toByteArray();
    } catch (XmlException e) {
      throw new RuntimeException(e);
    }
  }
}