chore(issuer): archive GPM when switch toggled

Author: Fahd

packages/polymath-issuer/src/actions/compliance.js

@@ -86,6 +86,20 @@ export type InvestorCSVRow = [
   string,
 ];
 
+// make more functional and switch transfermanager to module
+async function getDelegateDetails(permissionManager, transferManager) {
+  const delegates = await permissionManager.getAllDelegates(
+    transferManager.address,
+    PERM_TYPES.ADMIN
+  );
+  let delegateDetails = [];
+  for (const delegate of delegates) {
+    let details = await permissionManager.getDelegateDetails(delegate);
+    delegateDetails.push({ id: delegate, address: delegate, details });
+  }
+  return delegateDetails;
+}
+
 export const fetchManagers = () => async (
   dispatch: Function,
   getState: GetState
@@ -95,18 +109,17 @@ export const fetchManagers = () => async (
   try {
     const st: SecurityToken = getState().token.token.contract;
     const permissionManager = await st.getPermissionManager();
-    if (permissionManager) {
+    if (!permissionManager) {
+      return;
+    }
+    const moduleMetadata = await st.getModule(permissionManager.address);
+    if (permissionManager && !moduleMetadata.isArchived) {
       const transferManager = await st.getTransferManager();
       if (transferManager) {
-        const delegates = await permissionManager.getAllDelegates(
-          transferManager.address,
-          PERM_TYPES.ADMIN
+        const delegateDetails = await getDelegateDetails(
+          permissionManager,
+          transferManager
         );
-        let delegateDetails = [];
-        for (const delegate of delegates) {
-          let details = await permissionManager.getDelegateDetails(delegate);
-          delegateDetails.push({ id: delegate, address: delegate, details });
-        }
         dispatch(loadManagers(delegateDetails));
       }
       dispatch(toggleWhitelistManagement(true));
@@ -212,24 +225,22 @@ export const removeAddressFromTransferManager = (delegate: Address) => async (
   );
 };
 
-export const addGeneralPermissionModule = () => async (
+export const archiveGeneralPermissionModule = () => async (
   dispatch: Function,
   getState: GetState
 ) => {
   const st: SecurityToken = getState().token.token.contract;
-  if (await st.getPermissionManager()) {
-    dispatch(toggleWhitelistManagement(true));
-    return;
-  }
   dispatch(
     ui.tx(
-      ['Enabling General Permissions Manager for General Transfer Manager'],
+      ['Disabling General Permissions Manager'],
       async () => {
-        await st.setPermissionManager();
+        const permissionManager = await st.getPermissionManager();
+        await st.archiveModule(permissionManager.address);
       },
-      'General Permissions Manager for General Transfer Manager Enabled',
+      'General Permissions Manager Disabled',
       () => {
-        dispatch(toggleWhitelistManagement(true));
+        dispatch(toggleWhitelistManagement(false));
+        dispatch(loadManagers([]));
       },
       undefined,
       undefined,
@@ -239,6 +250,50 @@ export const addGeneralPermissionModule = () => async (
   );
 };
 
+export const addGeneralPermissionModule = () => async (
+  dispatch: Function,
+  getState: GetState
+) => {
+  try {
+    const st: SecurityToken = getState().token.token.contract;
+    const permissionManager = await st.getPermissionManager();
+    const transferManager = await st.getTransferManager();
+    let moduleMetadata = {};
+    let delegateDetails = [];
+
+    if (permissionManager)
+      moduleMetadata = await st.getModule(permissionManager.address);
+
+    dispatch(
+      ui.tx(
+        ['Enabling General Permissions Manager for General Transfer Manager'],
+        async () => {
+          if (moduleMetadata.isArchived) {
+            await st.unarchiveModule(permissionManager.address);
+            delegateDetails = await getDelegateDetails(
+              permissionManager,
+              transferManager
+            );
+          } else {
+            await st.setPermissionManager();
+          }
+        },
+        'General Permissions Manager for General Transfer Manager Enabled',
+        () => {
+          dispatch(loadManagers(delegateDetails));
+          dispatch(toggleWhitelistManagement(true));
+        },
+        undefined,
+        undefined,
+        undefined,
+        true
+      )
+    );
+  } catch (e) {
+    console.log(e);
+  }
+};
+
 export const fetchWhitelist = () => async (
   dispatch: Function,
   getState: GetState

packages/polymath-issuer/src/pages/compliance/CompliancePage.js

@@ -44,6 +44,7 @@ import {
   fetchManagers,
   toggleWhitelistManagement,
   addGeneralPermissionModule,
+  archiveGeneralPermissionModule,
 } from '../../actions/compliance';
 import Progress from '../token/components/Progress';
 import AddInvestorForm, {
@@ -106,6 +107,7 @@ type DispatchProps = {|
   updateOwnershipPercentage: (percentage: number) => any,
   toggleFreeze: () => any,
   addGeneralPermissionModule: () => any,
+  archiveGeneralPermissionModule: () => any,
 |};
 
 const mapStateToProps = (state: RootState) => ({
@@ -138,6 +140,7 @@ const mapDispatchToProps = {
   fetchManagers,
   toggleWhitelistManagement,
   addGeneralPermissionModule,
+  archiveGeneralPermissionModule,
 };
 
 type Props = StateProps & DispatchProps;
@@ -189,7 +192,6 @@ class CompliancePage extends Component<Props, State> {
       this.setState({ percentage: this.props.percentage });
     }
     this.props.fetchManagers();
-    // TODO: check redux for whitelist enable
   }
 
   componentWillReceiveProps(nextProps) {
@@ -199,7 +201,6 @@ class CompliancePage extends Component<Props, State> {
     ) {
       this.setState({ percentage: nextProps.percentage });
     }
-    // TODO: check redux for whitelist enable
   }
 
   handleChangePages = pc => {
@@ -422,7 +423,7 @@ class CompliancePage extends Component<Props, State> {
     if (isToggled) {
       await this.props.addGeneralPermissionModule();
     } else {
-      this.props.toggleWhitelistManagement(false);
+      await this.props.archiveGeneralPermissionModule();
     }
   };
 

packages/polymath-js/src/contracts/SecurityToken.js

@@ -515,4 +515,16 @@ export default class SecurityToken extends Contract {
       0
     );
   }
+
+  async getModule(at: Address): Promise<Web3Receipt> {
+    return this._methods.getModule(at).call();
+  }
+
+  async arhiveModule(at: Address): Promise<Web3Receipt> {
+    return this._tx(this._methods.archiveModule(at));
+  }
+
+  async unarchiveModule(at: Address): Promise<Web3Receipt> {
+    return this._tx(this._methods.unarchiveModule(at));
+  }
 }