Import code from internal repository to git

2025-05-20 19:29:35 +02:00 · 2016-03-01 17:18:14 -05:00 · 2016-03-01 17:18:14 -05:00 · 0ef0c933d2
commit 0ef0c933d2
2490 changed files with 281594 additions and 0 deletions
--- a/java/com/google/domain/registry/tools/AuctionStatusCommand.java
+++ b/java/com/google/domain/registry/tools/AuctionStatusCommand.java
@ -0,0 +1,121 @@
+// 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;
+
+import static com.google.common.base.Preconditions.checkNotNull;
+import static com.google.common.base.Preconditions.checkState;
+import static com.google.domain.registry.model.index.DomainApplicationIndex.loadActiveApplicationsByDomainName;
+import static com.google.domain.registry.model.ofy.ObjectifyService.ofy;
+import static com.google.domain.registry.model.registry.Registries.findTldForName;
+import static java.nio.charset.StandardCharsets.UTF_8;
+
+import com.google.common.base.Function;
+import com.google.common.base.Strings;
+import com.google.common.collect.ComparisonChain;
+import com.google.common.collect.FluentIterable;
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableSet;
+import com.google.common.collect.Lists;
+import com.google.common.collect.Ordering;
+import com.google.common.net.InternetDomainName;
+import com.google.domain.registry.model.contact.ContactResource;
+import com.google.domain.registry.model.domain.DomainApplication;
+import com.google.domain.registry.tools.Command.GtechCommand;
+import com.google.domain.registry.tools.Command.RemoteApiCommand;
+import com.google.domain.registry.tools.params.PathParameter;
+
+import com.beust.jcommander.Parameter;
+import com.beust.jcommander.Parameters;
+import com.googlecode.objectify.Work;
+
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.List;
+
+/** Command to check the status of domain applications. */
+@Parameters(separators = " =", commandDescription = "Check auction status")
+final class AuctionStatusCommand implements RemoteApiCommand, GtechCommand {
+
+  @Parameter(
+      description = "Domains(s) to check",
+      required = true)
+  private List<String> mainArguments;
+
+  @Parameter(
+      names = {"-o", "--output"},
+      description = "Output file.",
+      validateWith = PathParameter.OutputFile.class)
+  private Path output = Paths.get("/dev/stdout");
+
+  @Override
+  public void run() throws Exception {
+    final ImmutableSet<String> domains = ImmutableSet.copyOf(mainArguments);
+    Files.write(output, FluentIterable
+        .from(domains)
+        .transformAndConcat(new Function<String, Iterable<String>>() {
+            @Override
+            public Iterable<String> apply(String fullyQualifiedDomainName) {
+              checkState(
+                  findTldForName(InternetDomainName.from(fullyQualifiedDomainName)).isPresent(),
+                  "No tld found for %s", fullyQualifiedDomainName);
+              return ofy().transactNewReadOnly(new Work<Iterable<String>>() {
+                @Override
+                public Iterable<String> run() {
+                  ImmutableList.Builder<DomainApplication> applications =
+                      new ImmutableList.Builder<>();
+                  for (String domain : domains) {
+                    applications.addAll(
+                        loadActiveApplicationsByDomainName(domain, ofy().getTransactionTime()));
+                  }
+                  return Lists.transform(
+                      FluentIterable.from(applications.build()).toSortedList(ORDERING),
+                      APPLICATION_FORMATTER);
+                }});
+            }}), UTF_8);
+  }
+
+  private static final Ordering<DomainApplication> ORDERING = new Ordering<DomainApplication>() {
+        @Override
+        public int compare(DomainApplication left, DomainApplication right) {
+          return ComparisonChain.start()
+              .compare(left.getFullyQualifiedDomainName(), right.getFullyQualifiedDomainName())
+              .compareTrueFirst(
+                  left.getEncodedSignedMarks().isEmpty(), right.getEncodedSignedMarks().isEmpty())
+              .compare(left.getApplicationStatus(), right.getApplicationStatus())
+              .compare(left.getCreationTime(), right.getCreationTime())
+              .result();
+        }};
+
+  private static final Function<DomainApplication, String> APPLICATION_FORMATTER =
+      new Function<DomainApplication, String>() {
+        @Override
+        public String apply(DomainApplication app) {
+          ContactResource registrant = checkNotNull(app.loadRegistrant());
+          Object[] keysAndValues = new Object[] {
+              "Domain", app.getFullyQualifiedDomainName(),
+              "Type", app.getEncodedSignedMarks().isEmpty() ? "Landrush" : "Sunrise",
+              "Application Status", app.getApplicationStatus(),
+              "Application ID", app.getForeignKey(),
+              "Application Timestamp", app.getCreationTime(),
+              "Last Update", app.getLastEppUpdateTime(),
+              "Registrar Name", app.getCurrentSponsorClientId(),
+              "Registrant Email", registrant.getEmailAddress(),
+              "Registrant Phone", registrant.getVoiceNumber().getPhoneNumber()
+          };
+          return String.format(
+              Strings.repeat("%-25s= %s\n", keysAndValues.length / 2), keysAndValues);
+        }};
+}