add arch condition to check for any transitive dependency

Issue: TNG#780
Signed-off-by: e.solutions <[email protected]>
on-behalf-of: @e-esolutions-GmbH <[email protected]>

archunit/src/main/java/com/tngtech/archunit/lang/conditions/AnyTransitiveDependencyCondition.java

@@ -0,0 +1,120 @@
+/*
+ * Copyright 2014-2022 TNG Technology Consulting GmbH
+ *
+ * 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.tngtech.archunit.lang.conditions;
+
+import java.util.Collection;
+import java.util.HashSet;
+import java.util.LinkedList;
+import java.util.Objects;
+import java.util.Optional;
+import java.util.Set;
+import java.util.function.Predicate;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
+
+import com.tngtech.archunit.base.DescribedPredicate;
+import com.tngtech.archunit.core.domain.Dependency;
+import com.tngtech.archunit.core.domain.JavaClass;
+import com.tngtech.archunit.lang.ArchCondition;
+import com.tngtech.archunit.lang.ConditionEvent;
+import com.tngtech.archunit.lang.ConditionEvents;
+import com.tngtech.archunit.lang.SimpleConditionEvent;
+
+import static com.google.common.base.Preconditions.checkNotNull;
+
+public class AnyTransitiveDependencyCondition extends ArchCondition<JavaClass> {
+
+    private final DescribedPredicate<? super JavaClass> conditionPredicate;
+    private final Set<JavaClass> allClasses = new HashSet<>();
+
+    public AnyTransitiveDependencyCondition(DescribedPredicate<? super JavaClass> conditionPredicate) {
+        super("transitively depend on any classes that " + conditionPredicate.getDescription());
+
+        this.conditionPredicate = checkNotNull(conditionPredicate);
+    }
+
+    @Override
+    public void init(Collection<JavaClass> allObjectsToTest) {
+        allClasses.addAll(allObjectsToTest);
+    }
+
+    @Override
+    public void check(JavaClass item, ConditionEvents events) {
+        getDirectDependencyTargets(item)
+                .filter(not(allClasses::contains))
+                .map(this::getDependencyPathToMatchingClasses)
+                .filter(Objects::nonNull)
+                .map(dependencyPath -> createTransitivePathFoundEvent(item, dependencyPath))
+                .forEach(events::add);
+    }
+
+    /**
+     * @return the dependency path to a matching class including the source class or null if there is none
+     */
+    private LinkedList<JavaClass> getDependencyPathToMatchingClasses(JavaClass clazz) {
+        LinkedList<JavaClass> transitivePath = new LinkedList<>();
+        if (matchingTransitiveDependencies(clazz, new HashSet<>(), transitivePath)) {
+            transitivePath.add(clazz);
+            return transitivePath;
+        }
+        return null;
+    }
+
+    private boolean matchingTransitiveDependencies(
+            JavaClass clazz,
+            HashSet<JavaClass> analyzedClasses,
+            LinkedList<JavaClass> transitivePath
+    ) {
+        if (conditionPredicate.test(clazz)) {
+            return true;
+        }
+
+        analyzedClasses.add(clazz);
+
+        Optional<JavaClass> firstMatchingTransitiveDependency = getDirectDependencyTargets(clazz)
+                .filter(not(allClasses::contains))
+                .filter(not(analyzedClasses::contains))
+                .filter(it -> matchingTransitiveDependencies(it, analyzedClasses, transitivePath))
+                .findFirst();
+
+        firstMatchingTransitiveDependency.ifPresent(transitivePath::add);
+        return firstMatchingTransitiveDependency.isPresent();
+    }
+
+    private static Stream<JavaClass> getDirectDependencyTargets(JavaClass item) {
+        return item.getDirectDependenciesFromSelf().stream().map(Dependency::getTargetClass).map(JavaClass::getBaseComponentType).distinct();
+    }
+
+    private static ConditionEvent createTransitivePathFoundEvent(JavaClass clazz, LinkedList<JavaClass> dependencyPath) {
+        StringBuilder messageBuilder =
+                new StringBuilder("Class <" + clazz.getFullName() + "> accesses <" + dependencyPath.getLast().getFullName() + ">");
+        if (dependencyPath.size() > 1) {
+            messageBuilder
+                    .append(" which transitively accesses e.g. ")
+                    .append(dependencyPath.subList(0, dependencyPath.size()).stream()
+                            .map(it -> String.format("<%s>", it.getFullName()))
+                            .collect(Collectors.joining(" <- ")));
+
+        }
+        return SimpleConditionEvent.satisfied(clazz, messageBuilder.toString());
+    }
+
+    @SuppressWarnings("unchecked")
+    private static <T> Predicate<T> not(Predicate<? super T> target) {
+        Objects.requireNonNull(target);
+        return (Predicate<T>) target.negate();
+    }
+}

archunit/src/main/java/com/tngtech/archunit/lang/conditions/ArchConditions.java

@@ -309,6 +309,11 @@ public static ArchCondition<JavaClass> transitivelyDependOnClassesThat(final Des
                 GET_TRANSITIVE_DEPENDENCIES_FROM_SELF);
     }
 
+    @PublicAPI(usage = ACCESS)
+    public static ArchCondition<JavaClass> transitivelyDependOnAnyClassesThat(final DescribedPredicate<? super JavaClass> predicate) {
+        return new AnyTransitiveDependencyCondition(predicate);
+    }
+
     @PublicAPI(usage = ACCESS)
     public static ArchCondition<JavaClass> onlyDependOnClassesThat(final DescribedPredicate<? super JavaClass> predicate) {
         return new AllDependenciesCondition(

archunit/src/main/java/com/tngtech/archunit/lang/syntax/ClassesShouldInternal.java

@@ -524,6 +524,16 @@ public ClassesShouldConjunction onlyDependOnClassesThat(DescribedPredicate<? sup
         return addCondition(ArchConditions.onlyDependOnClassesThat(predicate));
     }
 
+    @Override
+    public ClassesThat<ClassesShouldConjunction> transitivelyDependOnAnyClassesThat() {
+        return new ClassesThatInternal<>(predicate -> addCondition(ArchConditions.transitivelyDependOnAnyClassesThat(predicate)));
+    }
+
+    @Override
+    public ClassesShouldConjunction transitivelyDependOnAnyClassesThat(DescribedPredicate<? super JavaClass> predicate) {
+        return addCondition(ArchConditions.transitivelyDependOnAnyClassesThat(predicate));
+    }
+
     @Override
     public ClassesThat<ClassesShouldConjunction> transitivelyDependOnClassesThat() {
         return new ClassesThatInternal<>(predicate -> addCondition(ArchConditions.transitivelyDependOnClassesThat(predicate)));

archunit/src/main/java/com/tngtech/archunit/lang/syntax/elements/ClassesShould.java

@@ -1015,6 +1015,21 @@ public interface ClassesShould {
     @PublicAPI(usage = ACCESS)
     ClassesShouldConjunction onlyDependOnClassesThat(DescribedPredicate<? super JavaClass> predicate);
 
+    /**
+     * TODO
+     * @return
+     */
+    @PublicAPI(usage = ACCESS)
+    ClassesThat<ClassesShouldConjunction> transitivelyDependOnAnyClassesThat();
+
+    /**
+     * TODO
+     * @param predicate
+     * @return
+     */
+    @PublicAPI(usage = ACCESS)
+    ClassesShouldConjunction transitivelyDependOnAnyClassesThat(DescribedPredicate<? super JavaClass> predicate);
+
     /**
      * Asserts that all classes selected by this rule transitively depend on certain classes.<br>
      * NOTE: This usually makes more sense the negated way, e.g.