# -*- coding: utf-8 -*-
# Description: NSD `nsd-control stats_noreset` netdata python.d module
# Author: <383c57 at gmail.com>
# SPDX-License-Identifier: GPL-3.0-or-later

import re

from bases.FrameworkServices.ExecutableService import ExecutableService


update_every = 30

NSD_CONTROL_COMMAND = 'nsd-control stats_noreset'
REGEX = re.compile(r'([A-Za-z0-9.]+)=(\d+)')

ORDER = [
    'queries',
    'zones',
    'protocol',
    'type',
    'transfer',
    'rcode',
]

CHARTS = {
    'queries': {
        'options': [None, 'queries', 'queries/s', 'queries', 'nsd.queries', 'line'],
        'lines': [
            ['num_queries', 'queries', 'incremental']
        ]
    },
    'zones': {
        'options': [None, 'zones', 'zones', 'zones', 'nsd.zones', 'stacked'],
        'lines': [
            ['zone_master', 'master', 'absolute'],
            ['zone_slave', 'slave', 'absolute']
        ]
    },
    'protocol': {
        'options': [None, 'protocol', 'queries/s', 'protocol', 'nsd.protocols', 'stacked'],
        'lines': [
            ['num_udp', 'udp', 'incremental'],
            ['num_udp6', 'udp6', 'incremental'],
            ['num_tcp', 'tcp', 'incremental'],
            ['num_tcp6', 'tcp6', 'incremental']
        ]
    },
    'type': {
        'options': [None, 'query type', 'queries/s', 'query type', 'nsd.type', 'stacked'],
        'lines': [
            ['num_type_A', 'A', 'incremental'],
            ['num_type_NS', 'NS', 'incremental'],
            ['num_type_CNAME', 'CNAME', 'incremental'],
            ['num_type_SOA', 'SOA', 'incremental'],
            ['num_type_PTR', 'PTR', 'incremental'],
            ['num_type_HINFO', 'HINFO', 'incremental'],
            ['num_type_MX', 'MX', 'incremental'],
            ['num_type_NAPTR', 'NAPTR', 'incremental'],
            ['num_type_TXT', 'TXT', 'incremental'],
            ['num_type_AAAA', 'AAAA', 'incremental'],
            ['num_type_SRV', 'SRV', 'incremental'],
            ['num_type_TYPE255', 'ANY', 'incremental']
        ]
    },
    'transfer': {
        'options': [None, 'transfer', 'queries/s', 'transfer', 'nsd.transfer', 'stacked'],
        'lines': [
            ['num_opcode_NOTIFY', 'NOTIFY', 'incremental'],
            ['num_type_TYPE252', 'AXFR', 'incremental']
        ]
    },
    'rcode': {
        'options': [None, 'return code', 'queries/s', 'return code', 'nsd.rcode', 'stacked'],
        'lines': [
            ['num_rcode_NOERROR', 'NOERROR', 'incremental'],
            ['num_rcode_FORMERR', 'FORMERR', 'incremental'],
            ['num_rcode_SERVFAIL', 'SERVFAIL', 'incremental'],
            ['num_rcode_NXDOMAIN', 'NXDOMAIN', 'incremental'],
            ['num_rcode_NOTIMP', 'NOTIMP', 'incremental'],
            ['num_rcode_REFUSED', 'REFUSED', 'incremental'],
            ['num_rcode_YXDOMAIN', 'YXDOMAIN', 'incremental']
        ]
    }
}


class Service(ExecutableService):
    def __init__(self, configuration=None, name=None):
        ExecutableService.__init__(self, configuration=configuration, name=name)
        self.order = ORDER
        self.definitions = CHARTS
        self.command = NSD_CONTROL_COMMAND

    def _get_data(self):
        lines = self._get_raw_data()
        if not lines:
            return None

        stats = dict(
            (k.replace('.', '_'), int(v)) for k, v in REGEX.findall(''.join(lines))
        )
        stats.setdefault('num_opcode_NOTIFY', 0)
        stats.setdefault('num_type_TYPE252', 0)
        stats.setdefault('num_type_TYPE255', 0)

        return stats