To get the water from the waterfall, just drag a villager over in front of it. Usually it says "getting a drink" but when the diamond is in the fire it will say "putting water on the firepit". It takes a lot of bowls to cool down the diamond.