Its so simple you don't need a sketch.That sounds like something to try, have a sketch handy for us picture thinkers?
Some hold the view that its safer to put the diode & switch in the neutral lead so the additions are not at live potential.
If you build it in a small die-cast box with proper cable glands and earth point, it might then probably makes no difference.