import * as FileSystem from 'expo-file-system'; import { Rocketchat as RocketchatClient } from '@rocket.chat/sdk'; import Model from '@nozbe/watermelondb/Model'; import { getDeviceToken } from '../../notifications/push'; import { extractHostname } from '../../utils/server'; import { BASIC_AUTH_KEY } from '../../utils/fetch'; import database, { getDatabase } from '../database'; import RocketChat from '../rocketchat'; import { useSsl } from '../../utils/url'; import log from '../../utils/log'; import { E2E_PRIVATE_KEY, E2E_PUBLIC_KEY, E2E_RANDOM_PASSWORD_KEY } from '../encryption/constants'; import UserPreferences from '../userPreferences'; import { ICertificate, IRocketChat } from '../../definitions'; import sdk from '../rocketchat/services/sdk'; function removeServerKeys({ server, userId }: { server: string; userId?: string | null }) { UserPreferences.removeItem(`${RocketChat.TOKEN_KEY}-${server}`); if (userId) { UserPreferences.removeItem(`${RocketChat.TOKEN_KEY}-${userId}`); } UserPreferences.removeItem(`${BASIC_AUTH_KEY}-${server}`); UserPreferences.removeItem(`${server}-${E2E_PUBLIC_KEY}`); UserPreferences.removeItem(`${server}-${E2E_PRIVATE_KEY}`); UserPreferences.removeItem(`${server}-${E2E_RANDOM_PASSWORD_KEY}`); } async function removeSharedCredentials({ server }: { server: string }) { // clear certificate for server - SSL Pinning try { const certificate = UserPreferences.getMap(extractHostname(server)) as ICertificate | null; if (certificate?.path) { UserPreferences.removeItem(extractHostname(server)); await FileSystem.deleteAsync(certificate.path); } } catch (e) { log(e); } } async function removeServerData({ server }: { server: string }) { try { const batch: Model[] = []; const serversDB = database.servers; const userId = UserPreferences.getString(`${RocketChat.TOKEN_KEY}-${server}`); const usersCollection = serversDB.get('users'); if (userId) { const userRecord = await usersCollection.find(userId); batch.push(userRecord.prepareDestroyPermanently()); } const serverCollection = serversDB.get('servers'); const serverRecord = await serverCollection.find(server); batch.push(serverRecord.prepareDestroyPermanently()); await serversDB.write(() => serversDB.batch(...batch)); await removeSharedCredentials({ server }); removeServerKeys({ server, userId }); } catch (e) { log(e); } } function removeCurrentServer() { UserPreferences.removeItem(RocketChat.CURRENT_SERVER); } async function removeServerDatabase({ server }: { server: string }) { try { const db = getDatabase(server); await db.write(() => db.unsafeResetDatabase()); } catch (e) { log(e); } } export async function removeServer({ server }: { server: string }): Promise { try { const userId = UserPreferences.getString(`${RocketChat.TOKEN_KEY}-${server}`); if (userId) { const resume = UserPreferences.getString(`${RocketChat.TOKEN_KEY}-${userId}`); const sdk = new RocketchatClient({ host: server, protocol: 'ddp', useSsl: useSsl(server) }); await sdk.login({ resume }); const token = getDeviceToken(); if (token) { await sdk.del('push.token', { token }); } await sdk.logout(); } await removeServerData({ server }); await removeServerDatabase({ server }); } catch (e) { log(e); } } export default async function logout(this: IRocketChat, { server }: { server: string }): Promise { if (this.roomsSub) { this.roomsSub.stop(); this.roomsSub = null; } if (this.activeUsersSubTimeout) { clearTimeout(this.activeUsersSubTimeout); this.activeUsersSubTimeout = false; } try { await this.removePushToken(); } catch (e) { log(e); } try { // RC 0.60.0 await sdk.current.logout(); } catch (e) { log(e); } if (sdk.current) { sdk.disconnect(); } await removeServerData({ server }); await removeCurrentServer(); await removeServerDatabase({ server }); }