/
backend_exec.go
95 lines (78 loc) · 2.56 KB
/
backend_exec.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
package main
import (
"context"
"encoding/json"
"fmt"
"os"
"strings"
"github.com/docker/docker/api/types"
"github.com/docker/docker/api/types/backend"
"github.com/docker/docker/api/types/container"
corev1 "k8s.io/api/core/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/util/rand"
"k8s.io/client-go/kubernetes/scheme"
"k8s.io/client-go/tools/remotecommand"
)
func (b *Backend) ContainerExecCreate(name string, config *types.ExecConfig) (string, error) {
ctx := context.TODO()
fmt.Println(name)
json.NewEncoder(os.Stdout).Encode(config)
s := strings.Split(name, ".")
if len(s) != 3 {
return "", fmt.Errorf("invalid container name %q", name)
}
ns, podName, image := s[0], s[1], s[2]
if image == "buildx_buildkit_default" {
image = "moby/buildkit"
}
pod, err := b.client.CoreV1().Pods(ns).Get(ctx, podName, metav1.GetOptions{})
if err != nil {
return "", err
}
ec := corev1.EphemeralContainer{
EphemeralContainerCommon: corev1.EphemeralContainerCommon{
Name: fmt.Sprintf("levias-%s", rand.String(8)),
Image: image,
Args: config.Cmd,
},
}
pod.Spec.EphemeralContainers = append(pod.Spec.EphemeralContainers, ec)
out, err := b.client.CoreV1().Pods(ns).UpdateEphemeralContainers(ctx, podName, pod, metav1.UpdateOptions{})
if err != nil {
return "", err
}
return strings.Join([]string{out.Namespace, out.Name, ec.Name}, "."), nil
}
func (b *Backend) ContainerExecInspect(id string) (*backend.ExecInspect, error) {
return nil, ErrUnimplemented
}
func (b *Backend) ContainerExecResize(name string, height, width int) error { return ErrUnimplemented }
func (b *Backend) ContainerExecStart(ctx context.Context, name string, options container.ExecStartOptions) error {
s := strings.Split(name, ".")
if len(s) != 3 {
return fmt.Errorf("invalid container name %q", name)
}
ns, pod, container := s[0], s[1], s[2]
fmt.Println("$", ns, pod, container)
req := b.client.CoreV1().RESTClient().Post().Resource("pods").Name(pod).Namespace(ns).SubResource("exec")
req.VersionedParams(&corev1.PodExecOptions{
Container: container,
Stdin: options.Stdin != nil,
Stdout: options.Stdout != nil,
Stderr: options.Stderr != nil,
}, scheme.ParameterCodec)
exec, err := remotecommand.NewSPDYExecutor(b.config, "POST", req.URL())
if err != nil {
return err
}
if err = exec.StreamWithContext(ctx, remotecommand.StreamOptions{
Stdin: options.Stdin,
Stdout: options.Stdout,
Stderr: options.Stderr,
}); err != nil {
return err
}
return nil
}
func (b *Backend) ExecExists(name string) (bool, error) { return true, nil }