"""
Get login statistics for a samba server.
NOTE: This file is now deprecated in favour of the newer journald mechanism
used in smbd-journald.py. This parser is still functional but is slower and
has less features. Please switch over if possible.
"""

import re
import glob

from logparse.formatting import *
from logparse.util import readlog, resolve
from logparse import config
from logparse.load_parsers import Parser

class Smbd(Parser):

    def __init__(self):
        super().__init__()
        self.name = "smbd"
        self.info = "Get login statistics for a samba server."
        self.deprecated = True
        self.successor = "smbd_journald"

    def parse_log(self):

        # Find list of logfiles

        logger.debug("Starting smbd section")
        section = Section("smbd")
        files = glob.glob(config.prefs.get("logs", "smbd") 
                + "/log.*[!\.gz][!\.old]")

        logger.debug("Found log files " + str(files))

        n_auths = 0         # total number of logins from all users
        sigma_auths = []    # contains users

        for file in files:  # one log file for each client

            logger.debug("Looking at file " + file)

            # Find the machine (IP or hostname) that this file represents

            # Get IP or hostname from file path (/var/log/samba/log.host)
            ip = re.search('log\.(.*)', file).group(1)    

            # If IP has disappeared, fall back to a hostname from logfile
            host = resolve(ip, fqdn=config.prefs.get("smbd", "smbd-resolve-domains"))
            if (host == ip and (
                        config.prefs.get("smbd", "smbd-resolve-domains") != "ip" 
                        or config.prefs.get("logparse", "resolve-domains") != "ip"):    
                newhost = re.findall('.*\]\@\[(.*)\]', readlog(file))
                if (len(set(newhost)) == 1):    # all hosts in one file should be the same
                    host = newhost[0].lower()

            # Count number of logins from each user-host pair
            matches = re.findall('.*(?:authentication for user \[|connect "
                "to service .* initially as user )(\S*)(?:\] .*succeeded| \()',
                readlog(file))

            for match in matches:
                userhost = match + "@" + host
                sigma_auths.append(userhost)
                n_auths += 1

        auth_data = Data(subtitle=plural("login", n_auths) + " from")
        if (len(sigma_auths) == 1):             
            # If only one user, do not display no of logins for this user
            auth_data.subtitle += ' ' + sigma_auths[0][0]
            section.append_data(auth_data)
        else:       # multiple users
            auth_data.items = sigma_auths
            auth_data.orderbyfreq()
            auth_data.truncl(config.prefs.getint("logparse", "maxlist"))
            logger.debug("Found {0} samba logins".format(str(n_auths)))
        section.append_data(auth_data)
        logger.info("Finished smbd section")
        return section