diff --git a/progress/writer.go b/progress/writer.go index 9803f382..9f386d19 100644 --- a/progress/writer.go +++ b/progress/writer.go @@ -59,6 +59,19 @@ type writer struct { mtx *sync.RWMutex } +type writerKey struct{} + +// WithContextWriter adds the writer to the context +func WithContextWriter(ctx context.Context, writer Writer) context.Context { + return context.WithValue(ctx, writerKey{}, writer) +} + +// ContextWriter returns the writer from the context +func ContextWriter(ctx context.Context) Writer { + s, _ := ctx.Value(writerKey{}).(Writer) + return s +} + // NewWriter returns a new multi-progress writer func NewWriter(out io.Writer) Writer { return &writer{