Compare commits

...

1 Commits

Author SHA1 Message Date
mantikoros
0b1d59a2ac initial commit 2021-12-17 19:47:46 -06:00
5 changed files with 81 additions and 50 deletions

16
functions/src/api-call.ts Normal file
View File

@ -0,0 +1,16 @@
import * as functions from 'firebase-functions'
import { placeBetLogic } from '.'
const apiFunctions = {
'place-bet': placeBetLogic,
}
export const apiCall = functions.firestore
.document('api-calls/{apiCallId}')
.onCreate(async (snap, context) => {
const data = snap.data() as any
console.log('data:', data)
const apiFunction = apiFunctions[data.name as keyof typeof apiFunctions]
await apiFunction(data)
})

View File

@ -3,4 +3,5 @@ import * as admin from 'firebase-admin'
admin.initializeApp()
export * from './place-bet'
export * from './resolve-market'
export * from './resolve-market'
export * from './api-call'

View File

@ -5,63 +5,62 @@ import { Contract } from './types/contract'
import { User } from './types/user'
import { Bet } from './types/bet'
export const placeBet = functions.runWith({ minInstances: 1 }).https.onCall(
async (
data: {
amount: number
outcome: string
contractId: string
},
context
) => {
const userId = context?.auth?.uid
if (!userId) return { status: 'error', message: 'Not authorized' }
export const placeBet = functions
.runWith({ minInstances: 1 })
.https.onCall((data, context) => {
placeBetLogic({ ...data, userId: context?.auth?.uid })
})
const { amount, outcome, contractId } = data
export const placeBetLogic = async (data: {
amount: number
outcome: string
contractId: string
userId: string
}) => {
console.log('placeBet called')
// if (!userId) return { status: 'error', message: 'Not authorized' }
if (outcome !== 'YES' && outcome !== 'NO')
return { status: 'error', message: 'Invalid outcome' }
const { amount, outcome, contractId, userId } = data
// run as transaction to prevent race conditions
return await firestore.runTransaction(async (transaction) => {
const userDoc = firestore.doc(`users/${userId}`)
const userSnap = await transaction.get(userDoc)
if (!userSnap.exists)
return { status: 'error', message: 'User not found' }
const user = userSnap.data() as User
if (outcome !== 'YES' && outcome !== 'NO')
return { status: 'error', message: 'Invalid outcome' }
if (user.balance < amount)
return { status: 'error', message: 'Insufficient balance' }
// run as transaction to prevent race conditions
return await firestore.runTransaction(async (transaction) => {
const userDoc = firestore.doc(`users/${userId}`)
const userSnap = await transaction.get(userDoc)
if (!userSnap.exists) return { status: 'error', message: 'User not found' }
const user = userSnap.data() as User
const contractDoc = firestore.doc(`contracts/${contractId}`)
const contractSnap = await transaction.get(contractDoc)
if (!contractSnap.exists)
return { status: 'error', message: 'Invalid contract' }
const contract = contractSnap.data() as Contract
if (user.balance < amount)
return { status: 'error', message: 'Insufficient balance' }
const newBetDoc = firestore
.collection(`contracts/${contractId}/bets`)
.doc()
const contractDoc = firestore.doc(`contracts/${contractId}`)
const contractSnap = await transaction.get(contractDoc)
if (!contractSnap.exists)
return { status: 'error', message: 'Invalid contract' }
const contract = contractSnap.data() as Contract
const { newBet, newPool, newDpmWeights, newBalance } = getNewBetInfo(
user,
outcome,
amount,
contract,
newBetDoc.id
)
const newBetDoc = firestore.collection(`contracts/${contractId}/bets`).doc()
transaction.create(newBetDoc, newBet)
transaction.update(contractDoc, {
pool: newPool,
dpmWeights: newDpmWeights,
})
transaction.update(userDoc, { balance: newBalance })
const { newBet, newPool, newDpmWeights, newBalance } = getNewBetInfo(
user,
outcome,
amount,
contract,
newBetDoc.id
)
return { status: 'success' }
transaction.create(newBetDoc, newBet)
transaction.update(contractDoc, {
pool: newPool,
dpmWeights: newDpmWeights,
})
}
)
transaction.update(userDoc, { balance: newBalance })
return { status: 'success' }
})
}
const firestore = admin.firestore()

View File

@ -15,6 +15,7 @@ import {
getDpmWeight,
getProbabilityAfterBet,
} from '../lib/calculation/contract'
import { apiCall } from '../lib/firebase/api-call'
export function BetPanel(props: { contract: Contract; className?: string }) {
const { contract, className } = props
@ -61,6 +62,7 @@ export function BetPanel(props: { contract: Contract; className?: string }) {
amount: betAmount,
outcome: betChoice,
contractId: contract.id,
userId: user.id,
}).then((r) => r.data as any)
console.log('placed bet. Result:', result)
@ -171,5 +173,9 @@ export function BetPanel(props: { contract: Contract; className?: string }) {
)
}
const functions = getFunctions()
export const placeBet = httpsCallable(functions, 'placeBet')
// const functions = getFunctions()
// export const placeBet = httpsCallable(functions, 'placeBet')
export const placeBet = async (bet: any) => {
await apiCall({ ...bet, name: 'place-bet' })
return { data: { success: true } }
}

View File

@ -0,0 +1,9 @@
import { app } from './init'
import { doc, collection, getFirestore, setDoc } from 'firebase/firestore'
const db = getFirestore(app)
export async function apiCall(apiCallDoc: any) {
const apiCallRef = doc(collection(db, 'api-calls'))
await setDoc(apiCallRef, apiCallDoc)
}