src/engine/entities/Entity.ts


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46

import { type Component } from "../components";

export abstract class Entity {
  static Id = 0;

  public id: string;
  public components: Map<string, Component>;
  public name: string;

  protected hooks: Map<string, { add: Function; remove: Function }>;

  constructor(name: string, id: string = (Entity.Id++).toString()) {
    this.name = name;
    this.id = id;
    this.components = new Map();
    this.hooks = new Map();
  }

  public addComponent(component: Component) {
    const hadBeforeSet = this.components.has(component.name);
    this.components.set(component.name, component);
    if (!hadBeforeSet) {
      this.hooks.get(component.name)?.add();
    }
  }

  public removeComponent(name: string) {
    this.components.delete(name);
    this.hooks.get(name)?.remove();
  }

  public getComponent<T extends Component>(name: string): T {
    if (!this.hasComponent(name)) {
      throw new Error("Entity does not have component " + name);
    }
    return this.components.get(name) as T;
  }

  public getComponents(): Component[] {
    return Array.from(this.components.values());
  }

  public hasComponent(name: string): boolean {
    return this.components.has(name);
  }
}