func ensureOffsetsAreByteOffsets(dir string, output *graph.Output) {
fset := fileset.NewFileSet()
files := make(map[string]*fileset.File)
addOrGetFile := func(filename string) *fileset.File {
if f, ok := files[filename]; ok {
return f
}
data, err := ioutil.ReadFile(filename)
if err != nil {
panic("ReadFile " + filename + ": " + err.Error())
}
f := fset.AddFile(filename, fset.Base(), len(data))
f.SetByteOffsetsForContent(data)
files[filename] = f
return f
}
fix := func(filename string, offsets ...*uint32) {
defer func() {
if e := recover(); e != nil {
log.Printf("failed to convert unicode offset to byte offset in file %s (did grapher output a nonexistent byte offset?) continuing anyway...", filename)
}
}()
if filename == "" {
return
}
filename = filepath.Join(dir, filename)
if fi, err := os.Stat(filename); err != nil || !fi.Mode().IsRegular() {
return
}
f := addOrGetFile(filename)
for _, offset := range offsets {
if *offset == 0 {
continue
}
before, after := *offset, uint32(f.ByteOffsetOfRune(int(*offset)))
if before != after {
log.Printf("Changed pos %d to %d in %s", before, after, filename)
}
*offset = uint32(f.ByteOffsetOfRune(int(*offset)))
}
}
for _, s := range output.Defs {
fix(s.File, &s.DefStart, &s.DefEnd)
}
for _, r := range output.Refs {
fix(r.File, &r.Start, &r.End)
}
for _, d := range output.Docs {
fix(d.File, &d.Start, &d.End)
}
for _, a := range output.Anns {
fix(a.File, &a.Start, &a.End)
}
}
// ComputeFileRange determines the actual file range according to the
// input range parameter. For example, if input has a line range set,
// the returned FileRange will contain the byte range that corresponds
// to the input line range.
func ComputeFileRange(data []byte, opt GetFileOptions) (*FileRange, *fileset.File, error) {
fr := opt.FileRange // alias for brevity
fset := fileset.NewFileSet()
f := fset.AddFile("", 1, len(data))
f.SetLinesForContent(data)
if opt.EntireFile || (fr.StartLine == 0 && fr.EndLine == 0 && fr.StartByte == 0 && fr.EndByte == 0) {
fr.StartLine, fr.EndLine = 0, 0
fr.StartByte, fr.EndByte = 0, int64(len(data))
}
lines := fr.StartLine != 0 || fr.EndLine != 0
bytes := fr.StartByte != 0 || fr.EndByte != 0
if lines && bytes {
return nil, nil, fmt.Errorf("must specify a line range OR a byte range, not both (%+v)", fr)
}
// TODO(sqs): fix up the sketchy int conversions
if lines {
// Given line range, validate it and return byte range.
if fr.StartLine == 0 {
fr.StartLine = 1 // 1-indexed
}
if fr.StartLine == 1 && fr.EndLine == 1 && f.LineCount() == 0 {
// Empty file.
return &fr, f, nil
}
if fr.StartLine < 0 || fr.StartLine > int64(f.LineCount()) {
return nil, nil, fmt.Errorf("start line %d out of bounds (%d lines total)", fr.StartLine, f.LineCount())
}
if fr.EndLine < 0 {
return nil, nil, fmt.Errorf("end line %d out of bounds (%d lines total)", fr.EndLine, f.LineCount())
}
if count := int64(f.LineCount()); fr.EndLine > count || fr.EndLine == 0 {
fr.EndLine = count
}
fr.StartByte, fr.EndByte = int64(f.LineOffset(int(fr.StartLine))), int64(f.LineEndOffset(int(fr.EndLine)))
} else if bytes {
if fr.StartByte < 0 || fr.StartByte > int64(len(data)-1) {
return nil, nil, fmt.Errorf("start byte %d out of bounds (%d bytes total)", fr.StartByte, len(data))
}
if fr.EndByte < 0 || fr.EndByte > int64(len(data)) {
return nil, nil, fmt.Errorf("end byte %d out of bounds (%d bytes total)", fr.EndByte, len(data))
}
fr.StartLine, fr.EndLine = int64(f.Line(f.Pos(int(fr.StartByte)))), int64(f.Line(f.Pos(int(fr.EndByte))))
if opt.ExpandContextLines > 0 {
fr.StartLine -= int64(opt.ExpandContextLines)
if fr.StartLine < 1 {
fr.StartLine = 1
}
fr.EndLine += int64(opt.ExpandContextLines)
if fr.EndLine > int64(f.LineCount()) {
fr.EndLine = int64(f.LineCount())
}
}
if opt.ExpandContextLines > 0 || opt.FullLines {
fr.StartByte, fr.EndByte = int64(f.LineOffset(int(fr.StartLine))), int64(f.LineEndOffset(int(fr.EndLine)))
}
}
return &fr, f, nil
}
