import { hookstate, none } from '@xrengine/hyperflux'

import { Entity } from '../../ecs/classes/Entity'
import { defineComponent } from '../../ecs/functions/ComponentFunctions'

export const NameComponent = defineComponent({
  name: 'NameComponent',

  onInit: () => '',

  onSet: (entity, component, name?: string) => {
    if (typeof name !== 'string') throw new Error('NameComponent expects a non-empty string')
    component.set(name)
    NameComponent.valueMap[entity] = name
    const namedEntities = NameComponent.entitiesByName[name]
    const exists = !!namedEntities.value
    exists && namedEntities.set([...namedEntities.value!, entity])
    !exists && namedEntities.set([entity])
  },

  onRemove: (entity, component) => {
    const name = component.value
    const namedEntities = NameComponent.entitiesByName[name]
    const isSingleton = namedEntities.length === 1
    isSingleton && namedEntities.set(none)
    !isSingleton && namedEntities.set(namedEntities.value.filter((namedEntity) => namedEntity !== entity))
  },

  entitiesByName: hookstate({} as Record<string, Entity[]>)
})