mirror of https://github.com/progval/irctest.git
86 lines
2.9 KiB
Python
86 lines
2.9 KiB
Python
import os
|
|
import subprocess
|
|
|
|
from irctest import authentication
|
|
from irctest import tls
|
|
from irctest.basecontrollers import NotImplementedByController
|
|
from irctest.basecontrollers import BaseClientController, DirectoryBasedController
|
|
|
|
TEMPLATE_CONFIG = """
|
|
supybot.directories.conf: {directory}/conf
|
|
supybot.directories.data: {directory}/data
|
|
supybot.directories.migrations: {directory}/migrations
|
|
supybot.log.level: DEBUG
|
|
supybot.log.stdout.level: {loglevel}
|
|
|
|
supybot.networks: testnet
|
|
supybot.networks.testnet.servers: {hostname}:{port}
|
|
|
|
supybot.protocols.ssl.verifyCertificates: True
|
|
supybot.networks.testnet.ssl: {enable_tls}
|
|
supybot.networks.testnet.ssl.serverFingerprints: {trusted_fingerprints}
|
|
|
|
supybot.networks.testnet.sasl.username: {username}
|
|
supybot.networks.testnet.sasl.password: {password}
|
|
supybot.networks.testnet.sasl.ecdsa_key: {directory}/ecdsa_key.pem
|
|
supybot.networks.testnet.sasl.mechanisms: {mechanisms}
|
|
"""
|
|
|
|
|
|
class LimnoriaController(BaseClientController, DirectoryBasedController):
|
|
software_name = "Limnoria"
|
|
supported_sasl_mechanisms = {
|
|
"PLAIN",
|
|
"ECDSA-NIST256P-CHALLENGE",
|
|
"SCRAM-SHA-256",
|
|
"EXTERNAL",
|
|
}
|
|
supported_capabilities = set(["sts"]) # Not exhaustive
|
|
|
|
def create_config(self):
|
|
create_config = super().create_config()
|
|
if create_config:
|
|
with self.open_file("bot.conf"):
|
|
pass
|
|
with self.open_file("conf/users.conf"):
|
|
pass
|
|
|
|
def run(self, hostname, port, auth, tls_config=None):
|
|
if tls_config is None:
|
|
tls_config = tls.TlsConfig(enable=False, trusted_fingerprints=[])
|
|
# Runs a client with the config given as arguments
|
|
assert self.proc is None
|
|
self.create_config()
|
|
if auth:
|
|
mechanisms = " ".join(
|
|
map(authentication.Mechanisms.as_string, auth.mechanisms)
|
|
)
|
|
if auth.ecdsa_key:
|
|
with self.open_file("ecdsa_key.pem") as fd:
|
|
fd.write(auth.ecdsa_key)
|
|
else:
|
|
mechanisms = ""
|
|
with self.open_file("bot.conf") as fd:
|
|
fd.write(
|
|
TEMPLATE_CONFIG.format(
|
|
directory=self.directory,
|
|
loglevel="CRITICAL",
|
|
hostname=hostname,
|
|
port=port,
|
|
username=auth.username if auth else "",
|
|
password=auth.password if auth else "",
|
|
mechanisms=mechanisms.lower(),
|
|
enable_tls=tls_config.enable if tls_config else "False",
|
|
trusted_fingerprints=" ".join(tls_config.trusted_fingerprints)
|
|
if tls_config
|
|
else "",
|
|
)
|
|
)
|
|
self.proc = subprocess.Popen(
|
|
["supybot", os.path.join(self.directory, "bot.conf")]
|
|
)
|
|
|
|
|
|
def get_irctest_controller_class():
|
|
return LimnoriaController
|