#!/usr/bin/env python3
import json
import ssl
import websocket

def format_query(query):
    if 'className' in query:
        return 'class ' + query['className']
    elif 'typeName' in query:
        return 'type ' + query['typeName']
    elif 'unify' in query:
        if 'name' in query and query['name'] != '':
            return query['name'] + ' :: ' + query['unify']
        else:
            return ':: ' + query['unify']
    else:
        return query['name']

def on_message(ws, msg):
    try:
        print(format_query(json.loads(msg)))
    except:
        pass

def on_error(ws, err):
    print(err)

def on_close(ws):
    print('Closed.')

if __name__ == '__main__':
    ws = websocket.WebSocketApp('wss://cloogle.org:31216',
            subprotocols=['cloogle-stats'],
            on_message=on_message, on_error=on_error, on_close=on_close)
    ws.run_forever(sslopt={'cert_reqs': ssl.CERT_NONE})