//
//  Nexus+Entity.swift
//  FirebladeECS
//
//  Created by Christian Treffs on 13.10.17.
//

extension Nexus {
    @discardableResult
    public func createEntity() -> Entity {
        let entityId: EntityIdentifier = entityIdGenerator.nextId()
        componentIdsByEntity[entityId] = []
        delegate?.nexusEvent(EntityCreated(entityId: entityId))
        return Entity(nexus: self, id: entityId)
    }

    @discardableResult
    public func createEntity(with components: Component...) -> Entity {
        let newEntity = createEntity()
        assign(components: components, to: newEntity.identifier)
        return newEntity
    }

    @discardableResult
    public func createEntity<C>(with components: C) -> Entity where C: Collection, C.Element == Component {
        let entity = self.createEntity()
        assign(components: components, to: entity.identifier)
        return entity
    }

    /// Number of entities in nexus.
    public var numEntities: Int {
        componentIdsByEntity.keys.count
    }

    public func exists(entity entityId: EntityIdentifier) -> Bool {
        componentIdsByEntity.keys.contains(entityId)
    }

    public func entity(from entityId: EntityIdentifier) -> Entity {
        Entity(nexus: self, id: entityId)
    }

    @discardableResult
    public func destroy(entity: Entity) -> Bool {
        self.destroy(entityId: entity.identifier)
    }

    @discardableResult
    public func destroy(entityId: EntityIdentifier) -> Bool {
        guard componentIdsByEntity.keys.contains(entityId) else {
            delegate?.nexusNonFatalError("EntityRemove failure: no entity \(entityId) to remove")
            return false
        }

        if removeAll(components: entityId) {
            update(familyMembership: entityId)
        }

        if let index = componentIdsByEntity.index(forKey: entityId) {
            componentIdsByEntity.remove(at: index)
        }

        entityIdGenerator.markUnused(entityId: entityId)

        delegate?.nexusEvent(EntityDestroyed(entityId: entityId))
        return true
    }
}