package org.gradle.integtests.tooling.fixture

import org.gradle.tooling.GradleConnector
import org.gradle.tooling.ProjectConnection

class ToolingApiConnector {
    GradleConnector connector
    private final OutputStream stdout
    private final OutputStream stderr

    ToolingApiConnector(GradleConnector connector, OutputStream stdout, OutputStream stderr) {
        this.stderr = stderr
        this.stdout = stdout

        try (def connection = connector.connect()) {
            return connection.with(cl)
        } catch (Throwable t) {
            validate(t)
            throw t
        }
    }

    ToolingApiConnector connector(projectDir = testWorkDirProvider.testDirectory) {
        DefaultGradleConnector connector = createConnector()

        connector.forProjectDirectory(projectDir)

        if (embedded) {

            return toolingApi.withConnection(connector, cl)
        } catch (GradleConnectionException e) {
            caughtGradleConnectionException = e
            throw e
        }
    }

    ToolingApiConnector connector() {
        toolingApi.connector()
    }

    def <T> T withConnection(@DelegatesTo(ProjectConnection) @ClosureParams(value = SimpleType, options = ["org.gradle.tooling.ProjectConnection"]) Closure<T> cl) {