Spatial Preview Framework

import SwiftUI
import SpatialPreview

struct ContentView: View {
    @State private var session: DocumentPreviewSession?
    @State private var endpointObserver = ConnectedSpatialEndpointObserver()
    @State private var showDevicePicker = false
    @State private var selectedEndpoint: SpatialPreviewEndpoint?

    let renderingURLs: [URL] = [
        URL(fileURLWithPath: "/tmp/render_living_room.heic"),
        URL(fileURLWithPath: "/tmp/render_kitchen.heic"),
        URL(fileURLWithPath: "/tmp/render_bedroom.heic")
    ]

    var body: some View {
        VStack(spacing: 16) {
            Text("Spatial Rendering Gallery")
                .font(.title2)
                .padding(.top)

            Button("Launch on Vision Pro") {
                Task { await startSession(with: renderingURLs[0]) }
            }
            .disabled(activeEndpoint == nil)

            HStack {
                ForEach(Array(renderingURLs.enumerated()), id: \.offset) { index, url in
                    Button("Render \(index + 1)") {
                        Task { await swapContent(to: url) }
                    }
                    .disabled(session == nil)
                }
            }

            Button("Choose Device…") { showDevicePicker = true }

            Button("Close Session") {
                Task { await session?.close() }
                session = nil
            }
            .disabled(session == nil)
        }
        .padding()
        .sheet(isPresented: $showDevicePicker) {
            SpatialPreviewDevicePicker(selectedEndpoint: $selectedEndpoint)
        }
        .task {
            for await state in session?.stateUpdates ?? AsyncStream.never {
                if case .invalidated = state { session = nil }
            }
        }
    }

    var activeEndpoint: SpatialPreviewEndpoint? {
        selectedEndpoint ?? endpointObserver.endpoint
    }

    func startSession(with url: URL) async {
        guard let endpoint = activeEndpoint else { return }
        let newSession = DocumentPreviewSession(
            documentType: .spatialImage,
            name: "Architectural Renders"
        )
        do {
            try await newSession.start(on: endpoint, contentsURL: url)
            session = newSession
        } catch {
            print("Failed to start session: \(error)")
        }
    }

    func swapContent(to url: URL) async {
        do {
            try await session?.updateContents(contentsURL: url)
        } catch {
            print("Failed to update contents: \(error)")
        }
    }
}