Method WalkFileInfos

private/bufpkg/bufmodule/module_read_bucket.go:345–437

(
	ctx context.Context,
	fn func(FileInfo) error,
	options ...WalkFileInfosOption,
)

}

func (b *moduleReadBucket) WalkFileInfos(
	ctx context.Context,
	fn func(FileInfo) error,
	options ...WalkFileInfosOption,
) error {
	// Note that we must verify that at least one file in this ModuleReadBucket is
	// a .proto file, per the documentation on Module.
	protoFileTracker := newProtoFileTracker()

	walkFileInfosOptions := newWalkFileInfosOptions()
	for _, option := range options {
		option(walkFileInfosOptions)
	}
	bucket, err := b.getBucket()
	if err != nil {
		return err
	}

	if !walkFileInfosOptions.onlyTargetFiles {
		// We only want to call trackModule if we are walking all the files, not just
		// the target files. By not calling trackModule outside of this if statement,
		// we will not produce NoProtoFilesErrors, per the documentation on trackModule.
		protoFileTracker.trackModule(b.module)
		if err := bucket.Walk(
			ctx,
			"",
			func(objectInfo storage.ObjectInfo) error {
				fileInfo, err := b.getFileInfo(ctx, objectInfo)
				if err != nil {
					return err
				}
				protoFileTracker.trackFileInfo(fileInfo)
				return fn(fileInfo)
			},
		); err != nil {
			return err
		}
		return protoFileTracker.validate()
	}

	// If we are walking all files, then we track the module if it is the target module
	if b.module.IsTarget() {
		protoFileTracker.trackModule(b.module)
	}

	targetFileWalkFunc := func(objectInfo storage.ObjectInfo) error {
		fileInfo, err := b.getFileInfo(ctx, objectInfo)
		if err != nil {
			return err
		}
		protoFileTracker.trackFileInfo(fileInfo)
		if !fileInfo.IsTargetFile() {
			return nil
		}
		return fn(fileInfo)
	}

	// If we have target paths, we do not want to walk the whole bucket.