-
Notifications
You must be signed in to change notification settings - Fork 9
/
terminal.go
100 lines (84 loc) · 3.04 KB
/
terminal.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
96
97
98
99
100
// Copyright 2022 Namespace Labs Inc; All rights reserved.
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
package kubernetes
import (
"context"
"fmt"
"os"
corev1 "k8s.io/api/core/v1"
k8serrors "k8s.io/apimachinery/pkg/api/errors"
k8srt "k8s.io/apimachinery/pkg/runtime"
"k8s.io/client-go/kubernetes"
"k8s.io/client-go/kubernetes/scheme"
"k8s.io/client-go/tools/remotecommand"
"namespacelabs.dev/foundation/framework/kubernetes/kubedef"
"namespacelabs.dev/foundation/internal/console"
"namespacelabs.dev/foundation/internal/console/termios"
"namespacelabs.dev/foundation/internal/fnerrors"
"namespacelabs.dev/foundation/internal/runtime"
"namespacelabs.dev/foundation/internal/runtime/kubernetes/client"
)
func (r *Cluster) attachTerminal(ctx context.Context, cli *kubernetes.Clientset, opaque *kubedef.ContainerPodReference, rio runtime.TerminalIO) error {
return r.lowLevelAttachTerm(ctx, cli, opaque.Namespace, opaque.PodName, rio, "attach", &corev1.PodAttachOptions{
Container: opaque.Container,
Stdin: true,
Stdout: true,
Stderr: true,
TTY: rio.TTY,
})
}
func (r *Cluster) lowLevelAttachTerm(ctx context.Context, cli *kubernetes.Clientset, ns, podname string, rio runtime.TerminalIO, subresource string, params k8srt.Object) error {
config, restClient, err := client.MakeGroupVersionBasedClientAndConfig(ctx, r.RESTConfig(), corev1.SchemeGroupVersion)
if err != nil {
return err
}
req := restClient.Post().
Resource("pods").
Name(podname).
Namespace(ns).
SubResource(subresource).
VersionedParams(params, scheme.ParameterCodec)
exec, err := remotecommand.NewSPDYExecutor(config, "POST", req.URL())
if err != nil {
return fnerrors.InvocationError("kubernetes", "creating executor failed: %w", err)
}
opts := remotecommand.StreamOptions{
Stdin: rio.Stdin,
Stdout: rio.Stdout,
Stderr: rio.Stderr,
Tty: rio.TTY,
TerminalSizeQueue: nil, // Set below.
}
if rio.ResizeQueue != nil {
opts.TerminalSizeQueue = readResizeQueue{rio.ResizeQueue}
}
// XXX move this check somewhere else.
if rio.Stdin == os.Stdin {
restore, err := termios.MakeRaw(os.Stdin.Fd())
if err != nil {
fmt.Fprintf(console.Warnings(ctx), "kubernetes/terminal: failed to set stdin to raw: %v\n", err)
} else {
defer func() {
_ = restore()
}()
}
done := console.EnterInputMode(ctx, "Press enter, if you don't see any output.\n")
defer done()
}
if err := exec.Stream(opts); err != nil {
if s, ok := err.(*k8serrors.StatusError); ok {
return fnerrors.InvocationError("kubernetes", "%+v: failed to attach terminal: %w", s.ErrStatus, err)
}
return fnerrors.InvocationError("kubernetes", "stream failed: %w", err)
}
return nil
}
type readResizeQueue struct{ ch chan termios.WinSize }
func (r readResizeQueue) Next() *remotecommand.TerminalSize {
v, ok := <-r.ch
if !ok {
return nil // Channel closed.
}
return &remotecommand.TerminalSize{Width: v.Width, Height: v.Height}
}