Rocket.Chat.ReactNative/app/lib/methods/getRoles.ts

132 lines
3.5 KiB
TypeScript

import { sanitizedRaw } from '@nozbe/watermelondb/RawRecord';
import Model from '@nozbe/watermelondb/Model';
import database from '../database';
import { getRoleById } from '../database/services/Role';
import log from './helpers/log';
import { store as reduxStore } from '../store/auxStore';
import { removeRoles, setRoles as setRolesAction, updateRoles } from '../../actions/roles';
import { TRoleModel } from '../../definitions';
import sdk from '../services/sdk';
import protectedFunction from './helpers/protectedFunction';
export async function setRoles(): Promise<void> {
const db = database.active;
const rolesCollection = db.get('roles');
const allRoles = await rolesCollection.query().fetch();
const parsed = allRoles.reduce((acc, item) => ({ ...acc, [item.id]: item.description || item.id }), {});
reduxStore.dispatch(setRolesAction(parsed));
}
interface IRolesChanged {
fields: {
args: {
type: string;
_id: string;
description: string;
}[];
};
}
export async function onRolesChanged(ddpMessage: IRolesChanged): Promise<void> {
const { type, _id, description } = ddpMessage.fields.args[0];
if (/changed/.test(type)) {
const db = database.active;
const rolesCollection = db.get('roles');
try {
const roleRecord = await getRoleById(_id);
if (roleRecord) {
await db.write(async () => {
await roleRecord.update(u => {
u.description = description;
});
});
} else {
await db.write(async () => {
await rolesCollection.create(post => {
post._raw = sanitizedRaw({ id: _id, description }, rolesCollection.schema);
});
});
}
reduxStore.dispatch(updateRoles(_id, description || _id));
} catch (e) {
log(e);
}
}
if (/removed/.test(type)) {
const db = database.active;
try {
const roleRecord = await getRoleById(_id);
if (roleRecord) {
await db.write(async () => {
await roleRecord.destroyPermanently();
});
reduxStore.dispatch(removeRoles(_id));
}
} catch (e) {
log(e);
}
}
}
export function getRoles(): Promise<void> {
const db = database.active;
return new Promise(async resolve => {
try {
// RC 0.70.0
const result = await sdk.get('roles.list');
if (!result.success) {
return resolve();
}
const { roles } = result;
if (roles && roles.length) {
await db.write(async () => {
const rolesCollections = db.get('roles');
const allRolesRecords = await rolesCollections.query().fetch();
// filter roles
const filteredRolesToCreate = roles.filter(i1 => !allRolesRecords.find(i2 => i1._id === i2.id));
const filteredRolesToUpdate = allRolesRecords.filter(i1 => roles.find(i2 => i1.id === i2._id));
// Create
const rolesToCreate = filteredRolesToCreate.map(role =>
rolesCollections.prepareCreate(
protectedFunction((r: TRoleModel) => {
r._raw = sanitizedRaw({ id: role._id }, rolesCollections.schema);
Object.assign(r, role);
})
)
);
// Update
const rolesToUpdate = filteredRolesToUpdate.map(role => {
const newRole = roles.find(r => r._id === role.id);
return role.prepareUpdate(
protectedFunction((r: TRoleModel) => {
Object.assign(r, newRole);
})
);
});
const allRecords: Model[] = [...rolesToCreate, ...rolesToUpdate];
try {
await db.batch(...allRecords);
} catch (e) {
log(e);
}
setRoles();
return allRecords.length;
});
return resolve();
}
} catch (e) {
log(e);
return resolve();
}
});
}