import os
import time
import subprocess

from irctest.basecontrollers import NotImplementedByController
from irctest.basecontrollers import BaseServerController, DirectoryBasedController

TEMPLATE_CONFIG = """
clients:
  # ping_frequency - client ping frequency
  ping_frequency:
    minutes: 1000

  # ping_timeout - ping timeout length
  ping_timeout:
    seconds: 10
data:
  format: json
  filename: {directory}/data.json
  save_frequency:
    minutes: 5
extensions:
- mammon.ext.rfc1459.42
- mammon.ext.rfc1459.ident
- mammon.ext.ircv3.account_notify
- mammon.ext.ircv3.server_time
- mammon.ext.ircv3.echo_message
- mammon.ext.ircv3.register
- mammon.ext.ircv3.sasl
- mammon.ext.misc.nopost
metadata:
  restricted_keys:
{restricted_keys}
  whitelist:
{authorized_keys}
monitor:
  limit: 20
motd:
  - "Hi"
limits:
  foo: bar
listeners:
- {{"host": "{hostname}", "port": {port}, "ssl": false}}
logs:
  {{
  }}
register:
  enabled_callbacks:
  - none
  verify_timeout:
    days: 1
roles:
  "placeholder":
    title: "Just a placeholder"
server:
  name: MyLittleServer
  network: MyLittleNetwork
  recvq_len: 20
"""


def make_list(l):
    return "\n".join(map("  - {}".format, l))


class MammonController(BaseServerController, DirectoryBasedController):
    software_name = "Mammon"
    supported_sasl_mechanisms = {
        "PLAIN",
        "ECDSA-NIST256P-CHALLENGE",
    }
    supported_capabilities = set()  # Not exhaustive

    def create_config(self):
        super().create_config()
        with self.open_file("server.conf"):
            pass

    def kill_proc(self):
        # Mammon does not seem to handle SIGTERM very well
        self.proc.kill()

    def run(
        self,
        hostname,
        port,
        password=None,
        ssl=False,
        restricted_metadata_keys=(),
        valid_metadata_keys=(),
        invalid_metadata_keys=(),
    ):
        if password is not None:
            raise NotImplementedByController("PASS command")
        if ssl:
            raise NotImplementedByController("SSL")
        assert self.proc is None
        self.port = port
        self.create_config()
        with self.open_file("server.yml") as fd:
            fd.write(
                TEMPLATE_CONFIG.format(
                    directory=self.directory,
                    hostname=hostname,
                    port=port,
                    authorized_keys=make_list(valid_metadata_keys),
                    restricted_keys=make_list(restricted_metadata_keys),
                )
            )
        # with self.open_file('server.yml', 'r') as fd:
        #    print(fd.read())
        self.proc = subprocess.Popen(
            [
                "mammond",
                "--nofork",  #'--debug',
                "--config",
                os.path.join(self.directory, "server.yml"),
            ]
        )

    def registerUser(self, case, username, password=None):
        # XXX: Move this somewhere else when
        # https://github.com/ircv3/ircv3-specifications/pull/152 becomes
        # part of the specification
        client = case.addClient(show_io=False)
        case.sendLine(client, "CAP LS 302")
        case.sendLine(client, "NICK registration_user")
        case.sendLine(client, "USER r e g :user")
        case.sendLine(client, "CAP END")
        while case.getRegistrationMessage(client).command != "001":
            pass
        list(case.getMessages(client))
        case.sendLine(client, "REG CREATE {} passphrase {}".format(username, password))
        msg = case.getMessage(client)
        assert msg.command == "920", msg
        list(case.getMessages(client))
        case.removeClient(client)


def get_irctest_controller_class():
    return MammonController