Use the snap modules from the following Salt PR:
This will both allow for more features as well as remove the current cmd.run states, to achieve a more robust solution.
As I've discussed in that PR's thread, I've already tested this out and it is working well. I'll provide a PR when have some time available.
This will also close #38.