avocado-old/packages/entity/trait/registry.js
2019-09-30 19:23:11 -05:00

28 lines
571 B
JavaScript

let traitId = 0;
const idToTraitMap = new Map();
const traitToIdMap = new Map();
const traitRegistry = new Map();
export function registerTrait(Trait) {
traitRegistry.set(Trait.type(), Trait);
traitToIdMap.set(Trait, traitId);
idToTraitMap.set(traitId, Trait);
traitId += 1;
}
export function hasTrait(type) {
return traitRegistry.has(type);
}
export function idFromTrait(Trait) {
return traitToIdMap.get(Trait);
}
export function lookupTrait(type) {
return traitRegistry.get(type);
}
export function traitFromId(id) {
return idToTraitMap.get(id);
}