func (e Env) runFleetCmdInternal(getOutput bool, args []string) (string, string, error) {
logs.WithF(e.fields).WithField("command", strings.Join(args, " ")).Debug("Running command on fleet")
if e.config.Fleet.Endpoint == "" {
return "", "", errors.New("Cannot find fleet.endpoint env config to call fleetctl")
}
envs := map[string]string{}
envs[FLEETCTL_ENDPOINT] = e.config.Fleet.Endpoint
if e.config.Fleet.Username != "" {
envs[FLEETCTL_SSH_USERNAME] = e.config.Fleet.Username
}
envs[FLEETCTL_STRICT_HOST_KEY_CHECKING] = fmt.Sprintf("%t", e.config.Fleet.Strict_host_key_checking)
envs[FLEETCTL_SUDO] = fmt.Sprintf("%t", e.config.Fleet.Sudo)
args = append([]string{"fleetctl"}, args...)
for key, val := range envs {
args = append([]string{key + "='" + val + "'"}, args...)
}
var stdout string
var stderr string
var err error
if getOutput {
stdout, stderr, err = common.ExecCmdGetStdoutAndStderr("bash", "-c", strings.Join(args, " "))
} else {
err = common.ExecCmd("bash", "-c", strings.Join(args, " "))
}
return stdout, stderr, err
}
func (p *Pod) Graph() error {
logs.WithF(p.fields).Info("Graphing")
os.MkdirAll(p.target, 0777)
var buffer bytes.Buffer
buffer.WriteString("digraph {\n")
buffer.WriteString(" {\n")
buffer.WriteString(" ")
buffer.WriteString("\"")
buffer.WriteString(p.manifest.Name.TinyNameId())
buffer.WriteString("\"")
buffer.WriteString(" [style=filled, fillcolor=yellow, shape=box]\n")
buffer.WriteString(" }\n")
for _, e := range p.manifest.Pod.Apps {
for _, d := range e.Dependencies {
buffer.WriteString(" ")
buffer.WriteString("\"")
buffer.WriteString(d.TinyNameId())
buffer.WriteString("\"")
buffer.WriteString(" -> ")
buffer.WriteString("\"")
buffer.WriteString(p.manifest.Name.TinyNameId())
buffer.WriteString("\"")
buffer.WriteString("\n")
}
}
buffer.WriteString("}\n")
if err := ioutil.WriteFile(p.target+pathGraphDot, buffer.Bytes(), 0644); err != nil {
return errs.WithEF(err, p.fields.WithField("file", p.target+pathGraphDot), "Failed to write file")
}
if _, _, err := common.ExecCmdGetStdoutAndStderr("dot", "-V"); err == nil {
if std, stderr, err := common.ExecCmdGetStdoutAndStderr("dot", "-Tpng", p.target+pathGraphDot, "-o", p.target+pathGraphPng); err != nil {
return errs.WithEF(err, p.fields.WithField("stdout", std).WithField("stderr", stderr), "Failed to create graph image")
}
}
return nil
}
func (aci *Aci) zipAci() error {
if _, err := os.Stat(aci.target + pathImageGzAci); err == nil {
return nil
}
if stdout, stderr, err := common.ExecCmdGetStdoutAndStderr("gzip", "-k", aci.target+pathImageAci); err != nil {
return errs.WithEF(err, aci.fields.WithField("path", aci.target+pathImageAci).WithField("stdout", stdout).WithField("stderr", stderr), "Failed to zip aci")
}
if err := common.ExecCmd("mv", aci.target+pathImageAci+".gz", aci.target+pathImageGzAci); err != nil {
return errs.WithEF(err, aci.fields.WithField("from", aci.target+pathImageAci+".gz").
WithField("to", aci.target+pathImageGzAci), "Failed to rename zip aci")
}
return nil
}
func (aci *Aci) Graph() error {
defer aci.giveBackUserRightsToTarget()
logs.WithF(aci.fields).Debug("Graphing")
os.MkdirAll(aci.target, 0777)
var buffer bytes.Buffer
buffer.WriteString("digraph {\n")
for _, dep := range aci.manifest.Aci.Dependencies {
buffer.WriteString(" ")
buffer.WriteString("\"")
buffer.WriteString(dep.TinyNameId())
buffer.WriteString("\"")
buffer.WriteString(" -> ")
buffer.WriteString("\"")
buffer.WriteString(aci.manifest.NameAndVersion.TinyNameId())
buffer.WriteString("\"")
buffer.WriteString("\n")
}
for _, dep := range aci.manifest.Builder.Dependencies {
buffer.WriteString(" ")
buffer.WriteString("\"")
buffer.WriteString(dep.TinyNameId())
buffer.WriteString("\"")
buffer.WriteString(" -> ")
buffer.WriteString("\"")
buffer.WriteString(aci.manifest.NameAndVersion.TinyNameId())
buffer.WriteString("\"")
buffer.WriteString("[color=red,penwidth=2.0]")
buffer.WriteString("\n")
}
for _, dep := range aci.manifest.Tester.Builder.Dependencies {
buffer.WriteString(" ")
buffer.WriteString("\"")
buffer.WriteString(dep.TinyNameId())
buffer.WriteString("\"")
buffer.WriteString(" -> ")
buffer.WriteString("\"")
buffer.WriteString(aci.manifest.NameAndVersion.TinyNameId())
buffer.WriteString("\"")
buffer.WriteString("[color=green,penwidth=2.0]")
buffer.WriteString("\n")
}
for _, dep := range aci.manifest.Tester.Aci.Dependencies {
buffer.WriteString(" ")
buffer.WriteString("\"")
buffer.WriteString(dep.TinyNameId())
buffer.WriteString("\"")
buffer.WriteString(" -> ")
buffer.WriteString("\"")
buffer.WriteString(aci.manifest.NameAndVersion.TinyNameId())
buffer.WriteString("\"")
buffer.WriteString("[color=blue,penwidth=2.0]")
buffer.WriteString("\n")
}
buffer.WriteString("}\n")
if err := ioutil.WriteFile(aci.target+pathGraphDot, buffer.Bytes(), 0644); err != nil {
return errs.WithEF(err, aci.fields.WithField("file", aci.target+pathGraphDot), "Failed to write file")
}
if _, _, err := common.ExecCmdGetStdoutAndStderr("dot", "-V"); err == nil {
if std, stderr, err := common.ExecCmdGetStdoutAndStderr("dot", "-Tpng", aci.target+pathGraphDot, "-o", aci.target+pathGraphPng); err != nil {
return errs.WithEF(err, aci.fields.WithField("stdout", std).WithField("stderr", stderr), "Failed to create graph image")
}
}
return nil
}