import json

from redis import StrictRedis

redis_client = StrictRedis()
redis_pubsub = redis_client.pubsub(ignore_subscribe_messages=True)

class TerminalColor:
    HEADER = '\033[95m'
    OKBLUE = '\033[94m'
    OKGREEN = '\033[92m'
    WARNING = '\033[93m'
    FAIL = '\033[91m'
    ENDC = '\033[0m'
    BOLD = '\033[1m'
    UNDERLINE = '\033[4m'

def on_warrior_message(message):
    data = json.loads(message["data"])

    # Split the message
    output_split = data["data"].split(" ")

    # Change the color depending on status.
    if "=404" in output_split[0]:
        status_color = TerminalColor.FAIL
    elif "=301" in output_split[0]:
        status_color = TerminalColor.WARNING
    else:
        status_color = TerminalColor.OKGREEN

    print("[{status_color}{host}:{port}{endc}]\t {data}{endc}".format(
        host=data["host"],
        port=data["port"],
        data=data["data"].strip(),
        # Colors
        status_color=status_color,
        endc=TerminalColor.ENDC
    ))

redis_pubsub.subscribe(**{"tumblr:warrior": on_warrior_message})
pubsub_thread = redis_pubsub.run_in_thread(sleep_time=0.001)

try:
    pubsub_thread.join()
except KeyboardInterrupt:
    pubsub_thread.stop()