2014-08-06 22:04:34 +00:00
|
|
|
package gitmedia
|
|
|
|
|
|
|
|
import (
|
2014-08-07 16:44:42 +00:00
|
|
|
"fmt"
|
2014-08-06 22:04:34 +00:00
|
|
|
"io"
|
2014-08-07 16:44:42 +00:00
|
|
|
"os"
|
|
|
|
"path/filepath"
|
2014-08-06 22:04:34 +00:00
|
|
|
)
|
|
|
|
|
2014-08-07 17:27:06 +00:00
|
|
|
type CallbackReader struct {
|
|
|
|
C CopyCallback
|
|
|
|
TotalSize int64
|
|
|
|
ReadSize int64
|
|
|
|
io.Reader
|
2014-08-06 22:04:34 +00:00
|
|
|
}
|
|
|
|
|
2014-08-07 14:53:13 +00:00
|
|
|
type CopyCallback func(int64, int64) error
|
|
|
|
|
2014-08-07 17:27:06 +00:00
|
|
|
func (w *CallbackReader) Read(p []byte) (int, error) {
|
|
|
|
n, err := w.Reader.Read(p)
|
2014-08-06 22:04:34 +00:00
|
|
|
|
|
|
|
if n > 0 {
|
2014-08-07 17:27:06 +00:00
|
|
|
w.ReadSize += int64(n)
|
2014-08-06 22:04:34 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if err == nil && w.C != nil {
|
2014-08-07 17:27:06 +00:00
|
|
|
err = w.C(w.TotalSize, w.ReadSize)
|
2014-08-06 22:04:34 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return n, err
|
|
|
|
}
|
|
|
|
|
2014-08-07 14:53:13 +00:00
|
|
|
func CopyWithCallback(writer io.Writer, reader io.Reader, totalSize int64, cb CopyCallback) (int64, error) {
|
|
|
|
if cb == nil {
|
|
|
|
return io.Copy(writer, reader)
|
|
|
|
}
|
|
|
|
|
2014-08-07 17:27:06 +00:00
|
|
|
cbReader := &CallbackReader{
|
2014-08-06 22:04:34 +00:00
|
|
|
C: cb,
|
|
|
|
TotalSize: totalSize,
|
2014-08-07 17:27:06 +00:00
|
|
|
Reader: reader,
|
2014-08-06 22:04:34 +00:00
|
|
|
}
|
2014-08-07 17:27:06 +00:00
|
|
|
return io.Copy(writer, cbReader)
|
2014-08-06 22:04:34 +00:00
|
|
|
}
|
2014-08-07 16:44:42 +00:00
|
|
|
|
2014-08-07 17:40:08 +00:00
|
|
|
func CopyCallbackFile(event, filename string, index, totalFiles int) (CopyCallback, *os.File, error) {
|
2014-08-07 16:44:42 +00:00
|
|
|
cbFilename := os.Getenv("GIT_MEDIA_PROGRESS")
|
|
|
|
if len(cbFilename) == 0 || len(filename) == 0 || len(event) == 0 {
|
|
|
|
return nil, nil, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
cbDir := filepath.Dir(cbFilename)
|
|
|
|
if err := os.MkdirAll(cbDir, 0755); err != nil {
|
2014-08-07 17:37:04 +00:00
|
|
|
return nil, nil, wrapProgressError(err, event, cbFilename)
|
2014-08-07 16:44:42 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
file, err := os.OpenFile(cbFilename, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0666)
|
|
|
|
if err != nil {
|
2014-08-07 17:37:04 +00:00
|
|
|
return nil, file, wrapProgressError(err, event, cbFilename)
|
2014-08-07 16:44:42 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
var prevProgress int
|
|
|
|
var progress int
|
|
|
|
|
|
|
|
cb := CopyCallback(func(total int64, written int64) error {
|
|
|
|
progress = 0
|
|
|
|
if total > 0 {
|
|
|
|
progress = int(float64(written) / float64(total) * 100)
|
|
|
|
}
|
|
|
|
|
|
|
|
if progress != prevProgress {
|
2014-08-07 17:40:08 +00:00
|
|
|
_, err := file.Write([]byte(fmt.Sprintf("%s %d/%d %d %s\n", event, index, totalFiles, progress, filename)))
|
2014-08-12 16:46:10 +00:00
|
|
|
file.Sync()
|
2014-08-07 16:44:42 +00:00
|
|
|
prevProgress = progress
|
2014-08-07 17:37:04 +00:00
|
|
|
return wrapProgressError(err, event, cbFilename)
|
2014-08-07 16:44:42 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
})
|
2014-08-07 17:40:08 +00:00
|
|
|
file.Write([]byte(fmt.Sprintf("%s %d/%d 0 %s\n", event, index, totalFiles, filename)))
|
2014-08-07 16:44:42 +00:00
|
|
|
|
|
|
|
return cb, file, nil
|
|
|
|
}
|
|
|
|
|
2014-08-07 17:37:04 +00:00
|
|
|
func wrapProgressError(err error, event, filename string) error {
|
2014-08-07 16:44:42 +00:00
|
|
|
if err == nil {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2014-08-07 17:37:04 +00:00
|
|
|
return fmt.Errorf("Error writing Git Media %s progress to %s: %s", event, filename, err.Error())
|
2014-08-07 16:44:42 +00:00
|
|
|
}
|