com.palantir.gradle.docker.DockerRunExtension.groovy Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gradle-docker Show documentation
Palantir open source project
There is a newer version: 0.36.0
/*
 * (c) Copyright 2020 Palantir Technologies Inc. All rights reserved.
 *
 * 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.palantir.gradle.docker

import static com.google.common.base.Preconditions.checkNotNull

import com.google.common.base.Preconditions
import com.google.common.collect.ImmutableList
import com.google.common.collect.ImmutableMap
import com.google.common.collect.ImmutableSet

class DockerRunExtension {

    private String name
    private String image
    private String network
    private List command = ImmutableList.of()
    private Set ports = ImmutableSet.of()
    private Map env = ImmutableMap.of()
    private List arguments = ImmutableList.of()
    private Map volumes = ImmutableMap.of()
    private boolean daemonize = true
    private boolean clean = false
    private boolean ignoreExitValue = false

    public String getName() {
        return name
    }

    public void setName(String name) {
        this.name = name
    }

    public boolean getDaemonize() {
        return daemonize
    }

    public void setDaemonize(boolean daemonize) {
        this.daemonize = daemonize
    }

    public boolean getClean() {
        return clean
    }

    public void setClean(boolean clean) {
        this.clean = clean
    }

    public String getImage() {
        return image
    }

    public void setImage(String image) {
        this.image = image
    }

    public Set getPorts() {
        return ports
    }

    public List getCommand() {
        return command
    }

    public Map getVolumes() {
        return volumes
    }

    public void command(String... command) {
        this.command = ImmutableList.copyOf(command)
    }

    public void setNetwork(String network) {
        this.network = network
    }

    public String getNetwork() {
        return network
    }

    private void setEnvSingle(String key, String value) {
        this.env.put(checkNotNull(key, "key"), checkNotNull(value, "value"))
    }

    public void env(Map env) {
        this.env = ImmutableMap.copyOf(env)
    }

    public Map getEnv() {
        return env
    }

    public void arguments(String... arguments) {
        this.arguments = ImmutableList.copyOf(arguments)
    }

    public List getArguments() {
        return arguments
    }

    public void ports(String... ports) {
        ImmutableSet.Builder builder = ImmutableSet.builder()
        for (String port : ports) {
            String[] mapping = port.split(':', 2)
            if (mapping.length == 1) {
                checkPortIsValid(mapping[0])
                builder.add("${mapping[0]}:${mapping[0]}")
            } else {
                checkPortIsValid(mapping[0])
                checkPortIsValid(mapping[1])
                builder.add("${mapping[0]}:${mapping[1]}")
            }
        }
        this.ports = builder.build()
    }

    public void volumes(Map volumes) {
      this.volumes = ImmutableMap.copyOf(volumes)
    }

    private static void checkPortIsValid(String port) {
        int val = Integer.parseInt(port)
        Preconditions.checkArgument(0 < val && val <= 65536, "Port must be in the range [1,65536]")
    }

    public boolean getIgnoreExitValue() {
        return ignoreExitValue
    }

    public void setIgnoreExitValue(boolean ignoreExitValue) {
        this.ignoreExitValue = ignoreExitValue
    }
}