This commit is contained in:
Nate Jones 2015-11-07 18:32:49 -08:00
parent b86486fdbc
commit f8106ccec6

View file

@ -30,7 +30,6 @@ type ImagesCommand struct {
var imagesCommand ImagesCommand var imagesCommand ImagesCommand
func (x *ImagesCommand) Execute(args []string) error { func (x *ImagesCommand) Execute(args []string) error {
var images *[]Image var images *[]Image
@ -83,7 +82,7 @@ func (x *ImagesCommand) Execute(args []string) error {
images = &ims images = &ims
} }
if imagesCommand.Tree || imagesCommand.Dot { if imagesCommand.Tree || imagesCommand.Dot {
var startImage *Image var startImage *Image
if len(args) > 0 { if len(args) > 0 {
@ -118,7 +117,7 @@ func (x *ImagesCommand) Execute(args []string) error {
return fmt.Errorf("Unable to find image %s = %s.", startImageArg, startImageRepo) return fmt.Errorf("Unable to find image %s = %s.", startImageArg, startImageRepo)
} }
} }
// select the start image of the tree // select the start image of the tree
var roots []Image var roots []Image
if startImage == nil { if startImage == nil {
@ -127,23 +126,23 @@ func (x *ImagesCommand) Execute(args []string) error {
startImage.ParentId = "" startImage.ParentId = ""
roots = []Image{*startImage} roots = []Image{*startImage}
} }
// build helper map (image -> children) // build helper map (image -> children)
var imagesByParent = make(map[string][]Image) var imagesByParent = make(map[string][]Image)
imagesByParent = collectChildren(images); imagesByParent = collectChildren(images)
// image ids truncate // image ids truncate
// initialize image informations // initialize image informations
// filter images // filter images
if imagesCommand.OnlyLabelled{ if imagesCommand.OnlyLabelled {
*images, imagesByParent = filterImages(images, &imagesByParent) *images, imagesByParent = filterImages(images, &imagesByParent)
} }
var buffer bytes.Buffer var buffer bytes.Buffer
if imagesCommand.Tree { if imagesCommand.Tree {
jsonToText(&buffer, imagesCommand.NoTruncate, roots, imagesByParent, "") jsonToText(&buffer, imagesCommand.NoTruncate, roots, imagesByParent, "")
} }
if imagesCommand.Dot { if imagesCommand.Dot {
buffer.WriteString("digraph docker {\n") buffer.WriteString("digraph docker {\n")
@ -151,7 +150,7 @@ func (x *ImagesCommand) Execute(args []string) error {
buffer.WriteString(" base [style=invisible]\n}\n") buffer.WriteString(" base [style=invisible]\n}\n")
buffer.String() buffer.String()
} }
fmt.Print(buffer.String()) fmt.Print(buffer.String())
} else if imagesCommand.Short { } else if imagesCommand.Short {
fmt.Printf(jsonToShort(images)) fmt.Printf(jsonToShort(images))
@ -162,7 +161,6 @@ func (x *ImagesCommand) Execute(args []string) error {
return nil return nil
} }
func collectChildren(images *[]Image) map[string][]Image { func collectChildren(images *[]Image) map[string][]Image {
var imagesByParent = make(map[string][]Image) var imagesByParent = make(map[string][]Image)
for _, image := range *images { for _, image := range *images {
@ -172,11 +170,10 @@ func collectChildren(images *[]Image) map[string][]Image {
imagesByParent[image.ParentId] = []Image{image} imagesByParent[image.ParentId] = []Image{image}
} }
} }
return imagesByParent return imagesByParent
} }
func collectRoots(images *[]Image) []Image { func collectRoots(images *[]Image) []Image {
var roots []Image var roots []Image
for _, image := range *images { for _, image := range *images {
@ -184,29 +181,28 @@ func collectRoots(images *[]Image) []Image {
roots = append(roots, image) roots = append(roots, image)
} }
} }
return roots return roots
} }
func filterImages(images *[]Image, byParent *map[string][]Image) (filteredImages []Image, filteredChildren map[string][]Image) {
func filterImages (images *[]Image, byParent *map[string][]Image) (filteredImages []Image, filteredChildren map[string][]Image) { for i := 0; i < len(*images); i++ {
for i := 0; i<len(*images); i++ { // image is visible
// image is visible
// 1. it has a label // 1. it has a label
// 2. it is root // 2. it is root
// 3. it is a node // 3. it is a node
var visible bool = (*images)[i].RepoTags[0] != "<none>:<none>" || (*images)[i].ParentId == "" || len((*byParent)[(*images)[i].Id]) > 1 var visible bool = (*images)[i].RepoTags[0] != "<none>:<none>" || (*images)[i].ParentId == "" || len((*byParent)[(*images)[i].Id]) > 1
if visible { if visible {
filteredImages = append(filteredImages, (*images)[i]) filteredImages = append(filteredImages, (*images)[i])
} else { } else {
// change childs parent id // change childs parent id
// if items are filtered with only one child // if items are filtered with only one child
for j := 0; j<len(filteredImages); j++ { for j := 0; j < len(filteredImages); j++ {
if filteredImages[j].ParentId == (*images)[i].Id { if filteredImages[j].ParentId == (*images)[i].Id {
filteredImages[j].ParentId = (*images)[i].ParentId filteredImages[j].ParentId = (*images)[i].ParentId
} }
} }
for j := 0; j<len(*images); j++ { for j := 0; j < len(*images); j++ {
if (*images)[j].ParentId == (*images)[i].Id { if (*images)[j].ParentId == (*images)[i].Id {
(*images)[j].ParentId = (*images)[i].ParentId (*images)[j].ParentId = (*images)[i].ParentId
} }
@ -215,7 +211,7 @@ func filterImages (images *[]Image, byParent *map[string][]Image) (filteredImage
} }
filteredChildren = collectChildren(&filteredImages) filteredChildren = collectChildren(&filteredImages)
return filteredImages, filteredChildren return filteredImages, filteredChildren
} }
@ -245,7 +241,6 @@ func jsonToText(buffer *bytes.Buffer, noTrunc bool, images []Image, byParent map
} }
} }
func PrintTreeNode(buffer *bytes.Buffer, noTrunc bool, image Image, prefix string) { func PrintTreeNode(buffer *bytes.Buffer, noTrunc bool, image Image, prefix string) {
var imageID string var imageID string
if noTrunc { if noTrunc {
@ -280,12 +275,10 @@ func humanSize(raw int64) string {
return fmt.Sprintf("%.01f %s", rawFloat, sizes[ind]) return fmt.Sprintf("%.01f %s", rawFloat, sizes[ind])
} }
func truncate(id string) string { func truncate(id string) string {
return id[0:12] return id[0:12]
} }
func parseImagesJSON(rawJSON []byte) (*[]Image, error) { func parseImagesJSON(rawJSON []byte) (*[]Image, error) {
var images []Image var images []Image
@ -298,7 +291,6 @@ func parseImagesJSON(rawJSON []byte) (*[]Image, error) {
return &images, nil return &images, nil
} }
func imagesToDot(buffer *bytes.Buffer, images []Image, byParent map[string][]Image) { func imagesToDot(buffer *bytes.Buffer, images []Image, byParent map[string][]Image) {
for _, image := range images { for _, image := range images {
if image.ParentId == "" { if image.ParentId == "" {
@ -315,7 +307,6 @@ func imagesToDot(buffer *bytes.Buffer, images []Image, byParent map[string][]Ima
} }
} }
func jsonToShort(images *[]Image) string { func jsonToShort(images *[]Image) string {
var buffer bytes.Buffer var buffer bytes.Buffer
@ -347,7 +338,6 @@ func jsonToShort(images *[]Image) string {
return buffer.String() return buffer.String()
} }
func init() { func init() {
parser.AddCommand("images", parser.AddCommand("images",
"Visualize docker images.", "Visualize docker images.",