From 610a8b5a408cb3d08f4e4e2d6cf9cbe7194490d5 Mon Sep 17 00:00:00 2001
From: Dave Henderson <dhenderson@gmail.com>
Date: Mon, 21 Feb 2022 22:41:32 -0500
Subject: Support piping input to plugin

Signed-off-by: Dave Henderson <dhenderson@gmail.com>
---
 plugins.go | 23 ++++++++++++++++++++---
 1 file changed, 20 insertions(+), 3 deletions(-)

(limited to 'plugins.go')

diff --git a/plugins.go b/plugins.go
index 89309c0e..3e6e1f04 100644
--- a/plugins.go
+++ b/plugins.go
@@ -19,11 +19,17 @@ import (
 
 func bindPlugins(ctx context.Context, cfg *config.Config, funcMap template.FuncMap) error {
 	for k, v := range cfg.Plugins {
+		timeout := cfg.PluginTimeout
+		if v.Timeout != 0 {
+			timeout = v.Timeout
+		}
+
 		plugin := &plugin{
 			ctx:     ctx,
 			name:    k,
-			path:    v,
-			timeout: cfg.PluginTimeout,
+			path:    v.Cmd,
+			timeout: timeout,
+			pipe:    v.Pipe,
 			stderr:  cfg.Stderr,
 		}
 		if _, ok := funcMap[plugin.name]; ok {
@@ -40,6 +46,7 @@ type plugin struct {
 	stderr     io.Writer
 	name, path string
 	timeout    time.Duration
+	pipe       bool
 }
 
 // builds a command that's appropriate for running scripts
@@ -79,8 +86,18 @@ func (p *plugin) run(args ...interface{}) (interface{}, error) {
 
 	ctx, cancel := context.WithTimeout(p.ctx, p.timeout)
 	defer cancel()
+
+	var stdin *bytes.Buffer
+	if p.pipe && len(a) > 0 {
+		stdin = bytes.NewBufferString(a[len(a)-1])
+		a = a[:len(a)-1]
+	}
+
 	c := exec.CommandContext(ctx, name, a...)
-	c.Stdin = nil
+	if stdin != nil {
+		c.Stdin = stdin
+	}
+
 	c.Stderr = p.stderr
 	outBuf := &bytes.Buffer{}
 	c.Stdout = outBuf
-- 
cgit v1.2.3