add commands support

Signed-off-by: Casey Lee <cplee@nektos.com>

pkg/runner/command.go

@@ -0,0 +1,91 @@
+package runner
+
+import (
+	"context"
+	"fmt"
+	"regexp"
+	"strings"
+
+	"github.com/nektos/act/pkg/common"
+)
+
+var commandPattern *regexp.Regexp
+
+func init() {
+	commandPattern = regexp.MustCompile("^::([^ ]+)( (.+))?::([^\r\n]*)[\r\n]+$")
+}
+
+func (rc *RunContext) commandHandler(ctx context.Context) common.LineHandler {
+	logger := common.Logger(ctx)
+	resumeCommand := ""
+	return func(line string) {
+		if m := commandPattern.FindStringSubmatch(line); m != nil {
+			command := m[1]
+			kvPairs := parseKeyValuePairs(m[3])
+			arg := m[4]
+
+			if resumeCommand != "" && command != resumeCommand {
+				return
+			}
+
+			switch command {
+			case "set-env":
+				rc.setEnv(ctx, kvPairs, arg)
+			case "set-output":
+				rc.setOutput(ctx, kvPairs, arg)
+			case "add-path":
+				rc.addPath(ctx, arg)
+			case "debug":
+				logger.Infof("  \U0001F4AC  %s", line)
+			case "warning":
+				logger.Infof("  \U0001F6A7  %s", line)
+			case "error":
+				logger.Infof("  \U00002757  %s", line)
+			case "add-mask":
+				logger.Infof("  \U00002699  %s", line)
+			case "stop-commands":
+				resumeCommand = arg
+				logger.Infof("  \U00002699  %s", line)
+			case resumeCommand:
+				resumeCommand = ""
+				logger.Infof("  \U00002699  %s", line)
+			default:
+				logger.Infof("  \U00002753  %s", line)
+			}
+		}
+	}
+}
+
+func (rc *RunContext) setEnv(ctx context.Context, kvPairs map[string]string, arg string) {
+	common.Logger(ctx).Infof("  \U00002699  ::set-env:: %s=%s", kvPairs["name"], arg)
+	if rc.Env == nil {
+		rc.Env = make(map[string]string)
+	}
+	rc.Env[kvPairs["name"]] = arg
+}
+func (rc *RunContext) setOutput(ctx context.Context, kvPairs map[string]string, arg string) {
+	common.Logger(ctx).Infof("  \U00002699  ::set-output:: %s=%s", kvPairs["name"], arg)
+	if rc.Outputs == nil {
+		rc.Outputs = make(map[string]string)
+	}
+	rc.Outputs[kvPairs["name"]] = arg
+}
+func (rc *RunContext) addPath(ctx context.Context, arg string) {
+	common.Logger(ctx).Infof("  \U00002699  ::add-path:: %s", arg)
+	if rc.Env == nil {
+		rc.Env = make(map[string]string)
+	}
+	rc.Env["PATH"] = fmt.Sprintf("%s:%s", arg, rc.Env["PATH"])
+}
+
+func parseKeyValuePairs(kvPairs string) map[string]string {
+	rtn := make(map[string]string)
+	kvPairList := strings.Split(kvPairs, ",")
+	for _, kvPair := range kvPairList {
+		kv := strings.Split(kvPair, "=")
+		if len(kv) == 2 {
+			rtn[kv[0]] = kv[1]
+		}
+	}
+	return rtn
+}