gitwrapper.go

package gitwrapper

import (
	"bytes"
	"io"
	"log"
	"os"
	"os/exec"
	"strings"
)

var (
	logger = log.New(os.Stdout, "[gw] ", 0)
)

func green(s string) string {
	return "\033[0;32m" + s + "\033[0m"
}

func runGitCmd(trace bool, gitCmd string) (string, error) {
	if trace {
		logger.Println(green(gitCmd))
	}

	cmd := exec.Command("bash", "-c", gitCmd)

	var stdout bytes.Buffer
	cmd.Stdout = &stdout
	cmd.Stderr = os.Stderr
	err := cmd.Run()

	// skip git error if std output exists
	if err != nil && stdout.Len() == 0 {
		return "", err
	}

	if trace {
		buffCopy := stdout
		_, _ = io.Copy(os.Stdout, &buffCopy)
	}

	return strings.TrimSpace(stdout.String()), nil
}

type Branch struct {
	Name                string
	IsCurrent, IsRemote bool
}

func getAllBranches() ([]Branch, error) {
	o, err := runGitCmd(false, "git branch -a")
	if err != nil {
		return nil, err
	}
	var bb []Branch
	for _, row := range strings.Split(o, "\n") {
		row = strings.TrimSpace(row)
		if len(row) == 0 {
			continue
		}

		b := Branch{Name: row}

		if strings.HasPrefix(b.Name, "* ") {
			b.IsCurrent = true
			b.Name = b.Name[2:]
		}

		if strings.HasPrefix(b.Name, "remotes/") {
			b.Name = b.Name[8:]
			b.IsRemote = true

			// ignore remote branches not from origin
			if !strings.HasPrefix(b.Name, "origin/") {
				continue
			}
			b.Name = b.Name[7:]
		}

		bb = append(bb, b)
	}
	return bb, nil
}