Golem that protects player and attacks others?

Hey, i was wondering if there is any known way a Golem could be triggered to be a friendly for the player that placed the last pumpkin block on the iron blocks and anyone else is an enemy that is automatically attacked or upon the player that owns the Golem attacking or being attacked.

Basically the nature of a wolf.

What events would i need here? A Block Place event and a Creature Spawn event?
Then what would i do to get what i need working? Any code that can be given would be very helpful.

Or, should i just use a wolf if its too much work? Iron Golem would be a lot more cooler though.

My code atm:

Code:

@EventHandler
    public void onCreatureSpawn(CreatureSpawnEvent event) {
        if(event.getEntityType() == EntityType.WITHER) {
            event.getEntity().setMaxHealth(1500);
            event.getEntity().setHealth(1500);
        }
       
        if(event.getEntityType() == EntityType.IRON_GOLEM) {
           
        }
    }

No code... lol because i am not sure which way is the best way to go about it.

Thanks!

 

@kayc01
https://bukkit.org/threads/making-iron-golem-target-a-player.75663/