2021-09-14 08:31:59 +00:00
|
|
|
/*
|
|
|
|
Copyright 2021 The Matrix.org Foundation C.I.C.
|
|
|
|
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
you may not use this file except in compliance with the License.
|
|
|
|
You may obtain a copy of the License at
|
|
|
|
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
See the License for the specific language governing permissions and
|
|
|
|
limitations under the License.
|
|
|
|
*/
|
2021-09-08 15:44:06 +00:00
|
|
|
import {
|
|
|
|
MatrixClient,
|
|
|
|
Permalinks,
|
2021-09-22 10:57:45 +00:00
|
|
|
PantalaimonClient,
|
|
|
|
MemoryStorageProvider
|
2021-09-08 15:44:06 +00:00
|
|
|
} from "matrix-bot-sdk";
|
|
|
|
import config from "../../src/config";
|
2021-09-22 10:57:45 +00:00
|
|
|
import * as HmacSHA1 from 'crypto-js/hmac-sha1';
|
|
|
|
import axios from 'axios';
|
|
|
|
import * as path from 'path';
|
|
|
|
import * as fs from 'fs/promises';
|
2021-09-08 15:44:06 +00:00
|
|
|
|
|
|
|
export async function createManagementRoom(client: MatrixClient) {
|
|
|
|
let roomId = await client.createRoom();
|
|
|
|
return await client.createRoomAlias(config.managementRoom, roomId);
|
|
|
|
}
|
|
|
|
|
|
|
|
export async function ensureManagementRoomExists(client: MatrixClient): Promise<string> {
|
|
|
|
return await client.resolveRoom(config.managementRoom).catch(async e => {
|
|
|
|
if (e?.body?.errcode === 'M_NOT_FOUND') {
|
|
|
|
console.info("moderation room hasn't been created yet, so we're making it now.")
|
|
|
|
return await createManagementRoom(client);
|
|
|
|
}
|
|
|
|
throw e;
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
export async function ensureLobbyRoomExists(client: MatrixClient): Promise<string> {
|
|
|
|
const alias = Permalinks.parseUrl(config.protectedRooms[0]).roomIdOrAlias;
|
|
|
|
return await client.resolveRoom(alias).catch(async e => {
|
|
|
|
if (e?.body?.errcode === 'M_NOT_FOUND') {
|
|
|
|
console.info(`${alias} hasn't been created yet, so we're making it now.`)
|
|
|
|
return await client.createRoomAlias(alias, await client.createRoom());
|
|
|
|
}
|
|
|
|
throw e;
|
|
|
|
});
|
2021-09-22 10:57:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
async function configureMjolnir() {
|
|
|
|
await fs.copyFile(path.join(__dirname, 'config', 'mjolnir', 'harness.yaml'), path.join(__dirname, '../../config/harness.yaml'));
|
|
|
|
await fs.rm(path.join(__dirname, 'mjolnir-data'), {recursive: true, force: true});
|
|
|
|
await registerUser('mjolnir', 'mjolnir', 'mjolnir', true);
|
|
|
|
const pantalaimon = new PantalaimonClient(config.homeserverUrl, new MemoryStorageProvider());
|
|
|
|
const client = await pantalaimon.createClientWithCredentials(config.pantalaimon.username, config.pantalaimon.password);
|
|
|
|
await ensureManagementRoomExists(client);
|
|
|
|
return await ensureLobbyRoomExists(client);
|
|
|
|
}
|
|
|
|
|
|
|
|
async function startMjolnir() {
|
|
|
|
await configureMjolnir();
|
|
|
|
console.info('starting mjolnir');
|
|
|
|
await import('../../src/index');
|
|
|
|
}
|
|
|
|
|
|
|
|
async function registerUser(username: string, displayname: string, password: string, admin: boolean) {
|
|
|
|
let registerUrl = `${config.homeserverUrl}/_synapse/admin/v1/register`
|
|
|
|
let { data } = await axios.get(registerUrl);
|
|
|
|
let nonce = data.nonce!;
|
|
|
|
let mac = HmacSHA1(`${nonce}\0${username}\0${password}\0${admin ? 'admin' : 'notadmin'}`, 'REGISTRATION_SHARED_SECRET');
|
|
|
|
return await axios.post(registerUrl, {
|
|
|
|
nonce,
|
|
|
|
username,
|
|
|
|
displayname,
|
|
|
|
password,
|
|
|
|
admin,
|
|
|
|
mac: mac.toString()
|
|
|
|
}).catch(e => {
|
|
|
|
if (e.isAxiosError && e.response.data.errcode === 'M_USER_IN_USE') {
|
|
|
|
console.log(`${username} already registered, skipping`)
|
|
|
|
} else {
|
|
|
|
throw e;
|
|
|
|
}
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|