mirror of
https://github.com/zu1k/nali.git
synced 2025-03-11 19:02:03 +08:00
29 lines
626 B
Go
29 lines
626 B
Go
|
package common
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"regexp"
|
||
|
|
)
|
||
|
|
|
||
|
|
var newlineReg = regexp.MustCompile(`\r?\n|\r\n?`)
|
||
|
|
|
||
|
|
func ScanLines(data []byte, atEOF bool) (advance int, token []byte, err error) {
|
||
|
|
if atEOF && len(data) == 0 {
|
||
|
|
return 0, nil, nil
|
||
|
|
}
|
||
|
|
if loc := newlineReg.FindIndex(data); len(loc) > 0 {
|
||
|
|
delimiterLen := 1
|
||
|
|
i := loc[0]
|
||
|
|
if i+1 < len(data) && data[i+1] == '\n' {
|
||
|
|
delimiterLen = 2
|
||
|
|
}
|
||
|
|
return i + delimiterLen, data[:i], nil
|
||
|
|
}
|
||
|
|
// If we're at EOF, we have a final, non-terminated line. Return it.
|
||
|
|
if atEOF {
|
||
|
|
return len(data), bytes.TrimSuffix(data, []byte{'\r'}), nil
|
||
|
|
}
|
||
|
|
// Request more data.
|
||
|
|
return 0, nil, nil
|
||
|
|
}
|