import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.CancellationException;

import static org.gradle.internal.FileUtils.hasExtension;

public class DistributionFactory {
    private final Clock clock;

    public DistributionFactory(Clock clock) {
        this.clock = clock;
    }

    /**
     * Returns the default distribution to use for the specified project.
     */

    public DefaultGradleConnector(ConnectionFactory connectionFactory, DistributionFactory distributionFactory) {
        this.connectionFactory = connectionFactory;
        this.distributionFactory = distributionFactory;
    }

    /**
     * Closes the tooling API, releasing all resources. Blocks until completed.
     *

    final ConnectionFactory connectionFactory = Mock()
    final DistributionFactory distributionFactory = Mock()
    final Distribution distribution = Mock()
    final File projectDir = new File('project-dir')
    final GradleConnector connector = new DefaultGradleConnector(connectionFactory, distributionFactory)

    def canCreateAConnectionGivenAProjectDirectory() {

            return new Factory<DefaultGradleConnector>() {
                @Override
                public DefaultGradleConnector create() {
                    return new DefaultGradleConnector(connectionFactory, distributionFactory);
                }
            };
        }

        @Provides

GradleConnector -> DefaultGradleConnector: create
DefaultGradleConnector --> User: instance
User -> DefaultGradleConnector: use distribution/version/URI
DefaultGradleConnector -> DistributionFactory: get distribution
DistributionFactory --> DefaultGradleConnector: Distribution
User -> DefaultGradleConnector: connect()
DefaultGradleConnector -> ConnectionFactory: create()
ConnectionFactory --> DefaultGradleConnector: DefaultProjectConnection

LoggingInitializer -> LazyConsumer: run()
LazyConsumer -> ToolingImplLoader: createConnection
create Connection
ToolingImplLoader -> Connection: new
Connection --> LazyConsumer: connection
note left
uses DistributionFactory to create classloader and instantiate DefaultConnection (provider side) using services lookup
then creates ConsumerConnection matching to capability of this DefaultConnection (DefaultToolingImplementationLoader.create())

    final ProgressLoggerFactory progressLoggerFactory = Mock()
    final ProgressLogger progressLogger = Mock()
    final BuildCancellationToken cancellationToken = Mock()
    final DistributionFactory factory = new DistributionFactory(Time.clock())
    final InternalBuildProgressListener buildProgressListener = Mock()

    def setup() {

Class <org.gradle.tooling.internal.consumer.DistributionFactory$ClasspathDistribution> is not annotated (directly or via its package) with @org.gradle.api.NonNullApi in (DistributionFactory.java:0)
Class <org.gradle.tooling.internal.consumer.DistributionFactory$InstalledDistribution$1> is not annotated (directly or via its package) with @org.gradle.api.NonNullApi in (DistributionFactory.java:0)