#!/usr/bin/env python

import os
import re

from yunohost.settings import settings_get
from yunohost.diagnosis import Diagnoser
from yunohost.regenconf import _get_regenconf_infos, _calculate_hash
from moulinette.utils.filesystem import read_file

class RegenconfDiagnoser(Diagnoser):

    id_ = os.path.splitext(os.path.basename(__file__))[0].split("-")[1]
    cache_duration = 300
    dependencies = []

    def run(self):

        regenconf_modified_files = list(self.manually_modified_files())

        if not regenconf_modified_files:
            yield dict(
                meta={"test": "regenconf"},
                status="SUCCESS",
                summary="diagnosis_regenconf_allgood",
            )
        else:
            for f in regenconf_modified_files:
                yield dict(
                    meta={
                        "test": "regenconf",
                        "category": f["category"],
                        "file": f["path"],
                    },
                    status="WARNING",
                    summary="diagnosis_regenconf_manually_modified",
                    details=["diagnosis_regenconf_manually_modified_details"],
                )

        if any(f["path"] == '/etc/ssh/sshd_config' for f in regenconf_modified_files) \
            and os.system("grep -q '^ *AllowGroups\\|^ *AllowUsers' /etc/ssh/sshd_config") != 0:
                yield dict(
                   meta={
                       "test": "sshd_config_insecure"
                   },
                   status="ERROR",
                   summary="diagnosis_sshd_config_insecure",
                )

        # Check consistency between actual ssh port in sshd_config vs. setting
        ssh_port_setting = settings_get('security.ssh.port')
        ssh_port_line = re.findall(
            r"\bPort *([0-9]{2,5})\b", read_file("/etc/ssh/sshd_config")
        )
        if len(ssh_port_line) == 1 and int(ssh_port_line[0]) != ssh_port_setting:
            yield dict(
               meta={
                   "test": "sshd_config_port_inconsistency"
               },
               status="WARNING",
               summary="diagnosis_sshd_config_inconsistent",
               details=["diagnosis_sshd_config_inconsistent_details"],
            )

    def manually_modified_files(self):

        for category, infos in _get_regenconf_infos().items():
            for path, hash_ in infos["conffiles"].items():
                if hash_ != _calculate_hash(path):
                    yield {"path": path, "category": category}


def main(args, env, loggers):
    return RegenconfDiagnoser(args, env, loggers).diagnose()