Refactor AttributeGraphVendor to RawRepresentable struct (#222)

Kyle-Ye · web-flow · commit 30d5d3449745 · 2026-03-11T00:03:24.000+08:00
diff --git a/Sources/OpenAttributeGraphShims/OAGShims.swift b/Sources/OpenAttributeGraphShims/OAGShims.swift
@@ -1,11 +1,27 @@
 //
-//  GraphShims.swift
+//  AGShims.swift
 //  OpenAttributeGraphShims
 
-public enum AttributeGraphVendor: String {
-    case oag = "org.OpenSwiftUIProject.OpenAttributeGraph"
-    case ag = "com.apple.AttributeGraph"
-    case compute = "dev.incrematic.compute"
+/// A type that identifies the underlying attribute graph implementation vendor.
+///
+/// Use `attributeGraphVendor` to check which vendor is active at runtime.
+public struct AttributeGraphVendor: RawRepresentable, Hashable, CaseIterable {
+    public let rawValue: String
+
+    public init(rawValue: String) {
+        self.rawValue = rawValue
+    }
+
+    /// OpenAttributeGraph — the open-source implementation by OpenSwiftUIProject.
+    public static let oag = AttributeGraphVendor(rawValue: "org.OpenSwiftUIProject.OpenAttributeGraph")
+
+    /// Apple's private AttributeGraph framework.
+    public static let ag = AttributeGraphVendor(rawValue: "com.apple.AttributeGraph")
+
+    /// An incremental computation library for Swift by @jcmosc
+    public static let compute = AttributeGraphVendor(rawValue: "dev.incrematic.compute")
+
+    public static var allCases: [AttributeGraphVendor] { [.oag, .ag, .compute] }
 }
 
 #if OPENATTRIBUTEGRAPH_COMPUTE
diff --git a/Tests/OpenAttributeGraphCompatibilityTests/OAGShims.swift b/Tests/OpenAttributeGraphCompatibilityTests/OAGShims.swift
@@ -1,5 +1,5 @@
 //
-//  GraphShims.swift
+//  OAGShims.swift
 //  OpenAttributeGraphCompatibilityTests
 
 #if OPENATTRIBUTEGRAPH
