[RDY]  Cleanup ui module

I'm using recreating the ui module in a private branch, and currently most of the code in ui.c doesn't relate to UI as Neovim knows it :smile: 

This PR is to reduce the number of changes necessary by my next PR(which will be big)