git_town_output_parser.go

package test

import (
	"strings"
)

// ExecutedGitCommand describes a Git command that was executed by Git Town during testing.
type ExecutedGitCommand struct {

	// Branch contains the branch in which this command ran.
	Branch string

	// Command contains the command executed.
	Command string
}

// GitCommandsInGitTownOutput provides the Git commands mentioned in the given Git Town output.
func GitCommandsInGitTownOutput(output string) (result []ExecutedGitCommand) {
	for _, line := range strings.Split(output, "\n") {
		line = strings.TrimSpace(line)
		if line == "" {
			continue
		}
		if lineContainsGitTownCommand(line) {
			result = append(result, parseLine(line))
		}
	}
	return result
}

// gitCommandLineBeginning contains the first few characters of lines containing Git commands in Git Town output.
const gitCommandLineBeginning = "\x1b[1m" // "\e[1m"

// lineContainsGitTownCommand indicates whether the given line contains a Git Town command.
func lineContainsGitTownCommand(line string) bool {
	return strings.HasPrefix(line, gitCommandLineBeginning)
}

// parseLine provides the Git Town command and branchname in the given line.
func parseLine(line string) ExecutedGitCommand {
	// NOTE: implementing this without regex
	// because the regex has gotten very complex and hard to maintain

	// remove the color codes at the beginning
	line = strings.Replace(line, gitCommandLineBeginning, "", 1)

	// extract branch name if it exists
	branchName := ""
	if line[0] == '[' {
		// line contains a branch name
		line = line[1:] // remove the leading "["
		parts := strings.SplitN(line, "]", 2)
		branchName = parts[0]
		line = parts[1]
	}

	return ExecutedGitCommand{Command: strings.TrimSpace(line), Branch: branchName}
}