#!/usr/bin/python
# check_vol_utilization.py -- nagios plugin uses libgfapi output for perf data
# Copyright (C) 2014 Red Hat Inc
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
#

import sys
import argparse
import capacity
from glusternagios import utils


def showVolumeUtilization(vname, warnLevel, critLevel):
    buf = {}
    try:
        buf = capacity.statvfs(vname, "localhost")
    except Exception as e:
        sys.stdout.write(("Volume Utilization UNKNOWN: %s\n" % str(e)))
        sys.exit(utils.PluginStatusCode.UNKNOWN)
#    print buf

####################################################################
#statvfs.frsize * statvfs.f_blocks# Size of filesystem in bytes    #
#statvfs.frsize * statvfs.f_bfree # Actual number of free bytes    #
#statvfs.frsize * statvfs.f_bavail# Number of free bytes that      #
#ordinary users are allowed to use (excl. reserved space           #
####################################################################
    #total size in KB
    total_size = (buf['f_bsize'] * buf['f_blocks']) * 0.000976563
    #Available free size in KB
    free_size = (buf['f_bsize'] * buf['f_bavail']) * 0.000976563
    #used size in KB
    used_size = total_size - ((buf['f_bsize'] * buf['f_bfree']) * 0.000976563)
    vol_utilization = (used_size / total_size) * 100
#    print int(total_size)
#    print int(free_size)
#    print int(used_size)
#    print vol_utilization
    perfLines = []
    perfLines.append(("utilization=%s%%;%s;%s total=%s "
                      "used=%s free=%s" % (str(int(vol_utilization)),
                                           str(warnLevel), str(critLevel),
                                           str(int(total_size)),
                                           str(int(used_size)),
                                           str(int(free_size)))))
#    print perfLines

    if int(vol_utilization) > critLevel:
        sys.stdout.write(
            ("Volume Utilization CRITICAL: Utilization:%s%%"
             "| %s\n" % (str(int(vol_utilization)), " ".join(perfLines))))
        sys.exit(utils.PluginStatusCode.CRITICAL)
    elif int(vol_utilization) > warnLevel:
        sys.stdout.write(
            ("Volume Utilization WARNING: Utilization:%s%%"
             "| %s\n" % (str(int(vol_utilization)), " ".join(perfLines))))
        sys.exit(utils.PluginStatusCode.WARNING)
    else:
        sys.stdout.write(
            ("Volume Utilization OK: Utilization:%s%%"
             "| %s\n" % (str(int(vol_utilization)), " ".join(perfLines))))
        sys.exit(utils.PluginStatusCode.OK)


def parse_input():

    parser = argparse.ArgumentParser(
        usage='%(prog)s [-h] <volume> -w <Warning> -c <Critical>')
    parser.add_argument("volume",
                        help="Name of the volume to get the Utilization")
    parser.add_argument("-w",
                        "--warning",
                        action="store",
                        type=int,
                        help="Warning Threshold in percentage")
    parser.add_argument("-c",
                        "--critical",
                        action="store",
                        type=int,
                        help="Critical Threshold in percentage")
    args = parser.parse_args()
    if not args.critical or not args.warning:
        print "UNKNOWN:Missing critical/warning threshold value."
        sys.exit(3)
    if args.critical <= args.warning:
        print "UNKNOWN:Critical must be greater than Warning."
        sys.exit(3)
    return args

if __name__ == '__main__':
    args = parse_input()
    showVolumeUtilization(args.volume, args.warning, args.critical)