Oh I remember those days
Simon was just like that and I became the queen of 2min showers. He grew out of it but I tried to occupy him while I was in the shower with a flossie or a kong toy filled with peanut butter. I would let him in the bathroom with me so I could keep an eye on him.
Good Luck, I know just how you feel