#
# gpg-mailgate
#
# This file is part of the gpg-mailgate source code.
#
# gpg-mailgate is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# gpg-mailgate source code is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with gpg-mailgate source code. If not, see <http://www.gnu.org/licenses/>.
#

import configparser
import logging
import subprocess
import os
import time


def _spawn(cmd):
    env_dict = {
        "PATH": os.getenv("PATH"),
        "PYTHONPATH": os.getcwd(),
        "GPG_MAILGATE_CONFIG": "test/gpg-mailgate-daemon-test.conf"
        }
    logging.debug(f"Spawning command: {cmd} with environment: {env_dict!r}")
    return subprocess.Popen(cmd,
                            stdin=None,
                            stdout=subprocess.PIPE,
                            env=env_dict)


def _interrupt(proc):
    # proc.send_signal(signal.SIGINT)
    proc.terminate()


def _load(name):
    logging.debug(f"Loading file {name}")
    f = open(name, "r")
    contents = f.read()
    f.close()
    return contents


def _send(host, port, mail_from, mail_to, message):
    logging.debug(f"Sending message to {host}:{port}")
    _spawn([os.getenv("PYTHON") or "python",
            "test/utils/sendmail.py",
            "-f", mail_from,
            "-t", mail_to,
            "-m", message])


def _load_test_config():
    cp = configparser.ConfigParser()
    cp.read("test/e2e.ini")
    return cp


def _identity(x):
    return x


def _inversion(x):
    return not(x)


def _report_result(message_file, expected, test_output, boolean_func=_identity):
    status = None
    expected_line = "\r\n" + expected  # + "\r\n"
    cond_met = boolean_func(expected_line in test_output)
    if cond_met:
        status = "Success"
    else:
        status = "Failure"

    print(message_file.ljust(35), status)


def _execute_case(config, case_name):
    logging.info(f"Executing case {case_name}")
    python = os.getenv("PYTHON", "python")

    relay_mock = _spawn([python, "test/utils/relay.py", "2500"])
    time.sleep(1)                            # Wait for the relay to start up.

    _send("localhost", 10025, "dave@disposlab",
          config.get(case_name, 'to'), config.get(case_name, 'in'))

    relay_mock.wait()
    (test_out, _) = relay_mock.communicate()

    test_out = test_out.decode('utf-8')
    logging.debug(f"Read {len(test_out)} characters of output: '{test_out}'")

    if 'out' in config[case_name]:
        _report_result(config.get(case_name, "in"), config.get(case_name, "out"), test_out)
    else:
        _report_result(config.get(case_name, "in"), config.get(case_name, "out-not"), test_out, boolean_func=_inversion)


def _main():
    conf = _load_test_config()

    logging.basicConfig(filename="test/logs/daemon-test.log",
                        format="%(asctime)s %(pathname)s:%(lineno)d %(levelname)s [%(funcName)s] %(message)s",
                        datefmt="%Y-%m-%d %H:%M:%S",
                        level=logging.DEBUG)

    logging.info("Starting Lacre Daemon tests...")
    python = os.getenv("PYTHON", "python")

    server = _spawn([python, "-m", "lacre.daemon"])

    for case_no in range(1, conf.getint("tests", "cases") + 1):
        _execute_case(conf, case_name=f"case-{case_no}")

    _interrupt(server)


if __name__ == '__main__':
    _main()