Import code from internal repository to git

java/com/google/domain/registry/tools/server/CreateGroupsTask.java

@@ -0,0 +1,130 @@
+// Copyright 2016 Google Inc. 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.server;
+
+import static com.google.domain.registry.export.SyncGroupMembersTask.getGroupEmailAddressForContactType;
+import static com.google.domain.registry.request.Action.Method.POST;
+import static java.util.Arrays.asList;
+import static javax.servlet.http.HttpServletResponse.SC_OK;
+
+import com.google.common.base.Function;
+import com.google.common.base.Optional;
+import com.google.domain.registry.config.ConfigModule.Config;
+import com.google.domain.registry.groups.GroupsConnection;
+import com.google.domain.registry.groups.GroupsConnection.Role;
+import com.google.domain.registry.model.registrar.Registrar;
+import com.google.domain.registry.model.registrar.RegistrarContact;
+import com.google.domain.registry.model.registrar.RegistrarContact.Type;
+import com.google.domain.registry.request.Action;
+import com.google.domain.registry.request.HttpException.BadRequestException;
+import com.google.domain.registry.request.HttpException.InternalServerErrorException;
+import com.google.domain.registry.request.Parameter;
+import com.google.domain.registry.request.Response;
+import com.google.domain.registry.util.Concurrent;
+import com.google.domain.registry.util.FormattingLogger;
+
+import java.io.PrintWriter;
+import java.io.StringWriter;
+import java.util.List;
+
+import javax.annotation.Nullable;
+import javax.inject.Inject;
+
+/** A task that creates Google Groups for a registrar's mailing lists. */
+@Action(path = CreateGroupsTask.PATH, method = POST)
+public class CreateGroupsTask implements Runnable {
+
+  public static final String PATH = "/_dr/admin/createGroups";
+  public static final String CLIENT_ID_PARAM = "clientId";
+
+  private static final FormattingLogger logger = FormattingLogger.getLoggerForCallerClass();
+
+  @Inject GroupsConnection groupsConnection;
+  @Inject Response response;
+  @Inject @Config("publicDomainName") String publicDomainName;
+  @Inject @Parameter("clientId") Optional<String> clientId;
+  @Inject CreateGroupsTask() {}
+
+  @Override
+  public void run() {
+    final Registrar registrar = initAndLoadRegistrar();
+    if (registrar == null) {
+      return;
+    }
+    List<RegistrarContact.Type> types = asList(RegistrarContact.Type.values());
+    // Concurrently create the groups for each RegistrarContact.Type, collecting the results from
+    // each call (which are either an Exception if it failed, or absent() if it succeeded).
+    List<Optional<Exception>> results = Concurrent.transform(types,
+        new Function<RegistrarContact.Type, Optional<Exception>>() {
+          @Override
+          public Optional<Exception> apply(Type type) {
+            try {
+              String groupKey = getGroupEmailAddressForContactType(
+                  registrar.getClientIdentifier(), type, publicDomainName);
+              String parentGroup =
+                  getGroupEmailAddressForContactType("registrar", type, publicDomainName);
+              // Creates the group, then adds it as a member to the global registrar group for
+              // that type.
+              groupsConnection.createGroup(groupKey);
+              groupsConnection.addMemberToGroup(parentGroup, groupKey, Role.MEMBER);
+              return Optional.<Exception> absent();
+            } catch (Exception e) {
+              return Optional.of(e);
+            }
+          }});
+    // Return the correct server response based on the results of the group creations.
+    if (Optional.presentInstances(results).iterator().hasNext()) {
+      StringWriter responseString = new StringWriter();
+      PrintWriter responseWriter = new PrintWriter(responseString);
+      for (int i = 0; i < results.size(); i++) {
+        Optional<Exception> e = results.get(i);
+        if (e.isPresent()) {
+          responseWriter.append(types.get(i).getDisplayName()).append(" => ");
+          e.get().printStackTrace(responseWriter);
+          logger.severefmt(
+              e.get(),
+              "Could not create Google Group for registrar %s for type %s",
+              registrar.getRegistrarName(),
+              types.get(i).toString());
+        } else {
+          responseWriter.printf("%s => Success%n", types.get(i).getDisplayName());
+        }
+      }
+      throw new InternalServerErrorException(responseString.toString());
+    } else {
+      response.setStatus(SC_OK);
+      response.setPayload("Success!");
+      logger.info("Successfully created groups for registrar: " + registrar.getRegistrarName());
+    }
+  }
+
+  @Nullable
+  private Registrar initAndLoadRegistrar() {
+    if (!clientId.isPresent()) {
+      respondToBadRequest("Error creating Google Groups, missing parameter: clientId");
+    }
+    final Registrar registrar = Registrar.loadByClientId(clientId.get());
+    if (registrar == null) {
+      respondToBadRequest(String.format(
+          "Error creating Google Groups; could not find registrar with id %s", clientId.get()));
+    }
+    return registrar;
+  }
+
+  private void respondToBadRequest(String message) {
+    logger.severe(message);
+    throw new BadRequestException(message);
+  }
+}