added giveStatus handler for better CM status stats

schema.graphql

@@ -20,6 +20,8 @@ type User @entity {
   reservationDayCount: BigInt!
   referralCount: BigInt!
   stakeCount: BigInt!
+  cmStatus: Boolean!
+  cmStatusInLaunch: Boolean!
 }
 
 type Reservation @entity {
@@ -78,7 +80,8 @@ type Global @entity {
   userCount: BigInt!
   reserverCount: BigInt!
   referrerCount: BigInt!
-  cmReferrerCount: BigInt!
+  cmStatusCount: BigInt!
+  cmStatusInLaunchCount: BigInt!
   reservationCount: BigInt!
   stakeCount: BigInt!
   stakerCount: BigInt!

src/mapping-liquidityTransformer.ts

@@ -1,7 +1,7 @@
 import { BigInt, ethereum } from "@graphprotocol/graph-ts"
 import {
   getOrCreateGlobal,
-  createUser
+  createUser,
 } from "./shared"
 import {
   ReferralAdded,
@@ -103,7 +103,8 @@ export function handleReferralAdded(event: ReferralAdded): void {
   referrer.referralCount = referrer.referralCount.plus(BigInt.fromI32(1))
   referrer.save()
   if (wasBelowCm && referrer.referralActualWei >= CM_REFERRER_THRESHOLD) {
-    global.cmReferrerCount = global.cmReferrerCount.plus(BigInt.fromI32(1))
+    referrer.cmStatusInLaunch = true
+    global.cmStatusInLaunchCount = global.cmStatusInLaunchCount.plus(BigInt.fromI32(1))
   }
   global.save()
 

src/mapping-wiseToken.ts

@@ -1,16 +1,26 @@
 import { BigInt, ethereum } from "@graphprotocol/graph-ts"
 import {
   getOrCreateGlobal,
-  createUser
+  createUser,
 } from "./shared"
 import {
-  StakeStart
+  GiveStatusCall,
+  StakeStart,
 } from "../generated/WiseToken/WiseToken"
 import {
   Stake,
-  User
+  User,
 } from "../generated/schema"
 
+export function handleGiveStatus (call: GiveStatusCall): void {
+  let referrer = new User(call.inputs._referrer.toHex())
+  referrer.cmStatus = true
+  referrer.save()
+
+  let global = getOrCreateGlobal();
+  global.cmStatusCount = global.cmStatusCount.plus(BigInt.fromI32(1))
+}
+
 export function handleStakeStart (event: StakeStart): void {
   let global = getOrCreateGlobal();
   global.stakeCount = global.stakeCount.plus(BigInt.fromI32(1))

src/shared.ts

@@ -1,14 +1,18 @@
 import {BigInt} from "@graphprotocol/graph-ts/index";
 import {Global, User} from "../generated/schema";
 
+export const ZERO = BigInt.fromI32(0)
+export const ONE = BigInt.fromI32(1)
+
 export function getOrCreateGlobal(): Global | null {
   let global = Global.load("0")
   if (global == null) {
     global = new Global("0")
-    global.userCount = BigInt.fromI32(0)
+    global.userCount = ZERO
     global.reserverCount = BigInt.fromI32(0)
     global.referrerCount = BigInt.fromI32(0)
-    global.cmReferrerCount = BigInt.fromI32(0)
+    global.cmStatusCount = BigInt.fromI32(0)
+    global.cmStatusInLaunchCount = BigInt.fromI32(0)
     global.reservationCount = BigInt.fromI32(0)
     global.stakeCount = BigInt.fromI32(0)
     global.stakerCount = BigInt.fromI32(0)
@@ -28,5 +32,7 @@ export function createUser(id: string): User | null {
   user.reservationDayCount = BigInt.fromI32(0)
   user.referralCount = BigInt.fromI32(0)
   user.stakeCount = BigInt.fromI32(0)
+  user.cmStatus = false
+  user.cmStatusInLaunch = false
   return user
 }

subgraph.template.yaml

@@ -21,6 +21,9 @@ dataSources:
       eventHandlers:
         - event: StakeStart(indexed bytes16,indexed address,indexed address,uint256,uint256,uint256,uint256,uint256,uint256)
           handler: handleStakeStart
+      callHandlers:
+        - function: giveStatus(address)
+          handler:  handleGiveStatus
       file: ./src/mapping-wiseToken.ts
   - kind: ethereum/contract
     name: LiquidityTransformer