Merge pull request #396 from Chia-Network/feature/addOrgSubscribe

feat: add org subscribe unsubscribe endpoints

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,
+    });
+  }
+};

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',

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 };

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)),
     );

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