wip: writerConsumers for plugins

Author: kuhe

smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/CommandGenerator.java

@@ -467,6 +467,7 @@ private void addCommandSpecificPlugins() {
         // applied automatically when the Command's middleware stack is copied from
         // the service's middleware stack.
         for (RuntimeClientPlugin plugin : runtimePlugins) {
+            plugin.getWriterConsumer().accept(writer);
             plugin.getPluginFunction().ifPresent(pluginSymbol -> {
                 // Construct additional parameters string
                 Map<String, Object> paramsMap = plugin.getAdditionalPluginFunctionParameters(

smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/ServiceBareBonesClientGenerator.java

@@ -387,6 +387,10 @@ private void generateConstructor() {
                     generateConfigVariable(configVariable - 1));
             }
 
+            for (RuntimeClientPlugin plugin : runtimePlugins) {
+                plugin.getWriterConsumer().accept(writer);
+            }
+
             // Add runtime plugin "resolve" method calls. These are invoked one
             // after the other until all of the runtime plugins have been called.
             // Only plugins that have configuration are called. Each time the

smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/auth/http/integration/AddHttpAuthSchemePlugin.java

@@ -67,6 +67,12 @@ public List<RuntimeClientPlugin> getClientPlugins() {
                         .name("this.getIdentityProviderConfigProvider()")
                         .build()
                 ))
+                .withWriter(writer -> {
+                    String httpAuthSchemeParametersProviderName = "default"
+                        + CodegenUtils.getServiceName(s.getSettings(), s.getModel(), s.getSymbolProvider())
+                        + "HttpAuthSchemeParametersProvider";
+                    w.addImport(httpAuthSchemeParametersProviderName, null, AuthUtils.AUTH_HTTP_PROVIDER_DEPENDENCY);
+                })
                 .build(),
             RuntimeClientPlugin.builder()
                 .servicePredicate((m, s) -> !s.hasTrait(EndpointRuleSetTrait.ID))
@@ -217,6 +223,14 @@ public void customize(TypeScriptCodegenContext codegenContext) {
         });
     }
 
+    private String inlineHttpAuthSchemeParametersProvider() {
+        return "";
+    }
+
+    private String inlineIdentityProviderConfigProvider() {
+        return "";
+    }
+
     /*
     export interface HttpAuthSchemeInputConfig {
       httpAuthSchemes?: HttpAuthScheme[];

smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/integration/RuntimeClientPlugin.java

@@ -20,7 +20,11 @@
 import java.util.Objects;
 import java.util.Optional;
 import java.util.Set;
+import java.util.function.BiConsumer;
 import java.util.function.BiPredicate;
+import java.util.function.Consumer;
+import java.util.function.UnaryOperator;
+
 import software.amazon.smithy.codegen.core.Symbol;
 import software.amazon.smithy.codegen.core.SymbolDependency;
 import software.amazon.smithy.codegen.core.SymbolReference;
@@ -29,10 +33,12 @@
 import software.amazon.smithy.model.shapes.ServiceShape;
 import software.amazon.smithy.typescript.codegen.TypeScriptDependency;
 import software.amazon.smithy.typescript.codegen.TypeScriptSettings;
+import software.amazon.smithy.typescript.codegen.TypeScriptWriter;
 import software.amazon.smithy.utils.SmithyBuilder;
 import software.amazon.smithy.utils.SmithyUnstableApi;
 import software.amazon.smithy.utils.StringUtils;
 import software.amazon.smithy.utils.ToSmithyBuilder;
+import software.amazon.smithy.utils.TriConsumer;
 
 /**
  * Represents a runtime plugin for a client that hooks into various aspects
@@ -56,6 +62,8 @@ public final class RuntimeClientPlugin implements ToSmithyBuilder<RuntimeClientP
     private final BiPredicate<Model, ServiceShape> servicePredicate;
     private final OperationPredicate operationPredicate;
     private final SettingsPredicate settingsPredicate;
+    private BiConsumer<TypeScriptWriter, BiConsumer<Model, ServiceShape>> clientWriterConsumer = (writer, consumer) -> {};
+    private BiConsumer<TypeScriptWriter, TriConsumer<Model, ServiceShape, OperationShape>> operationWriterConsumer = (writer, consumer) -> {};
 
     private RuntimeClientPlugin(Builder builder) {
         inputConfig = builder.inputConfig;
@@ -68,6 +76,8 @@ private RuntimeClientPlugin(Builder builder) {
         operationPredicate = builder.operationPredicate;
         servicePredicate = builder.servicePredicate;
         settingsPredicate = builder.settingsPredicate;
+        clientWriterConsumer = builder.clientWriterConsumer;
+        operationWriterConsumer = builder.operationWriterConsumer;
 
         boolean allNull = (inputConfig == null) && (resolvedConfig == null) && (resolveFunction == null);
         boolean allSet = (inputConfig != null) && (resolvedConfig != null) && (resolveFunction != null);
@@ -326,6 +336,18 @@ public boolean matchesSettings(Model model, ServiceShape service, TypeScriptSett
         return settingsPredicate.test(model, service, settings);
     }
 
+    /**
+     * @return the writer consumer for this RuntimeClientPlugin. May be used to add imports and dependencies
+     * used by this plugin.
+     */
+    public Consumer<TypeScriptWriter> getClientWriterConsumer() {
+        return this.clientWriterConsumer;
+    }
+
+    public Consumer<TypeScriptWriter> getClientWriterConsumer() {
+        return this.operationWriterConsumer;
+    }
+
     public static Builder builder() {
         return new Builder();
     }
@@ -398,6 +420,8 @@ public static final class Builder implements SmithyBuilder<RuntimeClientPlugin>
         private BiPredicate<Model, ServiceShape> servicePredicate = (model, service) -> true;
         private OperationPredicate operationPredicate = (model, service, operation) -> false;
         private SettingsPredicate settingsPredicate = (model, service, settings) -> true;
+        private BiConsumer<TypeScriptWriter, BiConsumer<Model, ServiceShape>> clientWriterConsumer = (writer, consumer) -> {};
+        private BiConsumer<TypeScriptWriter, TriConsumer<Model, ServiceShape, OperationShape>> operationWriterConsumer = (writer, consumer) -> {};
 
         @Override
         public RuntimeClientPlugin build() {
@@ -746,6 +770,28 @@ public Builder servicePredicate(BiPredicate<Model, ServiceShape> servicePredicat
             return this;
         }
 
+        /**
+         * Enables access to the writer for adding imports/dependencies.
+         */
+        public Builder withClientWriter(
+            BiConsumer<TypeScriptWriter,
+                BiConsumer<Model, ServiceShape>> clientWriterConsumer
+        ) {
+            this.clientWriterConsumer = clientWriterConsumer;
+            return this;
+        }
+
+        /**
+         * Enables access to the writer for adding imports/dependencies.
+         */
+        public Builder withOperationWriter(
+            BiConsumer<TypeScriptWriter,
+                TriConsumer<Model, ServiceShape, OperationShape>> operationWriterConsumer
+        ) {
+            this.operationWriterConsumer = operationWriterConsumer;
+            return this;
+        }
+
         /**
          * Configures various aspects of the builder based on naming conventions
          * defined by the provided {@link Convention} values.