google-nomulus/java/google/registry/rde/imports/XjcToContactResourceConverter.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.Predicates.equalTo;
import static com.google.common.base.Predicates.not;
import static google.registry.model.ofy.ObjectifyService.ofy;
import static google.registry.rde.imports.RdeImportUtils.generateTridForImport;

import com.google.common.base.Function;
import com.google.common.collect.FluentIterable;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Lists;
import com.googlecode.objectify.Key;
import google.registry.model.contact.ContactAddress;
import google.registry.model.contact.ContactPhoneNumber;
import google.registry.model.contact.ContactResource;
import google.registry.model.contact.Disclose;
import google.registry.model.contact.Disclose.PostalInfoChoice;
import google.registry.model.contact.PostalInfo;
import google.registry.model.eppcommon.StatusValue;
import google.registry.model.reporting.HistoryEntry;
import google.registry.model.transfer.TransferData;
import google.registry.model.transfer.TransferStatus;
import google.registry.util.XmlToEnumMapper;
import google.registry.xjc.contact.XjcContactAddrType;
import google.registry.xjc.contact.XjcContactDiscloseType;
import google.registry.xjc.contact.XjcContactE164Type;
import google.registry.xjc.contact.XjcContactIntLocType;
import google.registry.xjc.contact.XjcContactPostalInfoEnumType;
import google.registry.xjc.contact.XjcContactPostalInfoType;
import google.registry.xjc.contact.XjcContactStatusType;
import google.registry.xjc.rdecontact.XjcRdeContact;
import google.registry.xjc.rdecontact.XjcRdeContactElement;
import google.registry.xjc.rdecontact.XjcRdeContactTransferDataType;
import javax.annotation.Nullable;

/** Utility class that converts an {@link XjcRdeContact} into a {@link ContactResource}. */
final class XjcToContactResourceConverter extends XjcToEppResourceConverter {

  private static final XmlToEnumMapper<PostalInfo.Type> POSTAL_INFO_TYPE_MAPPER =
      XmlToEnumMapper.create(PostalInfo.Type.values());
  private static final XmlToEnumMapper<TransferStatus> TRANSFER_STATUS_MAPPER =
      XmlToEnumMapper.create(TransferStatus.values());

  private static final Function<XjcContactIntLocType, PostalInfoChoice> choiceConverter =
      new Function<XjcContactIntLocType, PostalInfoChoice>() {
        @Override
        public PostalInfoChoice apply(XjcContactIntLocType choice) {
          return convertPostalInfoChoice(choice);
        }
      };

  private static final Function<XjcContactStatusType, StatusValue> STATUS_VALUE_CONVERTER =
      new Function<XjcContactStatusType, StatusValue>() {
        @Override
        public StatusValue apply(XjcContactStatusType status) {
          return convertStatusValue(status);
        }
      };

  /** Converts {@link XjcRdeContact} to {@link ContactResource}. */
  static ContactResource convertContact(XjcRdeContact contact) {
    ofy().save().entity(
        new HistoryEntry.Builder()
            .setType(HistoryEntry.Type.RDE_IMPORT)
            .setClientId(contact.getClID())
            .setTrid(generateTridForImport())
            .setModificationTime(ofy().getTransactionTime())
            .setXmlBytes(getObjectXml(new XjcRdeContactElement(contact)))
            .setBySuperuser(true)
            .setReason("RDE Import")
            .setRequestedByRegistrar(false)
            .setParent(Key.create(null, ContactResource.class, contact.getRoid()))
            .build());
    return new ContactResource.Builder()
        .setRepoId(contact.getRoid())
        .setStatusValues(
            FluentIterable.from(contact.getStatuses())
                .transform(STATUS_VALUE_CONVERTER)
                // LINKED is implicit and should not be imported onto the new contact.
                .filter(not(equalTo(StatusValue.LINKED)))
                .toSet())
        .setLocalizedPostalInfo(
            getPostalInfoOfType(contact.getPostalInfos(), XjcContactPostalInfoEnumType.LOC))
        .setInternationalizedPostalInfo(
            getPostalInfoOfType(contact.getPostalInfos(), XjcContactPostalInfoEnumType.INT))
        .setContactId(contact.getId())
        .setPersistedCurrentSponsorClientId(contact.getClID())
        .setCreationClientId(contact.getCrRr() == null ? null : contact.getCrRr().getValue())
        .setLastEppUpdateClientId(contact.getUpRr() == null ? null : contact.getUpRr().getValue())
        .setCreationTime(contact.getCrDate())
        .setLastEppUpdateTime(contact.getUpDate())
        .setLastTransferTime(contact.getTrDate())
        .setVoiceNumber(convertPhoneNumber(contact.getVoice()))
        .setFaxNumber(convertPhoneNumber(contact.getFax()))
        .setEmailAddress(contact.getEmail())
        .setDisclose(convertDisclose(contact.getDisclose()))
        .setTransferData(convertTransferData(contact.getTrnData()))
        .build();
  }

  /**
   * Extracts a {@link PostalInfo} from an {@link Iterable} of {@link XjcContactPostalInfoEnumType}.
   */
  @Nullable
  private static PostalInfo getPostalInfoOfType(
      Iterable<XjcContactPostalInfoType> postalInfos, XjcContactPostalInfoEnumType type) {
    for (XjcContactPostalInfoType postalInfo : postalInfos) {
      if (postalInfo.getType() == type) {
        return convertPostalInfo(postalInfo);
      }
    }
    return null;
  }

  /** Converts {@link XjcRdeContactTransferDataType} to {@link TransferData}. */
  private static TransferData convertTransferData(
      @Nullable XjcRdeContactTransferDataType transferData) {
    if (transferData == null) {
      return TransferData.EMPTY;
    }
    return new TransferData.Builder()
        .setTransferStatus(TRANSFER_STATUS_MAPPER.xmlToEnum(transferData.getTrStatus().value()))
        .setGainingClientId(transferData.getReRr().getValue())
        .setLosingClientId(transferData.getAcRr().getValue())
        .setTransferRequestTime(transferData.getReDate())
        .setPendingTransferExpirationTime(transferData.getAcDate())
        .build();
  }

  /** Converts {@link XjcContactAddrType} to {@link ContactAddress}. */
  private static ContactAddress convertAddress(XjcContactAddrType address) {
    return new ContactAddress.Builder()
        .setStreet(ImmutableList.copyOf(address.getStreets()))
        .setCity(address.getCity())
        .setState(address.getSp())
        .setZip(address.getPc())
        .setCountryCode(address.getCc())
        .build();
  }

  /** Converts {@link XjcContactDiscloseType} to {@link Disclose}. */
  @Nullable
  private static Disclose convertDisclose(@Nullable XjcContactDiscloseType disclose) {
    if (disclose == null) {
      return null;
    }
    return new Disclose.Builder()
        .setFlag(disclose.isFlag())
        .setNames(ImmutableList.copyOf(Lists.transform(disclose.getNames(), choiceConverter)))
        .setOrgs(ImmutableList.copyOf(Lists.transform(disclose.getOrgs(), choiceConverter)))
        .setAddrs(ImmutableList.copyOf(Lists.transform(disclose.getAddrs(), choiceConverter)))
        .build();
  }

  /** Converts {@link XjcContactE164Type} to {@link ContactPhoneNumber}. */
  @Nullable
  private static ContactPhoneNumber convertPhoneNumber(@Nullable XjcContactE164Type phoneNumber) {
    if (phoneNumber == null) {
      return null;
    }
    return new ContactPhoneNumber.Builder()
        .setPhoneNumber(phoneNumber.getValue())
        .setExtension(phoneNumber.getX())
        .build();
  }

  /** Converts {@link PostalInfoChoice} to {@link XjcContactIntLocType}. */
  private static PostalInfoChoice convertPostalInfoChoice(XjcContactIntLocType choice) {
    return PostalInfoChoice.create(POSTAL_INFO_TYPE_MAPPER.xmlToEnum(choice.getType().value()));
  }

  /** Converts {@link XjcContactPostalInfoType} to {@link PostalInfo}. */
  private static PostalInfo convertPostalInfo(XjcContactPostalInfoType postalInfo) {
    return new PostalInfo.Builder()
        .setName(postalInfo.getName())
        .setOrg(postalInfo.getOrg())
        .setAddress(convertAddress(postalInfo.getAddr()))
        .setType(POSTAL_INFO_TYPE_MAPPER.xmlToEnum(postalInfo.getType().value()))
        .build();
  }

  /** Converts {@link XjcContactStatusType} to {@link StatusValue}. */
  private static StatusValue convertStatusValue(XjcContactStatusType statusType) {
    return StatusValue.fromXmlName(statusType.getS().value());
  }

  private XjcToContactResourceConverter() {}
}