save extensions from field options in message props

Author: artemeff

Makefile

@@ -16,4 +16,4 @@ gen-protos: protoc-gen-elixir
 	protoc -I src -I test/protobuf/protoc/proto --elixir_out=test/protobuf/protoc/proto_gen --plugin=./protoc-gen-elixir test/protobuf/protoc/proto/*.proto
 	protoc -I src --elixir_out=lib --plugin=./protoc-gen-elixir elixirpb.proto
 
-.PHONY: clean gen_google_proto gen_test_protos
+.PHONY: clean protoc-gen-elixir gen-google-protos gen-protos

lib/protobuf/dsl.ex

@@ -258,6 +258,10 @@ defmodule Protobuf.DSL do
     |> cal_encoded_fnum()
   end
 
+  defp parse_field_opts([{:extensions, map} | t], acc) do
+    parse_field_opts(t, Map.put(acc, :extensions, map))
+  end
+
   defp parse_field_opts([{:optional, true} | t], acc) do
     parse_field_opts(t, Map.put(acc, :optional?, true))
   end

lib/protobuf/field_props.ex

@@ -17,7 +17,8 @@ defmodule Protobuf.FieldProps do
           packed?: boolean,
           map?: boolean,
           deprecated?: boolean,
-          encoded_fnum: iodata
+          encoded_fnum: iodata,
+          extensions: map
         }
   defstruct fnum: nil,
             name: nil,
@@ -34,5 +35,6 @@ defmodule Protobuf.FieldProps do
             packed?: nil,
             map?: false,
             deprecated?: false,
-            encoded_fnum: nil
+            encoded_fnum: nil,
+            extensions: %{}
 end

lib/protobuf/protoc/generator/message.ex

@@ -288,6 +288,13 @@ defmodule Protobuf.Protoc.Generator.Message do
   end
 
   defp merge_field_options(opts, f) do
+    opts =
+      if map_size(f.options.__pb_extensions__) > 0 do
+        Map.put(opts, :extensions, inspect(f.options.__pb_extensions__))
+      else
+        opts
+      end
+
     opts
     |> Map.put(:packed, f.options.packed)
     |> Map.put(:deprecated, f.options.deprecated)

test/protobuf/protoc/generator/message_test.exs

@@ -143,13 +143,20 @@ defmodule Protobuf.Protoc.Generator.MessageTest do
             number: 1,
             type: :TYPE_INT32,
             label: :LABEL_OPTIONAL,
-            options: Google.Protobuf.FieldOptions.new(packed: true)
+            options:
+              Google.Protobuf.FieldOptions.new(
+                packed: true,
+                __pb_extensions__: %{{Mypkg.PbExtension, :myopt_bool} => true}
+              )
           )
         ]
       )
 
     {[], [msg]} = Generator.generate(ctx, desc)
-    assert msg =~ "field :a, 1, optional: true, type: :int32, packed: true\n"
+
+    assert msg =~
+             "field :a, 1, optional: true, type: :int32, " <>
+               "extensions: %{{Mypkg.PbExtension, :myopt_bool} => true}, packed: true\n"
   end
 
   test "generate/2 supports option :deprecated" do

test/protobuf/protoc/integration_test.exs

@@ -50,6 +50,10 @@ defmodule Protobuf.Protoc.IntegrationTest do
 
   test "options" do
     assert %{deprecated?: true} = My.Test.Options.__message_props__().field_props[1]
+
+    assert %{
+             extensions: %{{Mypkg.PbExtension, :myopt_bool} => true}
+           } = My.Test.Options.__message_props__().field_props[2]
   end
 
   test "extensions" do

test/protobuf/protoc/proto/extension.proto

@@ -4,9 +4,12 @@ package ext;
 
 // -I src is needed, see Makefile
 import "elixirpb.proto";
+import "mypkg.proto";
 
 option (elixirpb.file).module_prefix = "Protobuf.Protoc.ExtTest";
 
 message Foo {
   optional string a = 1;
+  optional string b = 2 [(mypkg.myopt_bool)=true];
+  optional string c = 3 [(mypkg.myopt_string)="test"];
 }

test/protobuf/protoc/proto/mypkg.proto

@@ -0,0 +1,10 @@
+syntax = "proto2";
+
+package mypkg;
+
+import "google/protobuf/descriptor.proto";
+
+extend google.protobuf.FieldOptions {
+  optional bool myopt_bool = 100000;
+  optional string myopt_string = 100001 [default="test"];
+}

test/protobuf/protoc/proto/test.proto

@@ -6,6 +6,9 @@ package my.test;  // dotted package name
 //import "imp.proto";
 import "multi/multi1.proto";  // unused import
 
+// extensions
+import "mypkg.proto";
+
 enum HatType {
   // deliberately skipping 0
   FEDORA = 1;
@@ -125,4 +128,5 @@ message Communique {
 
 message Options {
   optional string opt1 = 1 [deprecated=true];
+  optional string opt2 = 2 [(mypkg.myopt_bool)=true];
 }

test/protobuf/protoc/proto_gen/extension.pb.ex

@@ -3,9 +3,21 @@ defmodule Protobuf.Protoc.ExtTest.Foo do
   use Protobuf, syntax: :proto2
 
   @type t :: %__MODULE__{
-          a: String.t()
+          a: String.t(),
+          b: String.t(),
+          c: String.t()
         }
-  defstruct [:a]
+  defstruct [:a, :b, :c]
 
   field :a, 1, optional: true, type: :string
+
+  field :b, 2,
+    optional: true,
+    type: :string,
+    extensions: %{{Mypkg.PbExtension, :myopt_bool} => true}
+
+  field :c, 3,
+    optional: true,
+    type: :string,
+    extensions: %{{Mypkg.PbExtension, :myopt_string} => "test"}
 end