#!/usr/bin/env python3
"""
=head1 NAME

mongo_btree - MongoDB btree Plugin

=head1 APPLICABLE SYSTEMS

Works until MongoDB 2.7. The "indexCounters" field was removed in 2.8 version.

=head1 CONFIGURATION

    [mongo_btree]
    env.MONGO_DB_URI mongodb://user:password@host:port/dbname

=head1 AUTHOR

Original script there : https://github.com/comerford/mongo-munin

Doc added by Alban Espie-Guillon <alban.espie@alterway.fr>

=cut
"""

import json
import urllib.request
import sys


def getServerStatus():
    raw = urllib.request.urlopen("http://127.0.0.1:28017/_status").read()
    return json.loads(raw)["serverStatus"]


def get():
    status = getServerStatus()
    if status["version"] >= "2.4.0":
        return getServerStatus()["indexCounters"]
    else:
        return getServerStatus()["indexCounters"]["btree"]


def doData():
    for k, v in get().items():
        print(str(k) + ".value " + str(int(v)))


def doConfig():
    print("graph_title MongoDB btree stats")
    print("graph_args --base 1000 -l 0")
    print("graph_vlabel mb ${graph_period}")
    print("graph_category db")
    for k in get():
        print(k + ".label " + k)
        print(k + ".min 0")
        print(k + ".type COUNTER")
        print(k + ".max 500000")
        print(k + ".draw LINE1")


if __name__ == "__main__":
    if len(sys.argv) > 1 and sys.argv[1] == "config":
        doConfig()
    else:
        doData()