-
Notifications
You must be signed in to change notification settings - Fork 53
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
5 changed files
with
174 additions
and
6 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,18 @@ | ||
|
|
||
|
|
||
| export enum ConsequenceType { | ||
| alert, | ||
| redact, | ||
| ban | ||
| } | ||
|
|
||
| export class Consequence { | ||
| /* | ||
| * A description of an action to take when a protection detects abuse | ||
| * | ||
| * @param type Action to take | ||
| * @param reason Brief explanation of why we're taking an action, printed to management room. | ||
| * this wil be HTML escaped before printing, just in case it has user-provided data | ||
| */ | ||
| constructor(public readonly type: ConsequenceType, public readonly reason?: string) {} | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,111 @@ | ||
| import { strict as assert } from "assert"; | ||
|
|
||
| import config from "../../src/config"; | ||
| import { Mjolnir } from "../../src/Mjolnir"; | ||
| import { IProtection } from "../../src/protections/IProtection"; | ||
| import { newTestUser, noticeListener } from "./clientHelper"; | ||
| import { matrixClient, mjolnir } from "./mjolnirSetupUtils"; | ||
| import { ConsequenceType, Consequence } from "../../src/protections/consequence"; | ||
|
|
||
| describe("Test: standard consequences", function() { | ||
| let badUser; | ||
| this.beforeEach(async function () { | ||
| badUser = await newTestUser({ name: { contains: "standard-consequences" }}); | ||
| await badUser.start(); | ||
| }) | ||
| this.afterEach(async function () { | ||
| await badUser.stop(); | ||
| }) | ||
| it("Mjolnir applies a standard consequence redaction", async function() { | ||
| this.timeout(20000); | ||
|
|
||
| let protectedRoomId = await this.mjolnir.client.createRoom({ invite: [await badUser.getUserId()] }); | ||
| await badUser.joinRoom(this.mjolnir.managementRoomId); | ||
| await badUser.joinRoom(protectedRoomId); | ||
| await this.mjolnir.addProtectedRoom(protectedRoomId); | ||
|
|
||
| await this.mjolnir.registerProtection(new class implements IProtection { | ||
| name = "JY2TPN"; | ||
| description = "A test protection"; | ||
| settings = { }; | ||
| handleEvent = async (mjolnir: Mjolnir, roomId: string, event: any) => { | ||
| if (event.content.body === "ngmWkF") { | ||
| return new Consequence(ConsequenceType.redact, "asd"); | ||
| } | ||
| }; | ||
| }); | ||
| await this.mjolnir.enableProtection("JY2TPN"); | ||
|
|
||
| let reply = new Promise(async (resolve, reject) => { | ||
| const messageId = await badUser.sendMessage(protectedRoomId, {msgtype: "m.text", body: "ngmWkF"}); | ||
| let redaction; | ||
| badUser.on('room.event', (roomId, event) => { | ||
| if ( | ||
| roomId === protectedRoomId | ||
| && event?.type === "m.room.redaction" | ||
| && event.redacts === messageId | ||
| ) { | ||
| redaction = event | ||
| } | ||
| if ( | ||
| roomId === this.mjolnir.managementRoomId | ||
| && event?.type === "m.room.message" | ||
| && event?.content?.body?.startsWith("protection JY2TPN enacting redact against ") | ||
| && redaction !== undefined | ||
| ) { | ||
| resolve([redaction, event]) | ||
| } | ||
| }); | ||
| }); | ||
|
|
||
| const [eventRedact, eventMessage] = await reply | ||
| }); | ||
| it("Mjolnir applies a standard consequence ban", async function() { | ||
| this.timeout(20000); | ||
|
|
||
| let protectedRoomId = await this.mjolnir.client.createRoom({ invite: [await badUser.getUserId()] }); | ||
| await badUser.joinRoom(this.mjolnir.managementRoomId); | ||
| await badUser.joinRoom(protectedRoomId); | ||
| await this.mjolnir.addProtectedRoom(protectedRoomId); | ||
|
|
||
| await this.mjolnir.registerProtection(new class implements IProtection { | ||
| name = "0LxMTy"; | ||
| description = "A test protection"; | ||
| settings = { }; | ||
| handleEvent = async (mjolnir: Mjolnir, roomId: string, event: any) => { | ||
| if (event.content.body === "7Uga3d") { | ||
| return new Consequence(ConsequenceType.ban, "asd"); | ||
| } | ||
| }; | ||
| }); | ||
| await this.mjolnir.enableProtection("0LxMTy"); | ||
|
|
||
| let reply = new Promise(async (resolve, reject) => { | ||
| const messageId = await badUser.sendMessage(protectedRoomId, {msgtype: "m.text", body: "7Uga3d"}); | ||
| let ban; | ||
| badUser.on('room.leave', (roomId, event) => { | ||
| if ( | ||
| roomId === protectedRoomId | ||
| && event?.type === "m.room.member" | ||
| && event.content?.membership === "ban" | ||
| && event.state_key === badUser.userId | ||
| ) { | ||
| ban = event; | ||
| } | ||
| }); | ||
| badUser.on('room.event', (roomId, event) => { | ||
| if ( | ||
| roomId === this.mjolnir.managementRoomId | ||
| && event?.type === "m.room.message" | ||
| && event?.content?.body?.startsWith("protection 0LxMTy enacting ban against ") | ||
| && ban !== undefined | ||
| ) { | ||
| resolve([ban, event]) | ||
| } | ||
| }); | ||
| }); | ||
|
|
||
| const [eventBan, eventMessage] = await reply | ||
| }); | ||
| }); | ||
|
|