i'd agree its electrical .. Open whatever you need to open to access all the electrical connectors and switches.. let them all dry out. Use compressed air, if its avaialble, to blow-dry things.
Some corrosion might have already happened, depending on how long it was in the water and the type of water. All corrosion would need to be cleaned off of all wire connections, including switches. Afterwards spray connectors and switches with a bit of WD40 which repels water.
You may want to get into the brakes, steering, suspension components, etc., and remove any water .. and clean things up.
Although it involves a bit of work it's not a big deal .. you were comparatively lucky. When they found one of my bikes it was a total loss.