#!/usr/bin/env python
#
# Copyright (C) 2012 eNovance <licensing@enovance>
# 
# Authors: Mehdi Abaakouk <sileht@sileht.net>
#          Gael Lambert <gael.lambert@enovance.com>
#          Jesse Andrews <anotherjesse@rackspace.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
#
# Plugin to monitor status of Floating IPs in Nova
#
# To monitor a floating ips, link floating_ips to this file.
# E.g.
#    ln -s /usr/share/munin/plugins/nova_floating_ips /etc/munin/plugins/
#
# Needs following minimal configuration in plugin-conf.d/nova:
#   [nova_*]
#   user nova
#
# Magic markers
#%# capabilities=autoconf
#%# family=auto

import sys

try:
    from nova import context
    from nova import db
    from nova import flags
    from nova import utils
except ImportError:
    succesful_import = False
else:
    succesful_import = True



states = ['total', 'allocated', 'associated']

def print_config():
    global states
    print 'graph_title Nova Floating IPs'
    print 'graph_vlabel IPs'
    print 'graph_args --base 1000 --lower-limit 0'
    print 'graph_category nova'
    print 'graph_scale no'
    print 'graph_info This graph shows the number of Floating IPs in Nova and their status'
    for state in states:
        print '%s.label %s' % (state, state)
        print '%s.draw LINE2' % state
        print '%s.info %s IPs' % (state, state)


def get_status():
    status = {}
    for k in states:
        status[k] = 0

    ctxt = context.get_admin_context()
    floating_ips = db.floating_ip_get_all(ctxt)
    for floating_ip in floating_ips:
        status['total'] += 1
        if floating_ip['fixed_ip_id']:
            status['associated'] += 1
        if floating_ip['project_id']:
            status['allocated'] += 1

    return status


def print_values():
    status = get_status()
    for (state, value) in status.iteritems():
        print "%s.value %s" % (state, value)


if __name__ == '__main__':
    if len(sys.argv) > 1:
        if sys.argv[1] == "config":
            print_config()
        elif sys.argv[1] == "autoconf":
            if not succesful_import:  
                print 'no (failed import nova module)'
                sys.exit(0)
            else:
                print 'yes'
    elif succesful_import:
        utils.default_flagfile()
        flags.FLAGS(sys.argv)
        print_values()
