71 lines
2.0 KiB
Swift
71 lines
2.0 KiB
Swift
// Copyright 2020 Tokamak contributors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
// Created by Carson Katri on 7/10/21.
|
|
//
|
|
|
|
public protocol _ViewTraitKey {
|
|
associatedtype Value
|
|
static var defaultValue: Value { get }
|
|
}
|
|
|
|
public protocol _TraitWritingModifierProtocol {
|
|
func modifyViewTraitStore(_ viewTraitStore: inout _ViewTraitStore)
|
|
}
|
|
|
|
@frozen
|
|
public struct _TraitWritingModifier<Trait>: ViewModifier, _TraitWritingModifierProtocol
|
|
where Trait: _ViewTraitKey
|
|
{
|
|
public let value: Trait.Value
|
|
@inlinable
|
|
public init(value: Trait.Value) {
|
|
self.value = value
|
|
}
|
|
|
|
public func body(content: Content) -> some View {
|
|
content
|
|
}
|
|
|
|
public func modifyViewTraitStore(_ viewTraitStore: inout _ViewTraitStore) {
|
|
viewTraitStore.insert(value, forKey: Trait.self)
|
|
}
|
|
|
|
public static func _makeView(_ inputs: ViewInputs<_TraitWritingModifier<Trait>>)
|
|
-> ViewOutputs
|
|
{
|
|
var store = inputs.traits ?? .init()
|
|
store.insert(inputs.content.value, forKey: Trait.self)
|
|
return .init(inputs: inputs, traits: store)
|
|
}
|
|
}
|
|
|
|
extension ModifiedContent: _TraitWritingModifierProtocol
|
|
where Modifier: _TraitWritingModifierProtocol
|
|
{
|
|
public func modifyViewTraitStore(_ viewTraitStore: inout _ViewTraitStore) {
|
|
modifier.modifyViewTraitStore(&viewTraitStore)
|
|
}
|
|
}
|
|
|
|
public extension View {
|
|
@inlinable
|
|
func _trait<K>(
|
|
_ key: K.Type,
|
|
_ value: K.Value
|
|
) -> some View where K: _ViewTraitKey {
|
|
modifier(_TraitWritingModifier<K>(value: value))
|
|
}
|
|
}
|