ab934a2a08
Signed-off-by: Kerry Archibald <kerrya@element.io>
187 lines
6.1 KiB
TypeScript
187 lines
6.1 KiB
TypeScript
/*
|
|
Copyright 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 {
|
|
Beacon,
|
|
BeaconEvent,
|
|
MatrixEvent,
|
|
Room,
|
|
} from "matrix-js-sdk/src/matrix";
|
|
import {
|
|
BeaconInfoState, makeBeaconInfoContent,
|
|
} from "matrix-js-sdk/src/content-helpers";
|
|
|
|
import defaultDispatcher from "../dispatcher/dispatcher";
|
|
import { ActionPayload } from "../dispatcher/payloads";
|
|
import { AsyncStoreWithClient } from "./AsyncStoreWithClient";
|
|
|
|
const isOwnBeacon = (beacon: Beacon, userId: string): boolean => beacon.beaconInfoOwner === userId;
|
|
|
|
export enum OwnBeaconStoreEvent {
|
|
LivenessChange = 'OwnBeaconStore.LivenessChange'
|
|
}
|
|
|
|
type OwnBeaconStoreState = {
|
|
beacons: Map<string, Beacon>;
|
|
beaconsByRoomId: Map<Room['roomId'], Set<string>>;
|
|
liveBeaconIds: string[];
|
|
};
|
|
export class OwnBeaconStore extends AsyncStoreWithClient<OwnBeaconStoreState> {
|
|
private static internalInstance = new OwnBeaconStore();
|
|
public readonly beacons = new Map<string, Beacon>();
|
|
public readonly beaconsByRoomId = new Map<Room['roomId'], Set<string>>();
|
|
private liveBeaconIds = [];
|
|
|
|
public constructor() {
|
|
super(defaultDispatcher);
|
|
}
|
|
|
|
public static get instance(): OwnBeaconStore {
|
|
return OwnBeaconStore.internalInstance;
|
|
}
|
|
|
|
protected async onNotReady() {
|
|
this.matrixClient.removeListener(BeaconEvent.LivenessChange, this.onBeaconLiveness);
|
|
this.matrixClient.removeListener(BeaconEvent.New, this.onNewBeacon);
|
|
|
|
this.beacons.forEach(beacon => beacon.destroy());
|
|
|
|
this.beacons.clear();
|
|
this.beaconsByRoomId.clear();
|
|
this.liveBeaconIds = [];
|
|
}
|
|
|
|
protected async onReady(): Promise<void> {
|
|
this.matrixClient.on(BeaconEvent.LivenessChange, this.onBeaconLiveness);
|
|
this.matrixClient.on(BeaconEvent.New, this.onNewBeacon);
|
|
|
|
this.initialiseBeaconState();
|
|
}
|
|
|
|
protected async onAction(payload: ActionPayload): Promise<void> {
|
|
// we don't actually do anything here
|
|
}
|
|
|
|
public hasLiveBeacons(roomId?: string): boolean {
|
|
return !!this.getLiveBeaconIds(roomId).length;
|
|
}
|
|
|
|
public getLiveBeaconIds(roomId?: string): string[] {
|
|
if (!roomId) {
|
|
return this.liveBeaconIds;
|
|
}
|
|
return this.liveBeaconIds.filter(beaconId => this.beaconsByRoomId.get(roomId)?.has(beaconId));
|
|
}
|
|
|
|
public stopBeacon = async (beaconInfoId: string): Promise<void> => {
|
|
const beacon = this.beacons.get(beaconInfoId);
|
|
// if no beacon, or beacon is already explicitly set isLive: false
|
|
// do nothing
|
|
if (!beacon?.beaconInfo?.live) {
|
|
return;
|
|
}
|
|
|
|
return await this.updateBeaconEvent(beacon, { live: false });
|
|
};
|
|
|
|
private onNewBeacon = (_event: MatrixEvent, beacon: Beacon): void => {
|
|
if (!isOwnBeacon(beacon, this.matrixClient.getUserId())) {
|
|
return;
|
|
}
|
|
this.addBeacon(beacon);
|
|
this.checkLiveness();
|
|
};
|
|
|
|
private onBeaconLiveness = (isLive: boolean, beacon: Beacon): void => {
|
|
// check if we care about this beacon
|
|
if (!this.beacons.has(beacon.beaconInfoId)) {
|
|
return;
|
|
}
|
|
|
|
if (!isLive && this.liveBeaconIds.includes(beacon.beaconInfoId)) {
|
|
this.liveBeaconIds =
|
|
this.liveBeaconIds.filter(beaconId => beaconId !== beacon.beaconInfoId);
|
|
}
|
|
|
|
if (isLive && !this.liveBeaconIds.includes(beacon.beaconInfoId)) {
|
|
this.liveBeaconIds.push(beacon.beaconInfoId);
|
|
}
|
|
|
|
// beacon expired, update beacon to un-alive state
|
|
if (!isLive) {
|
|
this.stopBeacon(beacon.beaconInfoId);
|
|
}
|
|
|
|
// TODO start location polling here
|
|
|
|
this.emit(OwnBeaconStoreEvent.LivenessChange, this.hasLiveBeacons());
|
|
};
|
|
|
|
private initialiseBeaconState = () => {
|
|
const userId = this.matrixClient.getUserId();
|
|
const visibleRooms = this.matrixClient.getVisibleRooms();
|
|
|
|
visibleRooms
|
|
.forEach(room => {
|
|
const roomState = room.currentState;
|
|
const beacons = roomState.beacons;
|
|
const ownBeaconsArray = [...beacons.values()].filter(beacon => isOwnBeacon(beacon, userId));
|
|
ownBeaconsArray.forEach(beacon => this.addBeacon(beacon));
|
|
});
|
|
|
|
this.checkLiveness();
|
|
};
|
|
|
|
private addBeacon = (beacon: Beacon): void => {
|
|
this.beacons.set(beacon.beaconInfoId, beacon);
|
|
|
|
if (!this.beaconsByRoomId.has(beacon.roomId)) {
|
|
this.beaconsByRoomId.set(beacon.roomId, new Set<string>());
|
|
}
|
|
|
|
this.beaconsByRoomId.get(beacon.roomId).add(beacon.beaconInfoId);
|
|
|
|
beacon.monitorLiveness();
|
|
};
|
|
|
|
private checkLiveness = (): void => {
|
|
const prevLiveness = this.hasLiveBeacons();
|
|
this.liveBeaconIds = [...this.beacons.values()]
|
|
.filter(beacon => beacon.isLive)
|
|
.map(beacon => beacon.beaconInfoId);
|
|
|
|
const newLiveness = this.hasLiveBeacons();
|
|
|
|
if (prevLiveness !== newLiveness) {
|
|
this.emit(OwnBeaconStoreEvent.LivenessChange, newLiveness);
|
|
}
|
|
};
|
|
|
|
private updateBeaconEvent = async (beacon: Beacon, update: Partial<BeaconInfoState>): Promise<void> => {
|
|
const { description, timeout, timestamp, live, assetType } = {
|
|
...beacon.beaconInfo,
|
|
...update,
|
|
};
|
|
|
|
const updateContent = makeBeaconInfoContent(timeout,
|
|
live,
|
|
description,
|
|
assetType,
|
|
timestamp);
|
|
|
|
await this.matrixClient.unstable_setLiveBeacon(beacon.roomId, beacon.beaconInfoEventType, updateContent);
|
|
};
|
|
}
|