Texture or colour only on the displacement vector
You can just add in a mix node, another principled shader and a color ramp. 



Create a 2nd principled shader (this will be your lava).

Connect it with a mix shader, use a color ramp and have your displacement nodes connect to it. 

Then on your color ramp adjust the flags to your liking.