forked from Ivasoft/mattermost-mobile
109 lines
4.2 KiB
TypeScript
109 lines
4.2 KiB
TypeScript
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
import {Q} from '@nozbe/watermelondb';
|
|
import {map as map$, Subscription} from 'rxjs';
|
|
import {combineLatestWith} from 'rxjs/operators';
|
|
|
|
import {MM_TABLES} from '@constants/database';
|
|
import DatabaseManager from '@database/manager';
|
|
import {queryMyTeams} from '@queries/servers/team';
|
|
import {getIsCRTEnabled, observeThreadMentionCount, queryThreads} from '@queries/servers/thread';
|
|
import MyChannelModel from '@typings/database/models/servers/my_channel';
|
|
|
|
const {SERVER: {CHANNEL, MY_CHANNEL}} = MM_TABLES;
|
|
|
|
type ObserverArgs = {
|
|
myChannels: MyChannelModel[];
|
|
threadMentionCount: number;
|
|
}
|
|
|
|
export const subscribeServerUnreadAndMentions = (serverUrl: string, observer: ({myChannels, threadMentionCount}: ObserverArgs) => void) => {
|
|
const server = DatabaseManager.serverDatabases[serverUrl];
|
|
let subscription: Subscription|undefined;
|
|
|
|
if (server?.database) {
|
|
subscription = server.database.get<MyChannelModel>(MY_CHANNEL).
|
|
query(Q.on(CHANNEL, Q.where('delete_at', Q.eq(0)))).
|
|
observeWithColumns(['is_unread', 'mentions_count']).
|
|
pipe(
|
|
combineLatestWith(observeThreadMentionCount(server.database, undefined, false)),
|
|
map$(([myChannels, threadMentionCount]) => ({myChannels, threadMentionCount})),
|
|
).
|
|
subscribe(observer);
|
|
}
|
|
|
|
return subscription;
|
|
};
|
|
|
|
export const subscribeMentionsByServer = (serverUrl: string, observer: (serverUrl: string, {myChannels, threadMentionCount}: ObserverArgs) => void) => {
|
|
const server = DatabaseManager.serverDatabases[serverUrl];
|
|
let subscription: Subscription|undefined;
|
|
|
|
if (server?.database) {
|
|
subscription = server.database.
|
|
get(MY_CHANNEL).
|
|
query(Q.on(CHANNEL, Q.where('delete_at', Q.eq(0)))).
|
|
observeWithColumns(['mentions_count']).
|
|
pipe(
|
|
combineLatestWith(observeThreadMentionCount(server.database, undefined, false)),
|
|
map$(([myChannels, threadMentionCount]) => ({myChannels, threadMentionCount})),
|
|
).
|
|
subscribe(observer.bind(undefined, serverUrl));
|
|
}
|
|
|
|
return subscription;
|
|
};
|
|
|
|
export const subscribeUnreadAndMentionsByServer = (serverUrl: string, observer: (serverUrl: string, {myChannels, threadMentionCount}: ObserverArgs) => void) => {
|
|
const server = DatabaseManager.serverDatabases[serverUrl];
|
|
let subscription: Subscription|undefined;
|
|
|
|
if (server?.database) {
|
|
subscription = server.database.get<MyChannelModel>(MY_CHANNEL).
|
|
query(Q.on(CHANNEL, Q.where('delete_at', Q.eq(0)))).
|
|
observeWithColumns(['mentions_count', 'is_unread']).
|
|
pipe(
|
|
combineLatestWith(observeThreadMentionCount(server.database, undefined, false)),
|
|
map$(([myChannels, threadMentionCount]) => ({myChannels, threadMentionCount})),
|
|
).
|
|
subscribe(observer.bind(undefined, serverUrl));
|
|
}
|
|
|
|
return subscription;
|
|
};
|
|
|
|
export const getTotalMentionsForServer = async (serverUrl: string) => {
|
|
const server = DatabaseManager.serverDatabases[serverUrl];
|
|
let count = 0;
|
|
if (server?.database) {
|
|
const {database} = server;
|
|
const myChannels = await database.get<MyChannelModel>(MY_CHANNEL).
|
|
query(
|
|
Q.on(CHANNEL, Q.where('delete_at', Q.eq(0))),
|
|
Q.where('mentions_count', Q.gt(0)),
|
|
).fetch();
|
|
|
|
for (const mc of myChannels) {
|
|
count += mc.mentionsCount;
|
|
}
|
|
|
|
const isCRTEnabled = await getIsCRTEnabled(database);
|
|
if (isCRTEnabled) {
|
|
let includeDmGm = true;
|
|
const myTeamIds = await queryMyTeams(database).fetchIds();
|
|
for await (const teamId of myTeamIds) {
|
|
const threads = await queryThreads(database, teamId, false, includeDmGm).extend(
|
|
Q.where('unread_mentions', Q.gt(0)),
|
|
).fetch();
|
|
includeDmGm = false;
|
|
for (const t of threads) {
|
|
count += t.unreadMentions;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return count;
|
|
};
|