Add command to re-save all environment entities

This is the tool we'll run after dumping everything in sandbox except
for Registrars, Registries, and RegistrarContacts, in order to
re-create the commit logs on said entities.
-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=119369873

java/com/google/domain/registry/model/ofy/CommitLogMutation.java

@@ -16,6 +16,7 @@ package com.google.domain.registry.model.ofy;
 
 import static com.google.appengine.api.datastore.EntityTranslator.convertToPb;
 import static com.google.appengine.api.datastore.EntityTranslator.createFromPbBytes;
+import static com.google.common.base.Preconditions.checkNotNull;
 import static com.google.domain.registry.model.ofy.ObjectifyService.ofy;
 
 import com.google.appengine.api.datastore.KeyFactory;
@@ -79,7 +80,7 @@ public class CommitLogMutation extends ImmutableObject {
       Key<CommitLogManifest> parent,
       com.google.appengine.api.datastore.Entity rawEntity) {
     CommitLogMutation instance = new CommitLogMutation();
-    instance.parent = parent;
+    instance.parent = checkNotNull(parent);
     // Creates a web-safe key string.
     instance.entityKey = KeyFactory.keyToString(rawEntity.getKey());
     instance.entityProtoBytes = convertToPb(rawEntity).toByteArray();

java/com/google/domain/registry/tools/RegistryTool.java

@@ -62,6 +62,7 @@ public final class RegistryTool {
           .put("load_snapshot", LoadSnapshotCommand.class)
           .put("make_billing_tables", MakeBillingTablesCommand.class)
           .put("pending_escrow", PendingEscrowCommand.class)
+          .put("resave_environment_entities", ResaveEnvironmentEntitiesCommand.class)
           .put("send_escrow_report_to_icann", SendEscrowReportToIcannCommand.class)
           .put("update_application_status", UpdateApplicationStatusCommand.class)
           .put("update_claims_notice", UpdateClaimsNoticeCommand.class)

java/com/google/domain/registry/tools/ResaveEnvironmentEntitiesCommand.java

@@ -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 com.google.domain.registry.tools;
+
+import static com.google.common.collect.Iterables.concat;
+import static com.google.domain.registry.model.common.EntityGroupRoot.getCrossTldKey;
+import static com.google.domain.registry.model.ofy.ObjectifyService.ofy;
+
+import com.google.domain.registry.model.ImmutableObject;
+import com.google.domain.registry.model.registrar.Registrar;
+import com.google.domain.registry.model.registrar.RegistrarContact;
+import com.google.domain.registry.model.registry.Registry;
+import com.google.domain.registry.tools.Command.RemoteApiCommand;
+
+import com.beust.jcommander.Parameters;
+import com.googlecode.objectify.Key;
+import com.googlecode.objectify.VoidWork;
+
+/**
+ * Command to re-save all environment entities to ensure that they have valid commit logs.
+ *
+ * <p>The entities that are re-saved are those of type {@link Registry}, {@link Registrar}, and
+ * {@link RegistrarContact}.
+ */
+@Parameters(commandDescription = "Re-save all environment entities.")
+final class ResaveEnvironmentEntitiesCommand implements RemoteApiCommand {
+
+  @Override
+  public void run() throws Exception {
+    Iterable<Key<? extends ImmutableObject>> keys = concat(
+        ofy().load().type(Registrar.class).ancestor(getCrossTldKey()).keys(),
+        ofy().load().type(Registry.class).ancestor(getCrossTldKey()).keys(),
+        ofy().load().type(RegistrarContact.class).ancestor(getCrossTldKey()).keys());
+    for (final Key<? extends ImmutableObject> key : keys) {
+      ofy().transact(new VoidWork() {
+          @Override
+          public void vrun() {
+            ofy().save().entity(ofy().load().key(key).now());
+          }});
+      System.out.printf("Re-saved entity %s\n", key);
+    }
+  }
+}