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
|
package main
import (
"bytes"
"fmt"
"os"
"os/exec"
"strings"
"git.lukeshu.com/go/libfastimport"
)
func gitFastExport(args ...string) (*libfastimport.Frontend, error) {
cmd := exec.Command("git", append([]string{"fast-export"}, args...)...)
stdout, err := cmd.StdoutPipe()
if err != nil {
return nil, err
}
var stderr bytes.Buffer
cmd.Stderr = &stderr
err = cmd.Start()
if err != nil {
return nil, err
}
frontend := libfastimport.NewFrontend(stdout, nil, func(err error) error {
if _err := cmd.Wait(); _err != nil {
if ee, ok := _err.(*exec.ExitError); ok {
ee.Stderr = stderr.Bytes()
}
err = _err
}
return err
})
return frontend, nil
}
func gitFastImport(args ...string) (*libfastimport.Backend, error) {
cmd := exec.Command("git", append([]string{"fast-import", "--cat-blob-fd=3"}, args...)...)
pipr, pipw, err := os.Pipe()
if err != nil {
return nil, err
}
stdin, err := cmd.StdinPipe()
if err != nil {
return nil, err
}
cmd.Stdout = os.Stdout
var stderr bytes.Buffer
cmd.Stderr = &stderr
cmd.ExtraFiles = []*os.File{pipw}
err = cmd.Start()
if err != nil {
return nil, err
}
pipw.Close()
backend := libfastimport.NewBackend(stdin, pipr, func(err error) error {
pipr.Close()
stdin.Close()
if _err := cmd.Wait(); _err != nil {
if ee, ok := _err.(*exec.ExitError); ok {
ee.Stderr = stderr.Bytes()
}
err = _err
}
return err
})
return backend, nil
}
func gitRefs() (map[string]string, error) {
cmd := exec.Command("git", "for-each-ref")
b, err := cmd.Output()
if err != nil {
return nil, err
}
lines := strings.Split(string(b), "\n")
ret := make(map[string]string, len(lines)-1)
for _, line := range lines {
if line == "" {
continue
}
fields := strings.SplitN(line, " ", 3)
if len(fields) != 3 {
return nil, fmt.Errorf("malformed git for-each-ref line: %q", line)
}
refTarget := fields[0]
refType := fields[1]
refName := fields[2]
if refType != "commit" {
continue
}
ret[refName] = refTarget
}
return ret, nil
}
|