Function NewModuleSetForProtoc

private/buf/bufprotoc/bufprotoc.go:39–88

NewModuleSetForProtoc returns a new ModuleSet for protoc -I dirPaths and filePaths. The returned ModuleSet will have a single targeted Module, with target files matching the filePaths. Technically this will work with len(filePaths) == 0 but we should probably make sure that is banned in protoc.

(
	ctx context.Context,
	logger *slog.Logger,
	storageosProvider storageos.Provider,
	includeDirPaths []string,
	filePaths []string,
)

// Technically this will work with len(filePaths) == 0 but we should probably make sure
// that is banned in protoc.
func NewModuleSetForProtoc(
	ctx context.Context,
	logger *slog.Logger,
	storageosProvider storageos.Provider,
	includeDirPaths []string,
	filePaths []string,
) (bufmodule.ModuleSet, error) {
	absIncludeDirPaths, err := normalizeAndAbsolutePaths(includeDirPaths, "include directory")
	if err != nil {
		return nil, err
	}
	absFilePaths, err := normalizeAndAbsolutePaths(filePaths, "input file")
	if err != nil {
		return nil, err
	}
	var rootBuckets []storage.ReadBucket
	for _, includeDirPath := range includeDirPaths {
		rootBucket, err := storageosProvider.NewReadWriteBucket(
			includeDirPath,
			storageos.ReadWriteBucketWithSymlinksIfSupported(),
		)
		if err != nil {
			return nil, err
		}
		// need to do match extension here
		// https://github.com/bufbuild/buf/issues/113
		rootBuckets = append(rootBuckets, storage.FilterReadBucket(rootBucket, storage.MatchPathExt(".proto")))
	}
	targetPaths, err := xslices.MapError(
		absFilePaths,
		func(absFilePath string) (string, error) {
			return applyRootsToTargetPath(absIncludeDirPaths, absFilePath, normalpath.Absolute)
		},
	)
	if err != nil {
		return nil, err
	}

	moduleSetBuilder := bufmodule.NewModuleSetBuilder(ctx, logger, bufmodule.NopModuleDataProvider, bufmodule.NopCommitProvider)
	moduleSetBuilder.AddLocalModule(
		storage.MultiReadBucket(rootBuckets...),
		".",
		true,
		bufmodule.LocalModuleWithTargetPaths(
			targetPaths,
			nil,
		),
	)
	return moduleSetBuilder.Build()
}

// *** PRIVATE ***