103 lines
2.9 KiB
TypeScript
103 lines
2.9 KiB
TypeScript
import * as admin from 'firebase-admin'
|
|
|
|
import { initAdmin } from './script-init'
|
|
import { getAllUsers, getValues } from '../utils'
|
|
import { Contract } from 'common/contract'
|
|
import {
|
|
MarketCreatorBadge,
|
|
marketCreatorBadgeRarityThresholds,
|
|
StreakerBadge,
|
|
streakerBadgeRarityThresholds,
|
|
} from 'common/badge'
|
|
import { User } from 'common/user'
|
|
initAdmin()
|
|
|
|
const firestore = admin.firestore()
|
|
|
|
async function main() {
|
|
const users = await getAllUsers()
|
|
// const users = filterDefined([await getUser('6hHpzvRG0pMq8PNJs7RZj2qlZGn2')]) // dev ian
|
|
// const users = filterDefined([await getUser('AJwLWoo3xue32XIiAVrL5SyR1WB2')]) // prod ian
|
|
await Promise.all(
|
|
users.map(async (user) => {
|
|
if (!user.id) return
|
|
// Only backfill users without achievements
|
|
if (user.achievements === undefined) {
|
|
await firestore.collection('users').doc(user.id).update({
|
|
achievements: {},
|
|
})
|
|
user.achievements = {}
|
|
user.achievements = await awardMarketCreatorBadges(user)
|
|
user.achievements = await awardBettingStreakBadges(user)
|
|
console.log('Added achievements to user', user.id)
|
|
// going to ignore backfilling the proven correct badges for now
|
|
}
|
|
})
|
|
)
|
|
}
|
|
|
|
if (require.main === module) main().then(() => process.exit())
|
|
|
|
async function awardMarketCreatorBadges(user: User) {
|
|
// Award market maker badges
|
|
const contracts = await getValues<Contract>(
|
|
firestore
|
|
.collection(`contracts`)
|
|
.where('creatorId', '==', user.id)
|
|
.where('resolution', '!=', 'CANCEL')
|
|
)
|
|
|
|
const achievements = {
|
|
...user.achievements,
|
|
marketCreator: {
|
|
badges: [...(user.achievements.marketCreator?.badges ?? [])],
|
|
},
|
|
}
|
|
for (const threshold of marketCreatorBadgeRarityThresholds) {
|
|
if (contracts.length >= threshold) {
|
|
const badge = {
|
|
type: 'MARKET_CREATOR',
|
|
name: 'Market Creator',
|
|
data: {
|
|
totalContractsCreated: threshold,
|
|
},
|
|
createdTime: Date.now(),
|
|
} as MarketCreatorBadge
|
|
achievements.marketCreator.badges.push(badge)
|
|
}
|
|
}
|
|
// update user
|
|
await firestore.collection('users').doc(user.id).update({
|
|
achievements,
|
|
})
|
|
return achievements
|
|
}
|
|
|
|
async function awardBettingStreakBadges(user: User) {
|
|
const streak = user.currentBettingStreak ?? 0
|
|
const achievements = {
|
|
...user.achievements,
|
|
streaker: {
|
|
badges: [...(user.achievements?.streaker?.badges ?? [])],
|
|
},
|
|
}
|
|
for (const threshold of streakerBadgeRarityThresholds) {
|
|
if (streak >= threshold) {
|
|
const badge = {
|
|
type: 'STREAKER',
|
|
name: 'Streaker',
|
|
data: {
|
|
totalBettingStreak: threshold,
|
|
},
|
|
createdTime: Date.now(),
|
|
} as StreakerBadge
|
|
achievements.streaker.badges.push(badge)
|
|
}
|
|
}
|
|
// update user
|
|
await firestore.collection('users').doc(user.id).update({
|
|
achievements,
|
|
})
|
|
return achievements
|
|
}
|