#!/usr/bin/env python
import requests, json
import xml.etree.ElementTree as xml

r = requests.get("http://localhost:8081/stats")
r.raise_for_status()
out = {}
d = xml.fromstring(r.text)
assert d.tag == "inspircdstats"


def define(name, xps, vfilter=lambda x: x):
    global out
    out[name] = vfilter(d.findall(xps)[0].text)


define("usercount", "./general/usercount", int)
define("channelcount", "./general/channelcount", int)
schannels = d.findall("./channellist/channel")
# print(len(schannels))
channels = []
for schannel in schannels:
    channel = dict(
        name=schannel.findall("channelname")[0].text,
        usercount=int(schannel.findall("usercount")[0].text),
    )
    channel["topic"] = (
        schannel.findall("./channeltopic/topictext")[0].text
        if schannel.findall("./channeltopic/topictext")[0].text is not None
        else "No topic set"
    )
    # bugfix: if mode s is set and there's a topic, hide the topic
    if (
        "s" in schannel.findall("./channelmodes")[0].text.split()[0]
        and channel["topic"] != "No topic set"
    ):
        channel["topic"] = "Topic hidden"
    if channel["name"] == "#secret-sudoers":
        # no stat output for #secret-sudoers! it's a sekrit club of the ~team sysadmins!
        continue
    channel["webchatlink"] = "https://web.tilde.chat/?join=" + channel["name"].lstrip(
        "#"
    )
    channels.append(channel)
channels.sort(key=lambda x: x["name"].lower())
out["channels"] = channels
# print([x.text for x in d.findall("./channellist/channel/channeltopic/topictext")])
with open("/var/www/tilde.chat/stats.json", "w") as f:
    json.dump(out, f)