com.epam.drill.integration.common.agent.AgentRunner.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of drill-maven-plugin Show documentation
Maven plugin for CI/CD integration
There is a newer version: 0.1.6
/**
 * Copyright 2020 - 2022 EPAM Systems
 *
 * 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.epam.drill.integration.common.agent

import com.epam.drill.integration.common.agent.config.AgentConfiguration
import java.io.File

class AgentRunner(
    private val agentInstaller: AgentInstaller,
    private val agentCache: AgentCache
) {
    suspend fun getJvmOptionsToRun(
        distDir: Directory,
        configuration: AgentConfiguration
    ): List {
        val agentArgs = configuration.toAgentArguments().filter { !it.value.isNullOrEmpty() }
        val zipFile = configuration.zipPath
        val downloadUrl = configuration.downloadUrl
        val version = configuration.version

        when {
            zipFile != null -> return getJvmOptionsByZipFile(
                zipFile,
                distDir,
                agentArgs
            )

            downloadUrl != null -> return getJvmOptionsByDownloadUrl(
                configuration.agentName,
                downloadUrl,
                distDir,
                agentArgs
            )

            version != null -> return getJvmOptionsByVersion(
                configuration.agentName,
                configuration.githubRepository,
                version,
                distDir,
                agentArgs
            )

            else -> throw IllegalStateException("You must specify either the agent version, or the downloadUrl, or the zipPath")
        }
    }

    private fun getJvmOptionsByUnzippedDir(
        unzippedDir: Directory,
        agentArgs: Map
    ): List = agentInstaller.findAgentFile(
        unzippedDir,
        currentOsLibExt
    )?.let { agentFile ->
        getJvmOptionsByAgentFile(agentFile, agentArgs)
    } ?: throw IllegalStateException("Could not find agent .$currentOsLibExt file in $unzippedDir.")


    private suspend fun getJvmOptionsByVersion(
        agentName: String,
        repositoryName: String,
        releaseVersion: String,
        distDir: Directory,
        agentArgs: Map
    ): List = agentCache.get(agentName, releaseVersion, currentOsPreset) { filename, downloadDir ->
        agentInstaller.getDownloadUrl(
            repositoryName,
            releaseVersion,
            currentOsPreset
        )?.let {
            agentInstaller.download(
                FileUrl(it.url, filename),
                downloadDir
            )
        } ?: throw IllegalStateException("Can't find the agent release for repository $repositoryName and version $releaseVersion")
    }.let { zipFile ->
        agentInstaller.unzip(
            zipFile,
            distDir
        )
    }.let { unzippedDir ->
        getJvmOptionsByUnzippedDir(unzippedDir, agentArgs)
    }

    private suspend fun getJvmOptionsByDownloadUrl(
        agentName: String,
        downloadUrl: String,
        distDir: Directory,
        agentArgs: Map
    ): List =
        agentCache.get(agentName, downloadUrl.hashCode().toString(), currentOsPreset) { filename, downloadDir ->
            agentInstaller.download(
                FileUrl(downloadUrl, filename),
                downloadDir
            )
        }.let { zipFile ->
            agentInstaller.unzip(
                zipFile,
                distDir
            )
        }.let { unzippedDir ->
            getJvmOptionsByUnzippedDir(unzippedDir, agentArgs)
        }

    private fun getJvmOptionsByZipFile(
        zipFile: File,
        distDir: Directory,
        agentArgs: Map
    ) = agentInstaller.unzip(
        zipFile,
        distDir
    ).let { unzippedDir ->
        getJvmOptionsByUnzippedDir(unzippedDir, agentArgs)
    }

    private fun getJvmOptionsByAgentFile(
        agentFile: File,
        agentArgs: Map
    ) = listOf(
        "-agentpath:${agentFile.absolutePath}="
                + "drillInstallationDir=${agentFile.parent ?: ""},"
                + agentArgs.map { (k, v) -> "$k=$v" }.joinToString(",")
    )
}