Merge pull request #1151 from brendandburns/kubectl

Add support for KubectlCopy.

Author: k8s-ci-robot

examples/src/main/java/io/kubernetes/client/examples/KubectlExample.java

@@ -13,6 +13,7 @@
 package io.kubernetes.client.examples;
 
 import static io.kubernetes.client.extended.kubectl.Kubectl.apiResources;
+import static io.kubernetes.client.extended.kubectl.Kubectl.copy;
 import static io.kubernetes.client.extended.kubectl.Kubectl.exec;
 import static io.kubernetes.client.extended.kubectl.Kubectl.label;
 import static io.kubernetes.client.extended.kubectl.Kubectl.log;
@@ -43,7 +44,7 @@
  * A Java equivalent for the kubectl command line tool. Not nearly as complete.
  *
  * <p>Easiest way to run this: mvn exec:java
- * -Dexec.mainClass="io.kubernetes.client.examples.KubectlExample"
+ * -Dexec.mainClass="io.kubernetes.client.examples.KubectlExample" -Dexec.args="log some-pod"
  *
  * <p>From inside $REPO_DIR/examples
  */
@@ -87,6 +88,37 @@ public static void main(String[] args)
     String name = null;
 
     switch (verb) {
+      case "cp":
+        String from = args[1];
+        String to = args[2];
+        if (from.indexOf(":") != -1) {
+          String[] parts = from.split(":");
+          name = parts[0];
+          from = parts[1];
+          copy(client)
+              .namespace(ns)
+              .name(name)
+              .container(cli.getOptionValue("c", ""))
+              .fromPod(from)
+              .to(to)
+              .execute();
+        } else if (to.indexOf(":") != -1) {
+          String[] parts = to.split(":");
+          name = parts[0];
+          to = parts[1];
+          copy(client)
+              .namespace(ns)
+              .name(name)
+              .container(cli.getOptionValue("c", ""))
+              .from(from)
+              .toPod(to)
+              .execute();
+        } else {
+          System.err.println("Missing pod name for copy.");
+          System.exit(-1);
+        }
+        System.out.println("Copied " + from + " -> " + to);
+        System.exit(0);
       case "portforward":
         name = args[1];
         KubectlPortForward forward = portforward(client).name(name).namespace(ns);

extended/src/main/java/io/kubernetes/client/extended/kubectl/Kubectl.java

@@ -23,6 +23,25 @@
  */
 public class Kubectl {
 
+  /**
+   * Equivalence for `kubectl cp`.
+   *
+   * @return the kubectl copy
+   */
+  public static KubectlCopy copy() {
+    return copy(Configuration.getDefaultApiClient());
+  }
+
+  /**
+   * Equivalence for `kubectl cp`.
+   *
+   * @param apiClient the api client instance
+   * @return the kubectl copy
+   */
+  public static KubectlCopy copy(ApiClient apiClient) {
+    return new KubectlCopy(apiClient);
+  }
+
   /**
    * Equivalence for `kubectl label`.
    *

extended/src/main/java/io/kubernetes/client/extended/kubectl/KubectlCopy.java

@@ -0,0 +1,96 @@
+/*
+Copyright 2020 The Kubernetes Authors.
+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 io.kubernetes.client.extended.kubectl;
+
+import static com.google.common.base.Strings.isNullOrEmpty;
+
+import io.kubernetes.client.Copy;
+import io.kubernetes.client.extended.kubectl.exception.KubectlException;
+import io.kubernetes.client.openapi.ApiClient;
+import io.kubernetes.client.openapi.ApiException;
+import io.kubernetes.client.openapi.models.V1Pod;
+import io.kubernetes.client.util.exception.CopyNotSupportedException;
+import java.io.IOException;
+import java.nio.file.Paths;
+
+public class KubectlCopy extends Kubectl.ResourceAndContainerBuilder<V1Pod, KubectlCopy>
+    implements Kubectl.Executable<Boolean> {
+  private String from;
+  private String to;
+  private Boolean toPod;
+  private boolean dir;
+
+  KubectlCopy(ApiClient client) {
+    super(client, V1Pod.class);
+    this.toPod = null;
+  }
+
+  public KubectlCopy directory() {
+    this.dir = true;
+    return this;
+  }
+
+  public KubectlCopy from(String file) {
+    this.from = file;
+    return this;
+  }
+
+  public KubectlCopy fromPod(String fileInPod) {
+    this.from = fileInPod;
+    this.toPod = false;
+    return this;
+  }
+
+  public KubectlCopy to(String file) {
+    this.to = file;
+    return this;
+  }
+
+  public KubectlCopy toPod(String fileInPod) {
+    this.to = fileInPod;
+    this.toPod = true;
+    return this;
+  }
+
+  @Override
+  public Boolean execute() throws KubectlException {
+    validate();
+    Copy cp = new Copy(apiClient);
+    try {
+      if (toPod) {
+        cp.copyFileToPod(namespace, name, container, Paths.get(from), Paths.get(to));
+      } else {
+        if (this.dir) {
+          cp.copyDirectoryFromPod(namespace, name, container, from, Paths.get(to));
+        } else {
+          cp.copyFileFromPod(namespace, name, container, from, Paths.get(to));
+        }
+      }
+      return true;
+    } catch (ApiException | IOException | CopyNotSupportedException ex) {
+      throw new KubectlException(ex);
+    }
+  }
+
+  private void validate() throws KubectlException {
+    if (toPod == null) {
+      throw new KubectlException("You must specify either from or to a pod");
+    }
+    if (isNullOrEmpty(from)) {
+      throw new KubectlException("From file missing");
+    }
+    if (isNullOrEmpty(to)) {
+      throw new KubectlException("To file missing");
+    }
+  }
+}