You can use similar logic in the automation trigger to evaluate the states instead of the attributes on an entity update. The message would need to be tweaked also to utilize the correct data.
The template docs and the template dev tool are very helpful in figuring these things out.