manifold/functions/src/resolve-market.ts

131 lines
4.0 KiB
TypeScript
Raw Normal View History

2021-12-14 07:02:50 +00:00
import * as functions from 'firebase-functions'
import * as admin from 'firebase-admin'
import * as _ from 'lodash'
import { Contract } from './types/contract'
import { User } from './types/user'
import { Bet } from './types/bet'
export const PLATFORM_FEE = 0.01 // 1%
export const CREATOR_FEE = 0.01 // 1%
export const resolveMarket = functions
.runWith({ minInstances: 1 })
2021-12-17 22:15:09 +00:00
.https.onCall(
async (
data: {
outcome: string
contractId: string
},
context
) => {
const userId = context?.auth?.uid
if (!userId) return { status: 'error', message: 'Not authorized' }
const { outcome, contractId } = data
if (!['YES', 'NO', 'CANCEL'].includes(outcome))
return { status: 'error', message: 'Invalid outcome' }
const contractDoc = firestore.doc(`contracts/${contractId}`)
const contractSnap = await contractDoc.get()
if (!contractSnap.exists)
return { status: 'error', message: 'Invalid contract' }
const contract = contractSnap.data() as Contract
if (contract.creatorId !== userId)
return { status: 'error', message: 'User not creator of contract' }
if (contract.resolution)
return { status: 'error', message: 'Contract already resolved' }
await contractDoc.update({
isResolved: true,
resolution: outcome,
resolutionTime: Date.now(),
})
console.log('contract ', contractId, 'resolved to:', outcome)
const betsSnap = await firestore
.collection(`contracts/${contractId}/bets`)
.get()
const bets = betsSnap.docs.map((doc) => doc.data() as Bet)
const payouts =
outcome === 'CANCEL'
? getCancelPayouts(contract, bets)
2021-12-17 22:15:09 +00:00
: getPayouts(outcome, contract, bets)
console.log('payouts:', payouts)
const groups = _.groupBy(payouts, (payout) => payout.userId)
const userPayouts = _.mapValues(groups, (group) =>
_.sumBy(group, (g) => g.payout)
)
const payoutPromises = Object.entries(userPayouts).map(payUser)
return await Promise.all(payoutPromises)
.catch((e) => ({ status: 'error', message: e }))
.then(() => ({ status: 'success' }))
}
)
2021-12-14 07:02:50 +00:00
const firestore = admin.firestore()
const getCancelPayouts = (contract: Contract, bets: Bet[]) => {
const startPool = contract.startPool.YES + contract.startPool.NO
const truePool = contract.pool.YES + contract.pool.NO - startPool
const openBets = bets.filter((b) => !b.isSold && !b.sale)
const betSum = _.sumBy(openBets, (b) => b.amount)
return openBets.map((bet) => ({
userId: bet.userId,
payout: (bet.amount / betSum) * truePool,
}))
}
2021-12-14 07:02:50 +00:00
const getPayouts = (outcome: string, contract: Contract, bets: Bet[]) => {
const openBets = bets.filter((b) => !b.isSold && !b.sale)
const [yesBets, noBets] = _.partition(
openBets,
(bet) => bet.outcome === 'YES'
)
const startPool = contract.startPool.YES + contract.startPool.NO
const truePool = contract.pool.YES + contract.pool.NO - startPool
2021-12-14 07:02:50 +00:00
const [totalShares, winningBets] =
2021-12-17 22:15:09 +00:00
outcome === 'YES'
? [contract.totalShares.YES, yesBets]
: [contract.totalShares.NO, noBets]
2021-12-14 07:02:50 +00:00
const finalPool = (1 - PLATFORM_FEE - CREATOR_FEE) * truePool
const creatorPayout = CREATOR_FEE * truePool
2021-12-17 22:15:09 +00:00
console.log('final pool:', finalPool, 'creator fee:', creatorPayout)
2021-12-14 07:02:50 +00:00
2021-12-17 22:15:09 +00:00
const winnerPayouts = winningBets.map((bet) => ({
2021-12-14 07:02:50 +00:00
userId: bet.userId,
payout: (bet.shares / totalShares) * finalPool,
2021-12-14 07:02:50 +00:00
}))
2021-12-17 22:15:09 +00:00
return winnerPayouts.concat([
{ userId: contract.creatorId, payout: creatorPayout },
]) // add creator fee
2021-12-14 07:02:50 +00:00
}
const payUser = ([userId, payout]: [string, number]) => {
2021-12-17 22:15:09 +00:00
return firestore.runTransaction(async (transaction) => {
2021-12-14 07:02:50 +00:00
const userDoc = firestore.doc(`users/${userId}`)
const userSnap = await transaction.get(userDoc)
if (!userSnap.exists) return
const user = userSnap.data() as User
const newUserBalance = user.balance + payout
transaction.update(userDoc, { balance: newUserBalance })
})
}