Add @GraphContentBuilder<NodeID> to graph content body

SwiftGraphs · Jan 14, 2025 · 1b8799e · 1b8799e
1 parent 962016a
commit 1b8799e
Show file tree

Hide file tree

Showing 14 changed files with 60 additions and 42 deletions.
diff --git a/Sources/Grape/Contents/AnyGraphContent.swift b/Sources/Grape/Contents/AnyGraphContent.swift
@@ -9,9 +9,10 @@ public struct AnyGraphContent<NodeID: Hashable>: GraphContent {
         self.storage = storage
     }
 
+
     @inlinable
     public var body: _IdentifiableNever<NodeID> {
-        fatalError()
+        _IdentifiableNever<_>()
     }
 
     @inlinable

diff --git a/Sources/Grape/Contents/CompositedGraphContent.swift b/Sources/Grape/Contents/CompositedGraphContent.swift
diff --git a/Sources/Grape/Contents/GraphContent.swift b/Sources/Grape/Contents/GraphContent.swift
@@ -9,6 +9,7 @@ public protocol GraphContent<NodeID> {
     func _attachToGraphRenderingContext(_ context: inout _GraphRenderingContext<NodeID>)
 
     @inlinable
+    @GraphContentBuilder<NodeID>
     var body: Body { get }
 }
 

diff --git a/Sources/Grape/Contents/LinkMark.swift b/Sources/Grape/Contents/LinkMark.swift
@@ -5,8 +5,9 @@ public struct LinkMark<NodeID: Hashable>: GraphContent & Identifiable {
 
     @inlinable
     public var body: _IdentifiableNever<NodeID> {
-        fatalError()
+        _IdentifiableNever<_>()
     }
+
     // public enum LabelDisplayStrategy {
     //     case auto
     //     case specified(Bool)

diff --git a/Sources/Grape/Contents/ModifiedGraphContent.swift b/Sources/Grape/Contents/ModifiedGraphContent.swift
@@ -50,8 +50,9 @@ extension ModifiedGraphContent: GraphContent {
         modifier._exit(&context)
     }
 
-        @inlinable
+
+    @inlinable
     public var body: _IdentifiableNever<NodeID> {
-        fatalError()
+        _IdentifiableNever<_>()
     }
 }
diff --git a/Sources/Grape/Contents/NodeMark.swift b/Sources/Grape/Contents/NodeMark.swift
@@ -15,9 +15,9 @@ public struct NodeMark<NodeID: Hashable>: GraphContent, Identifiable, Equatable
 
     @inlinable
     public var body: _IdentifiableNever<NodeID> {
-        fatalError()
+        _IdentifiableNever<_>()
     }
-
+    
     @inlinable
     public func _attachToGraphRenderingContext(_ context: inout _GraphRenderingContext<NodeID>) {
         context.nodeOperations.append(

diff --git a/Sources/Grape/Contents/_ArrayGraphContent.swift b/Sources/Grape/Contents/_ArrayGraphContent.swift
@@ -13,9 +13,10 @@ where C: GraphContent {
         self.storage = storage
     }
 
+
     @inlinable
     public var body: _IdentifiableNever<NodeID> {
-        fatalError()
+        _IdentifiableNever<_>()
     }
 
     @inlinable

diff --git a/Sources/Grape/Contents/_ConditionalGraphContent.swift b/Sources/Grape/Contents/_ConditionalGraphContent.swift
@@ -18,9 +18,10 @@ where C1: GraphContent, C2: GraphContent, C1.NodeID == C2.NodeID {
         self.storage = storage
     }
 
-        @inlinable
+
+    @inlinable
     public var body: _IdentifiableNever<NodeID> {
-        fatalError()
+        _IdentifiableNever<_>()
     }
 
     @inlinable

diff --git a/Sources/Grape/Contents/_EmptyGraphContent.swift b/Sources/Grape/Contents/_EmptyGraphContent.swift
@@ -2,15 +2,15 @@
 struct _EmptyGraphContent<NodeID: Hashable>: GraphContent {
     @inlinable
     public init() {
-        
+
     }
     @inlinable
     public func _attachToGraphRenderingContext(_ context: inout _GraphRenderingContext<NodeID>) {
-        
+
     }
 
-        @inlinable
+    @inlinable
     public var body: _IdentifiableNever<NodeID> {
-        fatalError()
+        _IdentifiableNever<_>()
     }
-}
+}
diff --git a/Sources/Grape/Contents/_IdentifiableNever.swift b/Sources/Grape/Contents/_IdentifiableNever.swift
@@ -1,11 +1,22 @@
-public enum _IdentifiableNever<ID: Hashable> { }
+public enum _IdentifiableNever<ID: Hashable> {
+    @usableFromInline
+    internal init() {
+        fatalError()
+    }
+}
+
+
+@inlinable
+public func _fatalError<ID>(of identityType: ID) -> _IdentifiableNever<ID> where ID: Hashable {
+    _IdentifiableNever<ID>()
+}
 
 extension _IdentifiableNever: GraphContent {
     public typealias NodeID = ID
 
     @inlinable
     public var body: Self {
-        fatalError()
+        _IdentifiableNever<ID>()
     }
 
     @inlinable

diff --git a/Sources/Grape/Contents/_OptionalGraphContent.swift b/Sources/Grape/Contents/_OptionalGraphContent.swift
@@ -24,8 +24,9 @@ where C: GraphContent {
         }
     }
 
-        @inlinable
+
+    @inlinable
     public var body: _IdentifiableNever<NodeID> {
-        fatalError()
+        _IdentifiableNever<_>()
     }
 }
diff --git a/Sources/Grape/Contents/_PairedGraphContent.swift b/Sources/Grape/Contents/_PairedGraphContent.swift
@@ -21,8 +21,9 @@ where C1: GraphContent, C2: GraphContent, NodeID: Hashable, C1.NodeID == NodeID,
     }
 
 
+
     @inlinable
     public var body: _IdentifiableNever<NodeID> {
-        fatalError()
+        _IdentifiableNever<_>()
     }
 }
diff --git a/Tests/GrapeTests/ContentBuilderTests.swift b/Tests/GrapeTests/ContentBuilderTests.swift
@@ -50,16 +50,31 @@ final class ContentBuilderTests: XCTestCase {
     }
 
     func testForEach() {
-        let _ = [
+        let arr = [
             ID(id: 0),
             ID(id: 1),
             ID(id: 2),
         ]
 
-        // let _ = buildGraph {
-        //     ForEach(data: arr) { i in
-        //         NodeMark(id: i.id)
-        //     }
-        // }
+        let _ = buildGraph {
+            Series(arr) { i in
+                NodeMark(id: i.id)
+            }
+        }
+    }
+
+    func testComposing() {
+        struct MyGraphContent: GraphContent {
+            var body: some GraphContent<Int> {
+                NodeMark(id: 1)
+                AnnotationNodeMark(id: 3, radius: 4.0) {
+                    EmptyView()
+                }
+            }
+        }
+
+        let _ = buildGraph {
+            MyGraphContent()
+        }
     }
 }
diff --git a/Tests/GrapeTests/GraphContentBuilderTests.swift b/Tests/GrapeTests/GraphContentBuilderTests.swift
@@ -179,8 +179,7 @@ final class GraphContentBuilderTests: XCTestCase {
 
     }
 
-    struct MyGraphComponent: GraphComponent {
-        typealias NodeID = Int
+    struct MyGraphComponent: GraphContent {
         var body: some GraphContent<Int> {
             NodeMark(id: 0)
                 // .opacity(0.6)