Feature request: "caffeine" widget in top bar

In lineage there is a handy caffeine / coffee cup button in the top bar widget area that keeps the phone from sleeping while it is enabled.

The lineage one can be tapped several times to increase a timeout, after which it disables itself and the phone sleeps. I personally feel that is overkill. I would be happy with a simple toggle with the phone remaining awake indefinitely while enabled. Pressing the power button to explicitly sleep the phone would disable it as well.