#!/usr/bin/python

import os
import re
import sys

binarypath = "/usr/sbin/tw-cli"

if len(sys.argv) > 2:
    print("Usage: 3ware-status [--nagios]")
    sys.exit(1)

nagiosmode = False
nagiosoutput = ""
nagiosgoodarray = 0
nagiosbadarray = 0
nagiosgooddisk = 0
nagiosbaddisk = 0

if len(sys.argv) > 1:
    if sys.argv[1] == "--nagios":
        nagiosmode = True
    else:
        print("Usage: 3ware-status [--nagios]")
        sys.exit(1)

# Check binary exists (and +x), if not print an error message
# or return UNKNOWN nagios error code
if os.path.exists(binarypath) and os.access(binarypath, os.X_OK):
    pass
else:
    if nagiosmode:
        print("UNKNOWN - Cannot find " + binarypath)
    else:
        print("Cannot find " + binarypath + ". Please install it.")
    sys.exit(3)


# Get command output
def getOutput(cmd):
    output = os.popen(cmd)
    lines = []
    for line in output:
        if not re.match(r"^$", line.strip()):
            lines.append(line.strip())
    return lines


def returnControllerList(output):
    lines = []
    for line in output:
        if re.match(r"^c[0-9]+\s.*$", line.strip()):
            lines.append(line.split()[0])
    return lines


def returnDiskList(output):
    lines = []
    for line in output:
        if re.match(r"^[p][0-9]+\s.*$", line.strip()):
            # Shoudl contain something like 'u0'
            # '-' means the drive doesn't belong to any array
            # If is NOT PRESENT too, it just means this is an empty port
            if not line.split()[2].strip() == "-" and not line.split()[1].strip() == "NOT-PRESENT":
                lines.append(line.split())
    if fake_failure:
        lines[0][1] = "NOT PRESENT"
    return lines


def returnArrayList(output):
    lines = []
    for line in output:
        if re.match(r"^[u][0-9]+\s.*$", line.strip()):
            lines.append(line.split())
    if fake_failure:
        lines[0][2] = "DEGRADED"
    return lines


# A way to force a fake failure
fake_failure = False
if os.path.exists("/root/fake_3ware_failure"):
    fake_failure = True

cmd = binarypath + " info"
output = getOutput(cmd)
controllerlist = returnControllerList(output)

bad = False


# List available controller
if not nagiosmode:
    print("-- Controller informations --")
    print("-- ID | Model")
    for controller in controllerlist:
        cmd = binarypath + " info " + controller + " model"
        # https://github.com/eLvErDe/hwraid/issues/69
        try:
            model = getOutput(cmd)[0].split(" = ")[1].strip()
        except IndexError:
            model = "N/A"
        print(controller + " | " + model)
    print("")

# List arrays
if not nagiosmode:
    print("-- Arrays informations --")
    print("-- ID\tType\tSize\tStatus")
for controller in controllerlist:
    cmd = binarypath + " info " + controller
    output = getOutput(cmd)
    arraylist = returnArrayList(output)
    for array in arraylist:
        type = array[1].replace("-", "")
        id = controller + array[0]
        size = array[6].split(".")[0] + "G"
        status = array[2]
        if not status in ["OK", "VERIFYING"]:
            bad = True
            nagiosbadarray = nagiosbadarray + 1
        else:
            nagiosgoodarray = nagiosgoodarray + 1
        if not nagiosmode:
            print(id + "\t" + type + "\t" + size + "\t" + status)
if not nagiosmode:
    print("")

# List disks
if not nagiosmode:
    print("-- Disks informations")
    print("-- ID\tModel\t\t\tStatus")
for controller in controllerlist:
    cmd = binarypath + " info " + controller
    output = getOutput(cmd)
    disklist = returnDiskList(output)
    for disk in disklist:
        id = controller + disk[2] + disk[0]
        cmd = binarypath + " info " + controller + " " + disk[0] + " model"
        model = getOutput(cmd)[0].split(" = ")[1].strip()
        cmd = binarypath + " info " + controller + " " + disk[0] + " status"
        status = getOutput(cmd)[0].split(" = ")[1].strip()
        if not status == "OK":
            bad = True
            nagiosbaddisk = nagiosbaddisk + 1
        else:
            nagiosgooddisk = nagiosgooddisk + 1
        if not nagiosmode:
            print(id + "\t" + model + "\t" + status)

if nagiosmode:
    if bad:
        print(
            "RAID ERROR - Arrays: OK:"
            + str(nagiosgoodarray)
            + " Bad:"
            + str(nagiosbadarray)
            + " - Disks: OK:"
            + str(nagiosgooddisk)
            + " Bad:"
            + str(nagiosbaddisk)
        )
        sys.exit(2)
    else:
        print(
            "RAID OK - Arrays: OK:"
            + str(nagiosgoodarray)
            + " Bad:"
            + str(nagiosbadarray)
            + " - Disks: OK:"
            + str(nagiosgooddisk)
            + " Bad:"
            + str(nagiosbaddisk)
        )
else:
    if bad:
        print("\nThere is at least one disk/array in a NOT OPTIMAL state.")
        sys.exit(1)