India is a team who have a lot of quality players not being able to play just because there are too many quality players for the squad.
If what you said comes out true and Virat is not playing to show his displeasure, that will be a huge loss for him. As India surely has a batter to replace him and once you get out of the Indian squad, it's quite hard to make a return.
And with the form Virat is in right now, I don't think what he's thinking is the right decision. He might be the sole reason for his own downfall and no one can do anything about it.
Right now, India is full of quality and skilled players, so now it's time for Kohli to came down on earth and do his work for staying into time for some more years because I check many cases discipline is most important for any team and country instead of player and his records. So, if he is doing anything selfishly then management need to take some stick action against him as this will be good for many others in pipeline for having better and powerful role in future.
I never have any doubt about his commitment for country and skills in this game, but now it's time for having change his mind and play like a legendary player instead of doing foul plays against any other even he is doing something wrong he needs to be good.