manifold/functions/src/close-market.ts

import * as admin from 'firebase-admin'
import { z } from 'zod'

import { Contract } from '../../common/contract'
import { getUser } from './utils'

import { isAdmin, isManifoldId } from '../../common/envs/constants'
import { APIError, newEndpoint, validate } from './api'

const bodySchema = z.object({
  contractId: z.string(),
  closeTime: z.number().int().nonnegative().optional(),
})

export const closemarket = newEndpoint({}, async (req, auth) => {
  const { contractId, closeTime } = validate(bodySchema, req.body)
  const contractDoc = firestore.doc(`contracts/${contractId}`)
  const contractSnap = await contractDoc.get()
  if (!contractSnap.exists)
    throw new APIError(404, 'No contract exists with the provided ID')
  const contract = contractSnap.data() as Contract
  const { creatorId } = contract
  const firebaseUser = await admin.auth().getUser(auth.uid)

  if (
    creatorId !== auth.uid &&
    !isManifoldId(auth.uid) &&
    !isAdmin(firebaseUser.email)
  )
    throw new APIError(403, 'User is not creator of contract')

  const now = Date.now()
  if (!closeTime && contract.closeTime && contract.closeTime < now)
    throw new APIError(400, 'Contract already closed')

  if (closeTime && closeTime < now)
    throw new APIError(
      400,
      'Close time must be in the future. ' +
        'Alternatively, do not provide a close time to close immediately.'
    )

  const creator = await getUser(creatorId)
  if (!creator) throw new APIError(500, 'Creator not found')

  const updatedContract = {
    ...contract,
    closeTime: closeTime ? closeTime : now,
  }

  await contractDoc.update(updatedContract)

  console.log('contract ', contractId, 'closed')

  return updatedContract
})

const firestore = admin.firestore()