Solved Make controllable ender dragon

Hello
I'm trying to make a rideable ender dragon. To that, I have a command which spawns one and puts player as passenger, while also addimg them to a hashmap with their name and the dragon entity id. Everything fires correctly, till the last line, where the dragons velocity isnt changed and it doenst move at all.
Anyone have any ideas how to make it controllable and turn the head like the player flying in that direction?

Code (Java):

class DragonRunner extends BukkitRunnable{
    public void run(){
        if (BetterDragons.flyers.isEmpty())return;
     
     
     
        for (String pl : BetterDragons.flyers.keySet()){
            Player p = Bukkit.getPlayer(pl);
         
            System.out.println(pl);
         
            if (p == null)continue;
         
            Entity dr = null;
         
            for (Entity en : p.getNearbyEntities(10, 10, 10)){
             
                System.out.println(en.getEntityId() + ":" + BetterDragons.flyers.get(pl));
             
                if (dr != null)continue;
                System.out.println("found");
                if (en.getEntityId() == BetterDragons.flyers.get(pl))dr = en;
            }
         
            if (dr == null)continue;
         
            System.out.println("velocity work");
         
            dr.setVelocity(p.getLocation().getDirection().multiply(2).add(new Vector(0.0D, 0.2D, 0.0D)));
        }
    }
}

EDIT: Solved, made custom entity:

Code (Java):

public class RideableDragon extends EntityEnderDragon{

    public RideableDragon(World world) {
        super(world);
    }
   
   
    @Override
    public void n(){
        if (passengers.isEmpty()){
            setHealth(0);
            world.removeEntity(this);
            return;
        }
        Entity en = passengers.get(0);
        if (en == null){
            setHealth(0);
        }
       
        this.yaw = 180+en.yaw;
        this.pitch = en.pitch;
       
        setHeadRotation(en.pitch);
       
       
        Location loc = this.getBukkitEntity().getLocation().add(en.getBukkitEntity().getLocation().getDirection());
       
        if (loc.getBlock().getType().equals(Material.AIR)){
            this.setPosition(loc.getX(), loc.getY(), loc.getZ());
        }
       
       
       
       
    }

 

Hmm...
https://github.com/Bukkit/mc-dev/blob/master/net/minecraft/server/EntityEnderDragon.java
Do you know which fields are right? Not really easy reading obfuscated code.

 

Thanks, got it all figured out using MCP Mapping Viewer, and thanks for the heads up about that repository, thought it was updated but guess had that wrong.
Anyways, MCP Mapping Viewer and Spigot.jar source gets you a long way