/
pick.go
73 lines (68 loc) · 1.72 KB
/
pick.go
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
package main
import (
"fmt"
"log"
"os/exec"
"strings"
cli "gopkg.in/urfave/cli.v1"
)
const (
red = 31
green = 32
)
func color(code int, msg string) string {
return fmt.Sprintf("\x1b[%dm\x1b[49m%s\x1b[39m\x1b[49m", code, msg)
}
func pick(a *cli.Context) {
msg := "Cherry-Pick %s %s\n"
if a.NArg() < 2 {
log.Fatal("argument not enough")
}
sourceBranch := a.Args().Get(0)
targetBranch := a.Args().Get(1)
out, err := exec.Command("git", "checkout", targetBranch).CombinedOutput()
if err != nil {
log.Println(string(out))
return
}
out, err = exec.Command("git", "symbolic-ref", "--short", "HEAD").CombinedOutput()
if err != nil || !strings.Contains(string(out), targetBranch) {
fmt.Println(string(out))
fmt.Println(targetBranch)
log.Println("Git checkout targetBranch failed")
return
}
for _, c := range diffCommits(sourceBranch, targetBranch) {
if !strings.Contains(c.Author.String(), a.String("author")) {
continue
}
issues := strings.Split(a.String("issue"), ",")
find := false
for _, issue := range issues {
if strings.Contains(c.Message, issue) {
find = true
break
}
}
if !find {
continue
}
log.Printf("Cherry-Pick %s\n", c.Hash.String())
out, err := exec.Command("git", "cherry-pick", "-x", c.Hash.String()).CombinedOutput()
if err != nil {
log.Printf(msg, c.Hash.String(), color(red, "ERR"))
log.Println(string(out))
break
}
out, err = exec.Command("git", "log", "--grep", c.Hash.String()).CombinedOutput()
if err != nil || !strings.Contains(string(out), c.Hash.String()) {
log.Printf(msg, c.Hash.String(), color(red, "ERR"))
log.Println(string(out))
break
}
log.Printf(msg, c.Hash.String(), color(green, "OK"))
if a.Bool("step") {
break
}
}
}