From e99ac63b661ca7f4216e391f3f7a2da0e1f2f188 Mon Sep 17 00:00:00 2001
From: Carson Katri <Carson.katri@gmail.com>
Date: Thu, 25 Jul 2024 10:31:57 -0400
Subject: [PATCH] Support building for iOS 18

---
 Package.swift                                 |  1 +
 .../Components/AnchoringComponent.swift       |  4 ++
 .../EntityContentBuilder.swift                |  6 +++
 .../ARKitSession/ARKitSession+shared.swift    |  2 +
 .../Entities/ARKitSession/HandTracking.swift  |  2 +
 .../ARKitSession/SceneReconstruction.swift    |  2 +
 .../RealityViewCameraControlsModifier.swift   | 40 +++++++++++++++++++
 .../LiveViewNativeRealityKit/RealityKit.swift | 31 ++++++++++++++
 .../RealityView.swift                         | 10 +++++
 ...RealityViewCamera+AttributeDecodable.swift | 31 ++++++++++++++
 10 files changed, 129 insertions(+)
 create mode 100644 Sources/LiveViewNativeRealityKit/Modifiers/RealityViewCameraControlsModifier.swift
 create mode 100644 Sources/LiveViewNativeRealityKit/Types/RealityViewCamera+AttributeDecodable.swift

diff --git a/Package.swift b/Package.swift
index 691fb18..7d5abc2 100644
--- a/Package.swift
+++ b/Package.swift
@@ -5,6 +5,7 @@ import PackageDescription
 
 let package = Package(
     name: "LiveViewNativeRealityKit",
+    platforms: [.visionOS(.v1), .iOS("18.0")],
     products: [
         // Products define the executables and libraries a package produces, making them visible to other packages.
         .library(
diff --git a/Sources/LiveViewNativeRealityKit/Components/AnchoringComponent.swift b/Sources/LiveViewNativeRealityKit/Components/AnchoringComponent.swift
index bcb8462..b5a8acc 100644
--- a/Sources/LiveViewNativeRealityKit/Components/AnchoringComponent.swift
+++ b/Sources/LiveViewNativeRealityKit/Components/AnchoringComponent.swift
@@ -40,6 +40,7 @@ extension AnchoringComponent.Target: AttributeDecodable {
         guard let value = attribute?.value
         else { throw AttributeDecodingError.missingAttribute(Self.self) }
         switch value {
+        #if os(visionOS)
         case "hand":
             self = .hand(
                 try element.attributeValue(Chirality.self, for: "chirality"),
@@ -47,6 +48,7 @@ extension AnchoringComponent.Target: AttributeDecodable {
             )
         case "head":
             self = .head
+        #endif
         case "image":
             self = .image(
                 group: try element.attributeValue(String.self, for: "group"),
@@ -64,6 +66,7 @@ extension AnchoringComponent.Target: AttributeDecodable {
     }
 }
 
+#if os(visionOS)
 extension AnchoringComponent.Target.Chirality: AttributeDecodable {
     public init(from attribute: Attribute?, on element: ElementNode) throws {
         guard let value = attribute?.value
@@ -99,6 +102,7 @@ extension AnchoringComponent.Target.HandLocation: AttributeDecodable {
         }
     }
 }
+#endif
 
 extension AnchoringComponent.Target.Alignment: AttributeDecodable {
     public init(from attribute: Attribute?, on element: ElementNode) throws {
diff --git a/Sources/LiveViewNativeRealityKit/ContentBuilders/EntityContentBuilder.swift b/Sources/LiveViewNativeRealityKit/ContentBuilders/EntityContentBuilder.swift
index 4706de4..8ef418a 100644
--- a/Sources/LiveViewNativeRealityKit/ContentBuilders/EntityContentBuilder.swift
+++ b/Sources/LiveViewNativeRealityKit/ContentBuilders/EntityContentBuilder.swift
@@ -26,8 +26,10 @@ public struct EntityContentBuilder<Entities: EntityRegistry, Components: Compone
             case modelEntity = "ModelEntity"
             case anchorEntity = "AnchorEntity"
             
+            #if os(visionOS)
             case sceneReconstructionProvider = "SceneReconstructionProvider"
             case handTrackingProvider = "HandTrackingProvider"
+            #endif
         }
         
         public init?(rawValue: RawValue) {
@@ -69,10 +71,12 @@ public struct EntityContentBuilder<Entities: EntityRegistry, Components: Compone
                     entity = try ModelEntity(from: element, in: context)
                 case .anchorEntity:
                     entity = try AnchorEntity(from: element, in: context)
+                #if os(visionOS)
                 case .sceneReconstructionProvider:
                     entity = try SceneReconstructionEntity(from: element, in: context)
                 case .handTrackingProvider:
                     entity = HandTrackingEntity(from: element, in: context)
+                #endif
                 }
             case .custom:
                 guard let customEntity = (try Self.build([element.node], with: Entities.self, in: context)).first
@@ -213,7 +217,9 @@ extension Entity {
             self.components.set(elementNodeComponent)
         }
         
+        #if os(visionOS)
         guard !(self is SceneReconstructionEntity) else { return } // scene reconstruction creates its own child meshes
+        #endif
         
         /// The list of children previously part of this element.
         ///
diff --git a/Sources/LiveViewNativeRealityKit/Entities/ARKitSession/ARKitSession+shared.swift b/Sources/LiveViewNativeRealityKit/Entities/ARKitSession/ARKitSession+shared.swift
index 5d38457..69e111f 100644
--- a/Sources/LiveViewNativeRealityKit/Entities/ARKitSession/ARKitSession+shared.swift
+++ b/Sources/LiveViewNativeRealityKit/Entities/ARKitSession/ARKitSession+shared.swift
@@ -5,6 +5,7 @@
 //  Created by Carson.Katri on 5/30/24.
 //
 
+#if os(visionOS)
 import ARKit
 
 enum SharedARKitSessionStore {
@@ -19,3 +20,4 @@ enum SharedARKitSessionStore {
         }
     }
 }
+#endif
diff --git a/Sources/LiveViewNativeRealityKit/Entities/ARKitSession/HandTracking.swift b/Sources/LiveViewNativeRealityKit/Entities/ARKitSession/HandTracking.swift
index 1e1382d..be748d1 100644
--- a/Sources/LiveViewNativeRealityKit/Entities/ARKitSession/HandTracking.swift
+++ b/Sources/LiveViewNativeRealityKit/Entities/ARKitSession/HandTracking.swift
@@ -5,6 +5,7 @@
 //  Created by Carson Katri on 5/23/24.
 //
 
+#if os(visionOS)
 import LiveViewNative
 import LiveViewNativeCore
 import ARKit
@@ -152,3 +153,4 @@ class HandTrackingEntity: Entity {
         session.stop()
     }
 }
+#endif
diff --git a/Sources/LiveViewNativeRealityKit/Entities/ARKitSession/SceneReconstruction.swift b/Sources/LiveViewNativeRealityKit/Entities/ARKitSession/SceneReconstruction.swift
index c366978..1f97008 100644
--- a/Sources/LiveViewNativeRealityKit/Entities/ARKitSession/SceneReconstruction.swift
+++ b/Sources/LiveViewNativeRealityKit/Entities/ARKitSession/SceneReconstruction.swift
@@ -5,6 +5,7 @@
 //  Created by Carson Katri on 5/23/24.
 //
 
+#if os(visionOS)
 import LiveViewNative
 import ARKit
 import RealityKit
@@ -128,3 +129,4 @@ extension GeometrySource {
         return asArray(ofType: (T, T, T).self).map { .init($0.0, $0.1, $0.2) }
     }
 }
+#endif
diff --git a/Sources/LiveViewNativeRealityKit/Modifiers/RealityViewCameraControlsModifier.swift b/Sources/LiveViewNativeRealityKit/Modifiers/RealityViewCameraControlsModifier.swift
new file mode 100644
index 0000000..0e1ff76
--- /dev/null
+++ b/Sources/LiveViewNativeRealityKit/Modifiers/RealityViewCameraControlsModifier.swift
@@ -0,0 +1,40 @@
+//
+//  RealityViewCameraControlsModifier.swift
+//  
+//
+//  Created by Carson.Katri on 7/25/24.
+//
+
+#if os(iOS) || os(macOS)
+import LiveViewNative
+import LiveViewNativeStylesheet
+import SwiftUI
+import RealityKit
+
+@ParseableExpression
+struct _RealityViewCameraControlsModifier: ViewModifier {
+    public static var name: String { "realityViewCameraControls" }
+    
+    let controls: CameraControls
+    
+    init(_ controls: CameraControls) {
+        self.controls = controls
+    }
+    
+    func body(content: Content) -> some View {
+        content.realityViewCameraControls(controls)
+    }
+}
+
+extension CameraControls: ParseableModifierValue {
+    public static func parser(in context: ParseableModifierContext) -> some Parser<Substring.UTF8View, Self> {
+        ImplicitStaticMember([
+            "dolly": .dolly,
+            "none": .none,
+            "orbit": .orbit,
+            "pan": .pan,
+            "tilt": .tilt,
+        ])
+    }
+}
+#endif
diff --git a/Sources/LiveViewNativeRealityKit/RealityKit.swift b/Sources/LiveViewNativeRealityKit/RealityKit.swift
index 97eb896..879f229 100644
--- a/Sources/LiveViewNativeRealityKit/RealityKit.swift
+++ b/Sources/LiveViewNativeRealityKit/RealityKit.swift
@@ -1,6 +1,7 @@
 import LiveViewNative
 import LiveViewNativeStylesheet
 import SwiftUI
+import RealityKit
 
 public struct CustomizableRealityKitRegistry<
     Entities: EntityRegistry,
@@ -17,6 +18,36 @@ public struct CustomizableRealityKitRegistry<
                 _RealityView<Root, Entities, _ComponentContentBuilder<Components>>()
             }
         }
+        
+        #if os(iOS) || os(macOS)
+        public static func parseModifier(
+            _ input: inout Substring.UTF8View,
+            in context: ParseableModifierContext
+        ) throws -> CustomModifier {
+            try CustomModifier.parser(in: context).parse(&input)
+        }
+        
+        public struct CustomModifier: ViewModifier, ParseableModifierValue {
+            enum Storage {
+                case realityViewCameraControls(_RealityViewCameraControlsModifier)
+            }
+            
+            let storage: Storage
+            
+            public static func parser(in context: ParseableModifierContext) -> some Parser<Substring.UTF8View, Self> {
+                CustomModifierGroupParser(output: Self.self) {
+                    _RealityViewCameraControlsModifier.parser(in: context).map({ Self(storage: .realityViewCameraControls($0)) })
+                }
+            }
+            
+            public func body(content: Content) -> some View {
+                switch storage {
+                case .realityViewCameraControls(let modifier):
+                    content.modifier(modifier)
+                }
+            }
+        }
+        #endif
     }
 }
 
diff --git a/Sources/LiveViewNativeRealityKit/RealityView.swift b/Sources/LiveViewNativeRealityKit/RealityView.swift
index 1f80d51..d4455a6 100644
--- a/Sources/LiveViewNativeRealityKit/RealityView.swift
+++ b/Sources/LiveViewNativeRealityKit/RealityView.swift
@@ -80,6 +80,8 @@ struct _RealityView<Root: RootRegistry, Entities: EntityRegistry, Components: Co
     
     private var audibleClicks: Bool = false
     
+    private var camera: RealityViewCamera = .worldTracking
+    
     @LiveElementIgnored
     @State
     private var updateStorage = UpdateContextComponent<Root, Entities, Components>.Storage()
@@ -106,6 +108,10 @@ struct _RealityView<Root: RootRegistry, Entities: EntityRegistry, Components: Co
     
     var body: some View {
         RealityView { content in
+            #if os(iOS) || os(macOS)
+            content.camera = self.camera
+            #endif
+            
             self.subscriptions = [
                 content.subscribe(to: CollisionEvents.Began.self, componentType: PhysicsBodyChangeEventComponent.self) { collision in
                     for entity in [collision.entityA, collision.entityB] {
@@ -217,7 +223,11 @@ struct _RealityView<Root: RootRegistry, Entities: EntityRegistry, Components: Co
                             partialResult[String(attr.name.name.dropFirst(prefix.count))] = attr.value
                         }
                     
+                    #if os(visionOS)
                     let tapLocation = value.convert(value.location3D, from: .local, to: .scene)
+                    #else
+                    let tapLocation = value.hitTest(point: value.location, in: .local).first?.position ?? value.entity.position
+                    #endif
                     
                     payload["_location"] = [tapLocation.x, tapLocation.y, tapLocation.z]
                     
diff --git a/Sources/LiveViewNativeRealityKit/Types/RealityViewCamera+AttributeDecodable.swift b/Sources/LiveViewNativeRealityKit/Types/RealityViewCamera+AttributeDecodable.swift
new file mode 100644
index 0000000..c47424f
--- /dev/null
+++ b/Sources/LiveViewNativeRealityKit/Types/RealityViewCamera+AttributeDecodable.swift
@@ -0,0 +1,31 @@
+//
+//  RealityViewCamera+AttributeDecodable.swift
+//  
+//
+//  Created by Carson.Katri on 7/25/24.
+//
+
+#if os(iOS) || os(macOS)
+import LiveViewNative
+import LiveViewNativeCore
+import RealityKit
+import SwiftUI
+
+extension RealityViewCamera: AttributeDecodable {
+    public init(from attribute: Attribute?, on element: ElementNode) throws {
+        guard let value = attribute?.value
+        else { throw AttributeDecodingError.missingAttribute(Self.self) }
+        
+        switch value {
+        case "virtual":
+            self = .virtual
+        case "worldTracking":
+            self = .worldTracking
+        default:
+            throw AttributeDecodingError.badValue(Self.self)
+        }
+    }
+}
+#else
+typealias RealityViewCamera = String
+#endif