fix: register unregistered components on creating a query

.changeset/fresh-toes-run.md

@@ -0,0 +1,6 @@
+---
+"arancini": patch
+"@arancini/core": patch
+---
+
+fix: register unregistered components on creating a query

packages/arancini-core/src/world.ts

@@ -12,6 +12,7 @@ import {
 } from './entity-metadata'
 import {
   Query,
+  QueryCondition,
   QueryDescription,
   evaluateQueryBitSets,
   getFirstQueryResult,
@@ -367,6 +368,8 @@ export class World<E extends AnyEntity = any> extends EntityContainer<E> {
     const queryConditions = getQueryConditions(queryDescription)
     const key = getQueryDedupeString(this.componentRegistry, queryConditions)
 
+    this.registerQueryDescriptionComponents(queryConditions)
+
     const handle = options?.handle ?? DEFAULT_QUERY_HANDLE
 
     const queryUsages = this.queryUsages.get(key)
@@ -437,6 +440,9 @@ export class World<E extends AnyEntity = any> extends EntityContainer<E> {
     queryDescription: QueryDescription<E, ResultEntity>
   ): ResultEntity[] {
     const conditions = getQueryConditions(queryDescription)
+
+    this.registerQueryDescriptionComponents(conditions)
+
     const queryDedupe = getQueryDedupeString(this.componentRegistry, conditions)
 
     const query = this.queries.get(queryDedupe)
@@ -464,6 +470,9 @@ export class World<E extends AnyEntity = any> extends EntityContainer<E> {
     queryDescription: QueryDescription<E, ResultEntity>
   ): ResultEntity | undefined {
     const conditions = getQueryConditions(queryDescription)
+
+    this.registerQueryDescriptionComponents(conditions)
+
     const queryDedupe = getQueryDedupeString(this.componentRegistry, conditions)
 
     const query = this.queries.get(queryDedupe)
@@ -507,6 +516,23 @@ export class World<E extends AnyEntity = any> extends EntityContainer<E> {
     return ids
   }
 
+  private registerQueryDescriptionComponents(queryConditions: QueryCondition<E>[]) {
+    const queryComponents = new Set(
+      queryConditions.flatMap((condition) => condition.components)
+    )
+
+    const unregisteredComponents: string[] = []
+    for (const component of queryComponents) {
+      if (this.componentRegistry[component as string] === undefined) {
+        unregisteredComponents.push(component as string)
+      }
+    }
+
+    if (unregisteredComponents.length > 0) {
+      this.registerComponents(unregisteredComponents)
+    }
+  }
+
   private index(entity: E) {
     if (!this.has(entity)) return
 

packages/arancini-core/tst/world.spec.ts

@@ -20,14 +20,18 @@ describe('World', () => {
   })
 
   it('supports automatically registering components', () => {
-    const world = new World<Entity>({ components: ['foo'] })
+    const world = new World<Entity>()
 
-    const entity = { foo: 'test', bar: 1 }
+    const entity = { foo: 'test' }
 
     world.create(entity)
 
     const query = world.query((q) => q.has('bar'))
 
+    expect(query.entities.length).toBe(0)
+
+    world.add(entity, 'bar', 1)
+
     expect(query.entities.length).toBe(1)
   })