This repository has been archived by the owner on Mar 14, 2019. It is now read-only.
forked from libgit2/git2go
/
packbuilder.go
111 lines (95 loc) · 2.23 KB
/
packbuilder.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
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
package git
/*
#cgo pkg-config: libgit2
#include <git2.h>
#include <git2/errors.h>
#include <git2/pack.h>
#include <stdlib.h>
extern int _go_git_packbuilder_foreach(git_packbuilder *pb, void *payload);
*/
import "C"
import (
"runtime"
"unsafe"
"io"
)
type Packbuilder struct {
ptr *C.git_packbuilder
}
func (repo *Repository) NewPackbuilder() (*Packbuilder, error) {
builder := &Packbuilder{}
ret := C.git_packbuilder_new(&builder.ptr, repo.ptr)
if ret != 0 {
return nil, LastError()
}
runtime.SetFinalizer(builder, (*Packbuilder).Free)
return builder, nil
}
func (pb *Packbuilder) Free() {
runtime.SetFinalizer(pb, nil)
C.git_packbuilder_free(pb.ptr)
}
func (pb *Packbuilder) Insert(id *Oid, name string) error {
cname := C.CString(name)
defer C.free(unsafe.Pointer(cname))
ret := C.git_packbuilder_insert(pb.ptr, id.toC(), cname)
if ret != 0 {
return LastError()
}
return nil
}
func (pb *Packbuilder) InsertCommit(id *Oid) error {
ret := C.git_packbuilder_insert_commit(pb.ptr, id.toC())
if ret != 0 {
return LastError()
}
return nil
}
func (pb *Packbuilder) InsertTree(id *Oid) error {
ret := C.git_packbuilder_insert_tree(pb.ptr, id.toC())
if ret != 0 {
return LastError()
}
return nil
}
func (pb *Packbuilder) ObjectCount() uint32 {
return uint32(C.git_packbuilder_object_count(pb.ptr))
}
func (pb *Packbuilder) WriteToFile(name string) error {
cname := C.CString(name)
defer C.free(unsafe.Pointer(cname))
ret := C.git_packbuilder_write(pb.ptr, cname)
if ret != 0 {
return LastError()
}
return nil
}
func (pb *Packbuilder) Write(w io.Writer) error {
ch := pb.ForEach()
for slice := range ch {
_, err := w.Write(slice)
if err != nil {
return err
}
}
return nil
}
func (pb *Packbuilder) Written() uint32 {
return uint32(C.git_packbuilder_written(pb.ptr))
}
//export packbuilderForEachCb
func packbuilderForEachCb(buf unsafe.Pointer, size C.size_t, payload unsafe.Pointer) int {
ch := *(*chan []byte)(payload)
slice := C.GoBytes(buf, C.int(size))
ch <- slice
return 0
}
func (pb *Packbuilder) forEachWrap(ch chan []byte) {
C._go_git_packbuilder_foreach(pb.ptr, unsafe.Pointer(&ch))
close(ch)
}
func (pb *Packbuilder) ForEach() chan []byte {
ch := make(chan []byte, 0)
go pb.forEachWrap(ch)
return ch
}