import { world, system } from "@minecraft/server";

const speed = 0.25;
const grav = 0.08;

system.runInterval(() => {
    const dimension = world.getDimension("overworld");
    const entities = dimension.getEntities({ tags: ["my_mob"] });
    
    for (const entity of entities) {
        const vel = entity.getVelocity();
        let impulseX = 0;
        let impulseY = 0;
        let impulseZ = 0;

        if (entity.hasTag("x")) {
            impulseX = speed - vel.x;
        }

        if (entity.hasTag("y")) {
            impulseY = speed - vel.y + grav;
        }

        if (entity.hasTag("z")) {
            impulseZ = speed - vel.z;
        }

        if (impulseX !== 0 || impulseY !== 0 || impulseZ !== 0) {
            entity.applyImpulse({
                x: impulseX,
                y: impulseY,
                z: impulseZ
            });
        }
    }
});