57 lines
1.4 KiB
TypeScript
57 lines
1.4 KiB
TypeScript
import { Bet } from './bet'
|
|
import { Contract } from './contract'
|
|
import { User } from './user'
|
|
|
|
export const getNewBetInfo = (
|
|
user: User,
|
|
outcome: 'YES' | 'NO',
|
|
amount: number,
|
|
contract: Contract,
|
|
newBetId: string
|
|
) => {
|
|
const { YES: yesPool, NO: noPool } = contract.pool
|
|
|
|
const newPool =
|
|
outcome === 'YES'
|
|
? { YES: yesPool + amount, NO: noPool }
|
|
: { YES: yesPool, NO: noPool + amount }
|
|
|
|
const shares =
|
|
outcome === 'YES'
|
|
? amount + (amount * noPool ** 2) / (yesPool ** 2 + amount * yesPool)
|
|
: amount + (amount * yesPool ** 2) / (noPool ** 2 + amount * noPool)
|
|
|
|
const { YES: yesShares, NO: noShares } = contract.totalShares
|
|
|
|
const newTotalShares =
|
|
outcome === 'YES'
|
|
? { YES: yesShares + shares, NO: noShares }
|
|
: { YES: yesShares, NO: noShares + shares }
|
|
|
|
const { YES: yesBets, NO: noBets } = contract.totalBets
|
|
|
|
const newTotalBets =
|
|
outcome === 'YES'
|
|
? { YES: yesBets + amount, NO: noBets }
|
|
: { YES: yesBets, NO: noBets + amount }
|
|
|
|
const probBefore = yesPool ** 2 / (yesPool ** 2 + noPool ** 2)
|
|
const probAfter = newPool.YES ** 2 / (newPool.YES ** 2 + newPool.NO ** 2)
|
|
|
|
const newBet: Bet = {
|
|
id: newBetId,
|
|
userId: user.id,
|
|
contractId: contract.id,
|
|
amount,
|
|
shares,
|
|
outcome,
|
|
probBefore,
|
|
probAfter,
|
|
createdTime: Date.now(),
|
|
}
|
|
|
|
const newBalance = user.balance - amount
|
|
|
|
return { newBet, newPool, newTotalShares, newTotalBets, newBalance }
|
|
}
|