Service Integrations (specify integrations in smithy-build.json)

codegen/smithy-ruby-codegen-test-utils/src/main/java/software/amazon/smithy/ruby/codegen/integrations/WhiteLabelTestIntegration.java → codegen/smithy-ruby-codegen-test-utils/src/main/java/software/amazon/smithy/ruby/codegen/integrations/PluginTestIntegration.java

@@ -16,24 +16,21 @@
 package software.amazon.smithy.ruby.codegen.integrations;
 
 import java.util.List;
-import software.amazon.smithy.model.Model;
-import software.amazon.smithy.model.shapes.ServiceShape;
 import software.amazon.smithy.ruby.codegen.GenerationContext;
-import software.amazon.smithy.ruby.codegen.RubyIntegration;
 import software.amazon.smithy.ruby.codegen.RubyRuntimePlugin;
+import software.amazon.smithy.ruby.codegen.RubyServiceIntegration;
 import software.amazon.smithy.ruby.codegen.config.ClientConfig;
 import software.amazon.smithy.ruby.codegen.middleware.Middleware;
 import software.amazon.smithy.ruby.codegen.middleware.MiddlewareBuilder;
 import software.amazon.smithy.ruby.codegen.middleware.MiddlewareStackStep;
 
-public class WhiteLabelTestIntegration implements RubyIntegration {
+public class PluginTestIntegration extends RubyServiceIntegration {
 
     @Override
-    public boolean includeFor(ServiceShape service, Model model) {
-        return service.toShapeId().getName().equals("WhiteLabel");
+    public String name() {
+        return "plugin-test";
     }
 
-
     @Override
     public List<RubyRuntimePlugin> getRuntimePlugins(GenerationContext context) {
         return List.of(RubyRuntimePlugin.builder()

codegen/smithy-ruby-codegen-test-utils/src/main/resources/META-INF/services/software.amazon.smithy.ruby.codegen.RubyIntegration

@@ -1 +1 @@
-software.amazon.smithy.ruby.codegen.integrations.WhiteLabelTestIntegration
+software.amazon.smithy.ruby.codegen.integrations.PluginTestIntegration

codegen/smithy-ruby-codegen-test/smithy-build.json

@@ -30,6 +30,7 @@
       "plugins": {
         "ruby-codegen": {
           "service": "smithy.ruby.tests#WhiteLabel",
+          "integrations": ["plugin-test"],
           "module": "WhiteLabel",
           "gemspec": {
             "gemName": "white_label",

codegen/smithy-ruby-codegen/src/main/java/software/amazon/smithy/ruby/codegen/DirectedRubyCodegen.java

@@ -75,7 +75,8 @@ public GenerationContext createContext(CreateContextDirective<RubySettings, Ruby
         Model model = directive.model();
         List<RubyIntegration> integrations = directive.integrations().stream()
                 .filter((integration) -> integration
-                        .includeFor(service, model))
+                        .includeFor(service, model)
+                        || directive.settings().getIntegrations().contains(integration.name()))
                 .collect(Collectors.toList());
 
         Map<ShapeId, ProtocolGenerator> supportedProtocols = ProtocolGenerator

codegen/smithy-ruby-codegen/src/main/java/software/amazon/smithy/ruby/codegen/RubyServiceIntegration.java

@@ -0,0 +1,31 @@
+/*
+ * Copyright Amazon.com, Inc. or its affiliates. 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.
+ * A copy of the License is located at
+ *
+ *  http://aws.amazon.com/apache2.0
+ *
+ * or in the "license" file accompanying this file. This file 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 software.amazon.smithy.ruby.codegen;
+
+import software.amazon.smithy.model.Model;
+import software.amazon.smithy.model.shapes.ServiceShape;
+import software.amazon.smithy.utils.SmithyUnstableApi;
+
+/**
+ * A RubyIntegration that should be applied to services by specifying `integrations`
+ * in the smithy-build.json for the applicable services.
+ */
+@SmithyUnstableApi
+public abstract class RubyServiceIntegration implements  RubyIntegration {
+    public boolean includeFor(ServiceShape service, Model model) {
+        return false;
+    }
+}

codegen/smithy-ruby-codegen/src/main/java/software/amazon/smithy/ruby/codegen/RubySettings.java

@@ -19,6 +19,7 @@
 import java.util.Map;
 import java.util.Set;
 import java.util.logging.Logger;
+import java.util.stream.Collectors;
 import software.amazon.smithy.model.Model;
 import software.amazon.smithy.model.knowledge.ServiceIndex;
 import software.amazon.smithy.model.node.ObjectNode;
@@ -35,13 +36,15 @@ public final class RubySettings {
     private static final Logger LOGGER = Logger.getLogger(RubySettings.class.getName());
 
     private static final String SERVICE = "service";
+    private static final String INTEGRATIONS = "integrations";
     private static final String MODULE = "module";
     private static final String GEMSPEC = "gemspec";
     private static final String GEM_NAME = "gemName";
     private static final String GEM_VERSION = "gemVersion";
     private static final String GEM_SUMMARY = "gemSummary";
 
     private ShapeId service;
+    private Set<String> integrations = Set.of();
     private String module;
     private String gemName;
     private String gemVersion;
@@ -56,9 +59,15 @@ public final class RubySettings {
     public static RubySettings from(ObjectNode config) {
         RubySettings settings = new RubySettings();
         config.warnIfAdditionalProperties(
-                Arrays.asList(SERVICE, MODULE, GEMSPEC, GEM_NAME, GEM_VERSION, GEM_SUMMARY));
+                Arrays.asList(SERVICE, INTEGRATIONS, MODULE, GEMSPEC, GEM_NAME, GEM_VERSION, GEM_SUMMARY));
 
         settings.setService(config.expectStringMember(SERVICE).expectShapeId());
+        config.getArrayMember(INTEGRATIONS).ifPresent((integrations) -> {
+            settings.setIntegrations(
+                    integrations.getElements().stream().map(
+                            (n) -> n.expectStringNode().getValue()).collect(Collectors.toUnmodifiableSet())
+            );
+        });
         // module and namespace
         settings.setModule(config.expectStringMember(MODULE).getValue());
         // required gemspec values
@@ -80,12 +89,26 @@ public ShapeId getService() {
     }
 
     /**
-     * @param service service to generate for
+     * @param service service to generate for.
      */
     public void setService(ShapeId service) {
         this.service = service;
     }
 
+    /**
+     * @return list of integrations to apply to this service.
+     */
+    public Set<String> getIntegrations() {
+        return integrations;
+    }
+
+    /**
+     * @param integrations integrations to apply to this service.
+     */
+    public void setIntegrations(Set<String> integrations) {
+        this.integrations = integrations;
+    }
+
     /**
      * @return module (namespace) to generate in.
      */
@@ -151,9 +174,8 @@ public Set<RubyDependency> getBaseDependencies() {
     }
 
     /**
-     *
-     * @param service service to generate for
-     * @param model model used for generation
+     * @param service                 service to generate for
+     * @param model                   model used for generation
      * @param supportedProtocolTraits ordered list of all supported protocols
      * @return the resolved service protocol
      */