feat: added memo handler to contributions

project.yaml

@@ -28,6 +28,11 @@ dataSources:
           filter:
             module: crowdloan
             method: Contributed
+        - handler: handleCrowdloanMemo
+          kind: substrate/EventHandler
+          filter:
+            module: crowdloan
+            method: MemoUpdated
         # - handler: handleCrowdloanAllRefunded
         #   kind: substrate/EventHandler
         #   filter:
@@ -90,6 +95,11 @@ dataSources:
           filter:
             module: crowdloan
             method: Contributed
+        - handler: handleCrowdloanMemo
+          kind: substrate/EventHandler
+          filter:
+            module: crowdloan
+            method: MemoUpdated
         # - handler: handleCrowdloanAllRefunded
         #   kind: substrate/EventHandler
         #   filter:

schema.graphql

@@ -75,6 +75,7 @@ type CrowdloanSequence @entity {
 type Contribution @entity {
   id: ID!
   account: String! @index
+  memo: String @index
   parachain: Parachain!
   fund: Crowdloan!
   amount: BigInt! @index

src/handlers/parachain-handlers.ts

@@ -5,6 +5,7 @@ import { Chronicle } from '../types/models/Chronicle';
 import { ChronicleKey } from '../constants';
 import { parseNumber } from '../utils';
 import { CrowdloanStatus } from '../types';
+import { Contribution } from '../types/index';
 
 interface ParaInfo {
   manager: string;
@@ -69,6 +70,24 @@ export const onCrowdloanContributed = async (substrateEvent: SubstrateEvent) =>
   await Storage.save('Contribution', contribution);
 };
 
+export const onCrowdloanMemo = async (substrateEvent: SubstrateEvent) => {
+  const { event, block } = substrateEvent;
+  const { block: rawBlock } = block;
+
+  const blockNum = rawBlock.header.number.toNumber();
+  const [contributor, fundIdx, memo] = event.data.toJSON() as [string, number, string];
+  await Storage.ensureParachain(fundIdx);
+  await Storage.ensureFund(fundIdx);
+
+  const contributions = await Contribution.getByAccount(contributor);
+  const latestContributionBeforeMemo = contributions.filter((contrib) => contrib.blockNum <= blockNum).sort((a, b) => a.blockNum - b.blockNum).pop();
+
+  if (!latestContributionBeforeMemo) return;
+  latestContributionBeforeMemo.memo = memo;
+  logger.info(`Adding memo ${memo} for contributor ${contributor} at block ${blockNum}`);
+  await Storage.save('Contribution', latestContributionBeforeMemo);
+};
+
 export const onCrowdloanDissolved = async (substrateEvent: SubstrateEvent) => {
   const { event, block } = substrateEvent;
   const { timestamp, block: rawBlock } = block;

src/mappings/mappingHandlers.ts

@@ -7,6 +7,7 @@ import {
   onParachainRegistered,
   onCrowdloanCreated,
   onCrowdloanContributed,
+  onCrowdloanMemo,
   onCrowdloanDissolved,
 } from '../handlers/parachain-handlers';
 import { onSlotsLeased, onNewLeasePeriod } from '../handlers/lease-handlers';
@@ -33,6 +34,11 @@ export async function handleCrowdloanContributed(substrateEvent: SubstrateEvent)
 //   await onCrowdloanAllRefunded(substrateEvent);
 // }
 
+export async function handleCrowdloanMemo(substrateEvent: SubstrateEvent): Promise<void> {
+  getEventLogger(substrateEvent);
+  await onCrowdloanMemo(substrateEvent);
+}
+
 export async function handleCrowdloanDissolved(substrateEvent: SubstrateEvent): Promise<void> {
   getEventLogger(substrateEvent);
   await onCrowdloanDissolved(substrateEvent);

src/types/models/Contribution.ts

@@ -14,6 +14,8 @@ export class Contribution implements Entity {
 
     public account: string;
 
+    public memo?: string;
+
     public parachainId: string;
 
     public fundId: string;
@@ -53,6 +55,13 @@ export class Contribution implements Entity {
 
     }
 
+    static async getByMemo(memo: string): Promise<Contribution[] | undefined>{
+
+      const records = await store.getByField('Contribution', 'memo', memo);
+      return records.map(record => Contribution.create(record));
+
+    }
+
     static async getByParachainId(parachainId: string): Promise<Contribution[] | undefined>{
 
       const records = await store.getByField('Contribution', 'parachainId', parachainId);