From c0995e8552636820e64bce783f3d0c6c7ec4adf0 Mon Sep 17 00:00:00 2001
From: mantikoros <sgrugett@gmail.com>
Date: Sun, 17 Apr 2022 23:00:16 -0500
Subject: [PATCH] save dpm shares, pool

---
 functions/src/scripts/migrate-to-cfmm.ts | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/functions/src/scripts/migrate-to-cfmm.ts b/functions/src/scripts/migrate-to-cfmm.ts
index 2e119bf6..f86c9924 100644
--- a/functions/src/scripts/migrate-to-cfmm.ts
+++ b/functions/src/scripts/migrate-to-cfmm.ts
@@ -62,7 +62,11 @@ async function recalculateContract(contractRef: DocRef, isCommit = false) {
         shares
       )
 
-      if (isCommit) transaction.update(betsRef.doc(bet.id), { shares })
+      if (isCommit)
+        transaction.update(betsRef.doc(bet.id), {
+          shares,
+          dpmShares: bet.shares,
+        })
     }
 
     const prob = getDpmProbability(contract.totalShares)
@@ -78,6 +82,11 @@ async function recalculateContract(contractRef: DocRef, isCommit = false) {
       collectedFees: addObjects(contract.collectedFees ?? noFees, noFees),
     }
 
+    const additionalInfo = {
+      cfmmConversionTime: Date.now(),
+      dpmPool: contract.pool,
+    }
+
     const liquidityDocRef = contractRef.collection('liquidity').doc()
 
     const lp = getCpmmInitialLiquidity(
@@ -91,7 +100,10 @@ async function recalculateContract(contractRef: DocRef, isCommit = false) {
     )
 
     if (isCommit) {
-      transaction.update(contractRef, contractUpdate)
+      transaction.update(contractRef, {
+        ...contractUpdate,
+        ...additionalInfo,
+      })
       transaction.set(liquidityDocRef, lp)
 
       console.log('updated', contract.slug)