Merge pull request #53 from aavegotchi/feat/alchemicaToHarvest

feat: alchemica to harvest

Author: cinnabarhorse

abis/RealmDiamond.json

@@ -536,5 +536,27 @@
         ],
         "name": "ParcelWhitelistSet",
         "type": "event"
+    },
+    {
+        "anonymous": false,
+        "inputs": [
+            {
+                "indexed": false,
+                "name": "_tokenId",
+                "type": "uint256"
+            },
+            {
+                "indexed": false,
+                "name": "_round",
+                "type": "uint256"
+            },
+            {
+                "indexed": false,
+                "name": "_alchemicas",
+                "type": "uint256[]"
+            }
+        ],
+        "name": "SurveyParcel",
+        "type": "event"
     }
 ]

schema.graphql

@@ -416,6 +416,9 @@ type Parcel @entity {
     size: BigInt
 
     accessRights: [ParcelAccessRight!]! @derivedFrom(field: "parcel")
+
+    remainingAlchemica: [BigInt!]!
+    surveyRound: Int
 }
 
 type ParcelAccessRight @entity {

src/helper/realm.ts

@@ -13,6 +13,7 @@ import {
     NFTDisplayStatusUpdated,
     ParcelAccessRightSet,
     RealmDiamond,
+    SurveyParcel,
     Transfer,
     UnequipInstallation,
     UnequipTile,
@@ -39,7 +40,7 @@ import {
     UnequipInstallationEvent,
     UnequipTileEvent,
 } from "../../generated/schema";
-import { REALM_DIAMOND, StatCategory } from "./constants";
+import { BIGINT_ZERO, REALM_DIAMOND, StatCategory } from "./constants";
 import { getStat } from "./stats";
 
 export const getOrCreateParcel = (realmId: BigInt): Parcel => {
@@ -49,6 +50,13 @@ export const getOrCreateParcel = (realmId: BigInt): Parcel => {
         parcel = new Parcel(id);
         parcel.equippedInstallations = new Array<string>();
         parcel.equippedTiles = new Array<string>();
+        parcel.remainingAlchemica = [
+            BIGINT_ZERO,
+            BIGINT_ZERO,
+            BIGINT_ZERO,
+            BIGINT_ZERO,
+        ];
+        parcel.surveyRound = 0;
         parcel = updateParcelInfo(parcel);
     }
     return parcel;

src/mappings/realm.ts

@@ -18,6 +18,7 @@ import {
     EventCancelled,
     EventPriorityAndDurationUpdated,
     ParcelWhitelistSet,
+    SurveyParcel,
 } from "../../generated/RealmDiamond/RealmDiamond";
 import { ParcelWhitelistSetEvent } from "../../generated/schema";
 import { BIGINT_ONE, REALM_DIAMOND, StatCategory } from "../helper/constants";
@@ -158,6 +159,14 @@ export function handleAlchemicaClaimed(event: AlchemicaClaimed): void {
     // set last claim alchemica
     let parcel = getOrCreateParcel(event.params._realmId);
     parcel.lastClaimedAlchemica = event.block.timestamp;
+
+    let alchemicas = parcel.remainingAlchemica;
+    let entry = alchemicas[event.params._alchemicaType.toI32()];
+    alchemicas[event.params._alchemicaType.toI32()] = entry.minus(
+        event.params._amount
+    );
+    parcel.remainingAlchemica = alchemicas;
+
     parcel.save();
 
     // stats
@@ -540,3 +549,16 @@ export function handleParcelWhitelistSet(event: ParcelWhitelistSet): void {
     parEntity.whitelistId = event.params._whitelistId.toI32();
     parEntity.save();
 }
+
+export function handleSurveyParcel(event: SurveyParcel): void {
+    let entity = getOrCreateParcel(event.params._tokenId);
+
+    let alchemica = entity.remainingAlchemica;
+    for (let i = 0; i < event.params._alchemicas.length; i++) {
+        alchemica[i] = alchemica[i].plus(event.params._alchemicas[i]);
+    }
+
+    entity.surveyRound = entity.surveyRound + 1;
+    entity.remainingAlchemica = alchemica;
+    entity.save();
+}

subgraph.yaml

@@ -64,6 +64,8 @@ dataSources:
                 handler: handleBounceGateEventPriorityAndDurationUpdated
               - event: ParcelWhitelistSet(uint256,uint256,uint256)
                 handler: handleParcelWhitelistSet
+              - event: SurveyParcel(uint256,uint256,uint256[])
+                handler: handleSurveyParcel
           file: ./src/mappings/realm.ts
     - kind: ethereum/contract
       name: TileDiamond

tests/realm/handleSurveyParcel.test.ts

@@ -0,0 +1,106 @@
+import { ethereum } from "@graphprotocol/graph-ts";
+import {
+    afterAll,
+    assert,
+    beforeAll,
+    clearStore,
+    createMockedFunction,
+    describe,
+    newMockEvent,
+    test,
+} from "matchstick-as";
+import {
+    MintParcel,
+    SurveyParcel,
+} from "../../generated/RealmDiamond/RealmDiamond";
+import {
+    BIGINT_EIGHT,
+    BIGINT_FIVE,
+    BIGINT_FOUR,
+    BIGINT_ONE,
+    BIGINT_SEVEN,
+    BIGINT_SIX,
+    BIGINT_THREE,
+    BIGINT_TWO,
+    REALM_DIAMOND,
+} from "../../src/helper/constants";
+import { handleMintParcel, handleSurveyParcel } from "../../src/mappings/realm";
+
+let mockEvent = newMockEvent();
+let realmId = BIGINT_ONE;
+let round = BIGINT_TWO;
+describe("handleSurveyParcel", () => {
+    beforeAll(() => {
+        // prepare event
+        let event = new SurveyParcel(
+            mockEvent.address,
+            mockEvent.logIndex,
+            mockEvent.transactionLogIndex,
+            mockEvent.logType,
+            mockEvent.block,
+            mockEvent.transaction,
+            mockEvent.parameters,
+            null
+        );
+
+        event.parameters.push(
+            new ethereum.EventParam(
+                "_tokenId",
+                ethereum.Value.fromUnsignedBigInt(realmId)
+            )
+        );
+
+        event.parameters.push(
+            new ethereum.EventParam(
+                "_round",
+                ethereum.Value.fromUnsignedBigInt(round)
+            )
+        );
+
+        event.parameters.push(
+            new ethereum.EventParam(
+                "_alchemicas",
+                ethereum.Value.fromUnsignedBigIntArray([
+                    BIGINT_TWO,
+                    BIGINT_TWO,
+                    BIGINT_TWO,
+                    BIGINT_TWO,
+                ])
+            )
+        );
+
+        // mock getParcelInfo
+        let tuple: ethereum.Tuple = changetype<ethereum.Tuple>([
+            ethereum.Value.fromString("A"),
+            ethereum.Value.fromString("B"),
+            ethereum.Value.fromAddress(REALM_DIAMOND),
+            ethereum.Value.fromUnsignedBigInt(BIGINT_ONE),
+            ethereum.Value.fromUnsignedBigInt(BIGINT_TWO),
+            ethereum.Value.fromUnsignedBigInt(BIGINT_THREE),
+            ethereum.Value.fromUnsignedBigInt(BIGINT_FOUR),
+            ethereum.Value.fromUnsignedBigIntArray([
+                BIGINT_FIVE,
+                BIGINT_SIX,
+                BIGINT_SEVEN,
+                BIGINT_EIGHT,
+            ]),
+        ]);
+        createMockedFunction(
+            REALM_DIAMOND,
+            "getParcelInfo",
+            "getParcelInfo(uint256):((string,string,address,uint256,uint256,uint256,uint256,uint256[4]))"
+        )
+            .withArgs([ethereum.Value.fromUnsignedBigInt(realmId)])
+            .returns([ethereum.Value.fromTuple(tuple)]);
+
+        handleSurveyParcel(event);
+    });
+
+    test("it should has remainingAlchemica field", () => {
+        assert.fieldEquals("Parcel", "1", "remainingAlchemica", "[2, 2, 2, 2]");
+    });
+
+    afterAll(() => {
+        clearStore();
+    });
+});