Wolves

Is there a way to get wolves to teleport when the player is hurt, but only when they are hurt or attack something? Not when they move far away?

PS: This is for a project I'm working on to keep players alive.