Host Run Collector (#606)

Host Run Collector

Author: diamonwiggins

pkg/apis/troubleshoot/v1beta2/hostcollector_shared.go

@@ -150,6 +150,12 @@ type HostServices struct {
 	HostCollectorMeta `json:",inline" yaml:",inline"`
 }
 
+type HostRun struct {
+	HostCollectorMeta `json:",inline" yaml:",inline"`
+	Command           string   `json:"command"`
+	Args              []string `json:"args"`
+}
+
 type HostCollect struct {
 	CPU                   *CPU                   `json:"cpu,omitempty" yaml:"cpu,omitempty"`
 	Memory                *Memory                `json:"memory,omitempty" yaml:"memory,omitempty"`
@@ -169,6 +175,7 @@ type HostCollect struct {
 	Certificate           *Certificate           `json:"certificate,omitempty" yaml:"certificate,omitempty"`
 	HostServices          *HostServices          `json:"hostServices,omitempty" yaml:"hostServices,omitempty"`
 	HostOS                *HostOS                `json:"hostOS,omitempty" yaml:"hostOS,omitempty"`
+	HostRun               *HostRun               `json:"run,omitempty" yaml:"run,omitempty"`
 }
 
 func (c *HostCollect) GetName() string {

pkg/collect/host_collector.go

@@ -51,6 +51,8 @@ func GetHostCollector(collector *troubleshootv1beta2.HostCollect, bundlePath str
 		return &CollectHostServices{collector.HostServices, bundlePath}, true
 	case collector.HostOS != nil:
 		return &CollectHostOS{collector.HostOS, bundlePath}, true
+	case collector.HostRun != nil:
+		return &CollectHostRun{collector.HostRun, bundlePath}, true
 	default:
 		return nil, false
 	}

pkg/collect/host_run.go

@@ -0,0 +1,79 @@
+package collect
+
+import (
+	"bytes"
+	"encoding/json"
+	"os/exec"
+	"path/filepath"
+	"strings"
+
+	"github.com/pkg/errors"
+	troubleshootv1beta2 "github.com/replicatedhq/troubleshoot/pkg/apis/troubleshoot/v1beta2"
+)
+
+type HostRunInfo struct {
+	Command  string `json:"command"`
+	ExitCode string `json:"exitCode"`
+	Error    string `json:"error"`
+}
+
+type CollectHostRun struct {
+	hostCollector *troubleshootv1beta2.HostRun
+	BundlePath    string
+}
+
+func (c *CollectHostRun) Title() string {
+	return hostCollectorTitleOrDefault(c.hostCollector.HostCollectorMeta, "Run Host")
+}
+
+func (c *CollectHostRun) IsExcluded() (bool, error) {
+	return isExcluded(c.hostCollector.Exclude)
+}
+
+func (c *CollectHostRun) Collect(progressChan chan<- interface{}) (map[string][]byte, error) {
+	runHostCollector := c.hostCollector
+
+	cmd := exec.Command(runHostCollector.Command, runHostCollector.Args...)
+
+	var stdout, stderr bytes.Buffer
+	cmd.Stdout = &stdout
+	cmd.Stderr = &stderr
+
+	runInfo := HostRunInfo{
+		Command:  cmd.String(),
+		ExitCode: "0",
+	}
+
+	err := cmd.Run()
+	if err != nil {
+		if werr, ok := err.(*exec.ExitError); ok {
+			runInfo.ExitCode = strings.TrimPrefix(werr.Error(), "exit status ")
+			runInfo.Error = stderr.String()
+		} else {
+			return nil, errors.Wrap(err, "failed to run")
+		}
+	}
+
+	collectorName := c.hostCollector.CollectorName
+	if collectorName == "" {
+		collectorName = "run-host"
+	}
+	resultInfo := filepath.Join("host-collectors/run-host", collectorName+"-info.json")
+	result := filepath.Join("host-collectors/run-host", collectorName+".txt")
+
+	b, err := json.Marshal(runInfo)
+	if err != nil {
+		return nil, errors.Wrap(err, "failed to marshal run host result")
+	}
+
+	output := NewResult()
+	output.SaveResult(c.BundlePath, resultInfo, bytes.NewBuffer(b))
+	output.SaveResult(c.BundlePath, result, bytes.NewBuffer(stdout.Bytes()))
+
+	runHostOutput := map[string][]byte{
+		resultInfo: b,
+		result:     stdout.Bytes(),
+	}
+
+	return runHostOutput, nil
+}