An optimization framework to support water supply systems in the energy transition