obitools4/pkg/obiformats/rope_scanner.go

package obiformats

import "bytes"

// ropeScanner reads lines from a PieceOfChunk rope.
// The carry buffer handles lines that span two rope nodes; it grows as needed.
type ropeScanner struct {
	current *PieceOfChunk
	pos     int
	carry   []byte
}

func newRopeScanner(rope *PieceOfChunk) *ropeScanner {
	return &ropeScanner{current: rope}
}

// ReadLine returns the next line without the trailing \n (or \r\n).
// Returns nil at end of rope. The returned slice aliases carry[] or the node
// data and is valid only until the next ReadLine call.
func (s *ropeScanner) ReadLine() []byte {
	for {
		if s.current == nil {
			if len(s.carry) > 0 {
				line := s.carry
				s.carry = s.carry[:0]
				return line
			}
			return nil
		}

		data := s.current.data[s.pos:]
		idx := bytes.IndexByte(data, '\n')

		if idx >= 0 {
			var line []byte
			if len(s.carry) == 0 {
				line = data[:idx]
			} else {
				s.carry = append(s.carry, data[:idx]...)
				line = s.carry
				s.carry = s.carry[:0]
			}
			s.pos += idx + 1
			if s.pos >= len(s.current.data) {
				s.current = s.current.Next()
				s.pos = 0
			}
			if len(line) > 0 && line[len(line)-1] == '\r' {
				line = line[:len(line)-1]
			}
			return line
		}

		// No \n in this node: accumulate into carry and advance
		s.carry = append(s.carry, data...)
		s.current = s.current.Next()
		s.pos = 0
	}
}

// skipToNewline advances the scanner past the next '\n'.
func (s *ropeScanner) skipToNewline() {
	for s.current != nil {
		data := s.current.data[s.pos:]
		idx := bytes.IndexByte(data, '\n')
		if idx >= 0 {
			s.pos += idx + 1
			if s.pos >= len(s.current.data) {
				s.current = s.current.Next()
				s.pos = 0
			}
			return
		}
		s.current = s.current.Next()
		s.pos = 0
	}
}