Added container labels to ContainerReport interface

internal/actions/update.go

@@ -32,7 +32,7 @@ func Update(client container.Client, params types.UpdateParams) (types.Report, e
 
 	staleCheckFailed := 0
 
-	for i, targetContainer := range containers {
+	for _, targetContainer := range containers {
 		stale, newestImage, err := client.IsContainerStale(targetContainer)
 		shouldUpdate := stale && !params.NoRestart && !params.MonitorOnly && !targetContainer.IsMonitorOnly()
 		if err == nil && shouldUpdate {
@@ -57,7 +57,7 @@ func Update(client container.Client, params types.UpdateParams) (types.Report, e
 		} else {
 			progress.AddScanned(targetContainer, newestImage)
 		}
-		containers[i].SetStale(stale)
+		targetContainer.SetStale(stale)
 
 		if stale {
 			staleCount++

pkg/container/container.go

@@ -76,6 +76,11 @@ func (c Container) IsRestarting() bool {
 	return c.containerInfo.State.Restarting
 }
 
+// Labels returns a map of all the labels present on the container
+func (c Container) Labels() map[string]string {
+	return c.containerInfo.Config.Labels
+}
+
 // Name returns the Docker container name.
 func (c Container) Name() string {
 	return c.containerInfo.Name

pkg/session/container_status.go

@@ -26,6 +26,7 @@ type ContainerStatus struct {
 	imageName     string
 	error
 	state  State
+	labels map[string]string
 }
 
 // ID returns the container ID
@@ -61,6 +62,11 @@ func (u *ContainerStatus) Error() string {
 	return u.error.Error()
 }
 
+// Labels returns a map of all labels present on the container
+func (u *ContainerStatus) Labels() map[string]string {
+	return u.labels
+}
+
 // State returns the current State that the container is in
 func (u *ContainerStatus) State() string {
 	switch u.state {

pkg/session/progress.go

@@ -13,6 +13,7 @@ func UpdateFromContainer(cont types.Container, newImage types.ImageID, state Sta
 		containerID:   cont.ID(),
 		containerName: cont.Name(),
 		imageName:     cont.ImageName(),
+		labels:        cont.Labels(),
 		oldImage:      cont.SafeImageID(),
 		newImage:      newImage,
 		state:         state,

pkg/types/container.go

@@ -66,6 +66,7 @@ type Container interface {
 	GetLifecyclePostUpdateCommand() string
 	VerifyConfiguration() error
 	SetStale(bool)
+	Labels() map[string]string
 	IsStale() bool
 	IsNoPull() bool
 	SetLinkedToRestarting(bool)

pkg/types/report.go

@@ -20,4 +20,5 @@ type ContainerReport interface {
 	ImageName() string
 	Error() string
 	State() string
+	Labels() map[string]string
 }