dockviz/images.go
2014-04-21 21:04:38 -07:00

73 lines
1.6 KiB
Go

package main
import (
"encoding/json"
"fmt"
"io/ioutil"
"os"
"strings"
)
type Image struct {
Id string
ParentId string `json:",omitempty"`
RepoTags []string `json:",omitempty"`
VirtualSize int64
Size int64
Created int64
}
type ImagesCommand struct {
Dot bool `short:"d" long:"dot" description:"Show image information as Graphviz dot."`
Tree bool `short:"t" long:"tree" description:"Show image information as tree."`
}
var imagesCommand ImagesCommand
func (x *ImagesCommand) Execute(args []string) error {
// read in stdin
stdin, err := ioutil.ReadAll(os.Stdin)
if err != nil {
return fmt.Errorf("error reading all input", err)
}
var images []Image
err = json.Unmarshal(stdin, &images)
if err != nil {
return fmt.Errorf("Error reading JSON: ", err)
}
if imagesCommand.Dot {
fmt.Printf("digraph docker {\n")
for _, image := range images {
if image.ParentId == "" {
fmt.Printf(" base -> \"%s\" [style=invis]\n", truncate(image.Id))
} else {
fmt.Printf(" \"%s\" -> \"%s\"\n", truncate(image.ParentId), truncate(image.Id))
}
if image.RepoTags[0] != "<none>:<none>" {
fmt.Printf(" \"%s\" [label=\"%s\\n%s\",shape=box,fillcolor=\"paleturquoise\",style=\"filled,rounded\"];\n", truncate(image.Id), truncate(image.Id), strings.Join(image.RepoTags, "\\n"))
}
}
fmt.Printf(" base [style=invisible]\n}\n")
} else if imagesCommand.Tree {
fmt.Println("Tree output not implemented yet.")
}
return nil
}
func truncate(id string) string {
return id[0:12]
}
func init() {
parser.AddCommand("images",
"Visualize docker images.",
"",
&imagesCommand)
}