// Copyright 2017 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.dns;

import static google.registry.dns.RefreshDnsOnHostRenameAction.PARAM_HOST_KEY;
import static google.registry.request.RequestParameters.extractEnumParameter;
import static google.registry.request.RequestParameters.extractIntParameter;
import static google.registry.request.RequestParameters.extractOptionalIntParameter;
import static google.registry.request.RequestParameters.extractOptionalParameter;
import static google.registry.request.RequestParameters.extractRequiredParameter;
import static google.registry.request.RequestParameters.extractSetOfParameters;

import com.google.common.hash.HashFunction;
import com.google.common.hash.Hashing;
import dagger.Binds;
import dagger.Module;
import dagger.Provides;
import google.registry.dns.DnsUtils.TargetType;
import google.registry.dns.writer.DnsWriterZone;
import google.registry.request.Parameter;
import google.registry.request.RequestParameters;
import jakarta.servlet.http.HttpServletRequest;
import java.time.Instant;
import java.util.Optional;
import java.util.Set;

/** Dagger module for the dns package. */
@Module
public abstract class DnsModule {

  public static final String PARAM_DNS_WRITER = "dnsWriter";
  public static final String PARAM_LOCK_INDEX = "lockIndex";
  public static final String PARAM_NUM_PUBLISH_LOCKS = "numPublishLocks";
  public static final String PARAM_DOMAINS = "domains";
  public static final String PARAM_HOSTS = "hosts";
  public static final String PARAM_PUBLISH_TASK_ENQUEUED = "enqueued";
  public static final String PARAM_REFRESH_REQUEST_TIME = "requestTime";
  // This parameter cannot be named "jitterSeconds", which will be read by TldFanoutAction and not
  // be passed down to actions.
  public static final String PARAM_DNS_JITTER_SECONDS = "dnsJitterSeconds";

  @Binds
  @DnsWriterZone
  abstract String provideZoneName(@Parameter(RequestParameters.PARAM_TLD) String tld);

  /**
   * Provides a HashFunction used for generating sharded DNS publish queue tasks.
   *
   * <p>We use murmur3_32 because it isn't subject to change, and is fast (non-cryptographic, which
   * would be overkill in this situation.)
   */
  @Provides
  static HashFunction provideHashFunction() {
    return Hashing.murmur3_32_fixed();
  }

  @Provides
  @Parameter(PARAM_PUBLISH_TASK_ENQUEUED)
  static Instant provideCreateTime(HttpServletRequest req) {
    return Instant.parse(extractRequiredParameter(req, PARAM_PUBLISH_TASK_ENQUEUED));
  }

  // TODO: Retire the old header after DNS pull queue migration.
  @Provides
  @Parameter(PARAM_REFRESH_REQUEST_TIME)
  static Instant provideItemsCreateTime(HttpServletRequest req) {
    return Instant.parse(
        extractOptionalParameter(req, "itemsCreated")
            .orElse(extractRequiredParameter(req, PARAM_REFRESH_REQUEST_TIME)));
  }

  @Provides
  @Parameter(PARAM_DNS_WRITER)
  static String provideDnsWriter(HttpServletRequest req) {
    return extractRequiredParameter(req, PARAM_DNS_WRITER);
  }

  @Provides
  @Parameter(PARAM_LOCK_INDEX)
  static int provideLockIndex(HttpServletRequest req) {
    return extractIntParameter(req, PARAM_LOCK_INDEX);
  }

  @Provides
  @Parameter(PARAM_NUM_PUBLISH_LOCKS)
  static int provideMaxNumLocks(HttpServletRequest req) {
    return extractIntParameter(req, PARAM_NUM_PUBLISH_LOCKS);
  }

  @Provides
  @Parameter(PARAM_DOMAINS)
  static Set<String> provideDomains(HttpServletRequest req) {
    return extractSetOfParameters(req, PARAM_DOMAINS);
  }

  @Provides
  @Parameter(PARAM_HOSTS)
  static Set<String> provideHosts(HttpServletRequest req) {
    return extractSetOfParameters(req, PARAM_HOSTS);
  }

  @Provides
  @Parameter(PARAM_HOST_KEY)
  static String provideResourceKey(HttpServletRequest req) {
    return extractRequiredParameter(req, PARAM_HOST_KEY);
  }

  @Provides
  @Parameter(PARAM_DNS_JITTER_SECONDS)
  static Optional<Integer> provideJitterSeconds(HttpServletRequest req) {
    return extractOptionalIntParameter(req, PARAM_DNS_JITTER_SECONDS);
  }

  @Provides
  @Parameter("domainOrHostName")
  static String provideName(HttpServletRequest req) {
    return extractRequiredParameter(req, "name");
  }

  @Provides
  @Parameter("type")
  static TargetType provideType(HttpServletRequest req) {
    return extractEnumParameter(req, TargetType.class, "type");
  }
}
