mirror of https://github.com/progval/irctest.git
75 lines
2.8 KiB
Python
75 lines
2.8 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
|