Add complie method for EndpointRuleSet in RulesEngineBuilder

Author: joewyz

client/client-rulesengine/src/main/java/software/amazon/smithy/java/client/rulesengine/RulesEngineBuilder.java

@@ -17,6 +17,10 @@
 import java.util.ServiceLoader;
 import java.util.function.Function;
 import software.amazon.smithy.java.context.Context;
+import software.amazon.smithy.rulesengine.language.EndpointRuleSet;
+import software.amazon.smithy.rulesengine.logic.bdd.NodeReversal;
+import software.amazon.smithy.rulesengine.logic.bdd.SiftingOptimization;
+import software.amazon.smithy.rulesengine.logic.cfg.Cfg;
 import software.amazon.smithy.rulesengine.traits.EndpointBddTrait;
 
 /**
@@ -108,6 +112,20 @@ public Bytecode compile(EndpointBddTrait bdd) {
         return new BytecodeCompiler(extensions, bdd, functions, builtinProviders).compile();
     }
 
+    /**
+     * Compile BDD rules into a {@link Bytecode}.
+     *
+     * @param rules Endpoint Rules to compile.
+     * @return the compiled program.
+     */
+    public Bytecode compile(EndpointRuleSet rules) {
+        var cfg = Cfg.from(rules);
+        var originalTrait = EndpointBddTrait.from(cfg);
+        var optimizedTrait = SiftingOptimization.builder().cfg(cfg).build().apply(originalTrait);
+        var reversedTrait = new NodeReversal().apply(optimizedTrait);
+        return compile(reversedTrait);
+    }
+
     /**
      * Load bytecode from a file path.
      *

codegen/plugins/client-codegen/src/main/java/software/amazon/smithy/java/codegen/client/generators/BddFileGenerator.java

@@ -16,7 +16,6 @@
 import software.amazon.smithy.java.codegen.CodeGenerationContext;
 import software.amazon.smithy.java.codegen.JavaCodegenSettings;
 import software.amazon.smithy.model.shapes.ServiceShape;
-import software.amazon.smithy.rulesengine.logic.cfg.Cfg;
 import software.amazon.smithy.rulesengine.traits.EndpointBddTrait;
 import software.amazon.smithy.rulesengine.traits.EndpointRuleSetTrait;
 
@@ -40,15 +39,11 @@ public void accept(GenerateServiceDirective<CodeGenerationContext, JavaCodegenSe
     }
 
     private Bytecode compileBytecode(ServiceShape serviceShape) {
-        EndpointBddTrait bddTrait;
+        var engineBuilder = new RulesEngineBuilder();
         if (serviceShape.hasTrait(EndpointBddTrait.ID)) {
-            bddTrait = serviceShape.expectTrait(EndpointBddTrait.class);
+            return engineBuilder.compile(serviceShape.expectTrait(EndpointBddTrait.class));
         } else {
-            var endpointRuleSet = serviceShape.expectTrait(EndpointRuleSetTrait.class).getEndpointRuleSet();
-            var cfg = Cfg.from(endpointRuleSet);
-            bddTrait = EndpointBddTrait.from(cfg);
+            return engineBuilder.compile(serviceShape.expectTrait(EndpointRuleSetTrait.class).getEndpointRuleSet());
         }
-        var engineBuilder = new RulesEngineBuilder();
-        return engineBuilder.compile(bddTrait);
     }
 }