File: stash.go

package info (click to toggle)
gitbatch 0.6.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 644 kB
  • sloc: makefile: 5; sh: 1
file content (126 lines) | stat: -rw-r--r-- 3,350 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
package git

import (
	"os/exec"
	"regexp"
	"strconv"
	"strings"
)

// StashedItem holds the required fields for a stashed change
type StashedItem struct {
	StashID     int
	BranchName  string
	Hash        string
	Description string
	EntityPath  string
}

func (r *Repository) loadStashedItems() error {
	r.Stasheds = make([]*StashedItem, 0)
	output := stashGet(r, "list")
	stashIDRegex := regexp.MustCompile(`stash@{[\d]+}:`)
	stashIDRegexInt := regexp.MustCompile(`[\d]+`)
	stashBranchRegex := regexp.MustCompile(`^(.*?): `)
	stashMsgRegex := regexp.MustCompile(`WIP on \(?([^)]*)\)?`)
	stashHashRegex := regexp.MustCompile(`[\w|\d]{7}\s`)

	stashlist := strings.Split(output, "\n")
	for _, stashitem := range stashlist {
		// find id
		id := stashIDRegexInt.FindString(stashIDRegex.FindString(stashitem))
		i, err := strconv.Atoi(id)
		if err != nil {
			// probably something isn't right let's continue over this iteration
			continue
		}
		// trim id section
		trimmed := stashIDRegex.Split(stashitem, 2)[1]

		// let's ignore autostash
		if trimmed[1:] == "autostash" {
			continue
		}
		// find branch
		stashBranchRegexMatch := stashBranchRegex.FindString(trimmed)
		branchName := stashBranchRegexMatch[:len(stashBranchRegexMatch)-2]

		branchMatches := stashMsgRegex.FindStringSubmatch(branchName)
		if len(branchMatches) >= 2 {
			branchName = stashBranchRegexMatch[:len(stashBranchRegexMatch)-2]
		}

		// trim branch section
		trimmed = stashBranchRegex.Split(trimmed, 2)[1]
		hash := ""

		var desc string
		if stashHashRegex.MatchString(trimmed) {
			hash = stashHashRegex.FindString(trimmed)[:7]
			desc = stashHashRegex.Split(trimmed, 2)[1]
		} else {
			desc = trimmed
		}
		// trim hash

		r.Stasheds = append(r.Stasheds, &StashedItem{
			StashID:     i,
			BranchName:  branchName,
			Hash:        hash,
			Description: desc,
			EntityPath:  r.AbsPath,
		})
	}
	return nil
}

func stashGet(r *Repository, option string) string {
	args := make([]string, 0)
	args = append(args, "stash")
	args = append(args, option)
	cmd := exec.Command("git", args...)
	cmd.Dir = r.AbsPath
	output, err := cmd.CombinedOutput()
	if err != nil {
		return "?"
	}
	return string(output)
}

// Pop is the wrapper of "git stash pop" command that used for a file
func (stashedItem *StashedItem) Pop() (string, error) {
	args := make([]string, 0)
	args = append(args, "stash")
	args = append(args, "pop")
	args = append(args, "stash@{"+strconv.Itoa(stashedItem.StashID)+"}")
	cmd := exec.Command("git", args...)
	cmd.Dir = stashedItem.EntityPath
	output, err := cmd.CombinedOutput()
	return string(output), err
}

// Show is the wrapper of "git stash show -p " command
func (stashedItem *StashedItem) Show() (string, error) {
	args := make([]string, 0)
	args = append(args, "stash")
	args = append(args, "show")
	args = append(args, "-p")
	args = append(args, "stash@{"+strconv.Itoa(stashedItem.StashID)+"}")
	cmd := exec.Command("git", args...)
	cmd.Dir = stashedItem.EntityPath
	output, err := cmd.CombinedOutput()

	return string(output), err
}

// Stash is the wrapper of conventional "git stash" command
func (r *Repository) Stash() (string, error) {
	args := make([]string, 0)
	args = append(args, "stash")

	cmd := exec.Command("git", args...)
	cmd.Dir = r.AbsPath
	output, err := cmd.CombinedOutput()
	_ = r.Refresh()
	return string(output), err
}