2019-09-26 02:13:20 +00:00
|
|
|
/*
|
|
|
|
Copyright 2019 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.
|
|
|
|
*/
|
|
|
|
|
|
|
|
import * as path from "path";
|
|
|
|
import {
|
|
|
|
AutojoinRoomsMixin,
|
|
|
|
LogService,
|
|
|
|
MatrixClient,
|
2019-09-27 19:57:36 +00:00
|
|
|
Permalinks,
|
2019-09-26 02:13:20 +00:00
|
|
|
RichConsoleLogger,
|
|
|
|
SimpleFsStorageProvider
|
|
|
|
} from "matrix-bot-sdk";
|
|
|
|
import config from "./config";
|
2019-09-27 21:44:28 +00:00
|
|
|
import BanList from "./models/BanList";
|
2019-09-27 21:15:10 +00:00
|
|
|
import { Mjolnir } from "./Mjolnir";
|
2019-09-26 02:13:20 +00:00
|
|
|
|
|
|
|
LogService.setLogger(new RichConsoleLogger());
|
|
|
|
|
|
|
|
const storage = new SimpleFsStorageProvider(path.join(config.dataPath, "bot.json"));
|
|
|
|
const client = new MatrixClient(config.homeserverUrl, config.accessToken, storage);
|
|
|
|
|
|
|
|
if (config.autojoin) {
|
|
|
|
AutojoinRoomsMixin.setupOnClient(client);
|
|
|
|
}
|
|
|
|
|
2019-09-27 19:57:36 +00:00
|
|
|
(async function () {
|
2019-09-27 21:15:10 +00:00
|
|
|
const banLists: BanList[] = [];
|
2019-09-27 21:44:28 +00:00
|
|
|
const protectedRooms: { [roomId: string]: string } = {};
|
2019-09-27 21:15:10 +00:00
|
|
|
|
2019-09-27 19:57:36 +00:00
|
|
|
// Ensure we're in all the rooms we expect to be in
|
|
|
|
const joinedRooms = await client.getJoinedRooms();
|
|
|
|
for (const roomRef of config.banLists) {
|
|
|
|
const permalink = Permalinks.parseUrl(roomRef);
|
|
|
|
if (!permalink.roomIdOrAlias) continue;
|
|
|
|
|
|
|
|
const roomId = await client.resolveRoom(permalink.roomIdOrAlias);
|
|
|
|
if (!joinedRooms.includes(roomId)) {
|
|
|
|
await client.joinRoom(permalink.roomIdOrAlias, permalink.viaServers);
|
|
|
|
}
|
|
|
|
|
|
|
|
const list = new BanList(roomId, roomRef, client);
|
|
|
|
await list.updateList();
|
2019-09-27 21:15:10 +00:00
|
|
|
banLists.push(list);
|
2019-09-27 19:57:36 +00:00
|
|
|
}
|
|
|
|
|
2019-09-27 20:36:23 +00:00
|
|
|
// Ensure we're also joined to the rooms we're protecting
|
|
|
|
for (const roomRef of config.protectedRooms) {
|
|
|
|
const permalink = Permalinks.parseUrl(roomRef);
|
|
|
|
if (!permalink.roomIdOrAlias) continue;
|
|
|
|
|
|
|
|
let roomId = await client.resolveRoom(permalink.roomIdOrAlias);
|
|
|
|
if (!joinedRooms.includes(roomId)) {
|
|
|
|
roomId = await client.joinRoom(permalink.roomIdOrAlias, permalink.viaServers);
|
|
|
|
}
|
|
|
|
|
|
|
|
protectedRooms[roomId] = roomRef;
|
|
|
|
}
|
|
|
|
|
2019-09-27 21:44:28 +00:00
|
|
|
// Ensure we've joined the ban list we're publishing too
|
|
|
|
let banListRoomId = await client.resolveRoom(config.publishedBanListRoom);
|
|
|
|
if (!joinedRooms.includes(banListRoomId)) {
|
|
|
|
banListRoomId = await client.joinRoom(config.publishedBanListRoom);
|
|
|
|
}
|
|
|
|
|
2019-09-27 19:57:36 +00:00
|
|
|
// Ensure we're also in the management room
|
2019-09-27 21:15:10 +00:00
|
|
|
const managementRoomId = await client.joinRoom(config.managementRoom);
|
2019-09-27 19:57:36 +00:00
|
|
|
await client.sendNotice(managementRoomId, "Mjolnir is starting up. Use !mjolnir to query status.");
|
|
|
|
|
2019-09-27 21:44:28 +00:00
|
|
|
const bot = new Mjolnir(client, managementRoomId, banListRoomId, protectedRooms, banLists);
|
2019-09-27 21:15:10 +00:00
|
|
|
await bot.start();
|
|
|
|
|
2019-09-27 19:57:36 +00:00
|
|
|
// TODO: Check permissions for mjolnir in protected rooms
|
|
|
|
// TODO: Complain about permission changes in protected rooms (including after power levels change)
|
|
|
|
|
|
|
|
LogService.info("index", "Bot started!")
|
|
|
|
})();
|