Implement tryTakeItem in Belt

src/js/game/buildings/belt_base.js

@@ -60,6 +60,7 @@ export class MetaBeltBaseBuilding extends MetaBuilding {
                         pos: new Vector(0, 0),
                         directions: [enumDirection.bottom],
                         layer: this.getLayer(),
+                        processor: "Belt",
                     },
                 ],
                 animated: false,

src/js/game/components/belt.js

@@ -4,6 +4,7 @@ import { BeltPath } from "../belt_path";
 import { Component } from "../component";
 import { Entity } from "../entity";
 import { enumLayer } from "../root";
+import { BaseItem } from "../base_item";
 
 export const curvedBeltLength = /* Math.PI / 4 */ 0.78;
 
@@ -118,4 +119,19 @@ export class BeltComponent extends Component {
             }
         }
     }
+
+    /**
+     *
+     * @param {BaseItem} item
+     * @param {number} slot
+     */
+    tryTakeItem(item, slot) {
+        const path = this.assignedPath;
+        assert(path, "belt has no path");
+        if (path.tryAcceptItem(item)) {
+            return true;
+        }
+        // Belt can have nothing else
+        return false;
+    }
 }

src/js/game/systems/item_ejector.js

@@ -274,17 +274,6 @@ export class ItemEjectorSystem extends GameSystemWithFilter {
 
         const itemLayer = enumItemTypeToLayer[item.getItemType()];
 
-        const beltComp = receiver.components.Belt;
-        if (beltComp) {
-            const path = beltComp.assignedPath;
-            assert(path, "belt has no path");
-            if (path.tryAcceptItem(item)) {
-                return true;
-            }
-            // Belt can have nothing else
-            return false;
-        }
-
         const energyConsumerComp = receiver.components.EnergyConsumer;
         if (energyConsumerComp) {
             if (energyConsumerComp.tryAcceptItem(item, slotIndex)) {