mirror of
https://github.com/google/nomulus.git
synced 2025-07-22 18:55:58 +02:00
Add XjcToContactResourceConverter
XjcToContactResourceConverter is, as it name suggests, an inverse of ContactResourceToXjcConverter. This utility class is designed to support the TLD data import feature. EXTERNAL_REVIEW_URL=https://github.com/google/domain-registry/pull/19 GIT_AUTHOR=Wolfgang Meyers <wolfgang@donuts.co> (With some minor changes by Ben McIlwain.) ------------- Created by MOE: https://github.com/google/moe MOE_MIGRATED_REVID=125985714
This commit is contained in:
parent
9eeb0c43a1
commit
bdfa97b0ae
5 changed files with 492 additions and 0 deletions
|
@ -14,6 +14,7 @@
|
|||
|
||||
package google.registry.model;
|
||||
|
||||
import static com.google.common.base.Preconditions.checkState;
|
||||
import static com.google.common.collect.Sets.difference;
|
||||
import static com.google.common.collect.Sets.union;
|
||||
import static google.registry.util.CollectionUtils.difference;
|
||||
|
@ -211,6 +212,19 @@ public abstract class EppResource extends BackupGroupRoot implements Buildable,
|
|||
super(instance);
|
||||
}
|
||||
|
||||
/**
|
||||
* Set the time this resource was created.
|
||||
*
|
||||
* <p>Note: This can only be used if the creation time hasn't already been set, which it is in
|
||||
* normal EPP flows.
|
||||
*/
|
||||
public B setCreationTime(DateTime creationTime) {
|
||||
checkState(getInstance().creationTime.timestamp == null,
|
||||
"creationTime can only be set once for EppResource.");
|
||||
getInstance().creationTime = CreateAutoTimestamp.create(creationTime);
|
||||
return thisCastToDerived();
|
||||
}
|
||||
|
||||
/** Set the time this resource was created. Should only be used in tests. */
|
||||
@VisibleForTesting
|
||||
public B setCreationTimeForTest(DateTime creationTime) {
|
||||
|
|
196
java/google/registry/rde/XjcToContactResourceConverter.java
Normal file
196
java/google/registry/rde/XjcToContactResourceConverter.java
Normal file
|
@ -0,0 +1,196 @@
|
|||
// 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.
|
||||
|
||||
// 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.rde;
|
||||
|
||||
import com.google.common.base.Function;
|
||||
import com.google.common.collect.ImmutableList;
|
||||
import com.google.common.collect.ImmutableSet;
|
||||
import com.google.common.collect.Iterables;
|
||||
import com.google.common.collect.Lists;
|
||||
|
||||
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.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.XjcRdeContactTransferDataType;
|
||||
|
||||
import javax.annotation.Nullable;
|
||||
|
||||
/** Utility class that converts an {@link XjcRdeContact} into a {@link ContactResource}. */
|
||||
final class XjcToContactResourceConverter {
|
||||
|
||||
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_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) {
|
||||
return new ContactResource.Builder()
|
||||
.setRepoId(contact.getRoid())
|
||||
.setStatusValues(
|
||||
ImmutableSet.copyOf(Iterables.transform(contact.getStatuses(), STATUS_CONVERTER)))
|
||||
.setLocalizedPostalInfo(
|
||||
getPostalInfoOfType(contact.getPostalInfos(), XjcContactPostalInfoEnumType.LOC))
|
||||
.setInternationalizedPostalInfo(
|
||||
getPostalInfoOfType(contact.getPostalInfos(), XjcContactPostalInfoEnumType.INT))
|
||||
.setContactId(contact.getId())
|
||||
.setCurrentSponsorClientId(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() {}
|
||||
}
|
55
java/google/registry/util/XmlToEnumMapper.java
Normal file
55
java/google/registry/util/XmlToEnumMapper.java
Normal file
|
@ -0,0 +1,55 @@
|
|||
// 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.util;
|
||||
|
||||
import com.google.common.collect.ImmutableMap;
|
||||
|
||||
import javax.xml.bind.annotation.XmlEnumValue;
|
||||
|
||||
/** Efficient lookup from xml enums to java enums */
|
||||
public final class XmlToEnumMapper<T extends Enum<?>> {
|
||||
|
||||
private final ImmutableMap<String, T> map;
|
||||
|
||||
/** Look up T from the {@link XmlEnumValue} */
|
||||
public T xmlToEnum(String value) {
|
||||
return map.get(value);
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates a new {@link XmlToEnumMapper} from xml value to enum value.
|
||||
*/
|
||||
public static <T extends Enum<?>> XmlToEnumMapper<T> create(T[] enumValues) {
|
||||
return new XmlToEnumMapper<T>(enumValues);
|
||||
}
|
||||
|
||||
private XmlToEnumMapper(T[] enumValues) {
|
||||
ImmutableMap.Builder<String, T> mapBuilder = new ImmutableMap.Builder<>();
|
||||
for (T value : enumValues) {
|
||||
try {
|
||||
String xmlName =
|
||||
value
|
||||
.getDeclaringClass()
|
||||
.getField(value.name())
|
||||
.getAnnotation(XmlEnumValue.class)
|
||||
.value();
|
||||
mapBuilder = mapBuilder.put(xmlName, value);
|
||||
} catch (NoSuchFieldException e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
map = mapBuilder.build();
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue