Add components function builder implementation

This commit is contained in:
Christian Treffs 2020-07-30 15:33:49 +02:00
parent 09bfa595c3
commit 2440ec8ed9
No known key found for this signature in database
GPG Key ID: 49A4B4B460BE3ED4
1 changed files with 47 additions and 0 deletions

View File

@ -0,0 +1,47 @@
//
// Nexus+ComponentsBuilder.swift
//
//
// Created by Christian Treffs on 30.07.20.
//
#if swift(<5.3)
@_functionBuilder
public enum ComponentsBuilderPreview { }
public typealias ComponentsBuilder = ComponentsBuilderPreview
#else
@functionBuilder
public enum ComponentsBuilder { }
#endif
extension ComponentsBuilder {
public static func buildBlock(_ components: Component...) -> [Component] {
components
}
public struct Context {
public let index: Int
}
}
extension Nexus {
@discardableResult
public func createEntity(@ComponentsBuilder using componentBuilder: () -> Component) -> Entity {
self.createEntity(with: componentBuilder())
}
@discardableResult
public func createEntity(@ComponentsBuilder using componentBuilder: () -> [Component]) -> Entity {
self.createEntity(with: componentBuilder())
}
@discardableResult
public func createEntities(count: Int, @ComponentsBuilder using componentBuilder: (ComponentsBuilder.Context) -> Component) -> [Entity] {
(0..<count).map { self.createEntity(with: componentBuilder(ComponentsBuilder.Context(index: $0))) }
}
@discardableResult
public func createEntities(count: Int, @ComponentsBuilder using componentBuilder: (ComponentsBuilder.Context) -> [Component]) -> [Entity] {
(0..<count).map { self.createEntity(with: componentBuilder(ComponentsBuilder.Context(index: $0))) }
}
}