#!/usr/bin/env python3

"""
=head1 NAME


keystone_stats - monitor status of Keystone

=head1 CONFIGURATION


Needs following minimal configuration in plugin-conf.d/keystone:

  [keystone_*]
  user keystone

=head1 AUTHORS

Copyright 2012 - Mehdi Abaakouk <sileht@sileht.net>


=head1 MAGIC MARKERS

 #%# capabilities=autoconf
 #%# family=auto

=cut
"""

import sys
import traceback

try:
    from keystone.common import utils
    from keystone import config
    from keystone import exception
    from keystone import identity
except ImportError:
    successful_import = False
else:
    successful_import = True


stats = ['users', 'tenants']


def print_config():
    print('graph_title Keystone Stats')
    print('graph_vlabel count')
    print('graph_args --base 1000 --lower-limit 0')
    print('graph_category auth')
    print('graph_scale no')
    print('graph_info This graph shows stats about keystone: ' + (', ').join(stats))
    for field in stats:
        print('%s_enabled.label enabled %s' % (field, field))
        print('%s_enabled.draw LINE2' % field)
        print('%s_enabled.info %s enabled' % (field, field))
        print('%s_total.label total %s' % (field, field))
        print('%s_total.draw LINE2' % field)
        print('%s_total.info %s total' % (field, field))


def get_status():
    enabled = {}
    total = {}
    for k in stats:
        enabled[k] = 0
        total[k] = 0

    identity_api = identity.Manager()

    for user in identity_api.list_users(None):
        total['users'] += 1
        if user['enabled']:
            enabled['users'] += 1

    # Ldap and  pam driver don't support get_all_tenants()
    # kvs and sql implement get_tenants() instead of get_all_tenants()
    # Whoo: None of backend implements the correct function
    tenants = []
    for api_func in ['get_all_tenants', 'get_tenants']:
        try:
            tenants = getattr(identity_api, api_func)(None)
        except (exception.NotImplemented, NotImplementedError):
            pass

    for tenant in tenants:
        total['tenants'] += 1
        if tenant['enabled']:
            enabled['tenants'] += 1

    return {'enabled': enabled, 'total': total}


def print_values():
    stats = get_status()
    for state in stats.keys():
        for (field, value) in stats[state].items():
            print("%s_%s.value %s" % (field, state, value))


def load_conf():
    config.CONF(config_files=[utils.find_config('keystone.conf')])


if __name__ == '__main__':
    if len(sys.argv) > 1:
        if sys.argv[1] == "config":
            print_config()
        elif sys.argv[1] == "autoconf":
            if not successful_import:
                print('no (failed import keystone module)')
                sys.exit(0)
            try:
                load_conf()
                identity.Manager()
            except Exception:
                print('no (failed to connect keystone backend: %s' % traceback.format_exc())
                sys.exit(0)
            print('yes')
    elif successful_import:
        load_conf()
        print_values()