use rich instead of shipping custom formatting code

[keewis]

Since the tool was partially copied from black (which uses click), there's a few functions that implement custom formatting. Most if not all of that can also be done by rich, so we should be able to replace the custom code.