#!/usr/bin/env python3

import getopt
import sys
import requests
import urllib3
import json
import os
import cmk.utils.password_store
from pprint import pprint
from requests.structures import CaseInsensitiveDict
from requests.auth import HTTPBasicAuth
from time import (
    time, localtime, strftime,
)

def showUsage() -> None:
    sys.stderr.write("""CheckMK Mailcow Special Agent

USAGE: agent_nextcloud -H [hostname] -k [apikey]
       agent_nextcloud -h

OPTIONS:
  -H, --hostname                Hostname (FQDN or IP) of Nextcloud server
  -k, --apikey                  API Key
  -P, --port                    Port
  --no-https True|False         If "True": Disable HTTPS, use HTTP (not recommended!)
  --no-cert-check True|False    If "True": Disable TLS certificate check (not recommended!)
  -h, --help                    Show this help message and exit
""")

# set this to True to produce debug output (this clutters the agent output)
# be aware: activating this logs very sensitive information to debug files in ~/tmp 
# !!DO NOT FORGET to delete these files after debugging is done!!

DEBUG = False

mc_api_base = "api/v1/get"

opt_hostname = ""
opt_apikey = ""
opt_port = ""
opt_no_https = False
opt_no_cert_check = False

short_options = 'hH:k:P:'
long_options = [
    'hostname=', 'apikey=', 'port=', 'no-https=', 'no-cert-check=', 'help'
]

domain_data = {}
mailbox_data = {}

debugLogFilename = ""
SEP = "|"
#SEP = "\t"
TIMEFMT = "%Y-%m-%d %H:%M:%S"
FLOATFMT = "{:.4f}"

'''
Decorator function for debugging purposes
    creates a file with many information regarding the function call, like:
        timestamp
        name of function
        runtime
        number of arguments
        number of keyword arguments
        return value of function call
        type of return value
        all parameters given to function
'''
def debugLog(function):
    def wrapper(*args, **kwargs):
        # execute function and measure runtime
        start = time()
        value = function(*args, **kwargs)
        end = time()
        # get number of args and kwargs
        len_args = len(args)
        len_kwargs = len(kwargs)
        # format the output
        seconds = FLOATFMT.format(end - start)
        local_time = strftime(TIMEFMT,localtime(start))
        # get function name
        fname = function.__name__
        # create output
        # out1: Timestamp|Name of Function|Runtime|Num Args|Num Kwargs|Return Value|Return Value Type
        out1 = f"{local_time}{SEP}{fname}{SEP}{seconds}{SEP}{len_args}{SEP}{len_kwargs}{SEP}{value}{SEP}{type(value)}"
        # out2: all arguments
        out2 = ""
        for arg in args:
            out2 = out2 + SEP + str(arg)
        # out3: all keyword arguments
        out3 = ""
        if len_kwargs > 0:
            for key, val in kwargs.items():
                out3 = out3 + SEP + key + ":" + str(val)
        # write output to file
        if debugLogFilename != "":
            try:
                with open(debugLogFilename, "a+") as f:
                    f.write(f"{out1}{out2}{out3}\n")
            except:
                sys.stderr.write(f"Something went wrong when writing to file {debugLogFilename}\n")
                sys.exit(1)
        else:
            sys.stderr.write(f"Debug activated, but no debug filename given.\n")
            sys.exit(1)
        return value
    return wrapper


def getDebugFilename(hostname: str) -> str:
    home_path = os.getenv("HOME")
    tmp_path = f"{home_path}/tmp"
    file_name = f"{tmp_path}/mailcow_{hostname}_debug.log"
    return file_name


def getOptions() -> None:
    global opt_hostname
    global opt_apikey
    global opt_port
    global opt_no_https
    global opt_no_cert_check

    opts, args = getopt.getopt(sys.argv[1:], short_options, long_options)
    for opt, arg in opts:
        if opt in ['-H', '--hostname']:
            opt_hostname = arg
        elif opt in ['-k', '--apikey']:
            opt_apikey = arg
        elif opt in ['-P', '--port']:
            opt_port = arg
        elif opt in ['--no-https']:
            if arg == 'True':
                opt_no_https = True
            else:
                opt_no_https = False
        elif opt in ['--no-cert-check']:
            if arg == 'True':
                opt_no_cert_check = True
            else:
                opt_no_cert_check = False
        elif opt in ['-h', '--help']:
            showUsage()
            sys.exit(0)
    if DEBUG:
        home_path = os.getenv("HOME")
        tmp_path = f"{home_path}/tmp"
        help_file = f"{tmp_path}/mailcow_{opt_hostname}_debug.txt"
        with open(help_file, "a") as file:
            file.write(f"Number of Arguments: {len(sys.argv)}, Argument List: {str(sys.argv)}\n")


#@debugLog
def showOptions() -> None:
    print(f"Hostname: {opt_hostname}")
    print(f"Username: {opt_apikey}")
    print(f"Port: {opt_port}")
    print(f"No HTTPS: {opt_no_https}")
    print(f"No TLS Check: {opt_no_cert_check}")
    home_path = os.getenv("HOME")
    tmp_path = f"{home_path}/tmp"
    help_file = f"{tmp_path}/mailcow_{opt_hostname}_debug.txt"
    with open(help_file, "a") as file:
        file.write(f"Hostname: {opt_hostname}, Port: {opt_port}, No HTTPS: {opt_no_https}, No Cert Check: {opt_no_cert_check}\n")

#@debugLog
def getDomainInfo(headers: str, verify: bool, base_url: str) -> int:
    url = f"{base_url}/domain/all"
    response = requests.get(url, headers=headers, verify=verify)
    if (response.status_code == 200):
        jsdata = response.text
        data = json.loads(jsdata)   # returns a list of dictionaries
        i = 0
        while i < len(data):
            #pprint(data[i])
            # get domain name and status (active (1) or not (0))
            domain_name = data[i].get("domain_name")
            active = data[i].get("active")
            # get creation and last modification date
            created = data[i].get("created")
            modified = data[i].get("modified")      # returns "None" or date
            # get maximum and current number of mailboxes in this domain
            max_num_mboxes_for_domain = data[i].get("max_num_mboxes_for_domain")
            mboxes_in_domain = data[i].get("mboxes_in_domain")
            # get maximum and current number of aliases in this domain
            max_num_aliases_for_domain = data[i].get("max_num_aliases_for_domain")
            aliases_in_domain = data[i].get("aliases_in_domain")
            # get total messages in this domain
            msgs_total = data[i].get("msgs_total")
            # get total bytes used in this domain
            bytes_total = data[i].get("bytes_total")
            # get maximum quota for this domain
            max_quota_for_domain = data[i].get("max_quota_for_domain")
            # store all domain data in global dictionary
            dom = {
                "active": active,
                "created": created,
                "modified": modified,
                "max_num_mboxes_for_domain": max_num_mboxes_for_domain,
                "mboxes_in_domain": mboxes_in_domain,
                "max_num_aliases_for_domain": max_num_aliases_for_domain,
                "aliases_in_domain": aliases_in_domain,
                "msgs_total": msgs_total,
                "bytes_total": bytes_total,
                "max_quota_for_domain": max_quota_for_domain
            }
            domain_data[domain_name] = {}
            domain_data[domain_name] = dom
            i += 1
        # return number of email domains
        return i
    else:
        sys.stderr.write(f"Request response code is {response.status_code} with URL {url}\n")
        sys.exit(1)

#@debugLog
def getMailboxInfo(headers: str, verify: bool, base_url: str) -> tuple:
    url = f"{base_url}/mailbox/all"
    response = requests.get(url, headers=headers, verify=verify)
    if (response.status_code == 200):
        jsdata = response.text
        data = json.loads(jsdata)   # returns a list of dictionaries
        i = 0
        global_num_messages = 0
        while i < len(data):
            # get status of mailbox (0-->inactive or 1-->active)
            active = data[i].get("active")
            # get username of mailbox
            username = data[i].get("username")
            # get friendly name of user
            name = data[i].get("name")
            # get number of messages in mailbox
            num_messages = data[i].get("messages")
            # get quota used in percent (rounded to full percent)
            percent_in_use = data[i].get("percent_in_use")
            # get quota and quota used in bytes
            quota = data[i].get("quota")
            quota_used = data[i].get("quota_used")
            # get creation and last modification date
            created = data[i].get("created")
            modified = data[i].get("modified")
            # get number of messages
            messages = data[i].get("messages")
            # get last login dates
            last_imap_login = data[i].get("last_imap_login")
            last_pop3_login = data[i].get("last_pop3_login")
            last_smtp_login = data[i].get("last_smtp_login")
            mb = {
                "active": active,
                "created": created,
                "modified": modified,
                "name": name,
                "num_messages": num_messages,
                "percent_in_use": percent_in_use,
                "quota": quota,
                "quota_used": quota_used,
                "messages": messages,
                "last_imap_login": last_imap_login,
                "last_pop3_login": last_pop3_login,
                "last_smtp_login": last_smtp_login
            }
            mailbox_data[username] = {}
            mailbox_data[username] = mb
            i += 1
            global_num_messages += num_messages
        # return number of mailboxes and global number of messages
        return i, global_num_messages
    else:
        sys.stderr.write(f"Request response code is {response.status_code} with URL {url}\n")
        sys.exit(1)

#@debugLog
def getMailcowInfo(headers: str, verify: bool, base_url: str) -> str:
    url = f"{base_url}/status/version"
    response = requests.get(url, headers=headers, verify=verify)
    if (response.status_code == 200):
        jsdata = response.text
        data = json.loads(jsdata)   # returns a dictionary
        #pprint(data)
        # get Mailcow version
        mc_version = data["version"]
        return mc_version
    else:
        sys.stderr.write(f"Request response code is {response.status_code} with URL {url}\n")
        sys.exit(1)

#@debugLog
def getSolrInfo(headers: str, verify: bool, base_url: str) -> tuple:
    url = f"{base_url}/status/solr"
    response = requests.get(url, headers=headers, verify=verify)
    if (response.status_code == 200):
        jsdata = response.text
        data = json.loads(jsdata)   # returns a dictionary
        #pprint(data)
        # get Solr infos
        solr_enabled = data["solr_enabled"]
        solr_size = data["solr_size"]
        solr_documents = data["solr_documents"]
        return solr_enabled, solr_size, solr_documents
    else:
        sys.stderr.write(f"Request response code is {response.status_code} with URL {url}\n")
        sys.exit(1)


'''
Output is as follows:
0   mailbox name                email address used for login
1   active                      1 --> active, 0 --> not active
2   creation date               "None" if ???
3   last modified date          "None" if never modified
4   name                        display name
5   number of messages
6   percent in use              quota used, rounded to full percents
7   quota                       max quota in bytes
8   quota used                  quota used in bytes
9   last imap login             seconds since epoch, 0 if never logged in
10  last pop3 login             seconds since epoch, 0 if never logged in
11  last smtp login             seconds since epoch, 0 if never logged in

Example:
user1@dom1.de;1;2022-04-29 14:29:34;2022-04-29 14:29:34;Sarah;2433;2;21474836480;495481374;1692520168;0;1692281537
user2@dom1.de;1;2022-04-29 14:38:33;2022-04-29 14:38:33;Tom;271;0;21474836480;25895752;1657394782;1692519758;1681065713
user1@dom2.de;1;2022-04-30 09:55:37;2022-04-30 09:55:37;Melissa;53460;33;19964887040;6677677548;1692520066;0;1692510822
'''

#@debugLog
def doCmkOutputMailboxes() -> None:
    print("<<<mailcow_mailboxes:sep(59)>>>")
    for mb in mailbox_data:
        active = mailbox_data[mb]["active"]
        created = mailbox_data[mb]["created"]
        modified = mailbox_data[mb]["modified"]
        name = mailbox_data[mb]["name"]
        # strip semicolons, if present, since we use it as delimiter 
        name = name.replace(";", "")
        num_messages = mailbox_data[mb]["num_messages"]
        percent_in_use = mailbox_data[mb]["percent_in_use"]
        quota = mailbox_data[mb]["quota"]
        quota_used = mailbox_data[mb]["quota_used"]
        last_imap_login = mailbox_data[mb]["last_imap_login"]
        last_pop3_login = mailbox_data[mb]["last_pop3_login"]
        last_smtp_login = mailbox_data[mb]["last_smtp_login"]
        print(f"{mb};{active};{created};{modified};{name};{num_messages};{percent_in_use};{quota};{quota_used};{last_imap_login};{last_pop3_login};{last_smtp_login}")

#@debugLog
def doCmkOutputMailcow(version: str, num_domains: int, num_mailboxes: int, num_global_messages: int, solr_enabled: bool, solr_size: float, solr_documents: int) -> None:
    print("<<<mailcow_info:sep(59)>>>")
    # strip semicolons, if present, since we use it as delimiter
    version = version.replace(";", "")
    print(f"{version};{num_domains};{num_mailboxes};{num_global_messages};{solr_enabled};{solr_size};{solr_documents}")


'''
Output is as follows:
0   domain_name
1   active                      1 --> active, 0 --> not active
2   creation date               "None" if ???
3   last modified date          "None" if never modified
4   max number mailboxes
5   number of mailboxes
6   max number of aliases
7   number of aliases
8   total number of messages
9   total number of bytes used  in bytes
10  max quota                   in bytes

Example:
dom1.de;1;2022-04-23 22:54:57;None;10;0;400;6;0;0;10737418240
dom2.de;1;2022-04-29 13:38:42;2023-08-19 17:21:04;10;0;400;2;0;0;10737418240
dom3.de;1;2022-04-29 13:36:08;2022-04-29 21:26:19;10;1;100;3;28132;12852485367;21474836480
'''

#@debugLog
def doCmkOutputDomains() -> None:
    print("<<<mailcow_domains:sep(59)>>>")
    for dom in domain_data:
        active = domain_data[dom]["active"]
        created = domain_data[dom]["created"]
        modified = domain_data[dom]["modified"]
        max_num_mboxes_for_domain = domain_data[dom]["max_num_mboxes_for_domain"]
        mboxes_in_domain = domain_data[dom]["mboxes_in_domain"]
        max_num_aliases_for_domain = domain_data[dom]["max_num_aliases_for_domain"]
        aliases_in_domain = domain_data[dom]["aliases_in_domain"]
        msgs_total = domain_data[dom]["msgs_total"]
        bytes_total = domain_data[dom]["bytes_total"]
        max_quota_for_domain = domain_data[dom]["max_quota_for_domain"]
        print(f"{dom};{active};{created};{modified};{max_num_mboxes_for_domain};{mboxes_in_domain};{max_num_aliases_for_domain};{aliases_in_domain};{msgs_total};{bytes_total};{max_quota_for_domain}")


def main():
    global debugLogFilename
    cmk.utils.password_store.replace_passwords()
    getOptions()
    # do some parameter checks
    if (opt_hostname == ""):
        sys.stderr.write(f"No hostname given.\n")
        showUsage()
        sys.exit(1)
    else:
        hostname = opt_hostname
    if (opt_apikey == ""):
        sys.stderr.write(f"No API key given.\n")
        showUsage()
        sys.exit(1)
    if (opt_no_cert_check):
        # disable certificate warnings
        urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
        verify = False
    else:
        verify = True
    if (opt_port == ""):     
        if (opt_no_https):
            protocol = "http"
            port = "80"
        else:
            protocol = "https"
            port = "443"
    else:
        if (opt_no_https):
            protocol = "http"
        else:
            protocol = "https"
        port = opt_port
    if (protocol == "http" and port == "443"):
        sys.stderr.write(f"Combining HTTP with port 443 is not supported.\n")
        sys.exit(1)
    if (protocol == "https" and port == "80"):
        sys.stderr.write(f"Combining HTTPS with port 80 is not supported.\n")
        sys.exit(1)
    headers = CaseInsensitiveDict()
    headers["Accept"] = "application/json"
    headers["X-API-Key"] = opt_apikey
    # now "hostname" contains the FQDN of the host running Mailcow
    # now "protocol" is http or https
    # now "port" contains the port number to use
    # now "verify" signals whether certificate checking will be done (True) or not (False)
    # now "headers" contains the accepted format (JSON) and the API key to use
    debugLogFilename = getDebugFilename(hostname)
    if DEBUG:
        showOptions()
        print(f"hostname: {hostname}, protocol: {protocol}, port: {port}, verify: {verify}")
    base_url = f"{protocol}://{hostname}:{port}/{mc_api_base}"
    # get domain data
    num_domains = getDomainInfo(headers, verify, base_url)
    # get mailbox data
    num_mailboxes, num_global_messages = getMailboxInfo(headers, verify, base_url)
    # get global Mailcow info
    solr_enabled, solr_size, solr_documents = getSolrInfo(headers, verify, base_url)
    mailcow_version = getMailcowInfo(headers, verify, base_url)
    # create agent output
    doCmkOutputDomains()
    doCmkOutputMailboxes()
    doCmkOutputMailcow(mailcow_version, num_domains, num_mailboxes, num_global_messages, solr_enabled, solr_size, solr_documents)


if __name__ == "__main__":
    main()