Add plugin tasks (#651)

pkg/plugin/log.go

@@ -0,0 +1,79 @@
+package plugin
+
+import (
+	"bufio"
+	"io"
+	"strconv"
+
+	"github.com/stashapp/stash/pkg/logger"
+	"github.com/stashapp/stash/pkg/plugin/common/log"
+)
+
+func (t *pluginTask) handleStderrLine(line string, defaultLogLevel *log.Level) {
+	level, l := log.DetectLogLevel(line)
+
+	const pluginPrefix = "[Plugin] "
+	// if no log level, just output to info
+	if level == nil {
+		if defaultLogLevel != nil {
+			level = defaultLogLevel
+		} else {
+			level = &log.InfoLevel
+		}
+	}
+
+	switch *level {
+	case log.TraceLevel:
+		logger.Trace(pluginPrefix, l)
+	case log.DebugLevel:
+		logger.Debug(pluginPrefix, l)
+	case log.InfoLevel:
+		logger.Info(pluginPrefix, l)
+	case log.WarningLevel:
+		logger.Warn(pluginPrefix, l)
+	case log.ErrorLevel:
+		logger.Error(pluginPrefix, l)
+	case log.ProgressLevel:
+		progress, err := strconv.ParseFloat(l, 64)
+		if err != nil {
+			logger.Errorf("Error parsing progress value '%s': %s", l, err.Error())
+		} else {
+			// only pass progress through if channel present
+			if t.progress != nil {
+				// don't block on this
+				select {
+				case t.progress <- progress:
+				default:
+				}
+			}
+		}
+	}
+}
+
+func (t *pluginTask) handlePluginOutput(pluginOutputReader io.ReadCloser, defaultLogLevel *log.Level) {
+	// pipe plugin stderr to our logging
+	scanner := bufio.NewScanner(pluginOutputReader)
+	for scanner.Scan() {
+		str := scanner.Text()
+		if str != "" {
+			t.handleStderrLine(str, defaultLogLevel)
+		}
+	}
+
+	str := scanner.Text()
+	if str != "" {
+		t.handleStderrLine(str, defaultLogLevel)
+	}
+
+	pluginOutputReader.Close()
+}
+
+func (t *pluginTask) handlePluginStderr(pluginOutputReader io.ReadCloser) {
+	logLevel := log.LevelFromName(t.plugin.PluginErrLogLevel)
+	if logLevel == nil {
+		// default log level to error
+		logLevel = &log.ErrorLevel
+	}
+
+	t.handlePluginOutput(pluginOutputReader, logLevel)
+}