manifold/web/lib/firebase/contracts.ts

114 lines
3.5 KiB
TypeScript
Raw Normal View History

2021-12-10 04:54:40 +00:00
import { app } from './init'
import {
getFirestore,
doc,
setDoc,
deleteDoc,
where,
collection,
query,
getDocs,
2021-12-10 05:01:44 +00:00
onSnapshot,
2021-12-10 07:08:28 +00:00
orderBy,
getDoc,
limit,
2021-12-10 04:54:40 +00:00
} from 'firebase/firestore'
import dayjs from 'dayjs'
2021-12-09 22:05:55 +00:00
export type Contract = {
2021-12-10 04:54:40 +00:00
id: string // Chosen by creator; must be unique
creatorId: string
2021-12-10 06:21:12 +00:00
creatorName: string
2021-12-10 04:54:40 +00:00
2021-12-09 22:05:55 +00:00
question: string
2021-12-10 04:54:40 +00:00
description: string // More info about what the contract is about
outcomeType: 'BINARY' // | 'MULTI' | 'interval' | 'date'
// outcomes: ['YES', 'NO']
seedAmounts: { YES: number; NO: number } // seedBets: [number, number]
2021-12-11 00:40:23 +00:00
pot: { YES: number; NO: number }
2021-12-15 22:44:22 +00:00
dpmWeights: { YES: number; NO: number }
2021-12-10 04:54:40 +00:00
createdTime: number // Milliseconds since epoch
lastUpdatedTime: number // If the question or description was changed
closeTime?: number // When no more trading is allowed
2021-12-14 06:12:25 +00:00
isResolved: boolean
2021-12-16 06:44:04 +00:00
resolutionTime?: number // When the contract creator resolved the market; 0 if unresolved
2021-12-10 04:54:40 +00:00
resolution?: 'YES' | 'NO' | 'CANCEL' // Chosen by creator; must be one of outcomes
}
export function path(contract: Contract) {
// For now, derive username from creatorName
// Fix this when users can change their own names
const username = contract.creatorName.replace(/\s+/g, '')
2021-12-17 18:55:18 +00:00
return `/${username}/${contract.id}`
}
export function compute(contract: Contract) {
const { pot, seedAmounts, createdTime, resolutionTime, isResolved } = contract
const volume = pot.YES + pot.NO - seedAmounts.YES - seedAmounts.NO
const prob = pot.YES ** 2 / (pot.YES ** 2 + pot.NO ** 2)
const probPercent = Math.round(prob * 100) + '%'
const createdDate = dayjs(createdTime).format('MMM D')
2021-12-15 07:41:50 +00:00
const resolvedDate = isResolved
? dayjs(resolutionTime).format('MMM D')
: undefined
return { volume, probPercent, createdDate, resolvedDate }
}
2021-12-10 04:54:40 +00:00
const db = getFirestore(app)
2021-12-10 05:01:44 +00:00
const contractCollection = collection(db, 'contracts')
2021-12-10 04:54:40 +00:00
// Push contract to Firestore
export async function setContract(contract: Contract) {
const docRef = doc(db, 'contracts', contract.id)
await setDoc(docRef, contract)
}
2021-12-09 22:05:55 +00:00
export async function getContract(contractId: string) {
const docRef = doc(db, 'contracts', contractId)
const result = await getDoc(docRef)
return result.exists() ? (result.data() as Contract) : undefined
}
2021-12-10 04:54:40 +00:00
export async function deleteContract(contractId: string) {
const docRef = doc(db, 'contracts', contractId)
await deleteDoc(docRef)
}
export async function listContracts(creatorId: string): Promise<Contract[]> {
2021-12-10 07:08:28 +00:00
const q = query(
contractCollection,
where('creatorId', '==', creatorId),
orderBy('createdTime', 'desc')
)
2021-12-10 04:54:40 +00:00
const snapshot = await getDocs(q)
return snapshot.docs.map((doc) => doc.data() as Contract)
}
export async function listAllContracts(): Promise<Contract[]> {
const q = query(contractCollection, orderBy('createdTime', 'desc'), limit(25))
const snapshot = await getDocs(q)
return snapshot.docs.map((doc) => doc.data() as Contract)
2021-12-10 04:54:40 +00:00
}
2021-12-09 22:05:55 +00:00
export function listenForContracts(
setContracts: (contracts: Contract[]) => void
) {
const q = query(contractCollection, orderBy('createdTime', 'desc'), limit(25))
return onSnapshot(q, (snap) => {
setContracts(snap.docs.map((doc) => doc.data() as Contract))
})
}
2021-12-10 05:01:44 +00:00
export function listenForContract(
contractId: string,
2021-12-15 07:41:50 +00:00
setContract: (contract: Contract | null) => void
2021-12-10 05:01:44 +00:00
) {
const contractRef = doc(contractCollection, contractId)
return onSnapshot(contractRef, (contractSnap) => {
2021-12-15 07:41:50 +00:00
setContract((contractSnap.data() ?? null) as Contract | null)
2021-12-10 05:01:44 +00:00
})
}