//+build go1.7

// Copyright (C) 2016-2017 Vladimir Bauer
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package mpb

import "context"

// WithContext cancellation via context.
// pancis, if called on stopped Progress instance, i.e after (*Progress).Stop()
// or nil context passed
func (p *Progress) WithContext(ctx context.Context) *Progress {
	if isClosed(p.done) {
		panic(ErrCallAfterStop)
	}
	if ctx == nil {
		panic("nil context")
	}
	conf := <-p.userConf
	conf.cancel = ctx.Done()
	p.userConf <- conf
	return p
}