81 lines
1.9 KiB
Swift
81 lines
1.9 KiB
Swift
public protocol NodeModifier {
|
|
associatedtype Body: Node
|
|
// swiftlint:disable type_name
|
|
associatedtype _Content: Node
|
|
typealias Content = _Content
|
|
// swiftlint:enable type_name
|
|
|
|
func render(content: Content) -> Body
|
|
}
|
|
|
|
public extension NodeModifier {
|
|
func concat<T: NodeModifier>(
|
|
_ modifier: T
|
|
) -> ConcatModifier<Content, Self, T> {
|
|
ConcatModifier(first: self, second: modifier)
|
|
}
|
|
}
|
|
|
|
public extension Node {
|
|
|
|
func modifier<T: NodeModifier>(_ modifier: T) -> ModifiedNode<Self, T> {
|
|
.init(content: self, modifier: modifier)
|
|
}
|
|
}
|
|
|
|
public struct ConcatModifier<Content, Modifier1, Modifier2>: NodeModifier where Content: Node,
|
|
Modifier1: NodeModifier, Modifier2: NodeModifier,
|
|
Modifier1.Content == Content, Modifier1.Body == Modifier2.Content
|
|
{
|
|
let first: Modifier1
|
|
let second: Modifier2
|
|
|
|
public func render(content: Content) -> some Node {
|
|
second.render(content: first.render(content: content))
|
|
}
|
|
|
|
}
|
|
|
|
public struct ModifiedNode<Content, Modifier> {
|
|
var content: Content
|
|
var modifier: Modifier
|
|
}
|
|
|
|
extension ModifiedNode: NodeRepresentable where Content: NodeRepresentable,
|
|
Modifier: NodeModifier,
|
|
Modifier.Content == Content
|
|
{
|
|
public func render() -> String {
|
|
modifier.render(content: content).render()
|
|
}
|
|
}
|
|
|
|
extension ModifiedNode: Node where Content: Node,
|
|
Modifier: NodeModifier,
|
|
Modifier.Content == Content
|
|
{
|
|
public var body: some Node {
|
|
content
|
|
}
|
|
}
|
|
|
|
extension ModifiedNode: NodeModifier where
|
|
Content: NodeModifier,
|
|
Modifier: NodeModifier,
|
|
Content.Body == Modifier.Content,
|
|
Content: Node
|
|
{
|
|
public func render(content: Content) -> some Node {
|
|
let body = content.body
|
|
return modifier.render(content: body).render()
|
|
}
|
|
}
|
|
|
|
extension ModifiedNode where Modifier: NodeModifier {
|
|
|
|
func modifier<T: NodeModifier>(_ modifier: T) -> ModifiedNode<Content, ConcatModifier<Content, Modifier, T>> {
|
|
.init(content: content, modifier: self.modifier.concat(modifier))
|
|
}
|
|
|
|
}
|