forked from toolshed/abra
100 lines
2.7 KiB
Go
100 lines
2.7 KiB
Go
package container // https://github.com/docker/cli/blob/master/cli/command/container/tty.go
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
gosignal "os/signal"
|
|
"runtime"
|
|
"time"
|
|
|
|
"coopcloud.tech/abra/pkg/i18n"
|
|
"coopcloud.tech/abra/pkg/log"
|
|
"github.com/docker/cli/cli/command"
|
|
"github.com/docker/docker/api/types/container"
|
|
"github.com/docker/docker/client"
|
|
apiclient "github.com/docker/docker/client"
|
|
"github.com/moby/sys/signal"
|
|
)
|
|
|
|
// resizeTtyTo resizes tty to specific height and width
|
|
func resizeTtyTo(ctx context.Context, client client.ContainerAPIClient, id string, height, width uint, isExec bool) error {
|
|
if height == 0 && width == 0 {
|
|
return nil
|
|
}
|
|
|
|
options := container.ResizeOptions{
|
|
Height: height,
|
|
Width: width,
|
|
}
|
|
|
|
var err error
|
|
if isExec {
|
|
err = client.ContainerExecResize(ctx, id, options)
|
|
} else {
|
|
err = client.ContainerResize(ctx, id, options)
|
|
}
|
|
|
|
if err != nil {
|
|
log.Debug(fmt.Sprintf("%s\r", i18n.G("error resize: %s", err)))
|
|
}
|
|
return err
|
|
}
|
|
|
|
// resizeTty is to resize the tty with cli out's tty size
|
|
func resizeTty(ctx context.Context, client *apiclient.Client, cli command.Cli, id string, isExec bool) error {
|
|
height, width := cli.Out().GetTtySize()
|
|
return resizeTtyTo(ctx, client, id, height, width, isExec)
|
|
}
|
|
|
|
// initTtySize is to init the tty's size to the same as the window, if there is an error, it will retry 5 times.
|
|
func initTtySize(ctx context.Context, client *apiclient.Client, cli command.Cli, id string, isExec bool, resizeTtyFunc func(ctx context.Context, client *apiclient.Client, cli command.Cli, id string, isExec bool) error) {
|
|
rttyFunc := resizeTtyFunc
|
|
if rttyFunc == nil {
|
|
rttyFunc = resizeTty
|
|
}
|
|
if err := rttyFunc(ctx, client, cli, id, isExec); err != nil {
|
|
go func() {
|
|
var err error
|
|
for retry := 0; retry < 5; retry++ {
|
|
time.Sleep(10 * time.Millisecond)
|
|
if err = rttyFunc(ctx, client, cli, id, isExec); err == nil {
|
|
break
|
|
}
|
|
}
|
|
if err != nil {
|
|
fmt.Fprintln(cli.Err(), i18n.G("failed to resize tty, using default size"))
|
|
}
|
|
}()
|
|
}
|
|
}
|
|
|
|
// MonitorTtySize updates the container tty size when the terminal tty changes size
|
|
func MonitorTtySize(ctx context.Context, client *apiclient.Client, cli command.Cli, id string, isExec bool) error {
|
|
initTtySize(ctx, client, cli, id, isExec, resizeTty)
|
|
if runtime.GOOS == "windows" {
|
|
go func() {
|
|
prevH, prevW := cli.Out().GetTtySize()
|
|
for {
|
|
time.Sleep(time.Millisecond * 250)
|
|
h, w := cli.Out().GetTtySize()
|
|
|
|
if prevW != w || prevH != h {
|
|
resizeTty(ctx, client, cli, id, isExec)
|
|
}
|
|
prevH = h
|
|
prevW = w
|
|
}
|
|
}()
|
|
} else {
|
|
sigchan := make(chan os.Signal, 1)
|
|
gosignal.Notify(sigchan, signal.SIGWINCH)
|
|
go func() {
|
|
for range sigchan {
|
|
resizeTty(ctx, client, cli, id, isExec)
|
|
}
|
|
}()
|
|
}
|
|
return nil
|
|
}
|