Function extractLogs

cmd/errtrace/main_test.go:862–892

extractLogs parses the "// want" comments in src into a slice of logLine structs.

(src []byte)

// extractLogs parses the "// want" comments in src
// into a slice of logLine structs.
func extractLogs(src []byte) ([]logLine, error) {
	fset := token.NewFileSet()
	f, err := parser.ParseFile(fset, "", src, parser.ParseComments)
	if err != nil {
		return nil, errtrace.Wrap(fmt.Errorf("parse: %w", err))
	}

	var logs []logLine
	for _, c := range f.Comments {
		for _, l := range c.List {
			_, lit, ok := strings.Cut(l.Text, "// want:")
			if !ok {
				continue
			}

			pos := fset.Position(l.Pos())
			s, err := strconv.Unquote(lit)
			if err != nil {
				return nil, errtrace.Wrap(fmt.Errorf("%s:bad string literal: %s", pos, lit))
			}

			logs = append(logs, logLine{Line: pos.Line, Msg: s})
		}
	}

	sort.Slice(logs, func(i, j int) bool {
		return logs[i].Line < logs[j].Line
	})

	return logs, nil
}

func parseLogOutput(file, s string) ([]logLine, error) {
	var logs []logLine