I had the same idea, but looks like someone beat me and did a better job overall. Tested it once and the simulations look good.

Found one more, which seems more advanced : https://syde.cc/