#53: Support for oneof custom options

Author: kshchepanovskyi

build.gradle

@@ -32,7 +32,7 @@ dependencies {
         exclude group: 'com.jetbrains'
     }
     compile 'com.google.guava:guava:21.0'
-    compile 'io.protostuff:protostuff-parser:2.1.9'
+    compile 'io.protostuff:protostuff-parser:2.1.11'
 }
 
 apply plugin: 'idea'

src/main/java/io/protostuff/jetbrains/plugin/reference/FieldReferenceProviderImpl.java

@@ -6,6 +6,7 @@
 import static io.protostuff.compiler.model.ProtobufConstants.MSG_FILE_OPTIONS;
 import static io.protostuff.compiler.model.ProtobufConstants.MSG_MESSAGE_OPTIONS;
 import static io.protostuff.compiler.model.ProtobufConstants.MSG_METHOD_OPTIONS;
+import static io.protostuff.compiler.model.ProtobufConstants.MSG_ONEOF_OPTIONS;
 import static io.protostuff.compiler.model.ProtobufConstants.MSG_SERVICE_OPTIONS;
 
 import com.google.common.base.Strings;
@@ -27,6 +28,7 @@
 import io.protostuff.jetbrains.plugin.psi.MapNode;
 import io.protostuff.jetbrains.plugin.psi.MessageField;
 import io.protostuff.jetbrains.plugin.psi.MessageNode;
+import io.protostuff.jetbrains.plugin.psi.OneOfNode;
 import io.protostuff.jetbrains.plugin.psi.ProtoPsiFileRoot;
 import io.protostuff.jetbrains.plugin.psi.ProtoRootNode;
 import io.protostuff.jetbrains.plugin.psi.RpcMethodNode;
@@ -66,6 +68,7 @@ public class FieldReferenceProviderImpl implements FieldReferenceProvider {
             .put(RpcMethodNode.class, MSG_METHOD_OPTIONS)
             .put(ServiceNode.class, MSG_SERVICE_OPTIONS)
             .put(ProtoRootNode.class, MSG_FILE_OPTIONS)
+            .put(OneOfNode.class, MSG_ONEOF_OPTIONS)
             .build();
 
     public FieldReferenceProviderImpl(Project project) {

src/test/java/io/protostuff/jetbrains/plugin/rename/field/RenameFieldTest.java

@@ -25,6 +25,13 @@ public void testRenameField_caretAtFieldName() {
         Assert.assertNotNull(option);
     }
 
+    public void testRenameField_caretAtFieldName_OneofOption() {
+        myFixture.configureByFiles("rename/field/RenameField_CaretAtField_OneofOption.proto");
+        myFixture.renameElementAtCaret("NewName");
+        OptionNode option = myFixture.findElementByText("(foo).NewName", OptionNode.class);
+        Assert.assertNotNull(option);
+    }
+
     public void testRenameField_caretAtOption() {
         myFixture.configureByFiles("rename/field/RenameField_CaretAtOption.proto");
         myFixture.renameElementAtCaret("NewName");

src/test/resources/rename/field/RenameField_CaretAtField_OneofOption.proto

@@ -0,0 +1,21 @@
+syntax = "proto2";
+
+package rename.field;
+
+import "google/protobuf/descriptor.proto";
+
+
+extend google.protobuf.OneofOptions {
+    optional Foo foo = 52003;
+}
+
+message Foo {
+    optional int32 name<caret> = 1;
+}
+
+message Bar {
+    oneof oneof {
+        option (foo).name = 1;
+        optional string field = 1;
+    }
+}