Ulco, could you try your technique with the "Volume" controls in the water shader? Its density can be controlled by a function. "Volume 1 colour" can be white, but when this combines with the "Decay tint" which you would set to a blueish colour you get a more chromatically rich range of colours. White-ish where the volume is most dense and blueish where it it fades away. The intensity of the colour will depend on the relationship between Decay distance and Volume 1 density. That sounds a bit complicated, but think of it as some white material (the "volume") suspended in a blueish water ("decay distance" and "decay tint").
Matt