Preserve Package (#15)

* preserve package * bump version
alpancs · Nov 8, 2021 · a6e79ed · a6e79ed
1 parent 6651aa2
commit a6e79ed
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/content_builder.go b/content_builder.go
@@ -29,10 +29,11 @@ func (b *contentBuilder) build(protoFile *descriptorpb.FileDescriptorProto) (str
 	compVersion := b.request.GetCompilerVersion()
 	fmt.Fprintf(b.output, "// Code generated by protoc-gen-pubsub-schema. DO NOT EDIT.\n")
 	fmt.Fprintf(b.output, "// versions:\n")
-	fmt.Fprintf(b.output, "// 	protoc-gen-pubsub-schema v1.4.4\n")
+	fmt.Fprintf(b.output, "// 	protoc-gen-pubsub-schema v1.5.0\n")
 	fmt.Fprintf(b.output, "// 	protoc                   v%d.%d.%d%s\n", compVersion.GetMajor(), compVersion.GetMinor(), compVersion.GetPatch(), compVersion.GetSuffix())
 	fmt.Fprintf(b.output, "// source: %s\n\n", protoFile.GetName())
-	fmt.Fprintf(b.output, "syntax = \"%s\";\n", b.schemaSyntax)
+	fmt.Fprintf(b.output, "syntax = \"%s\";\n\n", b.schemaSyntax)
+	fmt.Fprintf(b.output, "package %s;\n", protoFile.GetPackage())
 	b.buildMessages(protoFile.GetMessageType(), 0)
 	b.buildEnums(protoFile.GetEnumType(), 0)
 	return b.output.String(), nil

diff --git a/example/user_add_comment.pps b/example/user_add_comment.pps
@@ -1,11 +1,13 @@
 // Code generated by protoc-gen-pubsub-schema. DO NOT EDIT.
 // versions:
-// 	protoc-gen-pubsub-schema v1.4.4
+// 	protoc-gen-pubsub-schema v1.5.0
 // 	protoc                   v3.19.1
 // source: example/user_add_comment.proto
 
 syntax = "proto2";
 
+package example;
+
 message UserAddComment {
   required User user = 1;
   required string comment = 2;