feat: script to update price connections

Author: chedieck

package.json

@@ -18,7 +18,8 @@
     "docker": "./scripts/docker-exec-shortcuts.sh",
     "ci:integration:test": "yarn pretest && dotenv -e .env.test -- ts-node -O '{\"module\":\"commonjs\"}' node_modules/jest/bin/jest.js tests/integration-tests --forceExit",
     "tarDebug": "tar cf debug.tar logs/ paybutton-config.json .env*",
-    "updateAllPrices": "./scripts/update-all-prices.sh"
+    "updateAllPrices": "./scripts/update-all-prices.sh",
+    "updateAllPriceConnections": "./scripts/update-all-price-connections.sh"
   },
   "dependencies": {
     "@emotion/react": "^11.8.2",

scripts/update-all-price-connections.sh

@@ -0,0 +1,9 @@
+set -e
+
+echo This will affect the database. Are you sure of what you\'re doing? [y/N]
+read ans 
+if [[ "$ans" = "Y" || "$ans" = "y" ]]; then
+  JOBS_ENV=true dotenv -e .env -c -- ts-node -O '{"module":"commonjs"}' -r tsconfig-paths/register ./scripts/updateAllPriceConnections.ts
+else 
+  echo Exited.
+fi

scripts/updateAllPriceConnections.ts

@@ -0,0 +1,98 @@
+import { flattenTimestamp } from '../services/priceService'
+import prisma from 'prisma/clientInstance'
+import { connectTransactionsListToPrices } from 'services/transactionService'
+import { Transaction } from '@prisma/client'
+import moment from 'moment'
+
+async function misconnectTxs (txsIds: string[]): Promise<void> {
+  if (process.env.ENVIRONMENT === 'production') {
+    return
+  }
+  if (txsIds.length === 0) {
+    return
+  }
+  for (const txId of txsIds) {
+    const tx = await prisma.transaction.findUniqueOrThrow({
+      where: {
+        id: txId
+      }
+    })
+    const txDayTimestamp = flattenTimestamp(tx.timestamp)
+    let signal = 1
+    if (txDayTimestamp === flattenTimestamp(moment.utc().unix())) {
+      signal = -1
+    }
+    await prisma.transaction.update({
+      where: {
+        id: txId
+      },
+      data: {
+        timestamp: tx.timestamp + (signal * 86400)
+      }
+    })
+  }
+}
+
+async function fixMisconnectedTxs (): Promise<void> {
+  const total = await prisma.transaction.count()
+  const pageSize = 1000
+  let page = 0
+
+  await misconnectTxs([
+    // ADD TXS IDS HERE TO TEST IT
+  ])
+
+  console.log('Fixing misconnected txs...')
+  while (true) {
+    const txsToFix: Transaction[] = []
+    // Get txs page
+    const txs = await prisma.transaction.findMany({
+      orderBy: {
+        timestamp: 'asc'
+      },
+      include: {
+        prices: {
+          select: {
+            price: {
+              select: {
+                timestamp: true
+              }
+            }
+          }
+        }
+      },
+      skip: page * pageSize,
+      take: pageSize
+    })
+
+    const viewedCount = page * pageSize + txs.length
+
+    // Finish if empty
+    if (txs.length === 0) {
+      break
+    }
+
+    // Find txs with misaligned timestamps
+    txs.forEach(tx => {
+      const txFlattenedTimestamp = flattenTimestamp(tx.timestamp)
+      const txPriceTimestamps = tx.prices.map(p => p.price.timestamp)
+      if (txPriceTimestamps.filter(t => t !== txFlattenedTimestamp).length > 0) {
+        txsToFix.push(tx)
+      }
+    })
+    if (txsToFix.length !== 0) {
+      console.log(`[${viewedCount}/${total}] Fixing ${txsToFix.length} txs...`)
+      await connectTransactionsListToPrices(txsToFix)
+      console.log(`[${viewedCount}/${total}] Finished fixing ${txsToFix.length} txs.`)
+    }
+    console.log(`[${viewedCount}/${total}]`)
+    page++
+  }
+  console.log('FINISHED')
+}
+
+async function run (): Promise<void> {
+  await fixMisconnectedTxs()
+}
+
+void run()