mirror of https://github.com/vector-im/riot-web
191 lines
6.6 KiB
TypeScript
191 lines
6.6 KiB
TypeScript
/*
|
|
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 { MatrixEvent } from "matrix-js-sdk/src/models/event";
|
|
import { MatrixClientPeg } from "../MatrixClientPeg";
|
|
import { ALL_RULE_TYPES, BanList } from "./BanList";
|
|
import SettingsStore from "../settings/SettingsStore";
|
|
import { _t } from "../languageHandler";
|
|
import dis from "../dispatcher/dispatcher";
|
|
import { SettingLevel } from "../settings/SettingLevel";
|
|
import { Preset } from "matrix-js-sdk/src/@types/partials";
|
|
import { ActionPayload } from "../dispatcher/payloads";
|
|
|
|
// TODO: Move this and related files to the js-sdk or something once finalized.
|
|
|
|
export class Mjolnir {
|
|
private static instance: Mjolnir = null;
|
|
|
|
private _lists: BanList[] = []; // eslint-disable-line @typescript-eslint/naming-convention
|
|
private _roomIds: string[] = []; // eslint-disable-line @typescript-eslint/naming-convention
|
|
private mjolnirWatchRef: string = null;
|
|
private dispatcherRef: string = null;
|
|
|
|
get roomIds(): string[] {
|
|
return this._roomIds;
|
|
}
|
|
|
|
get lists(): BanList[] {
|
|
return this._lists;
|
|
}
|
|
|
|
start() {
|
|
this.mjolnirWatchRef = SettingsStore.watchSetting("mjolnirRooms", null, this.onListsChanged.bind(this));
|
|
|
|
this.dispatcherRef = dis.register(this.onAction);
|
|
dis.dispatch({
|
|
action: 'do_after_sync_prepared',
|
|
deferred_action: { action: 'setup_mjolnir' },
|
|
});
|
|
}
|
|
|
|
private onAction = (payload: ActionPayload) => {
|
|
if (payload['action'] === 'setup_mjolnir') {
|
|
console.log("Setting up Mjolnir: after sync");
|
|
this.setup();
|
|
}
|
|
};
|
|
|
|
setup() {
|
|
if (!MatrixClientPeg.get()) return;
|
|
this.updateLists(SettingsStore.getValue("mjolnirRooms"));
|
|
MatrixClientPeg.get().on("RoomState.events", this.onEvent);
|
|
}
|
|
|
|
stop() {
|
|
if (this.mjolnirWatchRef) {
|
|
SettingsStore.unwatchSetting(this.mjolnirWatchRef);
|
|
this.mjolnirWatchRef = null;
|
|
}
|
|
|
|
if (this.dispatcherRef) {
|
|
dis.unregister(this.dispatcherRef);
|
|
this.dispatcherRef = null;
|
|
}
|
|
|
|
if (!MatrixClientPeg.get()) return;
|
|
MatrixClientPeg.get().removeListener("RoomState.events", this.onEvent);
|
|
}
|
|
|
|
async getOrCreatePersonalList(): Promise<BanList> {
|
|
let personalRoomId = SettingsStore.getValue("mjolnirPersonalRoom");
|
|
if (!personalRoomId) {
|
|
const resp = await MatrixClientPeg.get().createRoom({
|
|
name: _t("My Ban List"),
|
|
topic: _t("This is your list of users/servers you have blocked - don't leave the room!"),
|
|
preset: Preset.PrivateChat,
|
|
});
|
|
personalRoomId = resp['room_id'];
|
|
await SettingsStore.setValue(
|
|
"mjolnirPersonalRoom", null, SettingLevel.ACCOUNT, personalRoomId);
|
|
await SettingsStore.setValue(
|
|
"mjolnirRooms", null, SettingLevel.ACCOUNT, [personalRoomId, ...this._roomIds]);
|
|
}
|
|
if (!personalRoomId) {
|
|
throw new Error("Error finding a room ID to use");
|
|
}
|
|
|
|
let list = this._lists.find(b => b.roomId === personalRoomId);
|
|
if (!list) list = new BanList(personalRoomId);
|
|
// we don't append the list to the tracked rooms because it should already be there.
|
|
// we're just trying to get the caller some utility access to the list
|
|
|
|
return list;
|
|
}
|
|
|
|
// get without creating the list
|
|
getPersonalList(): BanList {
|
|
const personalRoomId = SettingsStore.getValue("mjolnirPersonalRoom");
|
|
if (!personalRoomId) return null;
|
|
|
|
let list = this._lists.find(b => b.roomId === personalRoomId);
|
|
if (!list) list = new BanList(personalRoomId);
|
|
// we don't append the list to the tracked rooms because it should already be there.
|
|
// we're just trying to get the caller some utility access to the list
|
|
|
|
return list;
|
|
}
|
|
|
|
async subscribeToList(roomId: string) {
|
|
const roomIds = [...this._roomIds, roomId];
|
|
await SettingsStore.setValue("mjolnirRooms", null, SettingLevel.ACCOUNT, roomIds);
|
|
this._lists.push(new BanList(roomId));
|
|
}
|
|
|
|
async unsubscribeFromList(roomId: string) {
|
|
const roomIds = this._roomIds.filter(r => r !== roomId);
|
|
await SettingsStore.setValue("mjolnirRooms", null, SettingLevel.ACCOUNT, roomIds);
|
|
this._lists = this._lists.filter(b => b.roomId !== roomId);
|
|
}
|
|
|
|
private onEvent = (event: MatrixEvent) => {
|
|
if (!MatrixClientPeg.get()) return;
|
|
if (!this._roomIds.includes(event.getRoomId())) return;
|
|
if (!ALL_RULE_TYPES.includes(event.getType())) return;
|
|
|
|
this.updateLists(this._roomIds);
|
|
};
|
|
|
|
private onListsChanged(settingName: string, roomId: string, atLevel: SettingLevel, newValue: string[]) {
|
|
// We know that ban lists are only recorded at one level so we don't need to re-eval them
|
|
this.updateLists(newValue);
|
|
}
|
|
|
|
private updateLists(listRoomIds: string[]) {
|
|
if (!MatrixClientPeg.get()) return;
|
|
|
|
console.log("Updating Mjolnir ban lists to: " + listRoomIds);
|
|
this._lists = [];
|
|
this._roomIds = listRoomIds || [];
|
|
if (!listRoomIds) return;
|
|
|
|
for (const roomId of listRoomIds) {
|
|
// Creating the list updates it
|
|
this._lists.push(new BanList(roomId));
|
|
}
|
|
}
|
|
|
|
isServerBanned(serverName: string): boolean {
|
|
for (const list of this._lists) {
|
|
for (const rule of list.serverRules) {
|
|
if (rule.isMatch(serverName)) {
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
isUserBanned(userId: string): boolean {
|
|
for (const list of this._lists) {
|
|
for (const rule of list.userRules) {
|
|
if (rule.isMatch(userId)) {
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static sharedInstance(): Mjolnir {
|
|
if (!Mjolnir.instance) {
|
|
Mjolnir.instance = new Mjolnir();
|
|
}
|
|
return Mjolnir.instance;
|
|
}
|
|
}
|
|
|