Wire Manage Tokens / Edit Colony motion

src/amplifyClient.ts

@@ -131,6 +131,27 @@ export const mutations = {
       }
     }
   `,
+  updateColonyMetadata: /* GraphQL */ `
+    mutation UpdateColonyMetadata($input: UpdateColonyMetadataInput!) {
+      updateColonyMetadata(input: $input) {
+        id
+      }
+    }
+  `,
+  createColonyTokens: /* GraphQL */ `
+    mutation CreateColonyTokens($input: CreateColonyTokensInput!) {
+      createColonyTokens(input: $input) {
+        id
+      }
+    }
+  `,
+  deleteColonyTokens: /* GraphQL */ `
+    mutation DeleteColonyTokens($input: DeleteColonyTokensInput!) {
+      deleteColonyTokens(input: $input) {
+        id
+      }
+    }
+  `,
 };
 
 /*
@@ -283,6 +304,26 @@ export const queries = {
               newDescription
             }
           }
+          pendingColonyMetadata {
+            id
+            displayName
+            avatar
+            thumbnail
+            changelog {
+              transactionHash
+              oldDisplayName
+              newDisplayName
+              hasAvatarChanged
+              hasWhitelistChanged
+              haveTokensChanged
+            }
+            isWhitelistActivated
+            whitelistedAddresses
+            modifiedTokenAddresses {
+              added
+              removed
+            }
+          }
         }
       }
     }
@@ -302,6 +343,52 @@ export const queries = {
       }
     }
   `,
+  getColonyMetadata: /* GraphQL */ `
+    query GetColonyMetadata($id: ID!) {
+      getColonyMetadata(id: $id) {
+        id
+        displayName
+        avatar
+        thumbnail
+        changelog {
+          transactionHash
+          oldDisplayName
+          newDisplayName
+          hasAvatarChanged
+          hasWhitelistChanged
+          haveTokensChanged
+        }
+        isWhitelistActivated
+        whitelistedAddresses
+        modifiedTokenAddresses {
+          added
+          removed
+        }
+      }
+    }
+  `,
+  getTokenFromEverywhere: /* GraphQL */ `
+    query GetTokenFromEverywhere($input: TokenFromEverywhereArguments!) {
+      getTokenFromEverywhere(input: $input) {
+        items {
+          id
+        }
+      }
+    }
+  `,
+  getColony: /* GraphQL */ `
+    query GetColony($id: ID!) {
+      getColony(id: $id) {
+        colonyAddress: id
+        tokens {
+          items {
+            id
+            tokenAddress: tokenID
+          }
+        }
+      }
+    }
+  `,
 };
 
 export default (): void => {

src/handlers/motions/motionCreated/handlers/editColony.ts

@@ -0,0 +1,19 @@
+import { TransactionDescription } from 'ethers/lib/utils';
+import { ContractEvent, motionNameMapping } from '~types';
+import { getPendingMetadataDatabaseId } from '~utils';
+import { createMotionInDB } from '../helpers';
+
+export const handleEditColonyMotion = async (
+  event: ContractEvent,
+  { name }: TransactionDescription,
+): Promise<void> => {
+  const { transactionHash, contractAddress: colonyAddress } = event;
+  const pendingColonyMetadataId = getPendingMetadataDatabaseId(
+    colonyAddress,
+    transactionHash,
+  );
+  await createMotionInDB(event, {
+    type: motionNameMapping[name],
+    pendingColonyMetadataId,
+  });
+};

src/handlers/motions/motionCreated/handlers/index.ts

@@ -5,3 +5,4 @@ export { handleUnlockTokenMotion } from './unlockToken';
 export { handlePaymentMotion } from './payment';
 export { handleMoveFundsMotion } from './moveFunds';
 export { handleDomainEditReputationMotion } from './reputation';
+export { handleEditColonyMotion } from './editColony';

src/handlers/motions/motionCreated/handlers/manageDomain.ts

@@ -1,6 +1,6 @@
 import { TransactionDescription } from 'ethers/lib/utils';
 import { ContractEvent, motionNameMapping } from '~types';
-import { getPendingMotionDomainDatabaseId } from '~utils';
+import { getPendingMetadataDatabaseId } from '~utils';
 import { createMotionInDB } from '../helpers';
 
 export const handleManageDomainMotion = async (
@@ -9,7 +9,7 @@ export const handleManageDomainMotion = async (
 ): Promise<void> => {
   const { transactionHash, contractAddress: colonyAddress } = event;
   const { name } = parsedAction;
-  const pendingDomainMetadataId = getPendingMotionDomainDatabaseId(
+  const pendingDomainMetadataId = getPendingMetadataDatabaseId(
     colonyAddress,
     transactionHash,
   );

src/handlers/motions/motionCreated/motionCreated.ts

@@ -12,6 +12,7 @@ import {
   handlePaymentMotion,
   handleMoveFundsMotion,
   handleDomainEditReputationMotion,
+  handleEditColonyMotion,
 } from './handlers';
 
 export default async (event: ContractEvent): Promise<void> => {
@@ -70,6 +71,11 @@ export default async (event: ContractEvent): Promise<void> => {
         break;
       }
 
+      case ColonyOperations.EditColony: {
+        await handleEditColonyMotion(event, parsedAction);
+        break;
+      }
+
       default: {
         break;
       }

src/handlers/motions/motionFinalized/helpers.ts

@@ -1,10 +1,23 @@
 import { BigNumber } from 'ethers';
 import { TransactionDescription } from 'ethers/lib/utils';
 
-import { ColonyOperations, MotionQuery, MotionVote, StakerReward } from '~types';
-import { getDomainDatabaseId, getVotingClient, verbose } from '~utils';
+import {
+  ColonyMetadata,
+  ColonyOperations,
+  MotionQuery,
+  MotionVote,
+  StakerReward,
+} from '~types';
+import {
+  getColonyFromDB,
+  getDomainDatabaseId,
+  getExistingTokenAddresses,
+  getVotingClient,
+  updateColonyTokens,
+  verbose,
+} from '~utils';
 import networkClient from '~networkClient';
-import { mutate } from '~amplifyClient';
+import { query, mutate } from '~amplifyClient';
 
 export const getStakerReward = async (
   motionId: string,
@@ -70,8 +83,15 @@ const getParsedActionFromDomainMotion = async (
   }
 };
 
-export const linkPendingDomainMetadataWithDomain = async (action: string, colonyAddress: string, finalizedMotion: MotionQuery): Promise<void> => {
-  const parsedDomainAction = await getParsedActionFromDomainMotion(action, colonyAddress);
+export const linkPendingDomainMetadataWithDomain = async (
+  action: string,
+  colonyAddress: string,
+  finalizedMotion: MotionQuery,
+): Promise<void> => {
+  const parsedDomainAction = await getParsedActionFromDomainMotion(
+    action,
+    colonyAddress,
+  );
   if (parsedDomainAction?.name === ColonyOperations.AddDomain) {
     const colonyClient = await networkClient.getColonyClient(colonyAddress);
     const domainCount = await colonyClient.getDomainCount();
@@ -96,3 +116,84 @@ export const linkPendingDomainMetadataWithDomain = async (action: string, colony
     });
   }
 };
+
+export const linkPendingColonyMetadataWithColony = async (
+  pendingColonyMetadata: ColonyMetadata,
+  colonyAddress: string,
+): Promise<void> => {
+  const currentColonyMetadata = await query<ColonyMetadata>(
+    'getColonyMetadata',
+    {
+      id: colonyAddress,
+    },
+  );
+
+  if (!currentColonyMetadata) {
+    console.error(
+      `Could not find the current metadata for the colony: ${colonyAddress}. This is a bug and should be investigated.`,
+    );
+    return;
+  }
+
+  const {
+    haveTokensChanged,
+    hasAvatarChanged,
+    hasWhitelistChanged,
+    newDisplayName,
+    oldDisplayName,
+  } = pendingColonyMetadata.changelog?.[0] ?? {};
+
+  const updatedMetadata = {
+    ...currentColonyMetadata,
+  };
+
+  /*
+   * Here, we update metadata as granularly as possible so that we don't overwrite state changes that occured
+   * after this motion was created.
+   */
+
+  if (hasAvatarChanged) {
+    // If avatar has changed, update avatar and thumbnail
+    updatedMetadata.avatar = pendingColonyMetadata.avatar;
+    updatedMetadata.thumbnail = pendingColonyMetadata.thumbnail;
+  }
+
+  if (hasWhitelistChanged) {
+    // If whitelist has changed, update whitelistedAddresses and isWhitelistActivated
+    updatedMetadata.isWhitelistActivated =
+      pendingColonyMetadata.isWhitelistActivated;
+
+    updatedMetadata.whitelistedAddresses =
+      pendingColonyMetadata.whitelistedAddresses;
+  }
+
+  if (newDisplayName !== oldDisplayName) {
+    // If displayName has changed, update displayName
+    updatedMetadata.displayName = pendingColonyMetadata.displayName;
+  }
+
+  if (haveTokensChanged && pendingColonyMetadata.modifiedTokenAddresses) {
+    // If tokens have changed, update colony tokens
+    const colony = await getColonyFromDB(colonyAddress);
+
+    if (colony) {
+      const existingTokenAddresses = getExistingTokenAddresses(colony);
+
+      await updateColonyTokens(
+        colony,
+        existingTokenAddresses,
+        pendingColonyMetadata.modifiedTokenAddresses,
+      );
+    }
+  }
+
+  await mutate('updateColonyMetadata', {
+    input: {
+      ...updatedMetadata,
+      changelog: [
+        ...(currentColonyMetadata.changelog ?? []),
+        pendingColonyMetadata.changelog?.[0],
+      ],
+    },
+  });
+};

src/handlers/motions/motionFinalized/motionFinalized.ts

@@ -9,7 +9,12 @@ import {
   updateMotionInDB,
   getMessageKey,
 } from '../helpers';
-import { getStakerReward, linkPendingDomainMetadataWithDomain } from './helpers';
+
+import {
+  getStakerReward,
+  linkPendingColonyMetadataWithColony,
+  linkPendingDomainMetadataWithDomain,
+} from './helpers';
 
 export default async (event: ContractEvent): Promise<void> => {
   const {
@@ -45,7 +50,9 @@ export default async (event: ContractEvent): Promise<void> => {
       motionData,
     } = finalizedMotion;
 
-    const yayWon = BigNumber.from(yayVotes).gt(nayVotes) || (Number(yayPercentage) > Number(nayPercentage));
+    const yayWon =
+      BigNumber.from(yayVotes).gt(nayVotes) ||
+      Number(yayPercentage) > Number(nayPercentage);
 
     /*
      * pendingDomainMetadata is a motion data prop that we use to store the metadata of a Domain that COULD be created/edited
@@ -54,7 +61,18 @@ export default async (event: ContractEvent): Promise<void> => {
      * a new DomainMetadata with the corresponding Domain item id.
      */
     if (finalizedMotion.pendingDomainMetadata && yayWon) {
-      await linkPendingDomainMetadataWithDomain(action, colonyAddress, finalizedMotion);
+      await linkPendingDomainMetadataWithDomain(
+        action,
+        colonyAddress,
+        finalizedMotion,
+      );
+    }
+
+    if (finalizedMotion.pendingColonyMetadata && yayWon) {
+      await linkPendingColonyMetadataWithColony(
+        finalizedMotion.pendingColonyMetadata,
+        colonyAddress,
+      );
     }
 
     const updatedStakerRewards = await Promise.all(

src/types/index.ts

@@ -105,6 +105,7 @@ export enum ColonyActionType {
   MoveFundsMotion = 'MOVE_FUNDS_MOTION',
   EmitDomainReputationPenaltyMotion = 'EMIT_DOMAIN_REPUTATION_PENALTY_MOTION',
   EmitDomainReputationRewardMotion = 'EMIT_DOMAIN_REPUTATION_REWARD_MOTION',
+  ColonyEditMotion = 'COLONY_EDIT_MOTION',
 }
 
 export type ColonyActionHandler = (event: ContractEvent) => Promise<void>;
@@ -136,3 +137,16 @@ interface VotingReputationParams {
   revealPeriod: string;
   escalationPeriod: string;
 }
+
+export interface ColonyQuery {
+  colonyAddress: string;
+  tokens:
+    | {
+        items: Array<{
+          id: string;
+          tokenAddress: string;
+        } | null>;
+      }
+    | null
+    | undefined;
+}