forked from git-lfs/git-lfs
/
clean.go
42 lines (33 loc) · 771 Bytes
/
clean.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
package pointer
import (
"crypto/sha256"
"encoding/hex"
"github.com/github/git-lfs/lfs"
"io"
"os"
)
type cleanedAsset struct {
File *os.File
mediafilepath string
*Pointer
}
func Clean(reader io.Reader, size int64, cb lfs.CopyCallback) (*cleanedAsset, error) {
tmp, err := lfs.TempFile("")
if err != nil {
return nil, err
}
oidHash := sha256.New()
writer := io.MultiWriter(oidHash, tmp)
if size == 0 {
cb = nil
}
written, err := lfs.CopyWithCallback(writer, reader, size, cb)
pointer := NewPointer(hex.EncodeToString(oidHash.Sum(nil)), written)
return &cleanedAsset{tmp, "", pointer}, err
}
func (a *cleanedAsset) Close() error {
return a.File.Close()
}
func (a *cleanedAsset) Teardown() error {
return os.Remove(a.File.Name())
}