giveStatus fix

wise-foundation · Dec 18, 2020 · ce79d2b · ce79d2b
1 parent ec856e7
commit ce79d2b
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/mapping-wiseToken.ts b/src/mapping-wiseToken.ts
@@ -15,17 +15,22 @@ import {
 } from "../generated/schema"
 
 export function handleGiveStatus (call: GiveStatusCall): void {
-  let referrer = createUser(call.inputs._referrer.toHex())
+  let referrerID = call.inputs._referrer.toHex()
+  let referrer = User.load(referrerID)
+  if (referrer == null) {
+    referrer = createUser(referrerID)
+  }
   if (referrer.cmStatus === false) {
     referrer.cmStatus = true
     referrer.cmStatusInLaunch = true
-    referrer.save()
 
     let global = getOrCreateGlobal()
     global.cmStatusCount = global.cmStatusCount.plus(ONE)
     global.cmStatusInLaunchCount = global.cmStatusInLaunchCount.plus(ONE)
     global.save()
   }
+
+  referrer.save()
 }
 
 export function handleStakeStart (event: StakeStart): void {