120 lines
3.3 KiB
TypeScript
120 lines
3.3 KiB
TypeScript
import { Rocketchat } from '@rocket.chat/sdk';
|
|
import isEmpty from 'lodash/isEmpty';
|
|
import EJSON from 'ejson';
|
|
|
|
import reduxStore from '../../createStore';
|
|
import { useSsl } from '../../../utils/url';
|
|
import { twoFactor } from '../../../utils/twoFactor';
|
|
|
|
class Sdk {
|
|
private sdk: typeof Rocketchat;
|
|
private code: any;
|
|
|
|
// TODO: We need to stop returning the SDK after all methods are dehydrated
|
|
initialize(server: string) {
|
|
this.code = null;
|
|
|
|
// The app can't reconnect if reopen interval is 5s while in development
|
|
this.sdk = new Rocketchat({ host: server, protocol: 'ddp', useSsl: useSsl(server), reopen: __DEV__ ? 20000 : 5000 });
|
|
return this.sdk;
|
|
}
|
|
|
|
/**
|
|
* TODO: evaluate the need for assigning "null" to this.sdk
|
|
* I'm returning "null" because we need to remove both instances of this.sdk here and on rocketchat.js
|
|
*/
|
|
disconnect() {
|
|
if (this.sdk) {
|
|
this.sdk.disconnect();
|
|
this.sdk = null;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
get(...args: any[]): Promise<any> {
|
|
return this.sdk.get(...args);
|
|
}
|
|
|
|
post(...args: any[]): Promise<any> {
|
|
return new Promise(async (resolve, reject) => {
|
|
const isMethodCall = args[0]?.startsWith('method.call/');
|
|
try {
|
|
const result = await this.sdk.post(...args);
|
|
|
|
/**
|
|
* if API_Use_REST_For_DDP_Calls is enabled and it's a method call,
|
|
* responses have a different object structure
|
|
*/
|
|
if (isMethodCall) {
|
|
const response = JSON.parse(result.message);
|
|
if (response?.error) {
|
|
throw response.error;
|
|
}
|
|
return resolve(response.result);
|
|
}
|
|
return resolve(result);
|
|
} catch (e: any) {
|
|
const errorType = isMethodCall ? e?.error : e?.data?.errorType;
|
|
const totpInvalid = 'totp-invalid';
|
|
const totpRequired = 'totp-required';
|
|
if ([totpInvalid, totpRequired].includes(errorType)) {
|
|
const { details } = isMethodCall ? e : e?.data;
|
|
try {
|
|
await twoFactor({ method: details?.method, invalid: errorType === totpInvalid });
|
|
return resolve(this.post(...args));
|
|
} catch {
|
|
// twoFactor was canceled
|
|
return resolve({});
|
|
}
|
|
} else {
|
|
reject(e);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
methodCall(...args: any[]): Promise<any> {
|
|
return new Promise(async (resolve, reject) => {
|
|
try {
|
|
const result = await this.sdk?.methodCall(...args, this.code || '');
|
|
return resolve(result);
|
|
} catch (e: any) {
|
|
if (e.error && (e.error === 'totp-required' || e.error === 'totp-invalid')) {
|
|
const { details } = e;
|
|
try {
|
|
this.code = await twoFactor({ method: details?.method, invalid: e.error === 'totp-invalid' });
|
|
return resolve(this.methodCall(...args));
|
|
} catch {
|
|
// twoFactor was canceled
|
|
return resolve({});
|
|
}
|
|
} else {
|
|
reject(e);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
methodCallWrapper(method: string, ...params: any[]): Promise<any> {
|
|
const { API_Use_REST_For_DDP_Calls } = reduxStore.getState().settings;
|
|
const { user } = reduxStore.getState().login;
|
|
if (API_Use_REST_For_DDP_Calls) {
|
|
const url = isEmpty(user) ? 'method.callAnon' : 'method.call';
|
|
return this.post(`${url}/${method}`, {
|
|
message: EJSON.stringify({ method, params })
|
|
});
|
|
}
|
|
const parsedParams = params.map(param => {
|
|
if (param instanceof Date) {
|
|
return { $date: new Date(param).getTime() };
|
|
}
|
|
return param;
|
|
});
|
|
return this.methodCall(method, ...parsedParams);
|
|
}
|
|
}
|
|
|
|
const sdk = new Sdk();
|
|
|
|
export default sdk;
|