Add an action to check the status of OT&E verification

-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=228333195

java/google/registry/model/OteAccountBuilder.java

@@ -361,7 +361,7 @@ public final class OteAccountBuilder {
   }
 
   /** Returns the ClientIds of the OT&E, with the TLDs each has access to. */
-  static ImmutableMap<String, String> createClientIdToTldMap(String baseClientId) {
+  public static ImmutableMap<String, String> createClientIdToTldMap(String baseClientId) {
     checkArgument(
         REGISTRAR_PATTERN.matcher(baseClientId).matches(),
         "Invalid registrar name: %s",
@@ -374,4 +374,17 @@ public final class OteAccountBuilder {
         .put(baseClientId + "-5", baseClientId + "-eap")
         .build();
   }
+
+  /** Returns the base client ID that correspond to a given OT&amp;E client ID. */
+  public static String getBaseClientId(String oteClientId) {
+    int index = oteClientId.lastIndexOf('-');
+    checkArgument(index > 0, "Invalid OT&E client ID: %s", oteClientId);
+    String baseClientId = oteClientId.substring(0, index);
+    checkArgument(
+        createClientIdToTldMap(baseClientId).containsKey(oteClientId),
+        "ID %s is not one of the OT&E client IDs for base %s",
+        oteClientId,
+        baseClientId);
+    return baseClientId;
+  }
 }

java/google/registry/model/OteStats.java

@@ -177,7 +177,7 @@ public class OteStats {
     }
 
     /** Returns a more human-readable translation of the enum constant. */
-    private String description() {
+    public String getDescription() {
       return Ascii.toLowerCase(this.name().replace('_', ' '));
     }
 
@@ -270,7 +270,7 @@ public class OteStats {
     return String.format(
         "%s\nTOTAL: %d",
         EnumSet.allOf(StatType.class).stream()
-            .map(stat -> String.format("%s: %d", stat.description(), statCounts.count(stat)))
+            .map(stat -> String.format("%s: %d", stat.getDescription(), statCounts.count(stat)))
             .collect(Collectors.joining("\n")),
         statCounts.size());
   }

java/google/registry/module/frontend/FrontendRequestComponent.java

@@ -27,6 +27,7 @@ import google.registry.request.RequestModule;
 import google.registry.request.RequestScope;
 import google.registry.ui.server.otesetup.ConsoleOteSetupAction;
 import google.registry.ui.server.registrar.ConsoleUiAction;
+import google.registry.ui.server.registrar.OteStatusAction;
 import google.registry.ui.server.registrar.RegistrarConsoleModule;
 import google.registry.ui.server.registrar.RegistrarSettingsAction;
 
@@ -46,6 +47,7 @@ interface FrontendRequestComponent {
   EppConsoleAction eppConsoleAction();
   EppTlsAction eppTlsAction();
   FlowComponent.Builder flowComponentBuilder();
+  OteStatusAction oteStatusAction();
   RegistrarSettingsAction registrarSettingsAction();
 
   @Subcomponent.Builder

java/google/registry/ui/server/registrar/OteStatusAction.java

@@ -0,0 +1,121 @@
+// Copyright 2019 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.ui.server.registrar;
+
+import static com.google.common.base.Preconditions.checkArgument;
+import static com.google.common.collect.ImmutableList.toImmutableList;
+import static google.registry.security.JsonResponseHelper.Status.ERROR;
+import static google.registry.security.JsonResponseHelper.Status.SUCCESS;
+
+import com.google.common.base.Strings;
+import com.google.common.collect.ImmutableMap;
+import com.google.common.flogger.FluentLogger;
+import google.registry.model.OteAccountBuilder;
+import google.registry.model.OteStats;
+import google.registry.model.OteStats.StatType;
+import google.registry.model.registrar.Registrar;
+import google.registry.model.registrar.Registrar.Type;
+import google.registry.request.Action;
+import google.registry.request.JsonActionRunner;
+import google.registry.request.auth.Auth;
+import google.registry.request.auth.AuthenticatedRegistrarAccessor;
+import google.registry.request.auth.AuthenticatedRegistrarAccessor.RegistrarAccessDeniedException;
+import google.registry.security.JsonResponseHelper;
+import java.util.Map;
+import java.util.Optional;
+import javax.inject.Inject;
+
+/**
+ * Admin servlet that allows creating or updating a registrar. Deletes are not allowed so as to
+ * preserve history.
+ */
+@Action(path = OteStatusAction.PATH, method = Action.Method.POST, auth = Auth.AUTH_PUBLIC_LOGGED_IN)
+public final class OteStatusAction implements Runnable, JsonActionRunner.JsonAction {
+
+  public static final String PATH = "/registrar-ote-status";
+
+  private static final FluentLogger logger = FluentLogger.forEnclosingClass();
+
+  private static final String CLIENT_ID_PARAM = "clientId";
+  private static final String COMPLETED_PARAM = "completed";
+  private static final String DETAILS_PARAM = "details";
+  private static final String STAT_TYPE_DESCRIPTION_PARAM = "description";
+  private static final String STAT_TYPE_REQUIREMENT_PARAM = "requirement";
+  private static final String STAT_TYPE_TIMES_PERFORMED_PARAM = "timesPerformed";
+
+  @Inject AuthenticatedRegistrarAccessor registrarAccessor;
+  @Inject JsonActionRunner jsonActionRunner;
+
+  @Inject
+  OteStatusAction() {}
+
+  @Override
+  public void run() {
+    jsonActionRunner.run(this);
+  }
+
+  @Override
+  public Map<String, ?> handleJsonRequest(Map<String, ?> input) {
+    try {
+      checkArgument(input != null, "Malformed JSON");
+
+      String oteClientId = (String) input.get(CLIENT_ID_PARAM);
+      checkArgument(
+          !Strings.isNullOrEmpty(oteClientId), "Missing key for OT&E client: %s", CLIENT_ID_PARAM);
+
+      String baseClientId = OteAccountBuilder.getBaseClientId(oteClientId);
+      Registrar oteRegistrar = registrarAccessor.getRegistrar(oteClientId);
+      verifyOteAccess(baseClientId);
+      checkArgument(
+          Type.OTE.equals(oteRegistrar.getType()),
+          "Registrar with ID %s is not an OT&E registrar",
+          oteClientId);
+
+      OteStats oteStats = OteStats.getFromRegistrar(baseClientId);
+      return JsonResponseHelper.create(
+          SUCCESS, "OT&E check completed successfully", convertOteStats(baseClientId, oteStats));
+    } catch (Throwable e) {
+      logger.atWarning().withCause(e).log(
+          "Failed to verify OT&E status for registrar with input %s", input);
+      return JsonResponseHelper.create(
+          ERROR, Optional.ofNullable(e.getMessage()).orElse("Unspecified error"));
+    }
+  }
+
+  private void verifyOteAccess(String baseClientId) throws RegistrarAccessDeniedException {
+    for (String oteClientId : OteAccountBuilder.createClientIdToTldMap(baseClientId).keySet()) {
+      registrarAccessor.verifyAccess(oteClientId);
+    }
+  }
+
+  private Map<String, Object> convertOteStats(String baseClientId, OteStats oteStats) {
+    return ImmutableMap.of(
+        CLIENT_ID_PARAM, baseClientId,
+        COMPLETED_PARAM, oteStats.getFailures().isEmpty(),
+        DETAILS_PARAM,
+            StatType.REQUIRED_STAT_TYPES.stream()
+                .map(statType -> convertSingleRequirement(statType, oteStats.getCount(statType)))
+                .collect(toImmutableList()));
+  }
+
+  private Map<String, Object> convertSingleRequirement(StatType statType, int count) {
+    int requirement = statType.getRequirement();
+    return ImmutableMap.of(
+        STAT_TYPE_DESCRIPTION_PARAM, statType.getDescription(),
+        STAT_TYPE_REQUIREMENT_PARAM, requirement,
+        STAT_TYPE_TIMES_PERFORMED_PARAM, count,
+        COMPLETED_PARAM, count >= requirement);
+  }
+}