You can give reflection different colors, but can't use any function I'm afraid. The only possibility I see is the reflectivity image or function input in a default shader. Try a colorful PF, and a small roughness and index of 7 or so, metalness on, see what that does. You probably need fake stones in several sizes to make up teh crystal hard shapes.