feat: Added support to get All segments used in the project

core-api/src/main/java/com/optimizely/ab/config/DatafileProjectConfig.java

@@ -74,6 +74,7 @@ public class DatafileProjectConfig implements ProjectConfig {
     private final List<Group> groups;
     private final List<Rollout> rollouts;
     private final List<Integration> integrations;
+    private final Set<String> allSegments;
 
     // key to entity mappings
     private final Map<String, Attribute> attributeKeyMapping;
@@ -204,6 +205,15 @@ public DatafileProjectConfig(String accountId,
         this.publicKeyForODP = publicKeyForODP;
         this.hostForODP = hostForODP;
 
+        Set<String> allSegments = new HashSet<>();
+        if (typedAudiences != null) {
+            for(Audience audience: typedAudiences) {
+                allSegments.addAll(audience.getSegments());
+            }
+        }
+
+        this.allSegments = allSegments;
+
         Map<String, Experiment> variationIdToExperimentMap = new HashMap<String, Experiment>();
         for (Experiment experiment : this.experiments) {
             for (Variation variation : experiment.getVariations()) {
@@ -424,6 +434,10 @@ public List<Experiment> getExperiments() {
         return experiments;
     }
 
+    public Set<String> getAllSegments() {
+        return this.allSegments;
+    }
+
     @Override
     public List<Experiment> getExperimentsForEventKey(String eventKey) {
         EventType event = eventNameMapping.get(eventKey);

core-api/src/main/java/com/optimizely/ab/config/audience/AndCondition.java

@@ -38,6 +38,7 @@ public AndCondition(@Nonnull List<Condition> conditions) {
         this.conditions = conditions;
     }
 
+    @Override
     public List<Condition> getConditions() {
         return conditions;
     }

core-api/src/main/java/com/optimizely/ab/config/audience/Audience.java

@@ -1,6 +1,6 @@
 /**
  *
- *    Copyright 2016-2017, 2019, Optimizely and contributors
+ *    Copyright 2016-2017, 2019, 2022, Optimizely and contributors
  *
  *    Licensed under the Apache License, Version 2.0 (the "License");
  *    you may not use this file except in compliance with the License.
@@ -22,6 +22,9 @@
 import com.optimizely.ab.config.IdKeyMapped;
 
 import javax.annotation.concurrent.Immutable;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
 
 /**
  * Represents the Optimizely Audience configuration.
@@ -69,4 +72,27 @@ public String toString() {
             ", conditions=" + conditions +
             '}';
     }
+
+    public Set<String> getSegments() {
+        return getSegments(conditions);
+    }
+
+    private static Set<String> getSegments(Condition conditions) {
+        List<Condition> nestedConditions = conditions.getConditions();
+        Set<String> segments = new HashSet<>();
+        if (nestedConditions != null) {
+            for (Condition nestedCondition : nestedConditions) {
+                Set<String> nestedSegments = getSegments(nestedCondition);
+                segments.addAll(nestedSegments);
+            }
+        } else {
+            if (conditions.getClass() == UserAttribute.class) {
+                UserAttribute userAttributeCondition = (UserAttribute) conditions;
+                if (UserAttribute.QUALIFIED.equals(userAttributeCondition.getMatch())) {
+                    segments.add((String)userAttributeCondition.getValue());
+                }
+            }
+        }
+        return segments;
+    }
 }

core-api/src/main/java/com/optimizely/ab/config/audience/AudienceIdCondition.java

@@ -27,6 +27,7 @@
 
 import javax.annotation.Nullable;
 import javax.annotation.concurrent.Immutable;
+import java.util.List;
 import java.util.Map;
 import java.util.Objects;
 
@@ -97,6 +98,11 @@ public boolean equals(Object o) {
                 (audienceId.equals(condition.audienceId)));
     }
 
+    @Override
+    public List<Condition> getConditions() {
+        return null;
+    }
+
     @Override
     public int hashCode() {
 

core-api/src/main/java/com/optimizely/ab/config/audience/Condition.java

@@ -20,6 +20,7 @@
 import com.optimizely.ab.config.ProjectConfig;
 
 import javax.annotation.Nullable;
+import java.util.List;
 import java.util.Map;
 
 /**
@@ -33,4 +34,6 @@ public interface Condition<T> {
     String toJson();
 
     String getOperandOrId();
+
+    List<Condition> getConditions();
 }

core-api/src/main/java/com/optimizely/ab/config/audience/EmptyCondition.java

@@ -21,7 +21,7 @@
 import javax.annotation.Nullable;
 import java.util.Map;
 
-public class EmptyCondition<T> implements Condition<T> {
+public class EmptyCondition<T> extends LeafCondition<T> {
     @Nullable
     @Override
     public Boolean evaluate(ProjectConfig config, OptimizelyUserContext user) {

core-api/src/main/java/com/optimizely/ab/config/audience/LeafCondition.java

@@ -0,0 +1,26 @@
+/**
+ *    Copyright 2022, Optimizely Inc. and contributors
+ *
+ *    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 com.optimizely.ab.config.audience;
+
+import java.util.List;
+
+public abstract class LeafCondition<T> implements Condition<T> {
+
+    @Override
+    public List<Condition> getConditions() {
+        return null;
+    }
+}

core-api/src/main/java/com/optimizely/ab/config/audience/NotCondition.java

@@ -22,6 +22,8 @@
 import javax.annotation.Nullable;
 import javax.annotation.concurrent.Immutable;
 import javax.annotation.Nonnull;
+import java.util.Arrays;
+import java.util.List;
 
 
 /**
@@ -41,6 +43,11 @@ public Condition getCondition() {
         return condition;
     }
 
+    @Override
+    public List<Condition> getConditions() {
+        return Arrays.asList(condition);
+    }
+
     @Nullable
     public Boolean evaluate(ProjectConfig config, OptimizelyUserContext user) {
         Boolean conditionEval = condition == null ? null : condition.evaluate(config, user);

core-api/src/main/java/com/optimizely/ab/config/audience/NullCondition.java

@@ -21,7 +21,7 @@
 import javax.annotation.Nullable;
 import java.util.Map;
 
-public class NullCondition<T> implements Condition<T> {
+public class NullCondition<T> extends LeafCondition<T> {
     @Nullable
     @Override
     public Boolean evaluate(ProjectConfig config, OptimizelyUserContext user) {

core-api/src/main/java/com/optimizely/ab/config/audience/OrCondition.java

@@ -38,6 +38,7 @@ public OrCondition(@Nonnull List<Condition> conditions) {
         this.conditions = conditions;
     }
 
+    @Override
     public List<Condition> getConditions() {
         return conditions;
     }

core-api/src/main/java/com/optimizely/ab/config/audience/UserAttribute.java

@@ -38,7 +38,7 @@
  */
 @Immutable
 @JsonIgnoreProperties(ignoreUnknown = true)
-public class UserAttribute<T> implements Condition<T> {
+public class UserAttribute<T> extends LeafCondition<T> {
     public static final String QUALIFIED = "qualified";
 
     private static final Logger logger = LoggerFactory.getLogger(UserAttribute.class);

core-api/src/test/java/com/optimizely/ab/config/audience/AudienceConditionEvaluationTest.java

@@ -1840,4 +1840,40 @@ public void nullValueEvaluate()  {
         assertNull(nullValueAttribute.evaluate(null, OTUtils.user(Collections.singletonMap(attributeName, attributeValue))));
         assertNull(nullValueAttribute.evaluate(null, OTUtils.user((Collections.singletonMap(attributeName, "")))));
     }
+
+    @Test
+    public void getAllSegmentsFromAudience() {
+        Condition condition = createMultipleConditionAudienceAndOrODP();
+        Audience audience = new Audience("1", "testAudience", condition);
+        assertEquals(new HashSet<>(Arrays.asList("odp-segment-1", "odp-segment-2", "odp-segment-3", "odp-segment-4")), audience.getSegments());
+
+        // ["and", "1", "2"]
+        List<Condition> audience1And2 = new ArrayList<>();
+        audience1And2.add(new UserAttribute("odp.audiences", "third_party_dimension", "qualified", "odp-segment-1"));
+        audience1And2.add(new UserAttribute("odp.audiences", "third_party_dimension", "qualified", "odp-segment-2"));
+        AndCondition audienceCondition1 = new AndCondition(audience1And2);
+
+        // ["and", "3", "4"]
+        List<Condition> audience3And4 = new ArrayList<>();
+        audience3And4.add(new UserAttribute("odp.audiences", "third_party_dimension", "qualified", "odp-segment-3"));
+        audience3And4.add(new UserAttribute("odp.audiences", "third_party_dimension", "qualified", "odp-segment-4"));
+        AndCondition audienceCondition2 = new AndCondition(audience3And4);
+
+        // ["or", "5", "6"]
+        List<Condition> audience5And6 = new ArrayList<>();
+        audience5And6.add(new UserAttribute("odp.audiences", "third_party_dimension", "qualified", "odp-segment-5"));
+        audience5And6.add(new UserAttribute("odp.audiences", "third_party_dimension", "qualified", "odp-segment-6"));
+        OrCondition audienceCondition3 = new OrCondition(audience5And6);
+
+
+        //['or', '1', '2', '3']
+        List<Condition> conditions = new ArrayList<>();
+        conditions.add(audienceCondition1);
+        conditions.add(audienceCondition2);
+        conditions.add(audienceCondition3);
+
+        OrCondition implicitOr = new OrCondition(conditions);
+        audience = new Audience("1", "testAudience", implicitOr);
+        assertEquals(new HashSet<>(Arrays.asList("odp-segment-1", "odp-segment-2", "odp-segment-3", "odp-segment-4", "odp-segment-5", "odp-segment-6")), audience.getSegments());
+    }
 }