bytes.go

package ybase

import "bytes"

type Bytes []byte

// LineColumn calculates line number and column from offset.
func (b Bytes) LineColumn(offset int) (int, int, bool) {
	if offset < 0 || offset >= len(b) {
		return 0, 0, false
	}

	target := b[:offset]
	xs := bytes.Split(target, []byte("\n"))
	line := len(xs)
	last := string(xs[len(xs)-1])
	column := len(last) + 1
	return line, column, true
}

// Offset calculates offset from line number and column.
func (b Bytes) Offset(line, column int) (int, bool) {
	if line < 1 || column < 1 {
		return 0, false
	}
	xs := bytes.Split(b, []byte("\n"))
	if line-1 >= len(xs) {
		return 0, false
	}
	lineRow := string(xs[line-1])
	if column-1 >= len(lineRow) {
		return 0, false
	}

	var offset int
	for _, x := range xs[:line-1] {
		offset += len(x)
	}
	offset += len(xs[:line-1]) - 1
	offset += len([]byte(lineRow[:column]))
	return offset, true
}

type ContextLine struct {
	Linum int
	Line  []byte
}

type Context struct {
	Target *ContextLine
	Lines  []*ContextLine
}

// Context retrieves lines before and after a specified line number.
//
// It returns the surrounding context, including the given number
// of lines before and after the specified line.
func (b Bytes) Context(line, count int) (*Context, bool) {
	if line < 1 || count < 0 {
		return nil, false
	}
	xs := bytes.Split(b, []byte("\n"))
	if line-1 >= len(xs) {
		return nil, false
	}

	target := &ContextLine{
		Linum: line,
		Line:  xs[line-1],
	}
	lines := []*ContextLine{}
	for linum := line - count; linum <= line+count; linum++ {
		index := linum - 1
		if 0 <= index && index < len(xs) {
			lines = append(lines, &ContextLine{
				Linum: linum,
				Line:  xs[index],
			})
		}
	}
	return &Context{
		Target: target,
		Lines:  lines,
	}, true
}

type Adjacency struct {
	Linum  int
	Column int
	Focus  rune
	String string
	Line   string
}

// Adjacency retrieves runes before and after a specified line and column.
func (b Bytes) Adjacency(line, column, count int) (*Adjacency, bool) {
	if line < 1 || column < 1 || count < 0 {
		return nil, false
	}
	xs := bytes.Split(b, []byte("\n"))
	if line-1 >= len(xs) {
		return nil, false
	}
	target := bytes.Runes(xs[line-1])
	if column-1 >= len(target) {
		return nil, false
	}
	focus := target[column-1]

	start := max(column-1-count, 0)
	end := min(column+count, len(target))
	targetString := string(target[start:end])
	return &Adjacency{
		Linum:  line,
		Column: column,
		Focus:  focus,
		String: targetString,
		Line:   string(target),
	}, true
}