From cc10e4afa0b56fff2f2c560529403bfe82003256 Mon Sep 17 00:00:00 2001
From: Umputun <umputun@gmail.com>
Date: Fri, 4 Aug 2023 03:26:58 -0500
Subject: [PATCH] refactor code extracting exporting key from shell line

---
 pkg/config/command.go | 26 +++++++++++++++-----------
 1 file changed, 15 insertions(+), 11 deletions(-)

diff --git a/pkg/config/command.go b/pkg/config/command.go
index 0e7f772b..b3b01dfe 100644
--- a/pkg/config/command.go
+++ b/pkg/config/command.go
@@ -201,17 +201,7 @@ func (cmd *Cmd) scriptFile(inp string) (r io.Reader) {
 		// if the line in the script is an export, add it to the list of exports.
 		// this is done to be able to print the variables set by the script to the console after the script is executed.
 		// the caller can use those variables to set environment variables for the next commands
-		trimmedLine := strings.TrimSpace(line)
-		if strings.HasPrefix(trimmedLine, "export") {
-			expKey := strings.TrimPrefix(trimmedLine, "export")
-			expElems := strings.Split(expKey, "=")
-			if len(expElems) != 2 {
-				continue
-			}
-			expKey = strings.TrimSpace(expElems[0])
-			if expKey == "" {
-				continue // skip empty exports
-			}
+		if expKey := cmd.exportKey(line); expKey != "" {
 			exports = append(exports, expKey)
 		}
 	}
@@ -226,6 +216,20 @@ func (cmd *Cmd) scriptFile(inp string) (r io.Reader) {
 	return &buf
 }
 
+func (cmd *Cmd) exportKey(line string) string {
+	trimmedLine := strings.TrimSpace(line)
+	if !strings.HasPrefix(trimmedLine, "export") {
+		return ""
+	}
+
+	expKey := strings.TrimPrefix(trimmedLine, "export")
+	expElems := strings.Split(expKey, "=")
+	if len(expElems) != 2 {
+		return ""
+	}
+	return strings.TrimSpace(expElems[0])
+}
+
 func (cmd *Cmd) hasShebang(inp string) bool {
 	lines := strings.Split(inp, "\n")
 	if len(lines) == 0 {