diff --git a/functions/src/add-liquidity.ts b/functions/src/add-liquidity.ts deleted file mode 100644 index e6090111..00000000 --- a/functions/src/add-liquidity.ts +++ /dev/null @@ -1,133 +0,0 @@ -import * as admin from 'firebase-admin' -import { z } from 'zod' - -import { Contract, CPMMContract } from '../../common/contract' -import { User } from '../../common/user' -import { removeUndefinedProps } from '../../common/util/object' -import { getNewLiquidityProvision } from '../../common/add-liquidity' -import { APIError, newEndpoint, validate } from './api' -import { - DEV_HOUSE_LIQUIDITY_PROVIDER_ID, - HOUSE_LIQUIDITY_PROVIDER_ID, -} from '../../common/antes' -import { isProd } from './utils' - -const bodySchema = z.object({ - contractId: z.string(), - amount: z.number().gt(0), -}) - -export const addliquidity = newEndpoint({}, async (req, auth) => { - const { amount, contractId } = validate(bodySchema, req.body) - - if (!isFinite(amount)) throw new APIError(400, 'Invalid amount') - - // run as transaction to prevent race conditions - return await firestore.runTransaction(async (transaction) => { - const userDoc = firestore.doc(`users/${auth.uid}`) - const userSnap = await transaction.get(userDoc) - if (!userSnap.exists) throw new APIError(400, 'User not found') - const user = userSnap.data() as User - - const contractDoc = firestore.doc(`contracts/${contractId}`) - const contractSnap = await transaction.get(contractDoc) - if (!contractSnap.exists) throw new APIError(400, 'Invalid contract') - const contract = contractSnap.data() as Contract - if ( - contract.mechanism !== 'cpmm-1' || - (contract.outcomeType !== 'BINARY' && - contract.outcomeType !== 'PSEUDO_NUMERIC') - ) - throw new APIError(400, 'Invalid contract') - - const { closeTime } = contract - if (closeTime && Date.now() > closeTime) - throw new APIError(400, 'Trading is closed') - - if (user.balance < amount) throw new APIError(400, 'Insufficient balance') - - const newLiquidityProvisionDoc = firestore - .collection(`contracts/${contractId}/liquidity`) - .doc() - - const { newLiquidityProvision, newPool, newP, newTotalLiquidity } = - getNewLiquidityProvision( - user.id, - amount, - contract, - newLiquidityProvisionDoc.id - ) - - if (newP !== undefined && !isFinite(newP)) { - return { - status: 'error', - message: 'Liquidity injection rejected due to overflow error.', - } - } - - transaction.update( - contractDoc, - removeUndefinedProps({ - pool: newPool, - p: newP, - totalLiquidity: newTotalLiquidity, - }) - ) - - const newBalance = user.balance - amount - const newTotalDeposits = user.totalDeposits - amount - - if (!isFinite(newBalance)) { - throw new APIError(500, 'Invalid user balance for ' + user.username) - } - - transaction.update(userDoc, { - balance: newBalance, - totalDeposits: newTotalDeposits, - }) - - transaction.create(newLiquidityProvisionDoc, newLiquidityProvision) - - return newLiquidityProvision - }) -}) - -const firestore = admin.firestore() - -export const addHouseLiquidity = (contract: CPMMContract, amount: number) => { - return firestore.runTransaction(async (transaction) => { - const newLiquidityProvisionDoc = firestore - .collection(`contracts/${contract.id}/liquidity`) - .doc() - - const providerId = isProd() - ? HOUSE_LIQUIDITY_PROVIDER_ID - : DEV_HOUSE_LIQUIDITY_PROVIDER_ID - - const { newLiquidityProvision, newPool, newP, newTotalLiquidity } = - getNewLiquidityProvision( - providerId, - amount, - contract, - newLiquidityProvisionDoc.id - ) - - if (newP !== undefined && !isFinite(newP)) { - throw new APIError( - 500, - 'Liquidity injection rejected due to overflow error.' - ) - } - - transaction.update( - firestore.doc(`contracts/${contract.id}`), - removeUndefinedProps({ - pool: newPool, - p: newP, - totalLiquidity: newTotalLiquidity, - }) - ) - - transaction.create(newLiquidityProvisionDoc, newLiquidityProvision) - }) -}