google.registry.tools.server.CreateGroupsAction.java Source code

Introduction

Here is the source code for google.registry.tools.server.CreateGroupsAction.java

Source

// Copyright 2016 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.tools.server;

import static google.registry.export.SyncGroupMembersAction.getGroupEmailAddressForContactType;
import static google.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 google.registry.config.RegistryConfig.Config;
import google.registry.groups.GroupsConnection;
import google.registry.groups.GroupsConnection.Role;
import google.registry.model.registrar.Registrar;
import google.registry.model.registrar.RegistrarContact;
import google.registry.model.registrar.RegistrarContact.Type;
import google.registry.request.Action;
import google.registry.request.HttpException.BadRequestException;
import google.registry.request.HttpException.InternalServerErrorException;
import google.registry.request.Parameter;
import google.registry.request.Response;
import google.registry.util.Concurrent;
import google.registry.util.FormattingLogger;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.List;
import javax.annotation.Nullable;
import javax.inject.Inject;

/** Action that creates Google Groups for a registrar's mailing lists. */
@Action(path = CreateGroupsAction.PATH, method = POST)
public class CreateGroupsAction 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();
    private static final int NUM_SIMULTANEOUS_CONNECTIONS = 5;

    @Inject
    GroupsConnection groupsConnection;
    @Inject
    Response response;
    @Inject
    @Config("publicDomainName")
    String publicDomainName;
    @Inject
    @Parameter("clientId")
    Optional<String> clientId;

    @Inject
    CreateGroupsAction() {
    }

    @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, NUM_SIMULTANEOUS_CONNECTIONS,
                new Function<RegistrarContact.Type, Optional<Exception>>() {
                    @Override
                    public Optional<Exception> apply(Type type) {
                        try {
                            String groupKey = getGroupEmailAddressForContactType(registrar.getClientId(), 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);
    }
}