Move more logic to main

cmd/gitprompt/gitprompt.go

@@ -110,5 +110,14 @@ func main() {
 		os.Exit(0)
 	}
 
-	gitprompt.Exec(format.String(), *zsh)
+	s, err := gitprompt.Parse()
+	if err != nil {
+		_, _ = fmt.Fprintln(os.Stderr, err)
+		os.Exit(1)
+	}
+	out, num := gitprompt.Print(s, format.String())
+	_, _ = fmt.Fprint(os.Stdout, out)
+	if *zsh {
+		_, _ = fmt.Fprintf(os.Stdout, "%%%dG", num)
+	}
 }

parser.go

@@ -9,6 +9,18 @@ import (
 	"strings"
 )
 
+// GitStatus is the parsed status for the current state in git.
+type GitStatus struct {
+	Sha       string
+	Branch    string
+	Untracked int
+	Modified  int
+	Staged    int
+	Conflicts int
+	Ahead     int
+	Behind    int
+}
+
 // Parse parses the status for the repository from git. Returns nil if the
 // current directory is not part of a git repository.
 func Parse() (*GitStatus, error) {