From 534ba0b6980d69d2318bef7e25149a735082c0c4 Mon Sep 17 00:00:00 2001
From: Tyler Goodlet <tgoodlet@gmail.com>
Date: Fri, 26 Jan 2018 11:31:11 -0500
Subject: [PATCH] Add json highlighting; make debug msgs white

---
 piker/log.py | 20 +++++++++++++++++---
 setup.py     |  4 +++-
 2 files changed, 20 insertions(+), 4 deletions(-)

diff --git a/piker/log.py b/piker/log.py
index c4304c1f..a849dbce 100644
--- a/piker/log.py
+++ b/piker/log.py
@@ -4,7 +4,9 @@ Log like a forester!
 """
 import sys
 import logging
+import json
 import colorlog
+from pygments import highlight, lexers, formatters
 
 _proj_name = 'piker'
 
@@ -12,7 +14,8 @@ _proj_name = 'piker'
 # (NOTE: we use the '{' format style)
 # Here, `thin_white` is just the laymen's gray.
 LOG_FORMAT = (
-    "{bold_white}{thin_white}{asctime}{reset}"
+    # "{bold_white}{log_color}{asctime}{reset}"
+    "{log_color}{asctime}{reset}"
     " {bold_white}{thin_white}({reset}"
     "{thin_white}{threadName}{reset}{bold_white}{thin_white})"
     " {reset}{log_color}[{reset}{bold_log_color}{levelname}{reset}{log_color}]"
@@ -32,9 +35,9 @@ STD_PALETTE = {
     'ERROR': 'red',
     'WARNING': 'yellow',
     'INFO': 'green',
-    'DEBUG': 'blue',
+    'DEBUG': 'white',
     'TRACE': 'cyan',
-    'GARBAGE': 'purple',
+    'GARBAGE': 'blue',
 }
 BOLD_PALETTE = {
     'bold': {
@@ -83,3 +86,14 @@ def get_console_log(level: str = None, name: str = None) -> logging.Logger:
         log.addHandler(handler)
 
     return log
+
+
+def colorize_json(data, style='algol_nu'):
+    """Colorize json output using ``pygments``.
+    """
+    formatted_json = json.dumps(data, sort_keys=True, indent=4)
+    return highlight(
+        formatted_json, lexers.JsonLexer(),
+        # likeable styles: algol_nu, tango, monokai
+        formatters.TerminalTrueColorFormatter(style=style)
+    )
diff --git a/setup.py b/setup.py
index ae606424..f5edd117 100755
--- a/setup.py
+++ b/setup.py
@@ -29,10 +29,12 @@ setup(
     entry_points={
         'console_scripts': [
             'pikerd = piker.brokers.cli:pikerd',
+            'piker = piker.brokers.cli:cli',
         ]
     },
     install_requires=[
-        'click', 'colorlog', 'trio', 'attrs', 'async_generator'
+        'click', 'colorlog', 'trio', 'attrs', 'async_generator',
+        'pygments',
     ],
     extras_require={
         'questrade': ['asks'],