feat: add --details flag to logs command

Signed-off-by: Ruihua Wen <[email protected]>

Author: SpiffyEight77

cmd/nerdctl/container/container_logs.go

@@ -53,6 +53,7 @@ The following containers are supported:
 	cmd.Flags().StringP("tail", "n", "all", "Number of lines to show from the end of the logs")
 	cmd.Flags().String("since", "", "Show logs since timestamp (e.g. 2013-01-02T13:23:37Z) or relative (e.g. 42m for 42 minutes)")
 	cmd.Flags().String("until", "", "Show logs before a timestamp (e.g. 2013-01-02T13:23:37Z) or relative (e.g. 42m for 42 minutes)")
+	cmd.Flags().Bool("details", false, "Show extra details provided to logs")
 	return cmd
 }
 
@@ -88,6 +89,10 @@ func logsOptions(cmd *cobra.Command) (types.ContainerLogsOptions, error) {
 	if err != nil {
 		return types.ContainerLogsOptions{}, err
 	}
+	details, err := cmd.Flags().GetBool("details")
+	if err != nil {
+		return types.ContainerLogsOptions{}, err
+	}
 	return types.ContainerLogsOptions{
 		Stdout:     cmd.OutOrStdout(),
 		Stderr:     cmd.OutOrStderr(),
@@ -97,6 +102,7 @@ func logsOptions(cmd *cobra.Command) (types.ContainerLogsOptions, error) {
 		Tail:       tail,
 		Since:      since,
 		Until:      until,
+		Details:    details,
 	}, nil
 }
 

cmd/nerdctl/container/container_logs_test.go

@@ -345,3 +345,35 @@ func TestNoneLoggerHasNoLogURI(t *testing.T) {
 	testCase.Expected = test.Expects(1, nil, nil)
 	testCase.Run(t)
 }
+
+func TestLogsWithDetails(t *testing.T) {
+	testCase := nerdtest.Setup()
+
+	testCase.Setup = func(data test.Data, helpers test.Helpers) {
+		helpers.Ensure("run", "-d", "--log-driver", "json-file",
+			"--log-opt", "max-size=10m",
+			"--log-opt", "max-file=3",
+			"--log-opt", "env=ENV",
+			"--env", "ENV=foo",
+			"--log-opt", "labels=LABEL",
+			"--label", "LABEL=bar",
+			"--name", data.Identifier(), testutil.CommonImage,
+			"sh", "-ec", "echo baz")
+	}
+
+	testCase.Cleanup = func(data test.Data, helpers test.Helpers) {
+		helpers.Anyhow("rm", "-f", data.Identifier())
+	}
+
+	testCase.Command = func(data test.Data, helpers test.Helpers) test.TestableCommand {
+		return helpers.Command("logs", "--details", data.Identifier())
+	}
+
+	testCase.Expected = test.Expects(0, nil, expect.All(
+		expect.Contains("ENV=foo"),
+		expect.Contains("LABEL=bar"),
+		expect.Contains("baz"),
+	))
+
+	testCase.Run(t)
+}

pkg/api/types/container_types.go

@@ -384,6 +384,8 @@ type ContainerLogsOptions struct {
 	Since string
 	// Show logs before a timestamp (e.g., 2013-01-02T13:23:37Z) or relative (e.g., 42m for 42 minutes).
 	Until string
+	// Details specifies whether to show extra details provided to logs
+	Details bool
 }
 
 // ContainerWaitOptions specifies options for `nerdctl (container) wait`.

pkg/cmd/container/logs.go

@@ -18,9 +18,11 @@ package container
 
 import (
 	"context"
+	"encoding/json"
 	"fmt"
 	"os"
 	"os/signal"
+	"strings"
 	"syscall"
 
 	containerd "github.com/containerd/containerd/v2/client"
@@ -102,6 +104,44 @@ func Logs(ctx context.Context, client *containerd.Client, container string, opti
 				}
 			}
 
+			detailPrefix := ""
+			if options.Details {
+				if logConfigJSON, ok := l["nerdctl/log-config"]; ok {
+					type LogConfig struct {
+						Opts map[string]string `json:"opts"`
+					}
+
+					e, err := getContainerEnvs(ctx, found.Container)
+					if err != nil {
+						return err
+					}
+
+					var logConfig LogConfig
+					if err := json.Unmarshal([]byte(logConfigJSON), &logConfig); err == nil {
+						var optPairs []string
+
+						for k, v := range logConfig.Opts {
+							lowerKey := strings.ToLower(k)
+							if lowerKey == "env" {
+								if env, ok := e[v]; ok {
+									optPairs = append(optPairs, fmt.Sprintf("%s=%s", v, env))
+								}
+							}
+
+							if lowerKey == "labels" {
+								if label, ok := l[v]; ok {
+									optPairs = append(optPairs, fmt.Sprintf("%s=%s", v, label))
+								}
+							}
+						}
+
+						if len(optPairs) > 0 {
+							detailPrefix = strings.Join(optPairs, ",")
+						}
+					}
+				}
+			}
+
 			logViewOpts := logging.LogViewOptions{
 				ContainerID:       found.Container.ID(),
 				Namespace:         l[labels.Namespace],
@@ -112,6 +152,8 @@ func Logs(ctx context.Context, client *containerd.Client, container string, opti
 				Tail:              options.Tail,
 				Since:             options.Since,
 				Until:             options.Until,
+				Details:           options.Details,
+				DetailPrefix:      detailPrefix,
 			}
 			logViewer, err := logging.InitContainerLogViewer(l, logViewOpts, stopChannel, options.GOptions.Experimental)
 			if err != nil {
@@ -146,3 +188,25 @@ func getLogPath(ctx context.Context, container containerd.Container) (string, er
 
 	return meta.LogPath, nil
 }
+
+func getContainerEnvs(ctx context.Context, container containerd.Container) (map[string]string, error) {
+	envMap := make(map[string]string)
+
+	spec, err := container.Spec(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	if spec.Process == nil {
+		return envMap, nil
+	}
+
+	for _, env := range spec.Process.Env {
+		parts := strings.SplitN(env, "=", 2)
+		if len(parts) == 2 {
+			envMap[parts[0]] = parts[1]
+		}
+	}
+
+	return envMap, nil
+}

pkg/logging/detail_writer.go

@@ -0,0 +1,42 @@
+/*
+   Copyright The containerd Authors.
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+*/
+
+package logging
+
+import "io"
+
+type DetailWriter struct {
+	w      io.Writer
+	prefix string
+}
+
+func NewDetailWriter(w io.Writer, prefix string) io.Writer {
+	return &DetailWriter{
+		w:      w,
+		prefix: prefix,
+	}
+}
+
+func (dw *DetailWriter) Write(p []byte) (n int, err error) {
+	if len(p) > 0 {
+		if _, err = dw.w.Write([]byte(dw.prefix)); err != nil {
+			return 0, err
+		}
+
+		return dw.w.Write(p)
+	}
+	return 0, nil
+}

pkg/logging/fluentd_logger.go

@@ -62,6 +62,8 @@ var FluentdLogOpts = []string{
 	fluentdAsyncReconnectInterval,
 	fluentRequestAck,
 	Tag,
+	Env,
+	Labels,
 }
 
 const (

pkg/logging/journald_logger.go

@@ -43,6 +43,8 @@ import (
 
 var JournalDriverLogOpts = []string{
 	Tag,
+	Env,
+	Labels,
 }
 
 func JournalLogOptsValidate(logOptMap map[string]string) error {

pkg/logging/json_logger.go

@@ -42,6 +42,8 @@ var JSONDriverLogOpts = []string{
 	LogPath,
 	MaxSize,
 	MaxFile,
+	Env,
+	Labels,
 }
 
 type JSONLogger struct {

pkg/logging/log_viewer.go

@@ -82,6 +82,12 @@ type LogViewOptions struct {
 	// Start/end timestampts to filter logs by.
 	Since string
 	Until string
+
+	// Details enables showing extra details(env and label) in logs.
+	Details bool
+
+	// DetailPrefix is the prefix added when Details is enabled.
+	DetailPrefix string
 }
 
 func (lvo *LogViewOptions) Validate() error {
@@ -150,6 +156,15 @@ func InitContainerLogViewer(containerLabels map[string]string, lvopts LogViewOpt
 
 // Prints all logs for this LogViewer's containers to the provided io.Writers.
 func (lv *ContainerLogViewer) PrintLogsTo(stdout, stderr io.Writer) error {
+	if lv.logViewingOptions.Details {
+		prefix := " "
+		if lv.logViewingOptions.DetailPrefix != "" {
+			prefix = lv.logViewingOptions.DetailPrefix + " "
+		}
+
+		stdout = NewDetailWriter(stdout, prefix)
+		stderr = NewDetailWriter(stderr, prefix)
+	}
 	viewerFunc, err := getLogViewer(lv.loggingConfig.Driver)
 	if err != nil {
 		return err

pkg/logging/logging.go

@@ -47,6 +47,8 @@ const (
 	MaxSize    = "max-size"
 	MaxFile    = "max-file"
 	Tag        = "tag"
+	Env        = "env"
+	Labels     = "labels"
 )
 
 type Driver interface {

pkg/logging/syslog_logger.go

@@ -56,6 +56,8 @@ var syslogOpts = []string{
 	syslogTLSSkipVerify,
 	syslogFormat,
 	Tag,
+	Env,
+	Labels,
 }
 
 var syslogFacilities = map[string]syslog.Priority{