mirror of https://github.com/progval/irctest.git
107 lines
3.5 KiB
Python
107 lines
3.5 KiB
Python
import json
|
|
import os
|
|
import subprocess
|
|
from typing import Optional, Type
|
|
|
|
from irctest import authentication, tls
|
|
from irctest.basecontrollers import (
|
|
BaseClientController,
|
|
DirectoryBasedController,
|
|
NotImplementedByController,
|
|
)
|
|
|
|
TEMPLATE_CONFIG = """
|
|
"use strict";
|
|
|
|
module.exports = {config};
|
|
"""
|
|
|
|
|
|
class TheLoungeController(BaseClientController, DirectoryBasedController):
|
|
software_name = "TheLounge"
|
|
supported_sasl_mechanisms = {
|
|
"PLAIN",
|
|
"ECDSA-NIST256P-CHALLENGE",
|
|
"SCRAM-SHA-256",
|
|
"EXTERNAL",
|
|
}
|
|
supports_sts = True
|
|
|
|
def create_config(self) -> None:
|
|
super().create_config()
|
|
with self.open_file("bot.conf"):
|
|
pass
|
|
with self.open_file("conf/users.conf"):
|
|
pass
|
|
|
|
def run(
|
|
self,
|
|
hostname: str,
|
|
port: int,
|
|
auth: Optional[authentication.Authentication],
|
|
tls_config: Optional[tls.TlsConfig] = None,
|
|
) -> None:
|
|
if tls_config is None:
|
|
tls_config = tls.TlsConfig(enable=False, trusted_fingerprints=[])
|
|
if tls_config and tls_config.trusted_fingerprints:
|
|
raise NotImplementedByController("Trusted fingerprints.")
|
|
if auth and any(
|
|
mech.to_string().startswith(("SCRAM-", "ECDSA-"))
|
|
for mech in auth.mechanisms
|
|
):
|
|
raise NotImplementedByController("ecdsa")
|
|
if auth and auth.password and len(auth.password) > 300:
|
|
# https://github.com/thelounge/thelounge/pull/4480
|
|
# Note that The Lounge truncates on 300 characters, not bytes.
|
|
raise NotImplementedByController("Passwords longer than 300 chars")
|
|
# Runs a client with the config given as arguments
|
|
assert self.proc is None
|
|
self.create_config()
|
|
if auth:
|
|
mechanisms = " ".join(mech.to_string() for mech in auth.mechanisms)
|
|
if auth.ecdsa_key:
|
|
with self.open_file("ecdsa_key.pem") as fd:
|
|
fd.write(auth.ecdsa_key)
|
|
else:
|
|
mechanisms = ""
|
|
|
|
assert self.directory
|
|
with self.open_file("config.js") as fd:
|
|
fd.write(
|
|
TEMPLATE_CONFIG.format(
|
|
config=json.dumps(
|
|
dict(
|
|
public=False,
|
|
host=f"unix:{self.directory}/sock", # prevents binding
|
|
)
|
|
)
|
|
)
|
|
)
|
|
with self.open_file("users/testuser.json") as fd:
|
|
json.dump(
|
|
dict(
|
|
networks=[
|
|
dict(
|
|
name="testnet",
|
|
host=hostname,
|
|
port=port,
|
|
tls=tls_config.enable if tls_config else "False",
|
|
sasl=mechanisms.lower(),
|
|
saslAccount=auth.username if auth else "",
|
|
saslPassword=auth.password if auth else "",
|
|
)
|
|
]
|
|
),
|
|
fd,
|
|
)
|
|
with self.open_file("users/testuser.json", "r") as fd:
|
|
print("config", json.load(fd)["networks"][0]["saslPassword"])
|
|
self.proc = subprocess.Popen(
|
|
[os.environ.get("THELOUNGE_BIN", "thelounge"), "start"],
|
|
env={**os.environ, "THELOUNGE_HOME": str(self.directory)},
|
|
)
|
|
|
|
|
|
def get_irctest_controller_class() -> Type[TheLoungeController]:
|
|
return TheLoungeController
|