Add core 'ports' functionality to docker context exporter, fix cross-platform Dockerfile write inconsistency

jib-core/src/main/java/com/google/cloud/tools/jib/docker/DockerContextGenerator.java

@@ -31,6 +31,7 @@
 import java.util.Collections;
 import java.util.List;
 import java.util.regex.Matcher;
+import java.util.stream.Collectors;
 import javax.annotation.Nullable;
 
 /**
@@ -46,12 +47,54 @@
  */
 public class DockerContextGenerator {
 
+  /**
+   * Formats a list for the Dockerfile's ENTRYPOINT or CMD.
+   *
+   * @see <a
+   *     href="https://docs.docker.com/engine/reference/builder/#exec-form-entrypoint-example">https://docs.docker.com/engine/reference/builder/#exec-form-entrypoint-example</a>
+   * @param items the list of items to join into an array.
+   * @return a string in the format: ["item1","item2",...]
+   */
+  @VisibleForTesting
+  static String joinAsJsonArray(List<String> items) {
+    StringBuilder resultString = new StringBuilder("[");
+    boolean firstComponent = true;
+    for (String item : items) {
+      if (!firstComponent) {
+        resultString.append(',');
+      }
+
+      // Escapes quotes.
+      item = item.replaceAll("\"", Matcher.quoteReplacement("\\\""));
+
+      resultString.append('"').append(item).append('"');
+      firstComponent = false;
+    }
+    resultString.append(']');
+
+    return resultString.toString();
+  }
+
+  /**
+   * Builds a list of Dockerfile "EXPOSE" instructions.
+   *
+   * @param exposedPorts the list of ports numbers/ranges to expose
+   * @return a string containing an EXPOSE instruction for each of the entries
+   */
+  @VisibleForTesting
+  static String makeExposeItems(List<String> exposedPorts) {
+    return String.join(
+        System.getProperty("line.separator"),
+        exposedPorts.stream().map(port -> "EXPOSE " + port).collect(Collectors.toList()));
+  }
+
   private final SourceFilesConfiguration sourceFilesConfiguration;
 
   @Nullable private String baseImage;
   private List<String> jvmFlags = Collections.emptyList();
   private String mainClass = "";
   private List<String> javaArguments = Collections.emptyList();
+  private List<String> exposedPorts = Collections.emptyList();
 
   public DockerContextGenerator(SourceFilesConfiguration sourceFilesConfiguration) {
     this.sourceFilesConfiguration = sourceFilesConfiguration;
@@ -102,6 +145,17 @@ public DockerContextGenerator setJavaArguments(List<String> javaArguments) {
     return this;
   }
 
+  /**
+   * Sets the exposed ports.
+   *
+   * @param exposedPorts the list of port numbers/port ranges to expose
+   * @return this
+   */
+  public DockerContextGenerator setExposedPorts(List<String> exposedPorts) {
+    this.exposedPorts = exposedPorts;
+    return this;
+  }
+
   /**
    * Creates the Docker context in {@code #targetDirectory}.
    *
@@ -159,38 +213,11 @@ String makeDockerfile() throws IOException {
             "@@DEPENDENCIES_PATH_ON_IMAGE@@", sourceFilesConfiguration.getDependenciesPathOnImage())
         .replace("@@RESOURCES_PATH_ON_IMAGE@@", sourceFilesConfiguration.getResourcesPathOnImage())
         .replace("@@CLASSES_PATH_ON_IMAGE@@", sourceFilesConfiguration.getClassesPathOnImage())
+        .replace("@@EXPOSE_INSTRUCTIONS@@", makeExposeItems(exposedPorts))
         .replace(
             "@@ENTRYPOINT@@",
             joinAsJsonArray(
                 EntrypointBuilder.makeEntrypoint(sourceFilesConfiguration, jvmFlags, mainClass)))
         .replace("@@CMD@@", joinAsJsonArray(javaArguments));
   }
-
-  /**
-   * Formats a list for the Dockerfile's ENTRYPOINT or CMD.
-   *
-   * @see <a
-   *     href="https://docs.docker.com/engine/reference/builder/#exec-form-entrypoint-example">https://docs.docker.com/engine/reference/builder/#exec-form-entrypoint-example</a>
-   * @param items the list of items to join into an array.
-   * @return a string in the format: ["item1","item2",...]
-   */
-  @VisibleForTesting
-  static String joinAsJsonArray(List<String> items) {
-    StringBuilder resultString = new StringBuilder("[");
-    boolean firstComponent = true;
-    for (String item : items) {
-      if (!firstComponent) {
-        resultString.append(',');
-      }
-
-      // Escapes quotes.
-      item = item.replaceAll("\"", Matcher.quoteReplacement("\\\""));
-
-      resultString.append('"').append(item).append('"');
-      firstComponent = false;
-    }
-    resultString.append(']');
-
-    return resultString.toString();
-  }
 }

jib-core/src/main/resources/DockerfileTemplate

@@ -4,5 +4,6 @@ COPY libs @@DEPENDENCIES_PATH_ON_IMAGE@@
 COPY resources @@RESOURCES_PATH_ON_IMAGE@@
 COPY classes @@CLASSES_PATH_ON_IMAGE@@
 
+@@EXPOSE_INSTRUCTIONS@@
 ENTRYPOINT @@ENTRYPOINT@@
 CMD @@CMD@@

jib-core/src/test/java/com/google/cloud/tools/jib/docker/DockerContextGeneratorTest.java

@@ -113,7 +113,7 @@ public void testGenerate() throws IOException, URISyntaxException {
   }
 
   @Test
-  public void testMakeDockerList() {
+  public void testJoinAsJsonArray() {
     Assert.assertEquals(
         "[\"java\",\"-cp\",\"/app/libs/*:/app/resources/:/app/classes/\",\"\"]",
         DockerContextGenerator.joinAsJsonArray(
@@ -128,19 +128,28 @@ public void testMakeDockerList() {
                 "AnotherMainClass")));
   }
 
+  @Test
+  public void testMakeExposeItems() {
+    Assert.assertEquals(
+        "EXPOSE 1000" + System.getProperty("line.separator") + "EXPOSE 2000-2010",
+        DockerContextGenerator.makeExposeItems(ImmutableList.of("1000", "2000-2010")));
+  }
+
   @Test
   public void testMakeDockerfile() throws IOException, URISyntaxException {
     String expectedBaseImage = "somebaseimage";
     List<String> expectedJvmFlags = Arrays.asList("-flag", "another\"Flag");
     String expectedMainClass = "SomeMainClass";
     List<String> expectedJavaArguments = Arrays.asList("arg1", "arg2");
+    List<String> exposedPorts = Arrays.asList("1000/tcp", "2000-2010/udp");
 
     String dockerfile =
         new DockerContextGenerator(mockSourceFilesConfiguration)
             .setBaseImage(expectedBaseImage)
             .setJvmFlags(expectedJvmFlags)
             .setMainClass(expectedMainClass)
             .setJavaArguments(expectedJavaArguments)
+            .setExposedPorts(exposedPorts)
             .makeDockerfile();
 
     Path sampleDockerfile = Paths.get(Resources.getResource("sampleDockerfile").toURI());

jib-core/src/test/resources/sampleDockerfile

@@ -4,5 +4,7 @@ COPY libs /app/libs/
 COPY resources /app/resources/
 COPY classes /app/classes/
 
+EXPOSE 1000/tcp
+EXPOSE 2000-2010/udp
 ENTRYPOINT ["java","-flag","another\"Flag","-cp","/app/libs/*:/app/resources/:/app/classes/","SomeMainClass"]
 CMD ["arg1","arg2"]