There is no restriction on how close a light switch or a plug can be to a shower stall or a whirlpool tub.

410.4d restricts pendant fixtures from within 3ft horizontally and 8ft vertically or the top of the bathtub rim.

A surface light in a shower should be suitable for wet location.

shortcircuit