fix: rework node/farm certification

Author: DylanVerstraete

src/mappings/farms.ts

@@ -2,7 +2,7 @@ import {
   EventHandlerContext,
 } from "@subsquid/substrate-processor";
 import { Farm, FarmCertification, PublicIp } from "../model";
-import { TfgridModuleFarmStoredEvent, TfgridModuleFarmDeletedEvent, TfgridModuleFarmUpdatedEvent, TfgridModuleFarmPayoutV2AddressRegisteredEvent, TfgridModuleFarmCertificationSetEvent } from "../types/events";
+import { TfgridModuleFarmStoredEvent, TfgridModuleFarmDeletedEvent, TfgridModuleFarmUpdatedEvent, TfgridModuleFarmPayoutV2AddressRegisteredEvent, TfgridModuleFarmCertificationSetEvent, TfgridModuleFarmingPolicySetEvent } from "../types/events";
 import * as v63 from '../types/v63'
 
 export async function farmStored(ctx: EventHandlerContext) {
@@ -68,17 +68,29 @@ export async function farmStored(ctx: EventHandlerContext) {
 export async function farmUpdated(ctx: EventHandlerContext) {
   const farmUpdatedEvent = new TfgridModuleFarmUpdatedEvent(ctx)
 
+  let certification = FarmCertification.NotCertified
+
   let farmUpdatedEventParsed
   if (farmUpdatedEvent.isV9) {
     farmUpdatedEventParsed = farmUpdatedEvent.asV9
   } else if (farmUpdatedEvent.isV50) {
     farmUpdatedEventParsed = farmUpdatedEvent.asV50
   } else if (farmUpdatedEvent.isV63) {
     farmUpdatedEventParsed = farmUpdatedEvent.asV63
+    switch(farmUpdatedEvent.asV101.certification.__kind) {
+      case "Gold": {
+        certification = FarmCertification.Gold
+      }
+    }
   } else if (farmUpdatedEvent.isV101) {
     let eventValue = ctx.event.params[0].value as v63.Farm
     eventValue.dedicatedFarm = false
     farmUpdatedEventParsed = farmUpdatedEvent.asV101
+    switch(farmUpdatedEvent.asV101.certification.__kind) {
+      case "Gold": {
+        certification = FarmCertification.Gold
+      }
+    }
   }
 
   if (!farmUpdatedEventParsed) return
@@ -90,6 +102,7 @@ export async function farmUpdated(ctx: EventHandlerContext) {
   savedFarm.name = farmUpdatedEventParsed.name.toString()
   savedFarm.twinID = farmUpdatedEventParsed.twinId
   savedFarm.pricingPolicyID = farmUpdatedEventParsed.pricingPolicyId
+  savedFarm.certification = certification
 
   await farmUpdatedEventParsed.publicIps.forEach(async ip => {
     if (ip.ip.toString().indexOf('\x00') >= 0) {

src/mappings/nodes.ts

@@ -36,6 +36,7 @@ export async function nodeStored(ctx: EventHandlerContext) {
 
   newNode.created = Number(nodeEvent.created)
   newNode.farmingPolicyId = nodeEvent.farmingPolicyId
+  newNode.certification = NodeCertification.Diy
 
   const newLocation = new Location()
   newLocation.id = ctx.event.id
@@ -45,45 +46,11 @@ export async function nodeStored(ctx: EventHandlerContext) {
 
   newNode.location = newLocation
 
-  if (node.isV28) {
-    const nodeAsV28 = node.asV28
-    if (nodeAsV28.certificationType) {
-      const certificationTypeAsString = nodeAsV28.certificationType.__kind.toString()
-      let certType = NodeCertification.Diy
-      switch (certificationTypeAsString) {
-        case 'Diy':
-          certType = NodeCertification.Diy
-          break
-        case 'Certified':
-          certType = NodeCertification.Certified
-          break
-      }
-      newNode.certification = certType
-    } else {
-      newNode.certification = NodeCertification.Diy
-    }
-  }
-
   if (node.isV43) {
     const nodeAsV43 = node.asV43
     newNode.secure = nodeAsV43.secureBoot ? true : false
     newNode.virtualized = nodeAsV43.virtualized ? true : false
     newNode.serialNumber = nodeAsV43.serialNumber.toString()
-    if (nodeAsV43.certificationType) {
-      const certificationTypeAsString = nodeAsV43.certificationType.__kind.toString()
-      let certType = NodeCertification.Diy
-      switch (certificationTypeAsString) {
-        case 'Diy':
-          certType = NodeCertification.Diy
-          break
-        case 'Certified':
-          certType = NodeCertification.Certified
-          break
-      }
-      newNode.certification = certType
-    } else {
-      newNode.certification = NodeCertification.Diy
-    }
   }
 
   if (node.isV101 || node.isV63) {
@@ -99,21 +66,6 @@ export async function nodeStored(ctx: EventHandlerContext) {
     newNode.virtualized = nodeEvent.virtualized ? true : false
     newNode.serialNumber = nodeEvent.serialNumber.toString()
     newNode.connectionPrice = nodeEvent.connectionPrice
-    if (nodeEvent.certification) {
-      const certificationTypeAsString = nodeEvent.certification.__kind.toString()
-      let certType = NodeCertification.Diy
-      switch (certificationTypeAsString) {
-        case 'Diy':
-          certType = NodeCertification.Diy
-          break
-        case 'Certified':
-          certType = NodeCertification.Certified
-          break
-      }
-      newNode.certification = certType
-    } else {
-      newNode.certification = NodeCertification.Diy
-    }
   }
 
   await ctx.store.save<Node>(newNode)

src/processor.ts

@@ -47,6 +47,7 @@ processor.addEventHandler('tfgridModule.FarmUpdated', ctx => farmUpdated(ctx));
 processor.addEventHandler('tfgridModule.FarmDeleted', ctx => farmDeleted(ctx));
 processor.addEventHandler('tfgridModule.FarmPayoutV2AddressRegistered', ctx => farmPayoutV2AddressRegistered(ctx));
 processor.addEventHandler('tfgridModule.FarmCertificationSet', ctx => farmCertificationSet(ctx));
+processor.addEventHandler('tfgridModule.FarmingPolicySet', ctx => farmCertificationSet(ctx));
 
 processor.addEventHandler('tfgridModule.EntityStored', ctx => entityStored(ctx));
 processor.addEventHandler('tfgridModule.EntityUpdated', ctx => entityUpdated(ctx));