deploydot.go

package libbusyna

import (
	"fmt"
	"io/ioutil"
	"log"
	"os"
	"path/filepath"
	"regexp"
	"strings"
)

// Create a dot graphviz file with db data read form the provided channel.
func DeployDot(c <-chan CmdData, outputfile string) {
	fd, err := ioutil.TempFile(filepath.Dir(outputfile), "busyna.dot-")
	if err != nil {
		log.Fatal(err)
	}
	defer os.Remove(fd.Name())

	fd.WriteString("digraph {\n\trankdir=LR\n\n")
	i := 0
	found := map[string]bool{}
	fixre := regexp.MustCompile(`[^\\]"+`)
	for cmddata := range c {
		fd.WriteString(fmt.Sprintf("\tnode%d [ label=\"%d\" tooltip=\"%s\" ]\n", i, i, fixre.ReplaceAllString(cmddata.Cmd.Line, `\"`)))

		for dep := range cmddata.Deps {
			if strings.HasPrefix(dep, "/") {
				continue
			}
			if _, ok := found[dep]; !ok {
				fd.WriteString(fmt.Sprintf("\t\"%s\" [ shape=rectangle ]\n", dep))
				found[dep] = true
			}
			fd.WriteString(fmt.Sprintf("\t\"%s\" -> node%d\n", dep, i))
		}
		for target := range cmddata.Targets {
			if strings.HasPrefix(target, "/") {
				continue
			}
			if _, ok := found[target]; !ok {
				fd.WriteString(fmt.Sprintf("\t\"%s\" [ shape=rectangle ]\n", target))
				found[target] = true
			}
			fd.WriteString(fmt.Sprintf("\tnode%d -> \"%s\"\n", i, target))
		}
		i++
	}
	fd.WriteString("}\n")
	fd.Close()

	if err = os.Rename(fd.Name(), outputfile); err != nil {
		log.Fatal(err)
	}
}