mv com/google/domain/registry google/registry

This change renames directories in preparation for the great package
rename. The repository is now in a broken state because the code
itself hasn't been updated. However this should ensure that git
correctly preserves history for each file.

java/google/registry/model/poll/PollMessage.java

@@ -0,0 +1,324 @@
+// 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 com.google.domain.registry.model.poll;
+
+import static com.google.common.base.Preconditions.checkNotNull;
+import static com.google.domain.registry.util.CollectionUtils.forceEmptyToNull;
+import static com.google.domain.registry.util.CollectionUtils.isNullOrEmpty;
+import static com.google.domain.registry.util.CollectionUtils.nullToEmpty;
+import static com.google.domain.registry.util.DateTimeUtils.END_OF_TIME;
+
+import com.google.common.annotations.VisibleForTesting;
+import com.google.common.base.Converter;
+import com.google.common.base.Optional;
+import com.google.common.collect.FluentIterable;
+import com.google.common.collect.ImmutableList;
+import com.google.domain.registry.model.Buildable;
+import com.google.domain.registry.model.EppResource;
+import com.google.domain.registry.model.ImmutableObject;
+import com.google.domain.registry.model.annotations.ExternalMessagingName;
+import com.google.domain.registry.model.contact.ContactResource;
+import com.google.domain.registry.model.domain.DomainBase;
+import com.google.domain.registry.model.domain.DomainRenewData;
+import com.google.domain.registry.model.domain.launch.LaunchInfoResponseExtension;
+import com.google.domain.registry.model.eppoutput.Response.ResponseData;
+import com.google.domain.registry.model.eppoutput.Response.ResponseExtension;
+import com.google.domain.registry.model.poll.PendingActionNotificationResponse.ContactPendingActionNotificationResponse;
+import com.google.domain.registry.model.poll.PendingActionNotificationResponse.DomainPendingActionNotificationResponse;
+import com.google.domain.registry.model.reporting.HistoryEntry;
+import com.google.domain.registry.model.transfer.TransferData.TransferServerApproveEntity;
+import com.google.domain.registry.model.transfer.TransferResponse.ContactTransferResponse;
+import com.google.domain.registry.model.transfer.TransferResponse.DomainTransferResponse;
+
+import com.googlecode.objectify.Key;
+import com.googlecode.objectify.annotation.Entity;
+import com.googlecode.objectify.annotation.EntitySubclass;
+import com.googlecode.objectify.annotation.Id;
+import com.googlecode.objectify.annotation.Index;
+import com.googlecode.objectify.annotation.Parent;
+
+import org.joda.time.DateTime;
+
+import java.util.List;
+
+/** A poll message that is pending for a registrar. */
+@Entity
+@ExternalMessagingName("message")
+public abstract class PollMessage
+    extends ImmutableObject implements Buildable, TransferServerApproveEntity {
+
+
+  public static final Converter<Key<PollMessage>, String> EXTERNAL_KEY_CONVERTER =
+      new PollMessageExternalKeyConverter();
+
+  /** Entity id. */
+  @Id
+  long id;
+
+  @Parent
+  Key<HistoryEntry> parent;
+
+  /** The registrar that this poll message will be delivered to. */
+  @Index
+  String clientId;
+
+  /** The time when the poll message should be delivered. May be in the future. */
+  @Index
+  DateTime eventTime;
+
+  /** Human readable message that will be returned with this poll message. */
+  String msg;
+
+  public Key<HistoryEntry> getParentKey() {
+    return parent;
+  }
+
+  public Long getId() {
+    return id;
+  }
+
+  public String getClientId() {
+    return clientId;
+  }
+
+  public DateTime getEventTime() {
+    return eventTime;
+  }
+
+  public String getMsg() {
+    return msg;
+  }
+
+  public abstract ImmutableList<ResponseData> getResponseData();
+
+  public abstract ImmutableList<ResponseExtension> getResponseExtensions();
+
+  /** Override Buildable.asBuilder() to give this method stronger typing. */
+  @Override
+  public abstract Builder<?, ?> asBuilder();
+
+  /** Builder for {@link PollMessage} because it is immutable. */
+  @VisibleForTesting
+  public abstract static class Builder<T extends PollMessage, B extends Builder<?, ?>>
+      extends GenericBuilder<T, B> {
+
+    protected Builder() {}
+
+    protected Builder(T instance) {
+      super(instance);
+    }
+
+    @VisibleForTesting
+    public B setId(Long id) {
+      getInstance().id = id;
+      return thisCastToDerived();
+    }
+
+    public B setClientId(String clientId) {
+      getInstance().clientId = clientId;
+      return thisCastToDerived();
+    }
+
+    public B setEventTime(DateTime eventTime) {
+      getInstance().eventTime = eventTime;
+      return thisCastToDerived();
+    }
+
+    public B setMsg(String msg) {
+      getInstance().msg = msg;
+      return thisCastToDerived();
+    }
+
+    public B setParent(HistoryEntry parent) {
+      getInstance().parent = Key.create(parent);
+      return thisCastToDerived();
+    }
+
+    public B setParentKey(Key<HistoryEntry> parentKey) {
+      getInstance().parent = parentKey;
+      return thisCastToDerived();
+    }
+
+    @Override
+    public T build() {
+      T instance = getInstance();
+      checkNotNull(instance.clientId);
+      checkNotNull(instance.eventTime);
+      checkNotNull(instance.parent);
+      return super.build();
+    }
+  }
+
+  /** A one-time poll message. */
+  @EntitySubclass(index = false)
+  public static class OneTime extends PollMessage {
+
+    // Response data. Objectify cannot persist a base class type, so we must have a separate field
+    // to hold every possible derived type of ResponseData that we might store.
+    List<ContactPendingActionNotificationResponse> contactPendingActionNotificationResponses;
+    List<ContactTransferResponse> contactTransferResponses;
+    List<DomainPendingActionNotificationResponse> domainPendingActionNotificationResponses;
+    List<DomainTransferResponse> domainTransferResponses;
+
+    // Extensions. Objectify cannot persist a base class type, so we must have a separate field
+    // to hold every possible derived type of ResponseExtensions that we might store.
+    //
+    // Note that we cannot store a list of LaunchInfoResponseExtension objects since it contains a
+    // list of embedded Mark objects, and embedded lists of lists are not allowed. This shouldn't
+    // matter since there's no scenario where multiple launch info response extensions are ever
+    // returned.
+    LaunchInfoResponseExtension launchInfoResponseExtension;
+
+    @Override
+    public Builder asBuilder() {
+      return new Builder(clone(this));
+    }
+
+    @Override
+    public ImmutableList<ResponseData> getResponseData() {
+      return new ImmutableList.Builder<ResponseData>()
+          .addAll(nullToEmpty(contactPendingActionNotificationResponses))
+          .addAll(nullToEmpty(contactTransferResponses))
+          .addAll(nullToEmpty(domainPendingActionNotificationResponses))
+          .addAll(nullToEmpty(domainTransferResponses))
+          .build();
+    }
+
+    /**
+     * Returns the class of the parent EppResource that this poll message is associated with,
+     * either DomainBase or ContactResource.
+     */
+    public Class<? extends EppResource> getParentResourceClass() {
+      if (!isNullOrEmpty(domainPendingActionNotificationResponses)
+          || !isNullOrEmpty(domainTransferResponses)) {
+        return DomainBase.class;
+      } else if (!isNullOrEmpty(contactPendingActionNotificationResponses)
+          || !isNullOrEmpty(contactTransferResponses)) {
+        return ContactResource.class;
+      } else {
+        throw new IllegalStateException(String.format(
+            "PollMessage.OneTime %s does not correspond with an EppResource of a known type",
+            Key.create(this)));
+      }
+    }
+
+    @Override
+    public ImmutableList<ResponseExtension> getResponseExtensions() {
+      return (launchInfoResponseExtension == null) ? ImmutableList.<ResponseExtension>of() :
+          ImmutableList.<ResponseExtension>of(launchInfoResponseExtension);
+    }
+
+    /** A builder for {@link OneTime} since it is immutable. */
+    @VisibleForTesting
+    public static class Builder extends PollMessage.Builder<OneTime, Builder> {
+
+      public Builder() {}
+
+      private Builder(OneTime instance) {
+        super(instance);
+      }
+
+      public Builder setResponseData(ImmutableList<? extends ResponseData> responseData) {
+        FluentIterable<? extends ResponseData> iterable = FluentIterable.from(responseData);
+        getInstance().contactPendingActionNotificationResponses = forceEmptyToNull(
+            iterable.filter(ContactPendingActionNotificationResponse.class).toList());
+        getInstance().contactTransferResponses = forceEmptyToNull(
+            iterable.filter(ContactTransferResponse.class).toList());
+        getInstance().domainPendingActionNotificationResponses = forceEmptyToNull(
+            iterable.filter(DomainPendingActionNotificationResponse.class).toList());
+        getInstance().domainTransferResponses = forceEmptyToNull(
+            iterable.filter(DomainTransferResponse.class).toList());
+        return this;
+      }
+
+      public Builder setResponseExtensions(
+          ImmutableList<? extends ResponseExtension> responseExtensions) {
+        getInstance().launchInfoResponseExtension = FluentIterable
+            .from(responseExtensions)
+            .filter(LaunchInfoResponseExtension.class)
+            .first()
+            .orNull();
+        return this;
+      }
+    }
+  }
+
+  /** An autorenew poll message which recurs annually. */
+  @EntitySubclass(index = false)
+  public static class Autorenew extends PollMessage {
+
+    /** The target id of the autorenew event. */
+    String targetId;
+
+    /** The autorenew recurs annually between {@link #eventTime} and this time. */
+    @Index
+    DateTime autorenewEndTime;
+
+    public String getTargetId() {
+      return targetId;
+    }
+
+    public DateTime getAutorenewEndTime() {
+      return autorenewEndTime;
+    }
+
+    @Override
+    public ImmutableList<ResponseData> getResponseData() {
+      // Note that the event time is when the auto-renew occured, so the expiration time in the
+      // response should be 1 year past that, since it denotes the new expiration time.
+      return ImmutableList.<ResponseData>of(
+          DomainRenewData.create(getTargetId(), getEventTime().plusYears(1)));
+    }
+
+    @Override
+    public ImmutableList<ResponseExtension> getResponseExtensions() {
+      return ImmutableList.of();
+    }
+
+    @Override
+    public Builder asBuilder() {
+      return new Builder(clone(this));
+    }
+
+    /** A builder for {@link Autorenew} since it is immutable. */
+    @VisibleForTesting
+    public static class Builder extends PollMessage.Builder<Autorenew, Builder> {
+
+      public Builder() {}
+
+      private Builder(Autorenew instance) {
+        super(instance);
+      }
+
+      public Builder setTargetId(String targetId) {
+        getInstance().targetId = targetId;
+        return this;
+      }
+
+      public Builder setAutorenewEndTime(DateTime autorenewEndTime) {
+        getInstance().autorenewEndTime = autorenewEndTime;
+        return this;
+      }
+
+      @Override
+      public Autorenew build() {
+        Autorenew instance = getInstance();
+        instance.autorenewEndTime =
+            Optional.fromNullable(instance.autorenewEndTime).or(END_OF_TIME);
+        return super.build();
+      }
+    }
+  }
+}