add auto init/unseal for vault

pkg/kubernetes_client/pod.go

@@ -0,0 +1,80 @@
+package kubernetes_client
+
+import (
+	"bytes"
+	"context"
+	"fmt"
+	"kube-forge/pkg/config"
+	"net/http"
+	"strings"
+	"time"
+
+	corev1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/client-go/kubernetes"
+	"k8s.io/client-go/tools/remotecommand"
+)
+
+func ExecuteCommandInPodContainer(command string, namespace string, podName string, container string) (string, error) {
+	clientset, err := kubernetes.NewForConfig(config.GetKubernetesConfig())
+	if err != nil {
+		panic(err.Error())
+	}
+	commandArray := strings.Split(command, " ")
+
+	execRequest := clientset.CoreV1().RESTClient().Post().
+		Resource("pods").
+		Name(podName).
+		Namespace(namespace).
+		SubResource("exec").
+		Param("container", container).
+		Param("stderr", "true").
+		Param("stdout", "true")
+	for _, com := range commandArray {
+		execRequest = execRequest.Param("command", com)
+	}
+
+	stderr := bytes.NewBufferString("")
+	stdout := bytes.NewBufferString("")
+
+	streamOptions := remotecommand.StreamOptions{
+		Stdout: stdout,
+		Stderr: stderr,
+		Tty:    false,
+	}
+
+	exec, err := remotecommand.NewSPDYExecutor(config.GetKubernetesConfig(), http.MethodPost, execRequest.URL())
+	if err != nil {
+		fmt.Println(err.Error())
+	}
+
+	err = exec.StreamWithContext(context.Background(), streamOptions)
+	if err != nil {
+		if stderr.Len() == 0 {
+			panic(err)
+		}
+		outputErr := stderr.String()
+		return outputErr, err
+	}
+	output := stdout.String()
+	return output, nil
+}
+
+func GetPodByName(podName string, podNamespace string) (*corev1.Pod, error) {
+	clientset, err := kubernetes.NewForConfig(config.GetKubernetesConfig())
+	if err != nil {
+		panic(err.Error())
+	}
+	for i := 15; i > 0; i-- {
+		time.Sleep(time.Second * 1)
+
+		pod, err := clientset.CoreV1().
+			Pods(podNamespace).
+			Get(context.Background(), podName, metav1.GetOptions{})
+		if err != nil || pod.Status.Phase != "Running" {
+			continue
+		}
+		return pod, nil
+	}
+	return nil, err
+}