Custom ability to work only under the influence of a pylon

[Vermia]

I've been trying to make an ability that allows the unit to die for a full mineral refund. I want it to work only under the influence of a pylon. Right now, when you use the ability, it actually works: when you click on the button, you get the unit out of the game and receive the resources, but it just doesnt feel right that you can do it anywhere on the map, so I thought that it would be great if you could only use it under the influence of a pylon or another ground-energy source.

Is there a way to do this without triggers?

I've looked the "Warp gate summon" ability because it looked like it was going to help, but I can't find anywhere that it requires ground-energy.

 

[X-maul]

Under the Ability: Commands + you will have to set a requirement that counts the amount of the powered behaviors that the pylon gives.

I've actually never checked how to pylon works, but I would gues that it periodically gives a hidden behavior to nearby units.