2022-01-10 21:07:57 +00:00
|
|
|
import dayjs from 'dayjs'
|
2021-12-10 04:54:40 +00:00
|
|
|
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,
|
2021-12-14 01:07:28 +00:00
|
|
|
getDoc,
|
2022-01-07 19:27:59 +00:00
|
|
|
updateDoc,
|
2022-01-09 20:26:51 +00:00
|
|
|
limit,
|
2021-12-10 04:54:40 +00:00
|
|
|
} from 'firebase/firestore'
|
2021-12-17 22:15:09 +00:00
|
|
|
|
2022-01-10 21:07:57 +00:00
|
|
|
import { app } from './init'
|
|
|
|
import { getValues, listenForValues } from './utils'
|
|
|
|
import { Contract } from '../../../common/contract'
|
2022-01-12 19:01:04 +00:00
|
|
|
import { getProbability } from '../../../common/calculate'
|
2022-01-15 06:43:15 +00:00
|
|
|
import { createRNG, shuffle } from '../../../common/util/random'
|
2022-01-10 21:07:57 +00:00
|
|
|
export type { Contract }
|
2021-12-10 04:54:40 +00:00
|
|
|
|
2022-01-12 19:01:04 +00:00
|
|
|
export function contractPath(contract: Contract) {
|
2021-12-16 00:52:19 +00:00
|
|
|
// For now, derive username from creatorName
|
2021-12-18 07:27:29 +00:00
|
|
|
return `/${contract.creatorUsername}/${contract.slug}`
|
2021-12-16 00:52:19 +00:00
|
|
|
}
|
|
|
|
|
2022-01-12 19:01:04 +00:00
|
|
|
export function contractMetrics(contract: Contract) {
|
|
|
|
const {
|
|
|
|
pool,
|
|
|
|
phantomShares,
|
|
|
|
totalShares,
|
|
|
|
createdTime,
|
|
|
|
resolutionTime,
|
|
|
|
isResolved,
|
|
|
|
} = contract
|
|
|
|
|
|
|
|
const truePool = pool.YES + pool.NO
|
|
|
|
const prob = getProbability(totalShares)
|
2021-12-14 10:23:32 +00:00
|
|
|
const probPercent = Math.round(prob * 100) + '%'
|
2022-01-12 19:01:04 +00:00
|
|
|
|
|
|
|
const startProb = getProbability(phantomShares)
|
|
|
|
|
2021-12-14 10:23:32 +00:00
|
|
|
const createdDate = dayjs(createdTime).format('MMM D')
|
2022-01-12 19:01:04 +00:00
|
|
|
|
2021-12-15 07:41:50 +00:00
|
|
|
const resolvedDate = isResolved
|
|
|
|
? dayjs(resolutionTime).format('MMM D')
|
|
|
|
: undefined
|
2022-01-12 19:01:04 +00:00
|
|
|
|
2021-12-24 21:06:01 +00:00
|
|
|
return { truePool, probPercent, startProb, createdDate, resolvedDate }
|
2021-12-14 10:23:32 +00:00
|
|
|
}
|
|
|
|
|
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
|
|
|
|
2022-01-07 19:27:59 +00:00
|
|
|
export async function updateContract(
|
|
|
|
contractId: string,
|
|
|
|
update: Partial<Contract>
|
|
|
|
) {
|
|
|
|
const docRef = doc(db, 'contracts', contractId)
|
|
|
|
await updateDoc(docRef, update)
|
|
|
|
}
|
|
|
|
|
2021-12-17 23:16:42 +00:00
|
|
|
export async function pushNewContract(contract: Omit<Contract, 'id'>) {
|
|
|
|
const newContractRef = doc(contractCollection)
|
|
|
|
const fullContract: Contract = { ...contract, id: newContractRef.id }
|
|
|
|
|
|
|
|
await setDoc(newContractRef, fullContract)
|
|
|
|
return fullContract
|
|
|
|
}
|
|
|
|
|
|
|
|
export async function getContractFromId(contractId: string) {
|
2021-12-14 01:07:28 +00:00
|
|
|
const docRef = doc(db, 'contracts', contractId)
|
|
|
|
const result = await getDoc(docRef)
|
|
|
|
|
2021-12-14 01:09:16 +00:00
|
|
|
return result.exists() ? (result.data() as Contract) : undefined
|
2021-12-14 01:07:28 +00:00
|
|
|
}
|
|
|
|
|
2021-12-17 23:16:42 +00:00
|
|
|
export async function getContractFromSlug(slug: string) {
|
|
|
|
const q = query(contractCollection, where('slug', '==', slug))
|
|
|
|
const snapshot = await getDocs(q)
|
|
|
|
|
|
|
|
return snapshot.empty ? undefined : (snapshot.docs[0].data() as Contract)
|
|
|
|
}
|
|
|
|
|
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)
|
2021-12-14 01:09:16 +00:00
|
|
|
return snapshot.docs.map((doc) => doc.data() as Contract)
|
|
|
|
}
|
|
|
|
|
|
|
|
export async function listAllContracts(): Promise<Contract[]> {
|
2021-12-30 18:35:29 +00:00
|
|
|
const q = query(contractCollection, orderBy('createdTime', 'desc'))
|
2021-12-14 01:09:16 +00:00
|
|
|
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
|
|
|
|
2021-12-17 04:44:48 +00:00
|
|
|
export function listenForContracts(
|
|
|
|
setContracts: (contracts: Contract[]) => void
|
|
|
|
) {
|
2021-12-30 18:35:29 +00:00
|
|
|
const q = query(contractCollection, orderBy('createdTime', 'desc'))
|
2021-12-17 04:44:48 +00:00
|
|
|
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
|
|
|
})
|
|
|
|
}
|
2022-01-05 06:32:52 +00:00
|
|
|
|
2022-01-09 20:26:51 +00:00
|
|
|
const hotContractsQuery = query(
|
|
|
|
contractCollection,
|
|
|
|
where('isResolved', '==', false),
|
2022-01-13 01:53:50 +00:00
|
|
|
where('visibility', '==', 'public'),
|
2022-01-09 20:26:51 +00:00
|
|
|
orderBy('volume24Hours', 'desc'),
|
2022-01-15 06:43:15 +00:00
|
|
|
limit(16)
|
2022-01-09 20:26:51 +00:00
|
|
|
)
|
|
|
|
|
2022-01-15 06:43:15 +00:00
|
|
|
function chooseHotContracts(contracts: Contract[]) {
|
|
|
|
const fiveMinutes = 5 * 60 * 1000
|
|
|
|
const seed = Math.round(Date.now() / fiveMinutes).toString()
|
|
|
|
shuffle(contracts, createRNG(seed))
|
|
|
|
return contracts.slice(0, 4)
|
|
|
|
}
|
|
|
|
|
2022-01-09 20:26:51 +00:00
|
|
|
export function listenForHotContracts(
|
|
|
|
setHotContracts: (contracts: Contract[]) => void
|
|
|
|
) {
|
2022-01-15 06:43:15 +00:00
|
|
|
return listenForValues<Contract>(hotContractsQuery, (contracts) =>
|
|
|
|
setHotContracts(chooseHotContracts(contracts))
|
|
|
|
)
|
2022-01-05 06:32:52 +00:00
|
|
|
}
|
|
|
|
|
2022-01-09 20:26:51 +00:00
|
|
|
export function getHotContracts() {
|
2022-01-15 06:43:15 +00:00
|
|
|
return getValues<Contract>(hotContractsQuery).then(chooseHotContracts)
|
2022-01-05 06:32:52 +00:00
|
|
|
}
|