From 7e7b08dc94cf41dfeb2418dd04671e05aee29d6a Mon Sep 17 00:00:00 2001
From: James Grugett <jahooma@gmail.com>
Date: Tue, 15 Mar 2022 00:12:39 -0500
Subject: [PATCH] In progress

---
 web/components/answers/answers-graph.tsx | 151 +++++++++++++++++++++++
 web/components/contract-overview.tsx     |   7 +-
 web/components/contract-prob-graph.tsx   |   2 +-
 3 files changed, 158 insertions(+), 2 deletions(-)
 create mode 100644 web/components/answers/answers-graph.tsx

diff --git a/web/components/answers/answers-graph.tsx b/web/components/answers/answers-graph.tsx
new file mode 100644
index 00000000..212de786
--- /dev/null
+++ b/web/components/answers/answers-graph.tsx
@@ -0,0 +1,151 @@
+import { DatumValue } from '@nivo/core'
+import { ResponsiveLine } from '@nivo/line'
+import dayjs from 'dayjs'
+import _ from 'lodash'
+
+import { Bet } from '../../../common/bet'
+import { Contract } from '../../../common/contract'
+import { getOutcomeProbability } from '../../../common/calculate'
+import { useBets } from '../../hooks/use-bets'
+import { useWindowSize } from '../../hooks/use-window-size'
+
+export function AnswersGraph(props: { contract: Contract; bets: Bet[] }) {
+  const { contract } = props
+  const { resolutionTime, closeTime, answers } = contract
+
+  const bets = (useBets(contract.id) ?? props.bets).filter((bet) => !bet.sale)
+  const betsByOutcome = _.groupBy(bets, (bet) => bet.outcome)
+
+  const isClosed = !!closeTime && Date.now() > closeTime
+  const latestTime = dayjs(
+    resolutionTime && isClosed
+      ? Math.min(resolutionTime, closeTime)
+      : isClosed
+      ? closeTime
+      : resolutionTime ?? Date.now()
+  )
+
+  const outcomes = Object.keys(betsByOutcome).filter((outcome) => {
+    const maxProb = Math.max(
+      ...betsByOutcome[outcome].map((bet) => bet.probAfter)
+    )
+    return outcome !== '0' && maxProb > 0.05
+  })
+
+  const { width } = useWindowSize()
+
+  const labelLength = !width || width > 800 ? 75 : 20
+
+  const sortedOutcomes = _.sortBy(
+    outcomes,
+    (outcome) => -1 * getOutcomeProbability(totalShares, outcome)
+  ).slice(0, 5)
+
+  const colors = ['#2a81e3', '#c72ae3', '#b91111', '#f3ad28', '#11b981']
+
+  const times = _.sortBy(bets.map((bet) => bet.createdTime))
+  const porbs = []
+  const totalShares = { '0': contract.totalShares['0'] }
+  for (const bet of bets) {
+  }
+
+  const dateTimes = times.map((time) => new Date(time))
+
+  const data = sortedOutcomes.map((outcome, i) => {
+    const bets = _.sortBy(betsByOutcome[outcome], (bet) => bet.createdTime)
+
+    const probs: number[] = []
+    let betIndex: number = 0
+    for (const time of times) {
+      if (betIndex === bets.length) probs.push(bets[betIndex - 1].probAfter)
+      else if (time < bets[betIndex].createdTime) {
+        probs.push(betIndex === 0 ? 0 : bets[betIndex - 1].probAfter)
+      } else {
+        probs.push(bets[betIndex].probAfter)
+        betIndex++
+      }
+    }
+
+    if (resolutionTime || isClosed) {
+      dateTimes.push(latestTime.toDate())
+      probs.push(probs[probs.length - 1])
+    } else {
+      // Add a fake datapoint in future so the line continues horizontally
+      // to the right.
+      dateTimes.push(latestTime.add(1, 'month').toDate())
+      probs.push(probs[probs.length - 1])
+    }
+
+    const points = probs.map((prob, i) => ({
+      x: dateTimes[i],
+      y: prob * 100,
+    }))
+
+    const answer =
+      answers?.find((answer) => answer.id === outcome)?.text ?? 'None'
+
+    const answerText =
+      answer.slice(0, labelLength) + (answer.length > labelLength ? '...' : '')
+    return { id: answerText, data: points, color: colors[i] }
+  })
+
+  data.reverse()
+
+  const yTickValues = [0, 25, 50, 75, 100]
+
+  const numXTickValues = !width || width < 800 ? 2 : 5
+  const hoursAgo = latestTime.subtract(5, 'hours')
+  const startDate = dayjs(contract.createdTime).isBefore(hoursAgo)
+    ? new Date(contract.createdTime)
+    : hoursAgo.toDate()
+
+  const lessThanAWeek = dayjs(startDate).add(1, 'week').isAfter(latestTime)
+
+  return (
+    <div
+      className="w-full overflow-hidden"
+      style={{ height: !width || width >= 800 ? 350 : 250 }}
+    >
+      <ResponsiveLine
+        data={data}
+        yScale={{ min: 0, max: 100, type: 'linear' }}
+        yFormat={formatPercent}
+        gridYValues={yTickValues}
+        axisLeft={{
+          tickValues: yTickValues,
+          format: formatPercent,
+        }}
+        xScale={{
+          type: 'time',
+          min: startDate,
+          max: latestTime.toDate(),
+        }}
+        xFormat={(d) => formatTime(+d.valueOf(), lessThanAWeek)}
+        axisBottom={{
+          tickValues: numXTickValues,
+          format: (time) => formatTime(+time, lessThanAWeek),
+        }}
+        colors={{ datum: 'color' }}
+        pointSize={0}
+        enableSlices="x"
+        enableGridX={!!width && width >= 800}
+        enableArea
+        margin={{ top: 20, right: 28, bottom: 22, left: 40 }}
+      />
+    </div>
+  )
+}
+
+function formatPercent(y: DatumValue) {
+  return `${Math.round(+y.toString())}%`
+}
+
+function formatTime(time: number, includeTime: boolean) {
+  const d = dayjs(time)
+
+  if (d.isSame(Date.now(), 'day')) return d.format('ha')
+
+  if (includeTime) return dayjs(time).format('MMM D, ha')
+
+  return dayjs(time).format('MMM D')
+}
diff --git a/web/components/contract-overview.tsx b/web/components/contract-overview.tsx
index d5a14573..f92f2ade 100644
--- a/web/components/contract-overview.tsx
+++ b/web/components/contract-overview.tsx
@@ -19,6 +19,7 @@ import BetRow from './bet-row'
 import { Fold } from '../../common/fold'
 import { FoldTagList } from './tags-list'
 import { ContractActivity } from './feed/contract-activity'
+import { AnswersGraph } from './answers/answers-graph'
 
 export const ContractOverview = (props: {
   contract: Contract
@@ -77,7 +78,11 @@ export const ContractOverview = (props: {
 
       <Spacer h={4} />
 
-      {isBinary && <ContractProbGraph contract={contract} bets={bets} />}
+      {isBinary ? (
+        <ContractProbGraph contract={contract} bets={bets} />
+      ) : (
+        <AnswersGraph contract={contract} bets={bets} />
+      )}
 
       {children}
 
diff --git a/web/components/contract-prob-graph.tsx b/web/components/contract-prob-graph.tsx
index d764179c..8ece4335 100644
--- a/web/components/contract-prob-graph.tsx
+++ b/web/components/contract-prob-graph.tsx
@@ -60,7 +60,7 @@ export function ContractProbGraph(props: { contract: Contract; bets: Bet[] }) {
 
   return (
     <div
-      className="w-full"
+      className="w-full overflow-hidden"
       style={{ height: !width || width >= 800 ? 400 : 250 }}
     >
       <ResponsiveLine