missing credits update on opt in

OriginProtocol · Dec 6, 2024 · 0aec5e0 · 0aec5e0
1 parent a7b465d
commit 0aec5e0
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/templates/otoken/otoken.ts b/src/templates/otoken/otoken.ts
@@ -696,6 +696,11 @@ export const createOTokenProcessor = (params: {
 
     owner.delegatedTo = null
     if (option === RebasingOption.OptIn) {
+      const afterHighResUpgrade = block.header.height >= (params.Upgrade_CreditsBalanceOfHighRes ?? 0)
+      const otokenContract = new otoken.Contract(ctx, block.header, params.otokenAddress)
+      owner.credits = afterHighResUpgrade
+        ? await otokenContract.creditsBalanceOfHighres(owner.address).then((c) => c._0)
+        : await otokenContract.creditsBalanceOf(owner.address).then((c) => c._0 * 1000000000n)
       rebaseOption.status = RebasingOption.OptIn
       owner.rebasingOption = RebasingOption.OptIn
       otokenObject.nonRebasingSupply -= owner.balance