package engine

import (
	"errors"
	"io/fs"
)

var errImportDirectoryEntryLimit = errors.New("import directory-entry visit limit reached")

// visitBoundedImportDirectory shares entry accounting across directory and glob searches.
func visitBoundedImportDirectory(access *filesystemAccess, directory string, limit int, accounting *importAccounting, visit func(fs.DirEntry) error) error {
	if accounting == nil {
		accounting = &importAccounting{}
	}
	var visitError error
	err := access.readDirectory(directory, func(entry fs.DirEntry) bool {
		accounting.entries++
		if accounting.entries > limit {
			visitError = errImportDirectoryEntryLimit
			return false
		}
		if err := visit(entry); err != nil {
			visitError = err
			return false
		}
		return true
	})
	if visitError != nil {
		return visitError
	}
	return err
}