SpongeDataService.java

Set<Module> goodModules = toImport.stream()
    .flatMap(n -> n.getChildren().stream())
    .flatMap(n -> {
        Object d = n.getData();
        if (d instanceof GradleSourceSetData) {
            return n.getChildren().stream()
                .filter(n1 -> n1.getData() instanceof DependencyData)
                .filter(n1 -> ((DependencyData) n1.getData()).getOwnerModule().getExternalName().contains("main"));
        } else {
            return Stream.of(n);
        }
    })
    .filter(n -> n.getData() instanceof DependencyData)
    .filter(n -> {
        final Object d = n.getData();
        if (d instanceof LibraryDependencyData) {
            return ((LibraryDependencyData) d).getExternalName()
                .startsWith(SpongeModuleType.getInstance().getGroupId() + ":" + SpongeModuleType.getInstance().getArtifactId());
        } else {
            return ((ModuleDependencyData) d).getExternalName().contains("SpongeCommon") ||
                ((ModuleDependencyData) d).getExternalName().contains("SpongeAPI");
        }
    })
    .map(n -> ((DependencyData) n.getData()).getOwnerModule())
    .map(modelsProvider::findIdeModule)
    .collect(Collectors.toSet());