#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright (C) 2019 tribe29 GmbH - License: GNU General Public License v2
# This file is part of Checkmk (https://checkmk.com). It is subject to the terms and
# conditions defined in the file COPYING, which is part of this source code package.

# NOTE: Careful when replacing the *-import below with a more specific import. This can cause
# problems because it might remove variables from the check-context which are necessary for
# resolving legacy discovery results such as [("SUMMARY", "diskstat_default_levels")]. Furthermore,
# it might also remove variables needed for accessing discovery rulesets.
from cmk.base.check_legacy_includes.liebert import *  # pylint: disable=wildcard-import,unused-wildcard-import

# example output
# .1.3.6.1.4.1.476.1.42.3.9.20.1.10.1.2.1.4868 Calculated Next Maintenance Month
# .1.3.6.1.4.1.476.1.42.3.9.20.1.20.1.2.1.4868 5
# .1.3.6.1.4.1.476.1.42.3.9.20.1.10.1.2.1.4869 Calculated Next Maintenance Year
# .1.3.6.1.4.1.476.1.42.3.9.20.1.20.1.2.1.4869 2017

factory_settings["liebert_maintenance_default_levels"] = {
    "levels": (10, 5)  # Remaining days until next maintenance
}


def inventory_liebert_maintenance(parsed):
    return [(None, {})]


def check_liebert_maintenance(_no_item, params, parsed):
    month, year = None, None
    for key, value in parsed.items():
        if "month" in key.lower():
            month = value
        elif "year" in key.lower():
            year = value

    if None in (month, year):
        return

    yield 0, "Next maintenance: %s/%s" % (month, year)

    time_left_seconds = time.mktime((year, month, 0, 0, 0, 0, 0, 0, 0)) - time.time()

    warn_days, crit_days = params["levels"]
    levels = (None, None, warn_days * 86400, crit_days * 86400)
    yield check_levels(time_left_seconds, None, levels, human_readable_func=get_age_human_readable)


check_info["liebert_maintenance"] = {
    "parse_function": lambda info: parse_liebert_without_unit_wrapper(info, int),
    "inventory_function": inventory_liebert_maintenance,
    "check_function": check_liebert_maintenance,
    "service_description": "Maintenance",
    "snmp_info": (
        ".1.3.6.1.4.1.476.1.42.3.9.20.1",
        [
            "10.1.2.1.4868",  # LIEBERT-GP-FLExible-MIB: lgpFlexibleEntryDataLabel
            "20.1.2.1.4868",  # LIEBERT-GP-FLExible-MIB: lgpFlexibleEntryValue
            "10.1.2.1.4869",  # LIEBERT-GP-FLExible-MIB: lgpFlexibleEntryDataLabel
            "20.1.2.1.4869",  # LIEBERT-GP-FLExible-MIB: lgpFlexibleEntryValue
        ],
    ),
    "snmp_scan_function": scan_liebert,
    "default_levels_variable": "liebert_maintenance_default_levels",
}
