Added a getter for System's GuidAttribute

tristanlabelle · Oct 17, 2023 · fedf343 · fedf343
1 parent f83920e
commit fedf343
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/Sources/DotNetMetadata/SystemAttributes.swift b/Sources/DotNetMetadata/SystemAttributes.swift
@@ -44,10 +44,20 @@ public enum SystemAttributes {
         let arguments = try attribute.arguments
         guard arguments.count == 1,
             case .constant(let constant) = arguments[0],
-            case .boolean(let value) = constant else {
-            throw InvalidMetadataError.attributeArguments
+            case .boolean(let value) = constant else { throw InvalidMetadataError.attributeArguments }
+
+        return value
+    }
+
+    public static func getGuid(_ type: TypeDefinition) throws -> String? {
+        guard let attribute = try type.firstAttribute(namespace: "System.Runtime.InteropServices", name: "GuidAttribute") else {
+            return nil
         }
 
+        let arguments = try attribute.arguments
+        guard arguments.count == 1,
+            case .constant(let constant) = arguments[0],
+            case .string(let value) = constant else { throw InvalidMetadataError.attributeArguments }
         return value
     }
 }