mattermost-mobile/app/queries/servers/user.ts

// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.

import {Database, Q} from '@nozbe/watermelondb';
import {combineLatest, of as of$} from 'rxjs';
import {distinctUntilChanged, switchMap} from 'rxjs/operators';

import {General} from '@constants';
import {MM_TABLES} from '@constants/database';
import {getTeammateNameDisplaySetting} from '@helpers/api/preference';
import {sanitizeLikeString} from '@helpers/database';

import {queryDisplayNamePreferences} from './preference';
import {observeCurrentUserId, observeLicense, getCurrentUserId, getConfig, getLicense, observeConfigValue} from './system';

import type ServerDataOperator from '@database/operator/server_data_operator';
import type ChannelMembershipModel from '@typings/database/models/servers/channel_membership';
import type TeamMembershipModel from '@typings/database/models/servers/team_membership';
import type UserModel from '@typings/database/models/servers/user';

const {SERVER: {CHANNEL_MEMBERSHIP, USER, TEAM_MEMBERSHIP, CHANNEL, MY_CHANNEL}} = MM_TABLES;
export const getUserById = async (database: Database, userId: string) => {
    try {
        const userRecord = (await database.get<UserModel>(USER).find(userId));
        return userRecord;
    } catch {
        return undefined;
    }
};

export const observeUser = (database: Database, userId: string) => {
    return database.get<UserModel>(USER).query(Q.where('id', userId), Q.take(1)).observe().pipe(
        switchMap((result) => (result.length ? result[0].observe() : of$(undefined))),
    );
};

export const getCurrentUser = async (database: Database) => {
    const currentUserId = await getCurrentUserId(database);
    if (currentUserId) {
        return getUserById(database, currentUserId);
    }

    return undefined;
};

export const observeCurrentUser = (database: Database) => {
    return observeCurrentUserId(database).pipe(
        switchMap((id) => observeUser(database, id)),
    );
};

export const observeCurrentUserRoles = (database: Database) => {
    return observeCurrentUser(database).pipe(
        switchMap((v) => of$(v?.roles)),
        distinctUntilChanged(),
    );
};

export const queryAllUsers = (database: Database) => {
    return database.get<UserModel>(USER).query();
};

export const queryUsersById = (database: Database, userIds: string[]) => {
    return database.get<UserModel>(USER).query(Q.where('id', Q.oneOf(userIds)));
};

export const queryUsersByUsername = (database: Database, usernames: string[]) => {
    return database.get<UserModel>(USER).query(Q.where('username', Q.oneOf(usernames)));
};

export async function prepareUsers(operator: ServerDataOperator, users: UserProfile[]): Promise<UserModel[]> {
    return operator.handleUsers({users, prepareRecordsOnly: true});
}

export const observeTeammateNameDisplay = (database: Database) => {
    const lockTeammateNameDisplay = observeConfigValue(database, 'LockTeammateNameDisplay');
    const teammateNameDisplay = observeConfigValue(database, 'TeammateNameDisplay');
    const license = observeLicense(database);
    const preferences = queryDisplayNamePreferences(database).
        observeWithColumns(['value']);
    return combineLatest([lockTeammateNameDisplay, teammateNameDisplay, license, preferences]).pipe(
        switchMap(
            ([ltnd, tnd, lcs, prefs]) => of$(getTeammateNameDisplaySetting(prefs, ltnd, tnd, lcs)),
        ),
    );
};

export async function getTeammateNameDisplay(database: Database) {
    const config = await getConfig(database);
    const license = await getLicense(database);
    const preferences = await queryDisplayNamePreferences(database).fetch();
    return getTeammateNameDisplaySetting(preferences, config?.LockTeammateNameDisplay, config?.TeammateNameDisplay, license);
}

export const queryUsersLike = (database: Database, likeUsername: string) => {
    return database.get<UserModel>(USER).query(
        Q.where('username', Q.like(
            `%${sanitizeLikeString(likeUsername)}%`,
        )),
    );
};

export const queryUsersByIdsOrUsernames = (database: Database, ids: string[], usernames: string[]) => {
    return database.get<UserModel>(USER).query(
        Q.or(
            Q.where('id', Q.oneOf(ids)),
            Q.where('username', Q.oneOf(usernames)),
        ));
};

export const observeUserIsTeamAdmin = (database: Database, userId: string, teamId: string) => {
    const id = `${teamId}-${userId}`;
    return database.get<TeamMembershipModel>(TEAM_MEMBERSHIP).query(
        Q.where('id', Q.eq(id)),
    ).observeWithColumns(['scheme_admin']).pipe(
        switchMap((tm) => of$(tm.length ? tm[0].schemeAdmin : false)),
    );
};

export const observeUserIsChannelAdmin = (database: Database, userId: string, channelId: string) => {
    const id = `${channelId}-${userId}`;
    return database.get<ChannelMembershipModel>(CHANNEL_MEMBERSHIP).query(
        Q.where('id', Q.eq(id)),
    ).observeWithColumns(['scheme_admin']).pipe(
        switchMap((cm) => of$(cm.length ? cm[0].schemeAdmin : false)),
        distinctUntilChanged(),
    );
};

export const observeDeactivatedUsers = (database: Database) => {
    return database.get<UserModel>(USER).query(
        Q.where('delete_at', Q.gt(0)),
    ).observe().pipe(
        switchMap((users) => {
            return of$(new Map(users.map((u) => [u.id, u])));
        }),
    );
};

export const getUsersFromDMSorted = async (database: Database, memberIds: string[]) => {
    try {
        return database.get<UserModel>(USER).query(
            Q.unsafeSqlQuery(`SELECT DISTINCT u.* FROM User u
            INNER JOIN ${CHANNEL_MEMBERSHIP} cm ON cm.user_id=u.id
            INNER JOIN ${CHANNEL} c ON c.id=cm.id AND c.type='${General.DM_CHANNEL}'
            INNER JOIN ${MY_CHANNEL} my
            WHERE cm.user_id IN (${`'${memberIds.join("','")}'`})
            ORDER BY my.last_viewed_at DESC`)).fetch();
    } catch (error) {
        return [];
    }
};

export const observeIsUserLanguageSupportedByAutotranslation = (database: Database) => {
    const currentUser = observeCurrentUser(database);
    const autoTranslationLanguages = observeConfigValue(database, 'AutoTranslationLanguages');

    return combineLatest([currentUser, autoTranslationLanguages]).pipe(
        switchMap(([user, languages]) => {
            if (!user?.locale || !languages) {
                return of$(false);
            }
            const userLocale = user.locale;
            const languagesList = languages.split(',');
            return of$(languagesList.includes(userLocale));
        }),
        distinctUntilChanged(),
    );
};