2016-08-18 01:26:37 +08:00
|
|
|
/*
|
|
|
|
Copyright 2016 OpenMarket Ltd
|
2019-06-19 18:48:47 +08:00
|
|
|
Copyright 2019 The Matrix.org Foundation C.I.C.
|
2016-08-18 01:26:37 +08:00
|
|
|
|
|
|
|
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-06-29 20:11:58 +08:00
|
|
|
import { PushProcessor } from "matrix-js-sdk/src/pushprocessor";
|
2021-09-27 20:32:04 +08:00
|
|
|
import { NotificationCountType, Room } from "matrix-js-sdk/src/models/room";
|
2022-05-14 03:13:21 +08:00
|
|
|
import {
|
|
|
|
ConditionKind,
|
|
|
|
IPushRule,
|
|
|
|
PushRuleActionName,
|
|
|
|
PushRuleKind,
|
|
|
|
TweakName,
|
|
|
|
} from "matrix-js-sdk/src/@types/PushRules";
|
2022-02-16 05:05:41 +08:00
|
|
|
import { EventType } from "matrix-js-sdk/src/@types/event";
|
2022-11-29 18:55:15 +08:00
|
|
|
import { MatrixClient } from "matrix-js-sdk/src/matrix";
|
2021-09-27 20:32:04 +08:00
|
|
|
|
2021-10-23 06:23:32 +08:00
|
|
|
import { MatrixClientPeg } from "./MatrixClientPeg";
|
|
|
|
|
2021-09-27 20:32:04 +08:00
|
|
|
export enum RoomNotifState {
|
|
|
|
AllMessagesLoud = "all_messages_loud",
|
|
|
|
AllMessages = "all_messages",
|
|
|
|
MentionsOnly = "mentions_only",
|
|
|
|
Mute = "mute",
|
|
|
|
}
|
2016-08-18 01:26:37 +08:00
|
|
|
|
2022-11-29 18:55:15 +08:00
|
|
|
export function getRoomNotifsState(client: MatrixClient, roomId: string): RoomNotifState {
|
|
|
|
if (client.isGuest()) return RoomNotifState.AllMessages;
|
2016-08-18 21:00:14 +08:00
|
|
|
|
2016-08-18 01:26:37 +08:00
|
|
|
// look through the override rules for a rule affecting this room:
|
|
|
|
// if one exists, it will take precedence.
|
|
|
|
const muteRule = findOverrideMuteRule(roomId);
|
2016-08-18 21:00:14 +08:00
|
|
|
if (muteRule) {
|
2021-09-27 20:32:04 +08:00
|
|
|
return RoomNotifState.Mute;
|
2016-08-18 01:26:37 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// for everything else, look at the room rule.
|
2018-02-07 01:50:53 +08:00
|
|
|
let roomRule = null;
|
|
|
|
try {
|
2022-11-29 18:55:15 +08:00
|
|
|
roomRule = client.getRoomPushRule("global", roomId);
|
2018-02-07 01:50:53 +08:00
|
|
|
} catch (err) {
|
|
|
|
// Possible that the client doesn't have pushRules yet. If so, it
|
2022-03-04 17:39:16 +08:00
|
|
|
// hasn't started either, so indicate that this room is not notifying.
|
2018-02-07 01:50:53 +08:00
|
|
|
return null;
|
|
|
|
}
|
2016-08-18 01:26:37 +08:00
|
|
|
|
|
|
|
// XXX: We have to assume the default is to notify for all messages
|
|
|
|
// (in particular this will be 'wrong' for one to one rooms because
|
|
|
|
// they will notify loudly for all messages)
|
2022-03-04 17:39:16 +08:00
|
|
|
if (!roomRule?.enabled) return RoomNotifState.AllMessages;
|
2016-08-18 01:26:37 +08:00
|
|
|
|
|
|
|
// a mute at the room level will still allow mentions
|
|
|
|
// to notify
|
2021-09-27 20:32:04 +08:00
|
|
|
if (isMuteRule(roomRule)) return RoomNotifState.MentionsOnly;
|
2016-08-18 01:26:37 +08:00
|
|
|
|
|
|
|
const actionsObject = PushProcessor.actionListToActionsObject(roomRule.actions);
|
2021-09-27 20:32:04 +08:00
|
|
|
if (actionsObject.tweaks.sound) return RoomNotifState.AllMessagesLoud;
|
2016-08-18 01:26:37 +08:00
|
|
|
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
2021-09-27 20:32:04 +08:00
|
|
|
export function setRoomNotifsState(roomId: string, newState: RoomNotifState): Promise<void> {
|
|
|
|
if (newState === RoomNotifState.Mute) {
|
2016-08-18 21:00:14 +08:00
|
|
|
return setRoomNotifsStateMuted(roomId);
|
|
|
|
} else {
|
|
|
|
return setRoomNotifsStateUnmuted(roomId, newState);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-10-24 14:50:21 +08:00
|
|
|
export function getUnreadNotificationCount(room: Room, type: NotificationCountType, threadId?: string): number {
|
|
|
|
let notificationCount = !!threadId
|
|
|
|
? room.getThreadUnreadNotificationCount(threadId, type)
|
|
|
|
: room.getUnreadNotificationCount(type);
|
2019-04-02 06:06:33 +08:00
|
|
|
|
|
|
|
// Check notification counts in the old room just in case there's some lost
|
|
|
|
// there. We only go one level down to avoid performance issues, and theory
|
|
|
|
// is that 1st generation rooms will have already been read by the 3rd generation.
|
2022-02-16 05:05:41 +08:00
|
|
|
const createEvent = room.currentState.getStateEvents(EventType.RoomCreate, "");
|
2022-10-24 14:50:21 +08:00
|
|
|
const predecessor = createEvent?.getContent().predecessor;
|
|
|
|
// Exclude threadId, as the same thread can't continue over a room upgrade
|
|
|
|
if (!threadId && predecessor) {
|
|
|
|
const oldRoomId = predecessor.room_id;
|
2019-04-02 06:06:33 +08:00
|
|
|
const oldRoom = MatrixClientPeg.get().getRoom(oldRoomId);
|
|
|
|
if (oldRoom) {
|
|
|
|
// We only ever care if there's highlights in the old room. No point in
|
|
|
|
// notifying the user for unread messages because they would have extreme
|
|
|
|
// difficulty changing their notification preferences away from "All Messages"
|
|
|
|
// and "Noisy".
|
2021-09-27 20:32:04 +08:00
|
|
|
notificationCount += oldRoom.getUnreadNotificationCount(NotificationCountType.Highlight);
|
2019-04-02 06:06:33 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return notificationCount;
|
|
|
|
}
|
|
|
|
|
2021-09-27 20:32:04 +08:00
|
|
|
function setRoomNotifsStateMuted(roomId: string): Promise<any> {
|
2016-08-18 01:26:37 +08:00
|
|
|
const cli = MatrixClientPeg.get();
|
|
|
|
const promises = [];
|
|
|
|
|
2016-08-18 21:00:14 +08:00
|
|
|
// delete the room rule
|
|
|
|
const roomRule = cli.getRoomPushRule("global", roomId);
|
|
|
|
if (roomRule) {
|
2021-09-27 20:32:04 +08:00
|
|
|
promises.push(cli.deletePushRule("global", PushRuleKind.RoomSpecific, roomRule.rule_id));
|
2016-08-18 21:00:14 +08:00
|
|
|
}
|
2016-08-18 01:26:37 +08:00
|
|
|
|
2016-08-18 22:21:46 +08:00
|
|
|
// add/replace an override rule to squelch everything in this room
|
|
|
|
// NB. We use the room ID as the name of this rule too, although this
|
|
|
|
// is an override rule, not a room rule: it still pertains to this room
|
|
|
|
// though, so using the room ID as the rule ID is logical and prevents
|
|
|
|
// duplicate copies of the rule.
|
2021-09-27 20:32:04 +08:00
|
|
|
promises.push(
|
|
|
|
cli.addPushRule("global", PushRuleKind.Override, roomId, {
|
2016-08-18 21:00:14 +08:00
|
|
|
conditions: [
|
|
|
|
{
|
2022-05-14 03:13:21 +08:00
|
|
|
kind: ConditionKind.EventMatch,
|
2016-08-18 21:00:14 +08:00
|
|
|
key: "room_id",
|
|
|
|
pattern: roomId,
|
2017-07-01 21:28:12 +08:00
|
|
|
},
|
2016-08-18 21:00:14 +08:00
|
|
|
],
|
2022-05-14 03:13:21 +08:00
|
|
|
actions: [PushRuleActionName.DontNotify],
|
2016-08-18 21:00:14 +08:00
|
|
|
}),
|
|
|
|
);
|
|
|
|
|
2017-07-12 21:04:20 +08:00
|
|
|
return Promise.all(promises);
|
2016-08-18 21:00:14 +08:00
|
|
|
}
|
|
|
|
|
2021-09-27 20:32:04 +08:00
|
|
|
function setRoomNotifsStateUnmuted(roomId: string, newState: RoomNotifState): Promise<any> {
|
2016-08-18 21:00:14 +08:00
|
|
|
const cli = MatrixClientPeg.get();
|
|
|
|
const promises = [];
|
|
|
|
|
|
|
|
const overrideMuteRule = findOverrideMuteRule(roomId);
|
|
|
|
if (overrideMuteRule) {
|
2021-09-27 20:32:04 +08:00
|
|
|
promises.push(cli.deletePushRule("global", PushRuleKind.Override, overrideMuteRule.rule_id));
|
2016-08-18 21:00:14 +08:00
|
|
|
}
|
|
|
|
|
2021-09-27 20:32:04 +08:00
|
|
|
if (newState === RoomNotifState.AllMessages) {
|
2016-08-18 23:59:25 +08:00
|
|
|
const roomRule = cli.getRoomPushRule("global", roomId);
|
|
|
|
if (roomRule) {
|
2021-09-27 20:32:04 +08:00
|
|
|
promises.push(cli.deletePushRule("global", PushRuleKind.RoomSpecific, roomRule.rule_id));
|
2016-08-18 23:59:25 +08:00
|
|
|
}
|
2021-09-27 20:32:04 +08:00
|
|
|
} else if (newState === RoomNotifState.MentionsOnly) {
|
|
|
|
promises.push(
|
|
|
|
cli.addPushRule("global", PushRuleKind.RoomSpecific, roomId, {
|
2022-05-14 03:13:21 +08:00
|
|
|
actions: [PushRuleActionName.DontNotify],
|
2016-08-18 01:26:37 +08:00
|
|
|
}),
|
|
|
|
);
|
2016-08-18 21:00:14 +08:00
|
|
|
// https://matrix.org/jira/browse/SPEC-400
|
2021-09-27 20:32:04 +08:00
|
|
|
promises.push(cli.setPushRuleEnabled("global", PushRuleKind.RoomSpecific, roomId, true));
|
|
|
|
} else if (newState === RoomNotifState.AllMessagesLoud) {
|
|
|
|
promises.push(
|
|
|
|
cli.addPushRule("global", PushRuleKind.RoomSpecific, roomId, {
|
2016-08-18 21:00:14 +08:00
|
|
|
actions: [
|
2022-05-14 03:13:21 +08:00
|
|
|
PushRuleActionName.Notify,
|
2016-08-18 21:00:14 +08:00
|
|
|
{
|
2022-05-14 03:13:21 +08:00
|
|
|
set_tweak: TweakName.Sound,
|
2016-08-18 21:00:14 +08:00
|
|
|
value: "default",
|
2017-07-01 21:28:12 +08:00
|
|
|
},
|
|
|
|
],
|
2016-08-18 21:00:14 +08:00
|
|
|
}),
|
|
|
|
);
|
|
|
|
// https://matrix.org/jira/browse/SPEC-400
|
2021-09-27 20:32:04 +08:00
|
|
|
promises.push(cli.setPushRuleEnabled("global", PushRuleKind.RoomSpecific, roomId, true));
|
2016-08-18 01:26:37 +08:00
|
|
|
}
|
|
|
|
|
2017-07-12 21:04:20 +08:00
|
|
|
return Promise.all(promises);
|
2016-08-18 01:26:37 +08:00
|
|
|
}
|
|
|
|
|
2021-11-20 01:35:11 +08:00
|
|
|
function findOverrideMuteRule(roomId: string): IPushRule {
|
2020-12-17 01:16:15 +08:00
|
|
|
const cli = MatrixClientPeg.get();
|
2021-11-20 01:35:11 +08:00
|
|
|
if (!cli?.pushRules?.global?.override) {
|
2018-02-07 01:50:53 +08:00
|
|
|
return null;
|
|
|
|
}
|
2021-11-20 01:35:11 +08:00
|
|
|
for (const rule of cli.pushRules.global.override) {
|
2022-03-04 17:39:16 +08:00
|
|
|
if (rule.enabled && isRuleForRoom(roomId, rule) && isMuteRule(rule)) {
|
|
|
|
return rule;
|
2016-08-18 01:26:37 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
2021-11-20 01:35:11 +08:00
|
|
|
function isRuleForRoom(roomId: string, rule: IPushRule): boolean {
|
2022-03-04 17:39:16 +08:00
|
|
|
if (rule.conditions?.length !== 1) {
|
2016-08-18 01:26:37 +08:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
const cond = rule.conditions[0];
|
2021-11-20 01:35:11 +08:00
|
|
|
return cond.kind === ConditionKind.EventMatch && cond.key === "room_id" && cond.pattern === roomId;
|
2016-08-18 01:26:37 +08:00
|
|
|
}
|
|
|
|
|
2021-11-20 01:35:11 +08:00
|
|
|
function isMuteRule(rule: IPushRule): boolean {
|
|
|
|
return rule.actions.length === 1 && rule.actions[0] === PushRuleActionName.DontNotify;
|
2016-08-18 01:26:37 +08:00
|
|
|
}
|