re

Author: Alfus

internal/protoschema/plugin/pluginsourceinfo/pluginsourceinfo.go

@@ -34,6 +34,12 @@ func GetSourceInfoPath(fileDescriptor protoreflect.FileDescriptor) string {
 	return fmt.Sprintf("%s%s", path, FileExtension)
 }
 
+func GenFileContents(fileDescriptor protoreflect.FileDescriptor) ([]byte, error) {
+	// Convert the file descriptor to a descriptorpb.FileDescriptorProto.
+	fileDescProto := protodesc.ToFileDescriptorProto(fileDescriptor)
+	return proto.Marshal(fileDescProto.SourceCodeInfo)
+}
+
 // Handle implements protoplugin.Handler and is the main entry point for the plugin.
 func Handle(
 	_ context.Context,
@@ -58,9 +64,3 @@ func Handle(
 	responseWriter.SetFeatureProto3Optional()
 	return nil
 }
-
-func GenFileContents(fileDescriptor protoreflect.FileDescriptor) ([]byte, error) {
-	// Convert the file descriptor to a descriptorpb.FileDescriptorProto.
-	fileDescProto := protodesc.ToFileDescriptorProto(fileDescriptor)
-	return proto.Marshal(fileDescProto.SourceCodeInfo)
-}

sourceinfo/sourceinfo.go

@@ -30,11 +30,11 @@ import (
 )
 
 var (
-	// SourceInfoGlobalTypes is a replacement for protoregistry.GlobalTypes that includes
+	// GlobalTypes is a replacement for protoregistry.GlobalTypes that includes
 	// all registered source info.
 	GlobalFiles protodesc.Resolver = sourceinfo.GlobalFiles
 
-	// SourceInfoGlobalFiles is a replacement for protoregistry.GlobalFiles that includes
+	// GlobalFiles is a replacement for protoregistry.GlobalFiles that includes
 	// all registered source info.
 	GlobalTypes interface {
 		protoregistry.MessageTypeResolver
@@ -43,12 +43,12 @@ var (
 	} = sourceinfo.GlobalTypes
 )
 
-// RegisterAllSourceInfo registers all sourceinfo files under the given output path.
+// RegisterAll registers all sourceinfo files under the given output path.
 func RegisterAll(root string) error {
 	return RegisterAllFS(os.DirFS(root), ".")
 }
 
-// RegisterAllSourceInfoFS registers all sourceinfo files under the given output path, using the given fs.FS.
+// RegisterAllFS registers all sourceinfo files under the given output path, using the given fs.FS.
 func RegisterAllFS(fsys fs.FS, root string) error {
 	return fs.WalkDir(fsys, root, func(path string, d fs.DirEntry, err error) error {
 		if err != nil {