2021-12-11 00:06:17 +00:00
|
|
|
import * as functions from 'firebase-functions'
|
|
|
|
import * as admin from 'firebase-admin'
|
|
|
|
|
2022-01-10 21:07:57 +00:00
|
|
|
import { Contract } from '../../common/contract'
|
|
|
|
import { User } from '../../common/user'
|
2022-03-02 03:31:48 +00:00
|
|
|
import {
|
|
|
|
getLoanAmount,
|
|
|
|
getNewBinaryBetInfo,
|
|
|
|
getNewMultiBetInfo,
|
|
|
|
} from '../../common/new-bet'
|
|
|
|
import { Bet } from '../../common/bet'
|
|
|
|
import { getValues } from './utils'
|
2021-12-11 00:06:17 +00:00
|
|
|
|
2021-12-12 21:32:06 +00:00
|
|
|
export const placeBet = functions.runWith({ minInstances: 1 }).https.onCall(
|
|
|
|
async (
|
|
|
|
data: {
|
|
|
|
amount: number
|
|
|
|
outcome: string
|
|
|
|
contractId: string
|
|
|
|
},
|
|
|
|
context
|
|
|
|
) => {
|
2021-12-11 03:45:05 +00:00
|
|
|
const userId = context?.auth?.uid
|
2021-12-12 21:32:06 +00:00
|
|
|
if (!userId) return { status: 'error', message: 'Not authorized' }
|
2021-12-11 03:45:05 +00:00
|
|
|
|
|
|
|
const { amount, outcome, contractId } = data
|
|
|
|
|
2022-01-08 17:51:31 +00:00
|
|
|
if (amount <= 0 || isNaN(amount) || !isFinite(amount))
|
|
|
|
return { status: 'error', message: 'Invalid amount' }
|
|
|
|
|
2022-02-17 23:00:19 +00:00
|
|
|
if (outcome !== 'YES' && outcome !== 'NO' && isNaN(+outcome))
|
2021-12-11 03:45:05 +00:00
|
|
|
return { status: 'error', message: 'Invalid outcome' }
|
|
|
|
|
|
|
|
// run as transaction to prevent race conditions
|
2021-12-12 21:32:06 +00:00
|
|
|
return await firestore.runTransaction(async (transaction) => {
|
2021-12-11 03:45:05 +00:00
|
|
|
const userDoc = firestore.doc(`users/${userId}`)
|
|
|
|
const userSnap = await transaction.get(userDoc)
|
2021-12-12 21:32:06 +00:00
|
|
|
if (!userSnap.exists)
|
|
|
|
return { status: 'error', message: 'User not found' }
|
2021-12-11 03:45:05 +00:00
|
|
|
const user = userSnap.data() as User
|
|
|
|
|
2021-12-13 17:58:47 +00:00
|
|
|
if (user.balance < amount)
|
2021-12-11 03:45:05 +00:00
|
|
|
return { status: 'error', message: 'Insufficient balance' }
|
|
|
|
|
|
|
|
const contractDoc = firestore.doc(`contracts/${contractId}`)
|
|
|
|
const contractSnap = await transaction.get(contractDoc)
|
2021-12-12 21:32:06 +00:00
|
|
|
if (!contractSnap.exists)
|
|
|
|
return { status: 'error', message: 'Invalid contract' }
|
2021-12-11 03:45:05 +00:00
|
|
|
const contract = contractSnap.data() as Contract
|
|
|
|
|
2022-02-17 23:00:19 +00:00
|
|
|
const { closeTime, outcomeType } = contract
|
2022-01-10 16:32:20 +00:00
|
|
|
if (closeTime && Date.now() > closeTime)
|
|
|
|
return { status: 'error', message: 'Trading is closed' }
|
|
|
|
|
2022-03-02 03:31:48 +00:00
|
|
|
const yourBetsSnap = await transaction.get(
|
|
|
|
contractDoc.collection('bets').where('userId', '==', userId)
|
|
|
|
)
|
|
|
|
const yourBets = yourBetsSnap.docs.map((doc) => doc.data() as Bet)
|
|
|
|
|
2022-02-17 23:00:19 +00:00
|
|
|
if (outcomeType === 'FREE_RESPONSE') {
|
|
|
|
const answerSnap = await transaction.get(
|
|
|
|
contractDoc.collection('answers').doc(outcome)
|
|
|
|
)
|
|
|
|
if (!answerSnap.exists)
|
|
|
|
return { status: 'error', message: 'Invalid contract' }
|
|
|
|
}
|
|
|
|
|
2021-12-12 21:32:06 +00:00
|
|
|
const newBetDoc = firestore
|
|
|
|
.collection(`contracts/${contractId}/bets`)
|
|
|
|
.doc()
|
2021-12-11 03:45:05 +00:00
|
|
|
|
2022-03-02 03:31:48 +00:00
|
|
|
const loanAmount = getLoanAmount(yourBets, amount)
|
|
|
|
|
2022-01-02 01:03:18 +00:00
|
|
|
const { newBet, newPool, newTotalShares, newTotalBets, newBalance } =
|
2022-02-17 23:00:19 +00:00
|
|
|
outcomeType === 'BINARY'
|
|
|
|
? getNewBinaryBetInfo(
|
|
|
|
user,
|
|
|
|
outcome as 'YES' | 'NO',
|
|
|
|
amount,
|
2022-03-02 03:31:48 +00:00
|
|
|
loanAmount,
|
|
|
|
contract,
|
|
|
|
newBetDoc.id
|
|
|
|
)
|
|
|
|
: getNewMultiBetInfo(
|
|
|
|
user,
|
|
|
|
outcome,
|
|
|
|
amount,
|
|
|
|
loanAmount,
|
2022-02-17 23:00:19 +00:00
|
|
|
contract,
|
|
|
|
newBetDoc.id
|
|
|
|
)
|
2021-12-11 03:45:05 +00:00
|
|
|
|
|
|
|
transaction.create(newBetDoc, newBet)
|
2021-12-17 22:15:09 +00:00
|
|
|
transaction.update(contractDoc, {
|
|
|
|
pool: newPool,
|
2021-12-24 21:06:01 +00:00
|
|
|
totalShares: newTotalShares,
|
2022-01-02 01:03:18 +00:00
|
|
|
totalBets: newTotalBets,
|
2021-12-17 22:15:09 +00:00
|
|
|
})
|
2021-12-13 17:58:47 +00:00
|
|
|
transaction.update(userDoc, { balance: newBalance })
|
2021-12-11 03:45:05 +00:00
|
|
|
|
2022-01-06 18:55:31 +00:00
|
|
|
return { status: 'success', betId: newBetDoc.id }
|
2021-12-11 03:45:05 +00:00
|
|
|
})
|
2021-12-12 21:32:06 +00:00
|
|
|
}
|
|
|
|
)
|
2021-12-11 00:06:17 +00:00
|
|
|
|
|
|
|
const firestore = admin.firestore()
|