Add command widget that displays the output of commands

zMoooooritz · muesli · commit 69571deda289 · 2021-06-08T13:28:11.000+02:00
diff --git a/widget.go b/widget.go
@@ -111,6 +111,9 @@ func NewWidget(base string, kc KeyConfig, bg image.Image) (Widget, error) {
 
 	case "top":
 		return NewTopWidget(*bw, kc.Widget), nil
+
+	case "command":
+		return NewCommandWidget(*bw, kc.Widget)
 	}
 
 	// unknown widget ID
diff --git a/widget_command.go b/widget_command.go
@@ -0,0 +1,76 @@
+package main
+
+import (
+	"fmt"
+	"image"
+	"os/exec"
+	"strings"
+
+	"github.com/muesli/streamdeck"
+)
+
+// CommandWidget is a widget displaying the output of command(s)
+type CommandWidget struct {
+	BaseWidget
+	command string
+	font    string
+}
+
+func NewCommandWidget(bw BaseWidget, opts WidgetConfig) (*CommandWidget, error) {
+	bw.setInterval(opts.Interval, 1000)
+
+	var command, font string
+	ConfigValue(opts.Config["command"], &command)
+	ConfigValue(opts.Config["font"], &font)
+
+	return &CommandWidget{
+		BaseWidget: bw,
+		command:    command,
+		font:       font,
+	}, nil
+}
+
+// Update renders the widget
+func (w *CommandWidget) Update(dev *streamdeck.Device) error {
+	size := int(dev.Pixels)
+	margin := size / 18
+	height := size - (margin * 2)
+	img := image.NewRGBA(image.Rect(0, 0, size, size))
+
+	commands := strings.Split(w.command, ";")
+	fonts := strings.Split(w.font, ";")
+
+	if len(commands) == 0 || len(w.command) == 0 {
+		return fmt.Errorf("no command(s) supplied")
+	}
+	for len(fonts) < len(commands) {
+		fonts = append(fonts, "regular")
+	}
+
+	for i := 0; i < len(commands); i++ {
+		str, err := runCommand(commands[i])
+		if err != nil {
+			return err
+		}
+		font := fontByName(fonts[i])
+		lower := margin + (height/len(commands))*i
+		upper := margin + (height/len(commands))*(i+1)
+
+		drawString(img, image.Rect(0, lower, size, upper),
+			font,
+			str,
+			dev.DPI,
+			-1,
+			image.Pt(-1, -1))
+	}
+
+	return w.render(dev, img)
+}
+
+func runCommand(command string) (string, error) {
+	output, err := exec.Command("sh", "-c", command).Output()
+	if err != nil {
+		return "", err
+	}
+	return strings.TrimSuffix(string(output), "\n"), nil
+}

Original file line number	Diff line number	Diff line change
`@@ -111,6 +111,9 @@ func NewWidget(base string, kc KeyConfig, bg image.Image) (Widget, error) {`
`111`	`111`
`112`	`112`	`case "top":`
`113`	`113`	`return NewTopWidget(*bw, kc.Widget), nil`
	`114`	`+`
	`115`	`+ case "command":`
	`116`	`+ return NewCommandWidget(*bw, kc.Widget)`
`114`	`117`	`}`
`115`	`118`
`116`	`119`	`// unknown widget ID`