meta/Sources/View/ContentModifier.swift

70 lines
2.0 KiB
Swift

//
// ContentModifier.swift
// Meta
//
// Created by david-swift on 29.06.24.
//
/// A widget which replaces views of a specific type in its content.
struct ContentModifier<Content>: ConvenienceWidget where Content: AnyView {
/// The wrapped view.
var content: AnyView
/// The closure for the modification.
var modify: (Content) -> AnyView
/// The view storage.
/// - Parameters:
/// - modifiers: Modify views before being updated.
/// - type: The type of the app storage.
func container<Storage>(
modifiers: [(any AnyView) -> any AnyView],
type: Storage.Type
) -> ViewStorage where Storage: AppStorage {
content.storage(modifiers: modifiers + [modifyView], type: type)
}
/// Update the stored content.
/// - Parameters:
/// - storage: The storage to update.
/// - modifiers: Modify views before being updated
/// - updateProperties: Whether to update the view's properties.
/// - type: The type of the app storage.
func update<Storage>(
_ storage: ViewStorage,
modifiers: [(any AnyView) -> any AnyView],
updateProperties: Bool,
type: Storage.Type
) where Storage: AppStorage {
content
.updateStorage(storage, modifiers: modifiers + [modifyView], updateProperties: updateProperties, type: type)
}
/// Apply the modifier to a view.
/// - Parameter view: The view.
func modifyView(_ view: AnyView) -> AnyView {
if let view = view as? Content {
return modify(view).stopModifiers()
} else {
return view
}
}
}
extension AnyView {
/// Replace every occurrence of a certain view type in the content.
/// - Parameters:
/// - type: The view type.
/// - modify: Modify the view.
/// - Returns: A view.
public func modifyContent<Content>(
_ type: Content.Type,
modify: @escaping (Content) -> AnyView
) -> AnyView where Content: AnyView {
ContentModifier(content: self, modify: modify)
}
}