mirror of
https://github.com/thomiceli/opengist
synced 2024-11-12 07:05:50 +01:00
40 lines
755 B
Go
40 lines
755 B
Go
package git
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
)
|
|
|
|
func truncateCommandOutput(out io.Reader, maxBytes int64) (string, bool, error) {
|
|
var (
|
|
buf []byte
|
|
err error
|
|
)
|
|
|
|
if maxBytes < 0 {
|
|
// read entire output
|
|
buf, err = io.ReadAll(out)
|
|
if err != nil {
|
|
return "", false, err
|
|
}
|
|
return string(buf), false, nil
|
|
}
|
|
|
|
// read up to maxBytes bytes
|
|
buf = make([]byte, maxBytes)
|
|
n, err := io.ReadFull(out, buf)
|
|
if err != nil && err != io.EOF && err != io.ErrUnexpectedEOF {
|
|
return "", false, err
|
|
}
|
|
bytesRead := int64(n)
|
|
|
|
// find index of last newline character
|
|
lastNewline := bytes.LastIndexByte(buf, '\n')
|
|
if lastNewline >= 0 {
|
|
// truncate buffer to exclude last line
|
|
buf = buf[:lastNewline]
|
|
}
|
|
|
|
return string(buf), bytesRead == maxBytes, nil
|
|
}
|