#!/usr/bin/env python3
# -*- coding: utf-8; py-indent-offset: 4 -*-
#
# Author:  Linuxfabrik GmbH, Zurich, Switzerland
# Contact: info (at) linuxfabrik (dot) ch
#          https://www.linuxfabrik.ch/
# License: The Unlicense, see LICENSE file.

# https://github.com/Linuxfabrik/monitoring-plugins/blob/main/CONTRIBUTING.md

"""See the check's README for more details.
"""

import argparse  # pylint: disable=C0413
import base64  # pylint: disable=C0413
import sys  # pylint: disable=C0413

import lib.base  # pylint: disable=C0413
import lib.redfish  # pylint: disable=C0413
import lib.txt  # pylint: disable=C0413
import lib.url  # pylint: disable=C0413
from lib.globals import (STATE_CRIT, STATE_OK,  # pylint: disable=C0413
                          STATE_UNKNOWN, STATE_WARN)

__author__ = 'Linuxfabrik GmbH, Zurich/Switzerland'
__version__ = '2024031401'

DESCRIPTION = """Checks the state of the Chassis collection containing resources that represent the
                 physical aspects of the infrastructure. A Chassis is roughly defined as a physical
                 view of a computer system as seen by a human. A single Chassis resource can house
                 sensors, fans, and other components."""

DEFAULT_INSECURE = False
DEFAULT_NO_PROXY = False
DEFAULT_TIMEOUT = 8
DEFAULT_URL = 'https://localhost:5000'


def parse_args():
    """Parse command line arguments using argparse.
    """
    parser = argparse.ArgumentParser(description=DESCRIPTION)

    parser.add_argument(
        '-V', '--version',
        action='version',
        version='%(prog)s: v{} by {}'.format(__version__, __author__)
    )

    parser.add_argument(
        '--always-ok',
        help='Always returns OK.',
        dest='ALWAYS_OK',
        action='store_true',
        default=False,
    )

    parser.add_argument(
        '--insecure',
        help='This option explicitly allows to perform "insecure" SSL connections. '
             'Default: %(default)s',
        dest='INSECURE',
        action='store_true',
        default=DEFAULT_INSECURE,
    )

    parser.add_argument(
        '--no-proxy',
        help='Do not use a proxy. '
             'Default: %(default)s',
        dest='NO_PROXY',
        action='store_true',
        default=DEFAULT_NO_PROXY,
    )

    parser.add_argument(
        '--password',
        help='Redfish API password.',
        dest='PASSWORD',
    )

    parser.add_argument(
        '--timeout',
        help='Network timeout in seconds. '
             'Default: %(default)s (seconds)',
        dest='TIMEOUT',
        type=int,
        default=DEFAULT_TIMEOUT,
    )

    parser.add_argument(
        '--url',
        help='Redfish API URL. '
             'Default: %(default)s',
        dest='URL',
        default=DEFAULT_URL,
    )

    parser.add_argument(
        '--username',
        help='Redfish API username.',
        dest='USERNAME',
    )

    return parser.parse_args()


def main():
    """The main function. Hier spielt die Musik.
    """

    # parse the command line, exit with UNKNOWN if it fails
    try:
        args = parse_args()
    except SystemExit:
        sys.exit(STATE_UNKNOWN)

    base_url = args.URL
    if not args.URL.startswith('http'):
        lib.base.cu('--url parameter has to start with "http://" or https://".')

    # Authorization (if needed)
    header = {}
    header['Accept'] = 'application/json'
    if args.USERNAME and args.PASSWORD:
        auth = '{}:{}'.format(args.USERNAME, args.PASSWORD)
        encoded_auth = lib.txt.to_text(base64.b64encode(lib.txt.to_bytes(auth)))
        header['Authorization'] = 'Basic {}'.format(encoded_auth)

    # Entry point: Get the main data
    url = '{}/redfish/v1/Chassis'.format(base_url)
    result = lib.base.coe(lib.url.fetch_json(
        url,
        header=header,
        insecure=args.INSECURE,
        no_proxy=args.NO_PROXY,
        timeout=args.TIMEOUT,
    ))
    # "Members": [
    #     {
    #         "@odata.id": "/redfish/v1/Chassis/1U"
    #     }
    # ],
    if len(result.get('Members', [])) == 0:
        lib.base.cu('Nothing to check, no Redfish members found.')

    # init some vars
    msg = ''
    state = STATE_OK
    perfdata = ''

    member_count = 0
    # fetch and analyze data by following the "Members" found
    for member in result.get('Members', []):
        url = '{}{}'.format(base_url, member['@odata.id'])  # "/redfish/v1/Chassis/1U"
        chassis = lib.base.coe(lib.url.fetch_json(
            url,
            header=header,
            insecure=args.INSECURE,
            no_proxy=args.NO_PROXY,
            timeout=args.TIMEOUT,
        ))
        chassis = lib.redfish.get_chassis(chassis)
        if chassis['Status_State'] not in ['Enabled', 'Quiesced']:
            continue
        member_count += 1
        chassis_state = lib.redfish.get_state(chassis)
        state = lib.base.get_worst(state, chassis_state)

        msg += 'Member:'
        msg += ' {}'.format(chassis['Manufacturer']) if chassis['Manufacturer'] else ''
        msg += ' {}'.format(chassis['Model']) if chassis['Model'] else ''
        msg += ', '
        msg += 'Power: {}, '.format(chassis['PowerState']) if chassis['PowerState'] else ''
        msg += 'LED: {}, '.format(chassis['IndicatorLED']) if chassis['IndicatorLED'] else ''
        msg += 'SKU: {}, '.format(chassis['SKU']) if chassis['SKU'] else ''
        msg += 'SerNo: {}, '.format(chassis['SerialNumber']) if chassis['SerialNumber'] else ''
        msg += 'PartNumber: {}, '.format(chassis['PartNumber']) if chassis['PartNumber'] else ''
        msg = msg[:-2] + lib.base.state2str(chassis_state, prefix=' ')

        # get all available sensor links for the member
        if not chassis['Sensors_@odata.id']:
            msg += '\n\n'
            continue

        url = '{}{}'.format(base_url, chassis['Sensors_@odata.id'])
        sensors = lib.base.coe(lib.url.fetch_json(
            url,
            header=header,
            insecure=args.INSECURE,
            no_proxy=args.NO_PROXY,
            timeout=args.TIMEOUT,
        ))
        table_data = []
        for sensor in sensors.get('Members', []):
            url = '{}{}'.format(base_url, sensor['@odata.id'])
            sensor_data = lib.base.coe(lib.url.fetch_json(
                url,
                header=header,
                insecure=args.INSECURE,
                no_proxy=args.NO_PROXY,
                timeout=args.TIMEOUT,
            ))
            sensor_data = lib.redfish.get_chassis_sensors(sensor_data)
            if sensor_data['Status_State'] not in ['Enabled', 'Quiesced']:
                continue
            perfdata += lib.redfish.get_perfdata(sensor_data, 'Reading')
            # is the sensor reading within its thresholds?
            sensor_data_value_state = lib.redfish.get_sensor_state(sensor_data, 'Reading')
            state = lib.base.get_worst(state, sensor_data_value_state)
            sensor_data['Value'] = lib.base.state2str(sensor_data_value_state, empty_ok=False)
            # is the sensor_data state healthy at all?
            sensor_data_state = lib.redfish.get_state(sensor_data)
            state = lib.base.get_worst(state, sensor_data_state)
            sensor_data['State'] = lib.base.state2str(sensor_data_state, empty_ok=False)
            table_data.append(sensor_data)
        if table_data:
            keys = ['Name', 'PhysicalContext', 'Reading', 'ReadingUnits', 'Value', 'State' ]
            headers = ['Sensor', 'Location', 'Reading', 'Unit', 'Value', 'State' ]
            msg += '\n\n' + lib.base.get_table(table_data, keys, header=headers)

        # get thermal values for the member
        url = '{}{}/Thermal'.format(base_url, member['@odata.id'])
        thermal = lib.base.coe(lib.url.fetch_json(
            url,
            header=header,
            insecure=args.INSECURE,
            no_proxy=args.NO_PROXY,
            timeout=args.TIMEOUT,
        ))

        # redundancy
        table_data = []
        for redundancy in thermal.get('Redundancy', []):
            redundancy = lib.redfish.get_chassis_thermal_redundancy(redundancy)
            if redundancy['Status_State'] not in ['Enabled', 'Quiesced']:
                continue
            perfdata += lib.redfish.get_perfdata(redundancy)
            # is the redundancy state healthy at all?
            redundancy_state = lib.redfish.get_state(redundancy)
            state = lib.base.get_worst(state, redundancy_state)
            redundancy['State'] = lib.base.state2str(redundancy_state, empty_ok=False)
            table_data.append(redundancy)
        if table_data:
            keys = ['Name', 'Mode', 'State' ]
            headers = ['Redundancy', 'Mode', 'State' ]
            msg += '\n\n' + lib.base.get_table(table_data, keys, header=headers)

        msg += '\n\n'

    if state == STATE_CRIT:
        msg = 'Checked sensors on {} {}. There are critical errors.\n\n'.format(
            member_count,
            lib.txt.pluralize('member', member_count)
        ) + msg
    elif state == STATE_WARN:
        msg = 'Checked sensors on {} {}. There are warnings.\n\n'.format(
            member_count,
            lib.txt.pluralize('member', member_count)
        ) + msg
    else:
        msg = 'Everything is ok, checked sensors on {} {}.\n\n'.format(
            member_count,
            lib.txt.pluralize('member', member_count)
        ) + msg

    # over and out
    lib.base.oao(msg, state, perfdata, always_ok=args.ALWAYS_OK)


if __name__ == '__main__':
    try:
        main()
    except Exception:   # pylint: disable=W0703
        lib.base.cu()
