Chore: Dehydrate login methods from rocketchat.js (#3759)

* dehydrate login methods from rocketchat.js
This commit is contained in:
Gerzon Z 2022-03-08 14:54:26 -04:00 committed by GitHub
parent 15f573a703
commit 80918f56fa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
9 changed files with 597 additions and 471 deletions

View File

@ -234,7 +234,7 @@ class LoginServices extends React.PureComponent<ILoginServicesProps, any> {
AppleAuthentication.AppleAuthenticationScope.EMAIL AppleAuthentication.AppleAuthenticationScope.EMAIL
] ]
}); });
// @ts-ignore
await RocketChat.loginOAuthOrSso({ fullName, email, identityToken }); await RocketChat.loginOAuthOrSso({ fullName, email, identityToken });
} catch { } catch {
logEvent(events.ENTER_WITH_APPLE_F); logEvent(events.ENTER_WITH_APPLE_F);

View File

@ -49,6 +49,7 @@ import { sanitizeLikeString } from '../../lib/database/utils';
import { CustomIcon } from '../../lib/Icons'; import { CustomIcon } from '../../lib/Icons';
import { IMessage } from '../../definitions/IMessage'; import { IMessage } from '../../definitions/IMessage';
import { forceJpgExtension } from './forceJpgExtension'; import { forceJpgExtension } from './forceJpgExtension';
import { IUser } from '../../definitions';
if (isAndroid) { if (isAndroid) {
require('./EmojiKeyboard'); require('./EmojiKeyboard');
@ -80,12 +81,7 @@ export interface IMessageBoxProps {
editing: boolean; editing: boolean;
threadsEnabled: boolean; threadsEnabled: boolean;
isFocused(): boolean; isFocused(): boolean;
user: { user: IUser;
id: string;
_id: string;
username: string;
token: string;
};
roomType: string; roomType: string;
tmid: string; tmid: string;
replyWithMention: boolean; replyWithMention: boolean;

View File

@ -0,0 +1,15 @@
export interface ICredentials {
user?: string;
password?: string;
username?: string;
ldapPass?: string;
ldap?: boolean;
ldapOptions?: object;
crowdPassword?: string;
crowd?: boolean;
code?: string;
totp?: {
login: ICredentials;
code: string;
};
}

View File

@ -1,13 +1,21 @@
import Model from '@nozbe/watermelondb/Model'; import Model from '@nozbe/watermelondb/Model';
import { IUserEmail, IUserSettings } from './IUser';
import { UserStatus } from './UserStatus';
export interface ILoggedUser { export interface ILoggedUser {
id: string; id: string;
token: string; token: string;
username: string; username: string;
name: string; name: string;
language?: string; language?: string;
status: string; status: UserStatus;
statusText?: string; statusText?: string;
customFields?: {
[key: string]: any;
};
statusLivechat?: string;
emails?: IUserEmail[];
roles?: string[]; roles?: string[];
avatarETag?: string; avatarETag?: string;
showMessageInMainThread?: boolean; showMessageInMainThread?: boolean;
@ -15,4 +23,16 @@ export interface ILoggedUser {
enableMessageParserEarlyAdoption?: boolean; enableMessageParserEarlyAdoption?: boolean;
} }
export interface ILoggedUserResultFromServer
extends Omit<ILoggedUser, 'enableMessageParserEarlyAdoption' | 'showMessageInMainThread'> {
settings: IUserSettings;
}
export interface ILoginResultFromServer {
status: string;
authToken: string;
userId: string;
me: ILoggedUserResultFromServer;
}
export type TLoggedUserModel = ILoggedUser & Model; export type TLoggedUserModel = ILoggedUser & Model;

View File

@ -3,6 +3,16 @@ import rocketchat from '../lib/rocketchat';
type TRocketChat = typeof rocketchat; type TRocketChat = typeof rocketchat;
export interface IRocketChat extends TRocketChat { export interface IRocketChat extends TRocketChat {
closeListener: any;
usersListener: any;
notifyAllListener: any;
rolesListener: any;
notifyLoggedListener: any;
activeUsers: any;
_setUserTimer: any;
connectedListener: any;
connectingListener: any;
connectTimeout: any;
sdk: any; sdk: any;
activeUsersSubTimeout: any; activeUsersSubTimeout: any;
roomsSub: any; roomsSub: any;

View File

@ -24,6 +24,7 @@ export * from './IServerHistory';
export * from './IRocketChat'; export * from './IRocketChat';
export * from './ICertificate'; export * from './ICertificate';
export * from './IUrl'; export * from './IUrl';
export * from './ICredentials';
export * from './ISearch'; export * from './ISearch';
export interface IBaseScreen<T extends Record<string, object | undefined>, S extends string> { export interface IBaseScreen<T extends Record<string, object | undefined>, S extends string> {

View File

@ -12,6 +12,7 @@ import log from '../../utils/log';
import { E2E_PRIVATE_KEY, E2E_PUBLIC_KEY, E2E_RANDOM_PASSWORD_KEY } from '../encryption/constants'; import { E2E_PRIVATE_KEY, E2E_PUBLIC_KEY, E2E_RANDOM_PASSWORD_KEY } from '../encryption/constants';
import UserPreferences from '../userPreferences'; import UserPreferences from '../userPreferences';
import { ICertificate, IRocketChat } from '../../definitions'; import { ICertificate, IRocketChat } from '../../definitions';
import sdk from '../rocketchat/services/sdk';
async function removeServerKeys({ server, userId }: { server: string; userId?: string | null }) { async function removeServerKeys({ server, userId }: { server: string; userId?: string | null }) {
await UserPreferences.removeItem(`${RocketChat.TOKEN_KEY}-${server}`); await UserPreferences.removeItem(`${RocketChat.TOKEN_KEY}-${server}`);
@ -116,13 +117,13 @@ export default async function logout(this: IRocketChat, { server }: { server: st
try { try {
// RC 0.60.0 // RC 0.60.0
await this.sdk.logout(); await sdk.current.logout();
} catch (e) { } catch (e) {
log(e); log(e);
} }
if (this.sdk) { if (sdk.current) {
this.sdk = null; sdk.disconnect();
} }
await removeServerData({ server }); await removeServerData({ server });

View File

@ -1,27 +1,15 @@
import { Q } from '@nozbe/watermelondb'; import { Q } from '@nozbe/watermelondb';
import { sanitizedRaw } from '@nozbe/watermelondb/RawRecord';
import AsyncStorage from '@react-native-community/async-storage'; import AsyncStorage from '@react-native-community/async-storage';
import { Rocketchat as RocketchatClient, settings as RocketChatSettings } from '@rocket.chat/sdk';
import { InteractionManager } from 'react-native'; import { InteractionManager } from 'react-native';
import RNFetchBlob from 'rn-fetch-blob';
import { setActiveUsers } from '../../actions/activeUsers'; import { setActiveUsers } from '../../actions/activeUsers';
import { connectRequest, connectSuccess, disconnect } from '../../actions/connect';
import { encryptionInit } from '../../actions/encryption'; import { encryptionInit } from '../../actions/encryption';
import { loginRequest, setLoginServices, setUser } from '../../actions/login'; import { setUser } from '../../actions/login';
import { updatePermission } from '../../actions/permissions';
import { selectServerFailure } from '../../actions/server';
import { updateSettings } from '../../actions/settings';
import { shareSelectServer, shareSetSettings, shareSetUser } from '../../actions/share'; import { shareSelectServer, shareSetSettings, shareSetUser } from '../../actions/share';
import defaultSettings from '../../constants/settings'; import defaultSettings from '../../constants/settings';
import I18n from '../../i18n';
import { getDeviceToken } from '../../notifications/push'; import { getDeviceToken } from '../../notifications/push';
import { getBundleId, isIOS } from '../../utils/deviceInfo'; import { getBundleId, isIOS } from '../../utils/deviceInfo';
import EventEmitter from '../../utils/events';
import fetch from '../../utils/fetch';
import log from '../../utils/log'; import log from '../../utils/log';
import SSLPinning from '../../utils/sslPinning'; import SSLPinning from '../../utils/sslPinning';
import { twoFactor } from '../../utils/twoFactor';
import { useSsl } from '../../utils/url';
import database from '../database'; import database from '../database';
import { sanitizeLikeString } from '../database/utils'; import { sanitizeLikeString } from '../database/utils';
import { Encryption } from '../encryption'; import { Encryption } from '../encryption';
@ -36,11 +24,10 @@ import {
} from '../methods/enterpriseModules'; } from '../methods/enterpriseModules';
import { getCustomEmojis, setCustomEmojis } from '../methods/getCustomEmojis'; import { getCustomEmojis, setCustomEmojis } from '../methods/getCustomEmojis';
import { getPermissions, setPermissions } from '../methods/getPermissions'; import { getPermissions, setPermissions } from '../methods/getPermissions';
import { getRoles, onRolesChanged, setRoles } from '../methods/getRoles'; import { getRoles, setRoles } from '../methods/getRoles';
import getRooms from '../methods/getRooms'; import getRooms from '../methods/getRooms';
import getSettings, { getLoginSettings, setSettings, subscribeSettings } from '../methods/getSettings'; import getSettings, { getLoginSettings, setSettings, subscribeSettings } from '../methods/getSettings';
import getSlashCommands from '../methods/getSlashCommands'; import getSlashCommands from '../methods/getSlashCommands';
import protectedFunction from '../methods/helpers/protectedFunction';
import loadMessagesForRoom from '../methods/loadMessagesForRoom'; import loadMessagesForRoom from '../methods/loadMessagesForRoom';
import loadMissedMessages from '../methods/loadMissedMessages'; import loadMissedMessages from '../methods/loadMissedMessages';
import loadNextMessages from '../methods/loadNextMessages'; import loadNextMessages from '../methods/loadNextMessages';
@ -66,6 +53,21 @@ import * as search from './methods/search';
// Services // Services
import sdk from './services/sdk'; import sdk from './services/sdk';
import toggleFavorite from './services/toggleFavorite'; import toggleFavorite from './services/toggleFavorite';
import {
login,
loginTOTP,
loginWithPassword,
loginOAuthOrSso,
getLoginServices,
determineAuthType,
disconnect,
checkAndReopen,
abort,
getServerInfo,
getWebsocketInfo,
stopListener,
connect
} from './services/connect';
import * as restAPis from './services/restApi'; import * as restAPis from './services/restApi';
const TOKEN_KEY = 'reactnativemeteor_usertoken'; const TOKEN_KEY = 'reactnativemeteor_usertoken';
@ -75,9 +77,8 @@ const CERTIFICATE_KEY = 'RC_CERTIFICATE_KEY';
export const THEME_PREFERENCES_KEY = 'RC_THEME_PREFERENCES_KEY'; export const THEME_PREFERENCES_KEY = 'RC_THEME_PREFERENCES_KEY';
export const CRASH_REPORT_KEY = 'RC_CRASH_REPORT_KEY'; export const CRASH_REPORT_KEY = 'RC_CRASH_REPORT_KEY';
export const ANALYTICS_EVENTS_KEY = 'RC_ANALYTICS_EVENTS_KEY'; export const ANALYTICS_EVENTS_KEY = 'RC_ANALYTICS_EVENTS_KEY';
const MIN_ROCKETCHAT_VERSION = '0.70.0'; export const MIN_ROCKETCHAT_VERSION = '0.70.0';
export const STATUSES = ['offline', 'online', 'away', 'busy'];
const STATUSES = ['offline', 'online', 'away', 'busy'];
const RocketChat = { const RocketChat = {
TOKEN_KEY, TOKEN_KEY,
@ -103,299 +104,14 @@ const RocketChat = {
} }
}, },
canOpenRoom, canOpenRoom,
async getWebsocketInfo({ server }) { getWebsocketInfo,
const sdk = new RocketchatClient({ host: server, protocol: 'ddp', useSsl: useSsl(server) }); getServerInfo,
stopListener,
try {
await sdk.connect();
} catch (err) {
if (err.message && err.message.includes('400')) {
return {
success: false,
message: I18n.t('Websocket_disabled', { contact: I18n.t('Contact_your_server_admin') })
};
}
}
sdk.disconnect();
return {
success: true
};
},
async getServerInfo(server) {
try {
const response = await RNFetchBlob.fetch('GET', `${server}/api/info`, { ...RocketChatSettings.customHeaders });
try {
// Try to resolve as json
const jsonRes = response.json();
if (!jsonRes?.success) {
return {
success: false,
message: I18n.t('Not_RC_Server', { contact: I18n.t('Contact_your_server_admin') })
};
}
if (compareServerVersion(jsonRes.version, 'lowerThan', MIN_ROCKETCHAT_VERSION)) {
return {
success: false,
message: I18n.t('Invalid_server_version', {
currentVersion: jsonRes.version,
minVersion: MIN_ROCKETCHAT_VERSION
})
};
}
return jsonRes;
} catch (error) {
// Request is successful, but response isn't a json
}
} catch (e) {
if (e?.message) {
if (e.message === 'Aborted') {
reduxStore.dispatch(selectServerFailure());
throw e;
}
return {
success: false,
message: e.message
};
}
}
return {
success: false,
message: I18n.t('Not_RC_Server', { contact: I18n.t('Contact_your_server_admin') })
};
},
stopListener(listener) {
return listener && listener.stop();
},
// Abort all requests and create a new AbortController // Abort all requests and create a new AbortController
abort() { abort,
if (this.controller) { checkAndReopen,
this.controller.abort(); disconnect,
if (this.sdk) { connect,
this.sdk.abort();
}
}
this.controller = new AbortController();
},
checkAndReopen() {
return this?.sdk?.checkAndReopen();
},
disconnect() {
this.sdk = sdk.disconnect();
},
connect({ server, user, logoutOnError = false }) {
return new Promise(resolve => {
if (this?.sdk?.client?.host === server) {
return resolve();
} else {
this.disconnect();
database.setActiveDB(server);
}
reduxStore.dispatch(connectRequest());
if (this.connectTimeout) {
clearTimeout(this.connectTimeout);
}
if (this.connectingListener) {
this.connectingListener.then(this.stopListener);
}
if (this.connectedListener) {
this.connectedListener.then(this.stopListener);
}
if (this.closeListener) {
this.closeListener.then(this.stopListener);
}
if (this.usersListener) {
this.usersListener.then(this.stopListener);
}
if (this.notifyAllListener) {
this.notifyAllListener.then(this.stopListener);
}
if (this.rolesListener) {
this.rolesListener.then(this.stopListener);
}
if (this.notifyLoggedListener) {
this.notifyLoggedListener.then(this.stopListener);
}
this.unsubscribeRooms();
EventEmitter.emit('INQUIRY_UNSUBSCRIBE');
this.sdk = sdk.initialize(server);
this.getSettings();
this.sdk
.connect()
.then(() => {
console.log('connected');
})
.catch(err => {
console.log('connect error', err);
});
this.connectingListener = this.sdk.onStreamData('connecting', () => {
reduxStore.dispatch(connectRequest());
});
this.connectedListener = this.sdk.onStreamData('connected', () => {
const { connected } = reduxStore.getState().meteor;
if (connected) {
return;
}
reduxStore.dispatch(connectSuccess());
const { server: currentServer } = reduxStore.getState().server;
if (user?.token && server === currentServer) {
reduxStore.dispatch(loginRequest({ resume: user.token }, logoutOnError));
}
});
this.closeListener = this.sdk.onStreamData('close', () => {
reduxStore.dispatch(disconnect());
});
this.usersListener = this.sdk.onStreamData(
'users',
protectedFunction(ddpMessage => RocketChat._setUser(ddpMessage))
);
this.notifyAllListener = this.sdk.onStreamData(
'stream-notify-all',
protectedFunction(async ddpMessage => {
const { eventName } = ddpMessage.fields;
if (/public-settings-changed/.test(eventName)) {
const { _id, value } = ddpMessage.fields.args[1];
const db = database.active;
const settingsCollection = db.get('settings');
try {
const settingsRecord = await settingsCollection.find(_id);
const { type } = defaultSettings[_id];
if (type) {
await db.action(async () => {
await settingsRecord.update(u => {
u[type] = value;
});
});
}
reduxStore.dispatch(updateSettings(_id, value));
} catch (e) {
log(e);
}
}
})
);
this.rolesListener = this.sdk.onStreamData(
'stream-roles',
protectedFunction(ddpMessage => onRolesChanged(ddpMessage))
);
// RC 4.1
this.sdk.onStreamData('stream-user-presence', ddpMessage => {
const userStatus = ddpMessage.fields.args[0];
const { uid } = ddpMessage.fields;
const [, status, statusText] = userStatus;
const newStatus = { status: STATUSES[status], statusText };
reduxStore.dispatch(setActiveUsers({ [uid]: newStatus }));
const { user: loggedUser } = reduxStore.getState().login;
if (loggedUser && loggedUser.id === uid) {
reduxStore.dispatch(setUser(newStatus));
}
});
this.notifyLoggedListener = this.sdk.onStreamData(
'stream-notify-logged',
protectedFunction(async ddpMessage => {
const { eventName } = ddpMessage.fields;
// `user-status` event is deprecated after RC 4.1 in favor of `stream-user-presence/${uid}`
if (/user-status/.test(eventName)) {
this.activeUsers = this.activeUsers || {};
if (!this._setUserTimer) {
this._setUserTimer = setTimeout(() => {
const activeUsersBatch = this.activeUsers;
InteractionManager.runAfterInteractions(() => {
reduxStore.dispatch(setActiveUsers(activeUsersBatch));
});
this._setUserTimer = null;
return (this.activeUsers = {});
}, 10000);
}
const userStatus = ddpMessage.fields.args[0];
const [id, , status, statusText] = userStatus;
this.activeUsers[id] = { status: STATUSES[status], statusText };
const { user: loggedUser } = reduxStore.getState().login;
if (loggedUser && loggedUser.id === id) {
reduxStore.dispatch(setUser({ status: STATUSES[status], statusText }));
}
} else if (/updateAvatar/.test(eventName)) {
const { username, etag } = ddpMessage.fields.args[0];
const db = database.active;
const userCollection = db.get('users');
try {
const [userRecord] = await userCollection.query(Q.where('username', Q.eq(username))).fetch();
await db.action(async () => {
await userRecord.update(u => {
u.avatarETag = etag;
});
});
} catch {
// We can't create a new record since we don't receive the user._id
}
} else if (/permissions-changed/.test(eventName)) {
const { _id, roles } = ddpMessage.fields.args[1];
const db = database.active;
const permissionsCollection = db.get('permissions');
try {
const permissionsRecord = await permissionsCollection.find(_id);
await db.action(async () => {
await permissionsRecord.update(u => {
u.roles = roles;
});
});
reduxStore.dispatch(updatePermission(_id, roles));
} catch (err) {
//
}
} else if (/Users:NameChanged/.test(eventName)) {
const userNameChanged = ddpMessage.fields.args[0];
const db = database.active;
const userCollection = db.get('users');
try {
const userRecord = await userCollection.find(userNameChanged._id);
await db.action(async () => {
await userRecord.update(u => {
Object.assign(u, userNameChanged);
});
});
} catch {
// User not found
await db.action(async () => {
await userCollection.create(u => {
u._raw = sanitizedRaw({ id: userNameChanged._id }, userCollection.schema);
Object.assign(u, userNameChanged);
});
});
}
}
})
);
resolve();
});
},
async shareExtensionInit(server) { async shareExtensionInit(server) {
database.setShareDB(server); database.setShareDB(server);
@ -489,103 +205,10 @@ const RocketChat = {
return this.methodCallWrapper('e2e.resetOwnE2EKey'); return this.methodCallWrapper('e2e.resetOwnE2EKey');
}, },
loginTOTP(params, loginEmailPassword, isFromWebView = false) { loginTOTP,
return new Promise(async (resolve, reject) => { loginWithPassword,
try { loginOAuthOrSso,
const result = await this.login(params, isFromWebView); login,
return resolve(result);
} catch (e) {
if (e.data?.error && (e.data.error === 'totp-required' || e.data.error === 'totp-invalid')) {
const { details } = e.data;
try {
const code = await twoFactor({ method: details?.method || 'totp', invalid: details?.error === 'totp-invalid' });
if (loginEmailPassword) {
reduxStore.dispatch(setUser({ username: params.user || params.username }));
// Force normalized params for 2FA starting RC 3.9.0.
const serverVersion = reduxStore.getState().server.version;
if (compareServerVersion(serverVersion, 'greaterThanOrEqualTo', '3.9.0')) {
const user = params.user ?? params.username;
const password = params.password ?? params.ldapPass ?? params.crowdPassword;
params = { user, password };
}
return resolve(this.loginTOTP({ ...params, code: code?.twoFactorCode }, loginEmailPassword));
}
return resolve(
this.loginTOTP({
totp: {
login: {
...params
},
code: code?.twoFactorCode
}
})
);
} catch {
// twoFactor was canceled
return reject();
}
} else {
reject(e);
}
}
});
},
loginWithPassword({ user, password }) {
let params = { user, password };
const state = reduxStore.getState();
if (state.settings.LDAP_Enable) {
params = {
username: user,
ldapPass: password,
ldap: true,
ldapOptions: {}
};
} else if (state.settings.CROWD_Enable) {
params = {
username: user,
crowdPassword: password,
crowd: true
};
}
return this.loginTOTP(params, true);
},
async loginOAuthOrSso(params, isFromWebView = true) {
const result = await this.loginTOTP(params, false, isFromWebView);
reduxStore.dispatch(loginRequest({ resume: result.token }, false, isFromWebView));
},
async login(credentials, isFromWebView = false) {
const sdk = this.shareSDK || this.sdk;
// RC 0.64.0
await sdk.login(credentials);
const { result } = sdk.currentLogin;
const user = {
id: result.userId,
token: result.authToken,
username: result.me.username,
name: result.me.name,
language: result.me.language,
status: result.me.status,
statusText: result.me.statusText,
customFields: result.me.customFields,
statusLivechat: result.me.statusLivechat,
emails: result.me.emails,
roles: result.me.roles,
avatarETag: result.me.avatarETag,
isFromWebView,
showMessageInMainThread: result.me.settings?.preferences?.showMessageInMainThread ?? true,
enableMessageParserEarlyAdoption: result.me.settings?.preferences?.enableMessageParserEarlyAdoption ?? true
};
return user;
},
logout, logout,
logoutOtherLocations() { logoutOtherLocations() {
const { id: userId } = reduxStore.getState().login.user; const { id: userId } = reduxStore.getState().login.user;
@ -843,59 +466,8 @@ const RocketChat = {
prefs = { ...prefs, ...param }; prefs = { ...prefs, ...param };
return UserPreferences.setMapAsync(SORT_PREFS_KEY, prefs); return UserPreferences.setMapAsync(SORT_PREFS_KEY, prefs);
}, },
async getLoginServices(server) { getLoginServices,
try { determineAuthType,
let loginServices = [];
const loginServicesResult = await fetch(`${server}/api/v1/settings.oauth`).then(response => response.json());
if (loginServicesResult.success && loginServicesResult.services) {
const { services } = loginServicesResult;
loginServices = services;
const loginServicesReducer = loginServices.reduce((ret, item) => {
const name = item.name || item.buttonLabelText || item.service;
const authType = this._determineAuthType(item);
if (authType !== 'not_supported') {
ret[name] = { ...item, name, authType };
}
return ret;
}, {});
reduxStore.dispatch(setLoginServices(loginServicesReducer));
} else {
reduxStore.dispatch(setLoginServices({}));
}
} catch (error) {
console.log(error);
reduxStore.dispatch(setLoginServices({}));
}
},
_determineAuthType(services) {
const { name, custom, showButton = true, service } = services;
const authName = name || service;
if (custom && showButton) {
return 'oauth_custom';
}
if (service === 'saml') {
return 'saml';
}
if (service === 'cas') {
return 'cas';
}
if (authName === 'apple' && isIOS) {
return 'apple';
}
// TODO: remove this after other oauth providers are implemented. e.g. Drupal, github_enterprise
const availableOAuth = ['facebook', 'github', 'gitlab', 'google', 'linkedin', 'meteor-developer', 'twitter', 'wordpress'];
return availableOAuth.includes(authName) ? 'oauth' : 'not_supported';
},
roomTypeToApiType, roomTypeToApiType,
readThreads(tmid) { readThreads(tmid) {
const serverVersion = reduxStore.getState().server.version; const serverVersion = reduxStore.getState().server.version;

View File

@ -0,0 +1,511 @@
import RNFetchBlob from 'rn-fetch-blob';
import { settings as RocketChatSettings } from '@rocket.chat/sdk';
import { sanitizedRaw } from '@nozbe/watermelondb/RawRecord';
import { InteractionManager } from 'react-native';
import { Q } from '@nozbe/watermelondb';
import log from '../../../utils/log';
import { onRolesChanged } from '../../methods/getRoles';
import { UserStatus } from '../../../definitions/UserStatus';
import { setActiveUsers } from '../../../actions/activeUsers';
import protectedFunction from '../../methods/helpers/protectedFunction';
import database from '../../database';
import { selectServerFailure } from '../../../actions/server';
import { twoFactor } from '../../../utils/twoFactor';
import { compareServerVersion } from '../../utils';
import { store } from '../../auxStore';
import { loginRequest, setLoginServices, setUser } from '../../../actions/login';
import sdk from './sdk';
import I18n from '../../../i18n';
import RocketChat, { MIN_ROCKETCHAT_VERSION, STATUSES } from '../rocketchat';
import { ICredentials, ILoggedUser, IRocketChat, IUser } from '../../../definitions';
import { isIOS } from '../../../utils/deviceInfo';
import { connectRequest, connectSuccess, disconnect as disconnectAction } from '../../../actions/connect';
import { updatePermission } from '../../../actions/permissions';
import EventEmitter from '../../../utils/events';
import { updateSettings } from '../../../actions/settings';
import defaultSettings from '../../../constants/settings';
interface IServices {
[index: string]: string | boolean;
name: string;
custom: boolean;
showButton: boolean;
buttonLabelText: string;
service: string;
}
// FIXME: Remove `this` context
function connect(
this: IRocketChat,
{ server, user, logoutOnError = false }: { server: string; user: IUser; logoutOnError: boolean }
) {
return new Promise<void>(resolve => {
if (sdk.current?.client?.host === server) {
return resolve();
}
disconnect();
database.setActiveDB(server);
store.dispatch(connectRequest());
if (this.connectTimeout) {
clearTimeout(this.connectTimeout);
}
if (this.connectingListener) {
this.connectingListener.then(stopListener);
}
if (this.connectedListener) {
this.connectedListener.then(stopListener);
}
if (this.closeListener) {
this.closeListener.then(stopListener);
}
if (this.usersListener) {
this.usersListener.then(stopListener);
}
if (this.notifyAllListener) {
this.notifyAllListener.then(stopListener);
}
if (this.rolesListener) {
this.rolesListener.then(stopListener);
}
if (this.notifyLoggedListener) {
this.notifyLoggedListener.then(stopListener);
}
this.unsubscribeRooms();
EventEmitter.emit('INQUIRY_UNSUBSCRIBE');
sdk.initialize(server);
this.getSettings();
sdk.current
.connect()
.then(() => {
console.log('connected');
})
.catch((err: unknown) => {
console.log('connect error', err);
});
this.connectingListener = sdk.current.onStreamData('connecting', () => {
store.dispatch(connectRequest());
});
this.connectedListener = sdk.current.onStreamData('connected', () => {
const { connected } = store.getState().meteor;
if (connected) {
return;
}
store.dispatch(connectSuccess());
const { server: currentServer } = store.getState().server;
if (user?.token && server === currentServer) {
store.dispatch(loginRequest({ resume: user.token }, logoutOnError));
}
});
this.closeListener = sdk.current.onStreamData('close', () => {
store.dispatch(disconnectAction());
});
this.usersListener = sdk.current.onStreamData(
'users',
protectedFunction((ddpMessage: any) => RocketChat._setUser(ddpMessage))
);
this.notifyAllListener = sdk.current.onStreamData(
'stream-notify-all',
protectedFunction(async (ddpMessage: { fields: { args?: any; eventName: string } }) => {
const { eventName } = ddpMessage.fields;
if (/public-settings-changed/.test(eventName)) {
const { _id, value } = ddpMessage.fields.args[1];
const db = database.active;
const settingsCollection = db.get('settings');
try {
const settingsRecord = await settingsCollection.find(_id);
// @ts-ignore
const { type } = defaultSettings[_id];
if (type) {
await db.write(async () => {
await settingsRecord.update(u => {
// @ts-ignore
u[type] = value;
});
});
}
store.dispatch(updateSettings(_id, value));
} catch (e) {
log(e);
}
}
})
);
this.rolesListener = sdk.current.onStreamData(
'stream-roles',
protectedFunction((ddpMessage: any) => onRolesChanged(ddpMessage))
);
// RC 4.1
sdk.current.onStreamData('stream-user-presence', (ddpMessage: { fields: { args?: any; uid?: any } }) => {
const userStatus = ddpMessage.fields.args[0];
const { uid } = ddpMessage.fields;
const [, status, statusText] = userStatus;
const newStatus = { status: STATUSES[status], statusText };
// @ts-ignore
store.dispatch(setActiveUsers({ [uid]: newStatus }));
const { user: loggedUser } = store.getState().login;
if (loggedUser && loggedUser.id === uid) {
// @ts-ignore
store.dispatch(setUser(newStatus));
}
});
this.notifyLoggedListener = sdk.current.onStreamData(
'stream-notify-logged',
protectedFunction(async (ddpMessage: { fields: { args?: any; eventName?: any } }) => {
const { eventName } = ddpMessage.fields;
// `user-status` event is deprecated after RC 4.1 in favor of `stream-user-presence/${uid}`
if (/user-status/.test(eventName)) {
this.activeUsers = this.activeUsers || {};
if (!this._setUserTimer) {
this._setUserTimer = setTimeout(() => {
const activeUsersBatch = this.activeUsers;
InteractionManager.runAfterInteractions(() => {
store.dispatch(setActiveUsers(activeUsersBatch));
});
this._setUserTimer = null;
return (this.activeUsers = {});
}, 10000);
}
const userStatus = ddpMessage.fields.args[0];
const [id, , status, statusText] = userStatus;
this.activeUsers[id] = { status: STATUSES[status], statusText };
const { user: loggedUser } = store.getState().login;
if (loggedUser && loggedUser.id === id) {
store.dispatch(setUser({ status: STATUSES[status] as UserStatus, statusText }));
}
} else if (/updateAvatar/.test(eventName)) {
const { username, etag } = ddpMessage.fields.args[0];
const db = database.active;
const userCollection = db.get('users');
try {
const [userRecord] = await userCollection.query(Q.where('username', Q.eq(username))).fetch();
await db.write(async () => {
await userRecord.update(u => {
u.avatarETag = etag;
});
});
} catch {
// We can't create a new record since we don't receive the user._id
}
} else if (/permissions-changed/.test(eventName)) {
const { _id, roles } = ddpMessage.fields.args[1];
const db = database.active;
const permissionsCollection = db.get('permissions');
try {
const permissionsRecord = await permissionsCollection.find(_id);
await db.write(async () => {
await permissionsRecord.update(u => {
u.roles = roles;
});
});
store.dispatch(updatePermission(_id, roles));
} catch (err) {
//
}
} else if (/Users:NameChanged/.test(eventName)) {
const userNameChanged = ddpMessage.fields.args[0];
const db = database.active;
const userCollection = db.get('users');
try {
const userRecord = await userCollection.find(userNameChanged._id);
await db.write(async () => {
await userRecord.update(u => {
Object.assign(u, userNameChanged);
});
});
} catch {
// User not found
await db.write(async () => {
await userCollection.create(u => {
u._raw = sanitizedRaw({ id: userNameChanged._id }, userCollection.schema);
Object.assign(u, userNameChanged);
});
});
}
}
})
);
resolve();
});
}
function stopListener(listener: any): boolean {
return listener && listener.stop();
}
async function login(credentials: ICredentials, isFromWebView = false): Promise<ILoggedUser | undefined> {
// RC 0.64.0
await sdk.current.login(credentials);
const result = sdk.current.currentLogin?.result;
if (result) {
const user: ILoggedUser = {
id: result.userId,
token: result.authToken,
username: result.me.username,
name: result.me.name,
language: result.me.language,
status: result.me.status,
statusText: result.me.statusText,
customFields: result.me.customFields,
statusLivechat: result.me.statusLivechat,
emails: result.me.emails,
roles: result.me.roles,
avatarETag: result.me.avatarETag,
isFromWebView,
showMessageInMainThread: result.me.settings?.preferences?.showMessageInMainThread ?? true,
enableMessageParserEarlyAdoption: result.me.settings?.preferences?.enableMessageParserEarlyAdoption ?? true
};
return user;
}
}
function loginTOTP(params: ICredentials, loginEmailPassword?: boolean, isFromWebView = false): Promise<ILoggedUser> {
return new Promise(async (resolve, reject) => {
try {
const result = await login(params, isFromWebView);
if (result) {
return resolve(result);
}
} catch (e: any) {
if (e.data?.error && (e.data.error === 'totp-required' || e.data.error === 'totp-invalid')) {
const { details } = e.data;
try {
const code = await twoFactor({ method: details?.method || 'totp', invalid: details?.error === 'totp-invalid' });
if (loginEmailPassword) {
store.dispatch(setUser({ username: params.user || params.username }));
// Force normalized params for 2FA starting RC 3.9.0.
const serverVersion = store.getState().server.version;
if (compareServerVersion(serverVersion as string, 'greaterThanOrEqualTo', '3.9.0')) {
const user = params.user ?? params.username;
const password = params.password ?? params.ldapPass ?? params.crowdPassword;
params = { user, password };
}
return resolve(loginTOTP({ ...params, code: code?.twoFactorCode }, loginEmailPassword));
}
return resolve(
loginTOTP({
totp: {
login: {
...params
},
code: code?.twoFactorCode
}
})
);
} catch {
// twoFactor was canceled
return reject();
}
} else {
reject(e);
}
}
});
}
function loginWithPassword({ user, password }: { user: string; password: string }): Promise<ILoggedUser> {
let params: ICredentials = { user, password };
const state = store.getState();
if (state.settings.LDAP_Enable) {
params = {
username: user,
ldapPass: password,
ldap: true,
ldapOptions: {}
};
} else if (state.settings.CROWD_Enable) {
params = {
username: user,
crowdPassword: password,
crowd: true
};
}
return loginTOTP(params, true);
}
async function loginOAuthOrSso(params: ICredentials, isFromWebView = true) {
const result = await loginTOTP(params, false, isFromWebView);
store.dispatch(loginRequest({ resume: result.token }, false, isFromWebView));
}
function abort() {
if (sdk.current) {
return sdk.current.abort();
}
return new AbortController();
}
function checkAndReopen() {
return sdk.current.checkAndReopen();
}
function disconnect() {
return sdk.disconnect();
}
async function getServerInfo(server: string) {
try {
const response = await RNFetchBlob.fetch('GET', `${server}/api/info`, { ...RocketChatSettings.customHeaders });
try {
// Try to resolve as json
const jsonRes = response.json();
if (!jsonRes?.success) {
return {
success: false,
message: I18n.t('Not_RC_Server', { contact: I18n.t('Contact_your_server_admin') })
};
}
if (compareServerVersion(jsonRes.version, 'lowerThan', MIN_ROCKETCHAT_VERSION)) {
return {
success: false,
message: I18n.t('Invalid_server_version', {
currentVersion: jsonRes.version,
minVersion: MIN_ROCKETCHAT_VERSION
})
};
}
return jsonRes;
} catch (error) {
// Request is successful, but response isn't a json
}
} catch (e: any) {
if (e?.message) {
if (e.message === 'Aborted') {
store.dispatch(selectServerFailure());
throw e;
}
return {
success: false,
message: e.message
};
}
}
return {
success: false,
message: I18n.t('Not_RC_Server', { contact: I18n.t('Contact_your_server_admin') })
};
}
async function getWebsocketInfo({ server }: { server: string }) {
sdk.initialize(server);
try {
await sdk.current.connect();
} catch (err: any) {
if (err.message && err.message.includes('400')) {
return {
success: false,
message: I18n.t('Websocket_disabled', { contact: I18n.t('Contact_your_server_admin') })
};
}
}
sdk.disconnect();
return {
success: true
};
}
async function getLoginServices(server: string) {
try {
let loginServices = [];
const loginServicesResult = await fetch(`${server}/api/v1/settings.oauth`).then(response => response.json());
if (loginServicesResult.success && loginServicesResult.services) {
const { services } = loginServicesResult;
loginServices = services;
const loginServicesReducer = loginServices.reduce((ret: IServices[], item: IServices) => {
const name = item.name || item.buttonLabelText || item.service;
const authType = determineAuthType(item);
if (authType !== 'not_supported') {
ret[name as unknown as number] = { ...item, name, authType };
}
return ret;
}, {});
store.dispatch(setLoginServices(loginServicesReducer));
} else {
store.dispatch(setLoginServices({}));
}
} catch (error) {
console.log(error);
store.dispatch(setLoginServices({}));
}
}
function determineAuthType(services: IServices) {
const { name, custom, showButton = true, service } = services;
const authName = name || service;
if (custom && showButton) {
return 'oauth_custom';
}
if (service === 'saml') {
return 'saml';
}
if (service === 'cas') {
return 'cas';
}
if (authName === 'apple' && isIOS) {
return 'apple';
}
// TODO: remove this after other oauth providers are implemented. e.g. Drupal, github_enterprise
const availableOAuth = ['facebook', 'github', 'gitlab', 'google', 'linkedin', 'meteor-developer', 'twitter', 'wordpress'];
return availableOAuth.includes(authName) ? 'oauth' : 'not_supported';
}
export {
login,
loginTOTP,
loginWithPassword,
loginOAuthOrSso,
checkAndReopen,
abort,
connect,
disconnect,
getServerInfo,
getWebsocketInfo,
stopListener,
getLoginServices,
determineAuthType
};