Scripted Actions On Selected Hosts with wait for log off

I would like to request a feature in Run Script and Run Script On Selected Hosts to have the same options as a re-image bulk task does, to allow the script to wait for users to log off and put the host into drain mode while it waits. There are scripted actions that needs to occur and then reboot the server, but I do not want to forcefully do it and I don't want it to execute until the last user is off of the machine.