From a8844f85855a031888f1f65723f3f7b0c8cde0fa Mon Sep 17 00:00:00 2001
From: Laura Brehm <laurabrehm@hey.com>
Date: Tue, 30 Jul 2024 11:38:30 +0100
Subject: [PATCH] WIP: add configurable detach delay to logs

Signed-off-by: Laura Brehm <laurabrehm@hey.com>
---
 cli/command/container/logs.go                | 46 ++++++++++++++++++++
 docs/reference/commandline/container_logs.md |  1 +
 docs/reference/commandline/logs.md           |  1 +
 3 files changed, 48 insertions(+)

diff --git a/cli/command/container/logs.go b/cli/command/container/logs.go
index 3d536f721b73..4ffe61dec309 100644
--- a/cli/command/container/logs.go
+++ b/cli/command/container/logs.go
@@ -3,11 +3,14 @@ package container
 import (
 	"context"
 	"io"
+	"strconv"
+	"time"
 
 	"github.com/docker/cli/cli"
 	"github.com/docker/cli/cli/command"
 	"github.com/docker/cli/cli/command/completion"
 	"github.com/docker/docker/api/types/container"
+	"github.com/docker/docker/api/types/events"
 	"github.com/docker/docker/pkg/stdcopy"
 	"github.com/spf13/cobra"
 )
@@ -20,6 +23,8 @@ type logsOptions struct {
 	details    bool
 	tail       string
 
+	detachDelay int
+
 	container string
 }
 
@@ -49,6 +54,7 @@ func NewLogsCommand(dockerCli command.Cli) *cobra.Command {
 	flags.BoolVarP(&opts.timestamps, "timestamps", "t", false, "Show timestamps")
 	flags.BoolVar(&opts.details, "details", false, "Show extra details provided to logs")
 	flags.StringVarP(&opts.tail, "tail", "n", "all", "Number of lines to show from the end of the logs")
+	flags.IntVarP(&opts.detachDelay, "delay", "d", 0, "Number of seconds to wait for container restart before exiting")
 	return cmd
 }
 
@@ -58,6 +64,45 @@ func runLogs(ctx context.Context, dockerCli command.Cli, opts *logsOptions) erro
 		return err
 	}
 
+	since := opts.since
+	for {
+		restarting := make(chan events.Message)
+		if opts.detachDelay != 0 {
+			go func() {
+				eventsCtx, eventsCtxCancel := context.WithCancel(ctx)
+				eventC, eventErrs := dockerCli.Client().Events(eventsCtx, events.ListOptions{})
+				defer eventsCtxCancel()
+				for {
+					select {
+					case event := <-eventC:
+						if event.Action == events.ActionRestart && event.Actor.ID == c.ID {
+							restarting <- event
+							return
+						}
+					case <-eventErrs:
+						return
+					}
+				}
+			}()
+		}
+
+		opts.since = since
+		err = streamLogs(ctx, dockerCli, opts, c)
+
+		if opts.detachDelay == 0 {
+			return err
+		}
+
+		select {
+		case restartEvent := <-restarting:
+			since = strconv.FormatInt(restartEvent.Time, 10)
+		case <-time.After(time.Duration(opts.detachDelay) * time.Second):
+			return err
+		}
+	}
+}
+
+func streamLogs(ctx context.Context, dockerCli command.Cli, opts *logsOptions, c container.InspectResponse) error {
 	responseBody, err := dockerCli.Client().ContainerLogs(ctx, c.ID, container.LogsOptions{
 		ShowStdout: true,
 		ShowStderr: true,
@@ -78,5 +123,6 @@ func runLogs(ctx context.Context, dockerCli command.Cli, opts *logsOptions) erro
 	} else {
 		_, err = stdcopy.StdCopy(dockerCli.Out(), dockerCli.Err(), responseBody)
 	}
+
 	return err
 }
diff --git a/docs/reference/commandline/container_logs.md b/docs/reference/commandline/container_logs.md
index ff8b92de50c2..a4f0151d4245 100644
--- a/docs/reference/commandline/container_logs.md
+++ b/docs/reference/commandline/container_logs.md
@@ -11,6 +11,7 @@ Fetch the logs of a container
 
 | Name                 | Type     | Default | Description                                                                                        |
 |:---------------------|:---------|:--------|:---------------------------------------------------------------------------------------------------|
+| `-d`, `--delay`      | `int`    | `0`     | Number of seconds to wait for container restart before exiting                                     |
 | `--details`          | `bool`   |         | Show extra details provided to logs                                                                |
 | `-f`, `--follow`     | `bool`   |         | Follow log output                                                                                  |
 | `--since`            | `string` |         | Show logs since timestamp (e.g. `2013-01-02T13:23:37Z`) or relative (e.g. `42m` for 42 minutes)    |
diff --git a/docs/reference/commandline/logs.md b/docs/reference/commandline/logs.md
index 56e7671c5c2a..47f777bbac4c 100644
--- a/docs/reference/commandline/logs.md
+++ b/docs/reference/commandline/logs.md
@@ -11,6 +11,7 @@ Fetch the logs of a container
 
 | Name                 | Type     | Default | Description                                                                                        |
 |:---------------------|:---------|:--------|:---------------------------------------------------------------------------------------------------|
+| `-d`, `--delay`      | `int`    | `0`     | Number of seconds to wait for container restart before exiting                                     |
 | `--details`          | `bool`   |         | Show extra details provided to logs                                                                |
 | `-f`, `--follow`     | `bool`   |         | Follow log output                                                                                  |
 | `--since`            | `string` |         | Show logs since timestamp (e.g. `2013-01-02T13:23:37Z`) or relative (e.g. `42m` for 42 minutes)    |