94 lines
3.5 KiB
TypeScript
94 lines
3.5 KiB
TypeScript
/*
|
|
Copyright 2021 - 2022 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 { Room } from "matrix-js-sdk/src/models/room";
|
|
|
|
import { NotificationColor } from "./NotificationColor";
|
|
import { arrayDiff } from "../../utils/arrays";
|
|
import { RoomNotificationState } from "./RoomNotificationState";
|
|
import { NotificationState, NotificationStateEvents } from "./NotificationState";
|
|
import { FetchRoomFn } from "./ListNotificationState";
|
|
import { DefaultTagID } from "../room-list/models";
|
|
import RoomListStore from "../room-list/RoomListStore";
|
|
|
|
export class SpaceNotificationState extends NotificationState {
|
|
public rooms: Room[] = []; // exposed only for tests
|
|
private states: { [spaceId: string]: RoomNotificationState } = {};
|
|
|
|
public constructor(private getRoomFn: FetchRoomFn) {
|
|
super();
|
|
}
|
|
|
|
public get symbol(): string | null {
|
|
return this._color === NotificationColor.Unsent ? "!" : null;
|
|
}
|
|
|
|
public setRooms(rooms: Room[]): void {
|
|
const oldRooms = this.rooms;
|
|
const diff = arrayDiff(oldRooms, rooms);
|
|
this.rooms = rooms;
|
|
for (const oldRoom of diff.removed) {
|
|
const state = this.states[oldRoom.roomId];
|
|
if (!state) continue; // We likely just didn't have a badge (race condition)
|
|
delete this.states[oldRoom.roomId];
|
|
state.off(NotificationStateEvents.Update, this.onRoomNotificationStateUpdate);
|
|
}
|
|
for (const newRoom of diff.added) {
|
|
const state = this.getRoomFn(newRoom);
|
|
state.on(NotificationStateEvents.Update, this.onRoomNotificationStateUpdate);
|
|
this.states[newRoom.roomId] = state;
|
|
}
|
|
|
|
this.calculateTotalState();
|
|
}
|
|
|
|
public getFirstRoomWithNotifications(): string | undefined {
|
|
return Object.values(this.states).find((state) => state.color >= this.color)?.room.roomId;
|
|
}
|
|
|
|
public destroy(): void {
|
|
super.destroy();
|
|
for (const state of Object.values(this.states)) {
|
|
state.off(NotificationStateEvents.Update, this.onRoomNotificationStateUpdate);
|
|
}
|
|
this.states = {};
|
|
}
|
|
|
|
private onRoomNotificationStateUpdate = (): void => {
|
|
this.calculateTotalState();
|
|
};
|
|
|
|
private calculateTotalState(): void {
|
|
const snapshot = this.snapshot();
|
|
|
|
this._count = 0;
|
|
this._color = NotificationColor.None;
|
|
for (const [roomId, state] of Object.entries(this.states)) {
|
|
const room = this.rooms.find((r) => r.roomId === roomId);
|
|
const roomTags = room ? RoomListStore.instance.getTagsForRoom(room) : [];
|
|
|
|
// We ignore unreads in LowPriority rooms, see https://github.com/vector-im/element-web/issues/16836
|
|
if (roomTags.includes(DefaultTagID.LowPriority) && state.color === NotificationColor.Bold) continue;
|
|
|
|
this._count += state.count;
|
|
this._color = Math.max(this.color, state.color);
|
|
}
|
|
|
|
// finally, publish an update if needed
|
|
this.emitIfUpdated(snapshot);
|
|
}
|
|
}
|