Merge pull request #672 from rolfbjarne/objcinterface-protocols

Add support for protocols in @interface definitions (ObjCInterfaceDecl).

Author: tannergooding

sources/libClangSharp/ClangSharp.cpp

@@ -3446,6 +3446,15 @@ int clangsharp_Cursor_getNumProtocols(CXCursor C) {
         if (const ObjCProtocolDecl* OCPD = dyn_cast<ObjCProtocolDecl>(D)) {
             return OCPD->protocol_size();
         }
+
+        if (const ObjCInterfaceDecl* OCID = dyn_cast<ObjCInterfaceDecl>(D)) {
+            unsigned n = 0;
+
+            for (auto protocol : OCID->protocols()) {
+                n++;
+            }
+            return n;
+        }
     }
 
     return -1;
@@ -3764,6 +3773,17 @@ CXCursor clangsharp_Cursor_getProtocol(CXCursor C, unsigned i) {
                 n++;
             }
         }
+
+        if (const ObjCInterfaceDecl* OCID = dyn_cast<ObjCInterfaceDecl>(D)) {
+            unsigned n = 0;
+
+            for (auto protocol : OCID->protocols()) {
+                if (n == i) {
+                    return MakeCXCursor(protocol, getCursorTU(C));
+                }
+                n++;
+            }
+        }
     }
 
     return clang_getNullCursor();

tests/ClangSharp.UnitTests/ObjectiveCTest.cs

@@ -64,6 +64,33 @@ @interface MyClass (MyCategory)
         }
     }
 
+    [Test]
+    public void ClassWithProtocols()
+    {
+        AssertNeedNewClangSharp();
+
+        var inputContents = $@"
+@protocol P1
+@end
+@protocol P2
+@end
+
+@interface MyClass <P1, P2>
+@end
+";
+
+        using var translationUnit = CreateTranslationUnit(inputContents, "objective-c++");
+
+        var classes = translationUnit.TranslationUnitDecl.Decls.OfType<ObjCInterfaceDecl>().ToList();
+        Assert.That(classes.Count, Is.GreaterThanOrEqualTo(1), $"At least one class");
+        var myClass = classes.SingleOrDefault(v => v.Name == "MyClass")!;
+        Assert.That(myClass, Is.Not.Null, "MyClass");
+        var protocols = myClass.Protocols.ToList();
+        Assert.That(protocols.Count, Is.EqualTo(2), "protocols.Count");
+        Assert.That(protocols[0].Name, Is.EqualTo("P1"), "protocols[0].Name");
+        Assert.That(protocols[1].Name, Is.EqualTo("P2"), "protocols[1].Name");
+    }
+
     [Test]
     public void Method_IsPropertyAccessor()
     {