103 lines
2.5 KiB
TypeScript
103 lines
2.5 KiB
TypeScript
import { groupBy, sumBy, mapValues, partition } from 'lodash'
|
|
|
|
import { Bet } from './bet'
|
|
import { getContractBetMetrics } from './calculate'
|
|
import { Contract } from './contract'
|
|
import { getPayouts } from './payouts'
|
|
|
|
export function scoreCreators(contracts: Contract[]) {
|
|
const creatorScore = mapValues(
|
|
groupBy(contracts, ({ creatorId }) => creatorId),
|
|
(contracts) =>
|
|
sumBy(
|
|
contracts.map((contract) => {
|
|
return contract.volume
|
|
})
|
|
)
|
|
)
|
|
|
|
return creatorScore
|
|
}
|
|
|
|
export function scoreTraders(contracts: Contract[], bets: Bet[][]) {
|
|
const userScoresByContract = contracts.map((contract, index) =>
|
|
scoreUsersByContract(contract, bets[index])
|
|
)
|
|
const userScores: { [userId: string]: number } = {}
|
|
for (const scores of userScoresByContract) {
|
|
addUserScores(scores, userScores)
|
|
}
|
|
return userScores
|
|
}
|
|
|
|
export function scoreUsersByContract(contract: Contract, bets: Bet[]) {
|
|
const { resolution } = contract
|
|
const resolutionProb =
|
|
contract.outcomeType == 'BINARY'
|
|
? contract.resolutionProbability
|
|
: undefined
|
|
|
|
const [closedBets, openBets] = partition(
|
|
bets,
|
|
(bet) => bet.isSold || bet.sale
|
|
)
|
|
const { payouts: resolvePayouts } = getPayouts(
|
|
resolution as string,
|
|
contract,
|
|
openBets,
|
|
[],
|
|
{},
|
|
resolutionProb
|
|
)
|
|
|
|
const salePayouts = closedBets.map((bet) => {
|
|
const { userId, sale } = bet
|
|
return { userId, payout: sale ? sale.amount : 0 }
|
|
})
|
|
|
|
const investments = bets
|
|
.filter((bet) => !bet.sale)
|
|
.map((bet) => {
|
|
const { userId, amount, loanAmount } = bet
|
|
const payout = -amount - (loanAmount ?? 0)
|
|
return { userId, payout }
|
|
})
|
|
|
|
const profits = openBets.map((bet) => {
|
|
const { userId } = bet
|
|
if (contract.isResolved) {
|
|
return { userId: userId, payout: 0 }
|
|
}
|
|
const payout = getContractBetMetrics(contract, [bet]).payout
|
|
console.log({
|
|
userId: userId,
|
|
metrics: getContractBetMetrics(contract, [bet]),
|
|
})
|
|
return { userId: userId, payout: payout }
|
|
})
|
|
|
|
const netPayouts = [
|
|
...resolvePayouts,
|
|
...salePayouts,
|
|
...investments,
|
|
...profits,
|
|
]
|
|
|
|
const userScore = mapValues(
|
|
groupBy(netPayouts, (payout) => payout.userId),
|
|
(payouts) => sumBy(payouts, ({ payout }) => payout)
|
|
)
|
|
|
|
return userScore
|
|
}
|
|
|
|
export function addUserScores(
|
|
src: { [userId: string]: number },
|
|
dest: { [userId: string]: number }
|
|
) {
|
|
for (const [userId, score] of Object.entries(src)) {
|
|
if (dest[userId] === undefined) dest[userId] = 0
|
|
dest[userId] += score
|
|
}
|
|
}
|