com.github.dockerjava.core.dockerfile.DockerfileStatement Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of docker-java Show documentation
Java API Client for Docker
There is a newer version: 3.4.1
package com.github.dockerjava.core.dockerfile;

import com.github.dockerjava.api.DockerClientException;

import java.net.URI;
import java.net.URISyntaxException;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import com.google.common.base.Objects;
import com.google.common.base.Optional;

/**
 * A statement present in a dockerfile.
 */
public abstract class DockerfileStatement {

  private DockerfileStatement() {
  }

  public T transform(Map env) {
    return (T) this;
  }

  protected String filterForEnvironmentVars(Map environmentMap,
                                            String extractedResource) {

    if (environmentMap.size() > 0) {

      String currentResourceContent = extractedResource;

      for (Map.Entry entry : environmentMap.entrySet()) {

        String variable = entry.getKey();

        String replacementValue = entry.getValue();

        // handle: $VARIABLE case
        currentResourceContent = currentResourceContent.replaceAll(
            "\\$" + variable, Matcher.quoteReplacement(replacementValue));

        // handle ${VARIABLE} case
        currentResourceContent = currentResourceContent.replaceAll(
            "\\$\\{" + variable + "\\}", Matcher.quoteReplacement(replacementValue));

      }

      return currentResourceContent;
    } else {
      return extractedResource;
    }
  }


  /**
   * A statement that we don't particularly care about.
   */
  public static class OtherLine extends DockerfileStatement {

    public final String statement;

    public OtherLine(String statement) {
      this.statement = statement;
    }

    @Override
    public String toString() {
      return statement;
    }
  }

  /**
   * An ADD or a COPY
   */
  public static class Add extends DockerfileStatement {

    private static final Pattern ADD_OR_COPY_PATTERN = Pattern
        .compile("^(ADD|COPY)\\s+(.*)\\s+(.*)$");

    public final String source;
    public final String destination;

    private Add(String source, String destination) {
      this.source = source;
      this.destination = destination;
    }

    private Add(final Matcher matcher) {
      source = matcher.group(2);
      destination = matcher.group(3);
    }

    @Override
    public Add transform(Map env) {
      String resource = filterForEnvironmentVars(env, source).trim();
      return new Add(resource, destination);
    }

    public boolean isFileResource() {
      URI uri;
      try {
        uri = new URI(source);
      } catch (URISyntaxException e) {
        return false;
      }
      return uri.getScheme() == null || "file".equals(uri.getScheme());
    }

    /**
     * Createa an Add if it matches, or missing if not.
     *
     * @param statement statement that may be an ADD or a COPY
     * @return optional typed item.
     */
    public static Optional create(String statement) {
      Matcher matcher = ADD_OR_COPY_PATTERN.matcher(statement.trim());
      if (!matcher.find()) {
        return Optional.absent();
      }

      if (matcher.groupCount() != 3) {
        throw new DockerClientException("Wrong ADD or COPY format");
      }

      return Optional.of(new Add(matcher));
    }


    @Override
    public String toString() {
      return Objects.toStringHelper(this)
          .add("source", source)
          .add("destination", destination)
          .toString();
    }
  }

  public static class Env extends DockerfileStatement {

    private static final Pattern ENV_PATTERN = Pattern
        .compile("^ENV\\s+(.*)\\s+(.*)$");

    public final String variable;
    public final String value;

    private Env(String variable, String value) {
      this.variable = variable;
      this.value = value;
    }

    private Env(Matcher envMatcher) {
      this.variable = envMatcher.group(1).trim();
      this.value = envMatcher.group(2).trim();
    }

    public static Optional create(String statement) {
      Matcher matcher = ENV_PATTERN.matcher(statement.trim());
      if (!matcher.find()) {
        return Optional.absent();
      }

      if (matcher.groupCount() != 2) {
        throw new DockerClientException("Wrong ENV format");
      }

      return Optional.of(new Env(matcher));
    }

    @Override
    public String toString() {
      return Objects.toStringHelper(this)
          .add("variable", variable)
          .add("value", value)
          .toString();
    }
  }

  /**
   * Return a dockerfile statement
   */
  public static Optional createFromLine(String cmd) {
    if (cmd.trim().isEmpty() || cmd.startsWith("#")) {
      return Optional.absent();
    }

    Optional line;

    line = Add.create(cmd);

    if (line.isPresent()) {
      return line;
    }

    line = Env.create(cmd);

    if (line.isPresent()) {
      return line;
    }

    return Optional.of(new OtherLine(cmd));


  }

}