mirror of
https://github.com/google/nomulus.git
synced 2025-05-13 16:07:15 +02:00
Add framework for customizable WHOIS commands
With some additional changes by Ben McIlwain. ------------- Created by MOE: https://github.com/google/moe MOE_MIGRATED_REVID=145447136
This commit is contained in:
parent
d3fe6be385
commit
bb3a0c78c5
15 changed files with 155 additions and 47 deletions
|
@ -15,7 +15,6 @@
|
|||
package google.registry.whois;
|
||||
|
||||
import static com.google.common.base.Preconditions.checkNotNull;
|
||||
import static google.registry.model.EppResourceUtils.loadByForeignKey;
|
||||
import static google.registry.model.registry.Registries.findTldForName;
|
||||
import static google.registry.model.registry.Registries.getTlds;
|
||||
import static javax.servlet.http.HttpServletResponse.SC_NOT_FOUND;
|
||||
|
@ -23,16 +22,13 @@ import static javax.servlet.http.HttpServletResponse.SC_NOT_FOUND;
|
|||
import com.google.common.annotations.VisibleForTesting;
|
||||
import com.google.common.base.Optional;
|
||||
import com.google.common.net.InternetDomainName;
|
||||
import google.registry.model.EppResource;
|
||||
import google.registry.util.TypeUtils.TypeInstantiator;
|
||||
import javax.annotation.Nullable;
|
||||
import org.joda.time.DateTime;
|
||||
|
||||
/** Represents a WHOIS lookup on a domain name (i.e. SLD) or a nameserver. */
|
||||
abstract class DomainOrHostLookupCommand<T extends EppResource> implements WhoisCommand {
|
||||
public abstract class DomainOrHostLookupCommand implements WhoisCommand {
|
||||
|
||||
@VisibleForTesting
|
||||
final InternetDomainName domainOrHostName;
|
||||
@VisibleForTesting final InternetDomainName domainOrHostName;
|
||||
|
||||
private final String errorPrefix;
|
||||
|
||||
|
@ -52,17 +48,15 @@ abstract class DomainOrHostLookupCommand<T extends EppResource> implements Whois
|
|||
}
|
||||
// Google Policy: Do not return records under TLDs for which we're not authoritative.
|
||||
if (tld.isPresent() && getTlds().contains(tld.get().toString())) {
|
||||
T domainOrHost = loadByForeignKey(
|
||||
new TypeInstantiator<T>(getClass()){}.getExactType(),
|
||||
domainOrHostName.toString(),
|
||||
now);
|
||||
if (domainOrHost != null) {
|
||||
return getSuccessResponse(domainOrHost, now);
|
||||
final Optional<WhoisResponse> response = getResponse(domainOrHostName, now);
|
||||
if (response.isPresent()) {
|
||||
return response.get();
|
||||
}
|
||||
}
|
||||
throw new WhoisException(now, SC_NOT_FOUND, errorPrefix + " not found.");
|
||||
}
|
||||
|
||||
/** Renders a response record, provided its successfully retrieved datastore entity. */
|
||||
abstract WhoisResponse getSuccessResponse(T domainOrHost, DateTime now);
|
||||
protected abstract Optional<WhoisResponse> getResponse(
|
||||
InternetDomainName domainName, DateTime now);
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue