Function SearchFile

pkg/comments/comments.go:28–65

SearchFile searches a file for comments. It infers the language

(filePath string, reader io.Reader, cb func(*Comment))

// SearchFile searches a file for comments. It infers the language
func SearchFile(filePath string, reader io.Reader, cb func(*Comment)) error {
	// create a preview reader that reads in some of the file for enry to better identify the language
	var buf bytes.Buffer
	tee := io.TeeReader(reader, &buf)
	previewReader := io.LimitReader(tee, 1000)
	preview, err := ioutil.ReadAll(previewReader)
	if err != nil {
		return err
	}

	// create a new reader concatenating the preview and the original reader (which has now been read from)
	fullReader := io.MultiReader(strings.NewReader(buf.String()), reader)

	lang := Language(enry.GetLanguage(filepath.Base(filePath), preview))
	if enry.IsVendor(filePath) {
		return nil
	}
	options, ok := LanguageParseOptions[lang]
	if !ok { // TODO provide a default parse option for when we don't know how to handle a language? I.e. default to CStyle comments say
		return nil
	}
	commentParser, err := lege.NewParser(options)
	if err != nil {
		return err
	}

	collections, err := commentParser.Parse(fullReader)
	if err != nil {
		return err
	}

	for _, c := range collections {
		comment := Comment{*c, filePath}
		cb(&comment)
	}

	return nil
}

// SearchDir searches a directory for comments
func SearchDir(dirPath string, cb func(comment *Comment)) error {