Cleanup: Remove all references to ColonyStake model

src/handlers/motions/helpers.ts

@@ -139,10 +139,3 @@ export const getMessageKey = (
 ): string => {
   return `${transactionHash}${logIndex}`;
 };
-
-export const getColonyStakeId = (
-  userAddress: string,
-  colonyAddress: string,
-): string => {
-  return `${userAddress}_${colonyAddress}`;
-};

src/handlers/motions/motionRewardClaimed/helpers.ts

@@ -1,26 +1,16 @@
-import { BigNumber } from 'ethers';
-import { mutate, query } from '~amplifyClient';
+import { mutate } from '~amplifyClient';
 import {
-  GetColonyStakeDocument,
-  GetColonyStakeQuery,
-  GetColonyStakeQueryVariables,
   StakerReward,
   UpdateColonyDocument,
   UpdateColonyMutation,
   UpdateColonyMutationVariables,
-  UpdateColonyStakeDocument,
-  UpdateColonyStakeMutation,
-  UpdateColonyStakeMutationVariables,
   UpdateUserStakeDocument,
   UpdateUserStakeMutation,
   UpdateUserStakeMutationVariables,
-  UserMotionStakes,
 } from '~graphql';
 import { getColonyFromDB, output } from '~utils';
 import { getUserStakeDatabaseId } from '~utils/stakes';
 
-import { getColonyStakeId } from '../helpers';
-
 export const getUpdatedStakerRewards = (
   stakerRewards: StakerReward[],
   stakerAddress: string,
@@ -106,42 +96,8 @@ export const updateColonyUnclaimedStakes = async (
  */
 export const reclaimUserStake = async (
   userAddress: string,
-  colonyAddress: string,
-  reclaimAmount: BigNumber,
   motionTransactionHash: string,
 ): Promise<void> => {
-  const colonyStakeId = getColonyStakeId(userAddress, colonyAddress);
-  const { data } =
-    (await query<GetColonyStakeQuery, GetColonyStakeQueryVariables>(
-      GetColonyStakeDocument,
-      {
-        colonyStakeId,
-      },
-    )) ?? {};
-
-  if (!data?.getColonyStake) {
-    output(
-      `Could not find user stake for user ${userAddress} in colony ${colonyAddress}. This is a bug and should be investigated.`,
-    );
-    return;
-  }
-
-  const totalAmount = data.getColonyStake.totalAmount;
-  let updatedAmount = BigNumber.from(totalAmount).sub(reclaimAmount);
-
-  // Should never be negative, but just in case.
-  if (updatedAmount.isNegative()) {
-    updatedAmount = BigNumber.from(0);
-  }
-
-  await mutate<UpdateColonyStakeMutation, UpdateColonyStakeMutationVariables>(
-    UpdateColonyStakeDocument,
-    {
-      colonyStakeId,
-      totalAmount: updatedAmount.toString(),
-    },
-  );
-
   // Update user stake status
   await mutate<UpdateUserStakeMutation, UpdateUserStakeMutationVariables>(
     UpdateUserStakeDocument,
@@ -153,21 +109,3 @@ export const reclaimUserStake = async (
     },
   );
 };
-
-export const getUserMotionStake = (
-  usersStakes: UserMotionStakes[],
-  userAddress: string,
-): BigNumber => {
-  const userStakes = usersStakes.find(({ address }) => address === userAddress);
-
-  if (!userStakes) {
-    output(
-      `Could not find the stakes for user ${userAddress}. This is a bug and should be investigated.`,
-    );
-    return BigNumber.from(0);
-  }
-
-  return BigNumber.from(userStakes.stakes.raw.yay).add(
-    userStakes.stakes.raw.nay,
-  );
-};

src/handlers/motions/motionRewardClaimed/motionRewardClaimed.ts

@@ -10,7 +10,6 @@ import {
   updateColonyUnclaimedStakes,
   reclaimUserStake,
   getUpdatedStakerRewards,
-  getUserMotionStake,
 } from './helpers';
 import { ColonyMotion } from '~graphql';
 import { ExtensionEventListener } from '~eventListeners';
@@ -41,9 +40,8 @@ export const handleMotionRewardClaimed: EventHandler = async (
   const claimedMotion = await getMotionFromDB(motionDatabaseId);
 
   if (claimedMotion) {
-    const { stakerRewards, usersStakes } = claimedMotion;
+    const { stakerRewards } = claimedMotion;
 
-    const userStake = getUserMotionStake(usersStakes, staker);
     const updatedStakerRewards = getUpdatedStakerRewards(stakerRewards, staker);
 
     const newMotionMessages = [
@@ -66,11 +64,6 @@ export const handleMotionRewardClaimed: EventHandler = async (
       motionDatabaseId,
       updatedStakerRewards,
     );
-    await reclaimUserStake(
-      staker,
-      colonyAddress,
-      userStake,
-      updatedMotionData.transactionHash,
-    );
+    await reclaimUserStake(staker, updatedMotionData.transactionHash);
   }
 };

src/handlers/motions/motionStaked/helpers.ts

@@ -2,31 +2,22 @@ import { BigNumber, constants } from 'ethers';
 import { MotionVote, MotionEvents } from '~types';
 import {
   ColonyMotion,
-  CreateColonyStakeDocument,
-  CreateColonyStakeMutation,
-  CreateColonyStakeMutationVariables,
   CreateMotionMessageInput,
   CreateUserStakeDocument,
   CreateUserStakeMutation,
   CreateUserStakeMutationVariables,
-  GetColonyStakeDocument,
-  GetColonyStakeQuery,
-  GetColonyStakeQueryVariables,
   GetUserStakeDocument,
   GetUserStakeQuery,
   GetUserStakeQueryVariables,
   MotionStakes,
-  UpdateColonyStakeDocument,
-  UpdateColonyStakeMutation,
-  UpdateColonyStakeMutationVariables,
   UpdateUserStakeDocument,
   UpdateUserStakeMutation,
   UpdateUserStakeMutationVariables,
   UserMotionStakes,
 } from '~graphql';
 import { mutate, query } from '~amplifyClient';
 import { getUserStakeDatabaseId } from '~utils/stakes';
-import { getMotionSide, getColonyStakeId } from '../helpers';
+import { getMotionSide } from '../helpers';
 
 export const getRequiredStake = (
   skillRep: BigNumber,
@@ -297,48 +288,6 @@ export const getUpdatedMessages = ({
   return updatedMessages;
 };
 
-/**
- * If it's the first time a user has staked in a colony, we create a Colony Stake record for the user,
- * else we update the amount they've currently got staked in the colony.
- */
-export const updateUserColonyStake = async (
-  userAddress: string,
-  colonyAddress: string,
-  stakeAmount: BigNumber,
-): Promise<void> => {
-  const colonyStakeId = getColonyStakeId(userAddress, colonyAddress);
-  const { data } =
-    (await query<GetColonyStakeQuery, GetColonyStakeQueryVariables>(
-      GetColonyStakeDocument,
-      {
-        colonyStakeId,
-      },
-    )) ?? {};
-
-  if (data?.getColonyStake) {
-    const { totalAmount } = data?.getColonyStake;
-    const updatedTotal = BigNumber.from(totalAmount).add(stakeAmount);
-
-    await mutate<UpdateColonyStakeMutation, UpdateColonyStakeMutationVariables>(
-      UpdateColonyStakeDocument,
-      {
-        totalAmount: updatedTotal.toString(),
-        colonyStakeId,
-      },
-    );
-  } else {
-    await mutate<CreateColonyStakeMutation, CreateColonyStakeMutationVariables>(
-      CreateColonyStakeDocument,
-      {
-        totalAmount: stakeAmount.toString(),
-        colonyStakeId,
-        colonyAddress,
-        userAddress,
-      },
-    );
-  }
-};
-
 /**
  * Function to update the user stake following staking on a motion
  */

src/handlers/motions/motionStaked/motionStaked.ts

@@ -12,7 +12,6 @@ import {
   getUpdatedMessages,
   updateMotionInDB,
   getMessageKey,
-  updateUserColonyStake,
   updateUserStake,
 } from '../helpers';
 
@@ -29,8 +28,6 @@ export const handleMotionStaked: EventHandler = async (
   } = event;
   const { colonyAddress } = listener as ExtensionEventListener;
 
-  await updateUserColonyStake(staker, colonyAddress, amount);
-
   const votingClient = await getVotingClient(colonyAddress);
 
   if (!votingClient) {