// 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.writer.clouddns;

import com.google.api.client.http.HttpRequestInitializer;
import com.google.api.client.http.HttpTransport;
import com.google.api.client.json.JsonFactory;
import com.google.api.services.dns.Dns;
import com.google.api.services.dns.DnsScopes;
import com.google.common.util.concurrent.RateLimiter;
import dagger.Module;
import dagger.Provides;
import dagger.multibindings.IntoMap;
import dagger.multibindings.IntoSet;
import dagger.multibindings.StringKey;
import google.registry.config.RegistryConfig.Config;
import google.registry.dns.writer.DnsWriter;
import java.util.Optional;
import java.util.Set;
import java.util.function.Function;
import javax.inject.Named;

/** Dagger module for Google Cloud DNS service connection objects. */
@Module
public final class CloudDnsWriterModule {

  @Provides
  static Dns provideDns(
      HttpTransport transport,
      JsonFactory jsonFactory,
      Function<Set<String>, ? extends HttpRequestInitializer> credential,
      @Config("projectId") String projectId,
      @Config("cloudDnsRootUrl") Optional<String> rootUrl,
      @Config("cloudDnsServicePath") Optional<String> servicePath) {
    Dns.Builder builder =
        new Dns.Builder(transport, jsonFactory, credential.apply(DnsScopes.all()))
            .setApplicationName(projectId);

    rootUrl.ifPresent(builder::setRootUrl);
    servicePath.ifPresent(builder::setServicePath);

    return builder.build();
  }

  @Provides
  @IntoMap
  @StringKey(CloudDnsWriter.NAME)
  static DnsWriter provideWriter(CloudDnsWriter writer) {
    return writer;
  }

  @Provides
  @IntoSet
  @Named("dnsWriterNames")
  static String provideWriterName() {
    return CloudDnsWriter.NAME;
  }

  @Provides
  @Named("cloudDns")
  static RateLimiter provideRateLimiter() {
    // This is the default max QPS for Cloud DNS. It can be increased by contacting the team
    // via the Quotas page on the Cloud Console.
    int cloudDnsMaxQps = 50;
    return RateLimiter.create(cloudDnsMaxQps);
  }

  @Provides
  @Named("cloudDnsNumThreads")
  static int provideNumThreads() {
    // TODO(b/70217860): find the "best" number of threads, taking into account running time, App
    // Engine constraints, and any Cloud DNS comsiderations etc.
    //
    // NOTE: any number below 2 will not use threading at all.
    return 10;
  }
}

@@ -948,24 +948,45 @@
 ateLimiter;%0A
+import dagger.Binds;%0A
 import dagge
@@ -995,16 +995,16 @@
 Module;%0A
-
 import d
@@ -1432,13 +1432,16 @@
 lic 
-final
+abstract
  cla
@@ -2092,39 +2092,36 @@
 uild();%0A  %7D%0A%0A  @
-Provide
+Bind
 s%0A  @IntoMap%0A  @
@@ -2153,22 +2153,24 @@
 NAME)%0A  
-static
+abstract
  DnsWrit
@@ -2212,33 +2212,9 @@
 ter)
- %7B%0A    return writer;%0A  %7D
+;
 %0A%0A  
@@ -2962,14 +2962,51 @@
 urn 10;%0A
-
   %7D
+%0A%0A  private CloudDnsWriterModule() %7B%7D
 %0A%7D%0A

Use @Binds instead of @Provides for DnsWriter