diff --git a/src/controllers/organization.controller.js b/src/controllers/organization.controller.js index 15686518..96d2a4b1 100644 --- a/src/controllers/organization.controller.js +++ b/src/controllers/organization.controller.js @@ -10,6 +10,8 @@ import { assertIfReadOnlyMode, } from '../utils/data-assertions'; +import { ModelKeys, Audit } from '../models'; + export const findAll = async (req, res) => { return res.json(await Organization.getOrgsMap()); }; @@ -129,3 +131,36 @@ export const subscribeToOrganization = async (req, res) => { }); } }; + +export const unsubscribeToOrganization = async (req, res) => { + try { + await assertIfReadOnlyMode(); + await assertDataLayerAvailable(); + await assertWalletIsAvailable(); + await assertWalletIsSynced(); + await assertHomeOrgExists(); + + await Organization.update( + { subscribed: false, registryHash: '0' }, + { where: { orgUid: req.body.orgUid } }, + ); + + await Promise.all([ + ...Object.keys(ModelKeys).map( + async (key) => + await ModelKeys[key].destroy({ where: { orgUid: req.body.orgUid } }), + ), + Audit.destroy({ where: { orgUid: req.body.orgUid } }), + ]); + + return res.json({ + message: + 'UnSubscribed to organization, you will no longer receive updates.', + }); + } catch (error) { + res.status(400).json({ + message: 'Error unsubscribing to organization', + error: error.message, + }); + } +}; diff --git a/src/models/organizations/organizations.model.js b/src/models/organizations/organizations.model.js index 6acd09f9..7aba7999 100644 --- a/src/models/organizations/organizations.model.js +++ b/src/models/organizations/organizations.model.js @@ -187,7 +187,7 @@ class Organization extends Model { static subscribeToOrganization = async (orgUid) => { const exists = await Organization.findOne({ where: { orgUid } }); if (exists) { - await Organization.update({ subscribed: true }, { orgUid }); + await Organization.update({ subscribed: true }, { where: { orgUid } }); } else { throw new Error( 'Can not subscribe, please import this organization first', diff --git a/src/routes/v1/resources/organization.js b/src/routes/v1/resources/organization.js index f51e6c16..ce77feca 100644 --- a/src/routes/v1/resources/organization.js +++ b/src/routes/v1/resources/organization.js @@ -4,7 +4,12 @@ import express from 'express'; import joiExpress from 'express-joi-validation'; import { OrganizationController } from '../../../controllers'; -import { newOrganizationSchema } from '../../../validations'; +import { + newOrganizationSchema, + unsubscribeOrganizationSchema, + subscribeOrganizationSchema, + importOrganizationSchema, +} from '../../../validations'; const validator = joiExpress.createValidator({ passError: true }); const OrganizationRouter = express.Router(); @@ -29,4 +34,28 @@ OrganizationRouter.put('/', (req, res) => { return OrganizationController.importOrg(req, res); }); +OrganizationRouter.put( + '/import', + validator.body(importOrganizationSchema), + (req, res) => { + return OrganizationController.importOrg(req, res); + }, +); + +OrganizationRouter.put( + '/subscribe', + validator.body(subscribeOrganizationSchema), + (req, res) => { + return OrganizationController.subscribeToOrganization(req, res); + }, +); + +OrganizationRouter.put( + '/unsubscribe', + validator.body(unsubscribeOrganizationSchema), + (req, res) => { + return OrganizationController.unsubscribeToOrganization(req, res); + }, +); + export { OrganizationRouter }; diff --git a/src/tasks/sync-audit-table.js b/src/tasks/sync-audit-table.js index e2b22685..c132c7f9 100644 --- a/src/tasks/sync-audit-table.js +++ b/src/tasks/sync-audit-table.js @@ -15,7 +15,10 @@ dotenv.config(); const task = new Task('sync-audit', async () => { log('Syncing Audit Information'); if (process.env.USE_SIMULATOR === 'false') { - const organizations = await Organization.findAll({ raw: true }); + const organizations = await Organization.findAll({ + where: { subscribed: true }, + raw: true, + }); await Promise.all( organizations.map((organization) => syncOrganizationAudit(organization)), ); diff --git a/src/validations/organizations.validations.js b/src/validations/organizations.validations.js index d1a1a236..bf60fd5b 100644 --- a/src/validations/organizations.validations.js +++ b/src/validations/organizations.validations.js @@ -8,3 +8,11 @@ export const newOrganizationSchema = Joi.object({ export const importOrganizationSchema = Joi.object({ orgUid: Joi.string().required(), }); + +export const unsubscribeOrganizationSchema = Joi.object({ + orgUid: Joi.string().required(), +}); + +export const subscribeOrganizationSchema = Joi.object({ + orgUid: Joi.string().required(), +});