40 lines
1.2 KiB
TypeScript
40 lines
1.2 KiB
TypeScript
|
import * as admin from 'firebase-admin'
|
||
|
|
import { initAdmin } from './script-init'
|
||
|
|
import { getValues, log, writeAsync } from '../utils'
|
||
|
|
import { Bet } from '../../../common/bet'
|
||
|
|
import { groupBy, mapValues, sortBy, uniq } from 'lodash'
|
||
|
|
|
||
|
|
initAdmin()
|
||
|
|
const firestore = admin.firestore()
|
||
|
|
|
||
|
|
const getBettorsByContractId = async () => {
|
||
|
|
const bets = await getValues<Bet>(firestore.collectionGroup('bets'))
|
||
|
|
log(`Loaded ${bets.length} bets.`)
|
||
|
|
const betsByContractId = groupBy(bets, 'contractId')
|
||
|
|
return mapValues(betsByContractId, (bets) =>
|
||
|
|
uniq(sortBy(bets, 'createdTime').map((bet) => bet.userId))
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
const updateUniqueBettors = async () => {
|
||
|
|
const bettorsByContractId = await getBettorsByContractId()
|
||
|
|
|
||
|
|
const updates = Object.entries(bettorsByContractId).map(
|
||
|
|
([contractId, userIds]) => {
|
||
|
|
const update = {
|
||
|
|
uniqueBettorIds: userIds,
|
||
|
|
uniqueBettorCount: userIds.length,
|
||
|
|
}
|
||
|
|
const docRef = firestore.collection('contracts').doc(contractId)
|
||
|
|
return { doc: docRef, fields: update }
|
||
|
|
}
|
||
|
|
)
|
||
|
|
log(`Updating ${updates.length} contracts.`)
|
||
|
|
await writeAsync(firestore, updates)
|
||
|
|
log(`Updated all contracts.`)
|
||
|
|
}
|
||
|
|
|
||
|
|
if (require.main === module) {
|
||
|
|
updateUniqueBettors()
|
||
|
|
}
|