meta/Sources/View/StateWrapper.swift
2024-06-17 17:05:26 +02:00

104 lines
3.3 KiB
Swift

//
// StateWrapper.swift
// Meta
//
// Created by david-swift on 09.06.24.
//
import Observation
/// A storage for `@State` properties.
public struct StateWrapper: ConvenienceWidget {
/// The content.
var content: () -> Body
/// The state information (from properties with the `State` wrapper).
var state: [String: StateProtocol] = [:]
/// The debug tree parameters.
public var debugTreeParameters: [(String, value: CustomStringConvertible)] {
[
("state", value: state)
]
}
/// The debug tree's content.
public var debugTreeContent: [(String, body: Body)] {
[("content", body: content())]
}
/// The identifier of the field storing whether to update the wrapper's content.
private var updateID: String { "update" }
/// Initialize a `StateWrapper`.
/// - Parameter content: The view content.
public init(@ViewBuilder content: @escaping () -> Body) {
self.content = content
}
/// Initialize a `StateWrapper`.
/// - Parameters:
/// - content: The view content.
/// - state: The state information.
init(content: @escaping () -> Body, state: [String: StateProtocol]) {
self.content = content
self.state = state
}
/// Update a view storage.
/// - Parameters:
/// - storage: The view storage.
/// - modifiers: Modify views before being updated.
/// - updateProperties: Whether to update properties.
/// - type: The type of the widgets.
public func update<WidgetType>(
_ storage: ViewStorage,
modifiers: [(AnyView) -> AnyView],
updateProperties: Bool,
type: WidgetType.Type
) {
var updateProperties = storage.fields[updateID] as? Bool ?? false
storage.fields[updateID] = false
for property in state {
if let storage = storage.state[property.key]?.content.storage {
property.value.content.storage = storage
}
if property.value.content.storage.update {
updateProperties = true
property.value.content.storage.update = false
}
}
if let storage = storage.content[.mainContent]?.first {
content()
.widget(modifiers: modifiers)
.update(storage, modifiers: modifiers, updateProperties: updateProperties, type: type)
}
}
/// Get a view storage.
/// - Parameters:
/// - modifiers: Modify views before being updated.
/// - type: The type of the widgets.
/// - Returns: The view storage.
public func container<WidgetType>(modifiers: [(AnyView) -> AnyView], type: WidgetType.Type) -> ViewStorage {
let content = content().storage(modifiers: modifiers, type: type)
let storage = ViewStorage(content.pointer, content: [.mainContent: [content]])
storage.state = state
observe(storage: storage)
return storage
}
/// Observe the observable properties accessed in the view.
/// - Parameter storage: The view storage
func observe(storage: ViewStorage) {
withObservationTracking {
_ = content().getDebugTree(parameters: true, type: AnyView.self)
} onChange: {
storage.fields[updateID] = true
UpdateManager.updateViews()
observe(storage: storage)
}
}
}