#!/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 datetime  # pylint: disable=C0413
import sys  # pylint: disable=C0413

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

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

DESCRIPTION = 'Warns before an expiration date is scheduled to occur.'

DEFAULT_WARN = 50      # days
DEFAULT_CRIT = 30      # days


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

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

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

    parser.add_argument(
        '--input',
        help='"Display Name 1, yyyy-mm-dd, warn, crit" (repeating)',
        dest='INPUT',
        type=lib.args.csv,
        required = True,
        action='append',
    )

    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)

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

    try:
        # ['Supermicro SYS1, 2022-01-10, 50, None', 'Contract XMPL.COM, 2021-12-24, 50, 30']
        for line in args.INPUT:
            item, expiration_date, warn, crit = line

            expiration_date = datetime.datetime.strptime(expiration_date, "%Y-%m-%d")
            today = lib.time.now(as_type='datetime')
            delta = expiration_date - today
            delta = int(delta.days) + 1

            if crit.lower() != 'none' and delta < int(crit):
                line_state = STATE_CRIT
            elif warn.lower() != 'none' and delta < int(warn):
                line_state = STATE_WARN
            else:
                line_state = STATE_OK

            if delta > 0:
                msg += '* {} expires in {} {} (thresholds {}/{}){}\n'.format(item, delta, lib.txt.pluralize('day', delta), warn, crit, lib.base.state2str(line_state, prefix=' '))
            elif delta == 0:
                msg += '* {} expires today{}\n'.format(item, lib.base.state2str(line_state, prefix=' '))
            else:
                msg += '* {} expired {} {} ago{}\n'.format(item, abs(delta), lib.txt.pluralize('day', abs(delta)), lib.base.state2str(line_state, prefix=' '))

            state = lib.base.get_worst(line_state, state)

        if state == STATE_CRIT:
            msg = 'There are one or more criticals.\n' + msg
        elif state == STATE_WARN:
            msg = 'There are one or more warnings.\n' + msg
        else:
            msg = 'Everything is ok.\n' + msg

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

    except Exception:
        lib.base.cu('Something seems to be wrong with the input parameter format or its timestamps')


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