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 { 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 { 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 { 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(); } }); }