from __future__ import print_function

import sys

import prometheus_client as node
from contextlib import contextmanager

import argparse, time
import Ice, Murmur


@contextmanager
def ice_connect(host, port):
    prxstr = "Meta:tcp -h %s -p %d -t 1000" % (host, port)

    ic = Ice.initialize(sys.argv)
    base = ic.stringToProxy(prxstr)
    meta = Murmur.MetaPrx.checkedCast(base)
    if not meta:
        print('cannot establish connection', file=sys.stderr)
        sys.exit(1)

    yield meta

    if ic:
        ic.destroy()


def main():
    parser = argparse.ArgumentParser(description='Prometheus statistics for a Mumble ICE interface')
    parser.add_argument('-H', '--host', help='Host of the Ice interface', default='127.0.0.1')
    parser.add_argument('-p', '--port', help='Port of the Ice interface', default=6502)
    parser.add_argument('-i', '--interval', help='Interval in seconds', default=60)
    args = parser.parse_args()

    node.start_http_server(8001)

    g = node.Gauge('users_connected', 'Number of connected users')
    while True:
        t1 = time.time()
        print('gathering statistics')
        with ice_connect(args.host, args.port) as meta:
            for server in meta.getBootedServers():
                g.set(len(server.getUsers()))
        time_to_wait = args.interval - (time.time() - t1)
        if time_to_wait > 0:
            time.sleep(time_to_wait)
    return 0

if __name__ == '__main__':
    sys.exit(main())