can now map and foreach with outh new entities input

src/ecs/Query.ts

@@ -10,31 +10,28 @@ export class Query {
     return this._filter;
   }
 
-  private readonly _entities: IEntity[];
-
   private readonly _entityManager: EntityManager;
 
   constructor(filter: QueryFilter, entityManager: EntityManager = EntityManager.instance) {
     this._filter = filter;
     this._entityManager = entityManager;
-    this._entities = entityManager.entities;
   }
 
   public find(entities: IEntity[]): IEntity | undefined {
     return entities.find(this._filter);
   }
 
-  public filter(entities: IEntity[] = this._entities): IEntity[] {
+  public filter(entities: IEntity[] = this._entityManager.entities): IEntity[] {
     return entities.filter(this._filter);
   }
 
-  public map(callback: (entity: IEntity) => IEntity, entities: IEntity[]): IEntity[] {
+  public map(callback: (entity: IEntity) => IEntity, entities: IEntity[] = this.filter()): IEntity[] {
     return entities.map((entity) => {
       return callback(entity);
     });
   }
 
-  public foreach(callback: (entity: IEntity) => void, entities: IEntity[]): void {
+  public foreach(callback: (entity: IEntity) => void, entities: IEntity[] = this.filter()): void {
     entities.forEach((entity) => {
       if (this._filter(entity)) {
         callback(entity);

tests/ecs/Query.test.ts

@@ -206,6 +206,25 @@ describe('>>> Query', () => {
         expect(afterEntities).toBeDefined();
         expect(afterEntities.length).toBe(beforeEntities.length - updateEntities.length);
       });
+
+      it('should map to add component C3 to all entities missing C3 component without filter', () => {
+        const hasC3Query = new Query((entity: IEntity) => entity.has(C3), entityManager);
+        const beforeEntities = hasC3Query.map((entity: IEntity) => entity);
+
+        expect(beforeEntities).toBeDefined();
+        expect(beforeEntities.length).toBe(1);
+
+        const updateQuery = new Query((entity: IEntity) => entity.hasNone(C3), entityManager);
+        const updateEntities = updateQuery.map((entity: IEntity) => entity.add(new C3()));
+
+        expect(updateEntities).toBeDefined();
+        expect(updateEntities.length).toBe(2);
+
+        const afterEntities = hasC3Query.map((entity: IEntity) => entity);
+
+        expect(afterEntities).toBeDefined();
+        expect(afterEntities.length).toBe(beforeEntities.length + updateEntities.length);
+      });
     });
 
     describe('>>> Query foreach', () => {