manifold/web/components/bet-panel.tsx

import { getFunctions, httpsCallable } from 'firebase/functions'
import clsx from 'clsx'
import React, { useEffect, useState } from 'react'

import { useUser } from '../hooks/use-user'
import { Contract } from '../lib/firebase/contracts'
import { Col } from './layout/col'
import { Row } from './layout/row'
import { Spacer } from './layout/spacer'
import { YesNoSelector } from './yes-no-selector'
import {
  formatMoney,
  formatPercent,
  formatWithCommas,
} from '../lib/util/format'
import { Title } from './title'
import {
  getProbability,
  calculateShares,
  getProbabilityAfterBet,
  calculatePayoutAfterCorrectBet,
} from '../lib/calculate'
import { firebaseLogin } from '../lib/firebase/users'
import { AddFundsButton } from './add-funds-button'
import { OutcomeLabel } from './outcome-label'
import { AdvancedPanel } from './advanced-panel'
import { Bet } from '../lib/firebase/bets'
import { placeBet } from '../lib/firebase/api-call'

export function BetPanel(props: { contract: Contract; className?: string }) {
  useEffect(() => {
    // warm up cloud function
    placeBet({}).catch()
  }, [])

  const { contract, className } = props

  const user = useUser()

  const [betChoice, setBetChoice] = useState<'YES' | 'NO'>('YES')
  const [betAmount, setBetAmount] = useState<number | undefined>(undefined)

  const [error, setError] = useState<string | undefined>()
  const [isSubmitting, setIsSubmitting] = useState(false)
  const [wasSubmitted, setWasSubmitted] = useState(false)

  function onBetChoice(choice: 'YES' | 'NO') {
    setBetChoice(choice)
    setWasSubmitted(false)
  }

  function onBetChange(str: string) {
    setWasSubmitted(false)

    const amount = parseInt(str.replace(/[^\d]/, ''))

    if (str && isNaN(amount)) return

    setBetAmount(str ? amount : undefined)

    if (user && user.balance < amount) setError('Insufficient balance')
    else setError(undefined)
  }

  async function submitBet() {
    if (!user || !betAmount) return

    if (user.balance < betAmount) {
      setError('Insufficient balance')
      return
    }

    setError(undefined)
    setIsSubmitting(true)

    const result = await placeBet({
      amount: betAmount,
      outcome: betChoice,
      contractId: contract.id,
    }).then((r) => r.data as any)

    console.log('placed bet. Result:', result)

    if (result?.status === 'success') {
      setIsSubmitting(false)
      setWasSubmitted(true)
      setBetAmount(undefined)
    } else {
      setError(result?.error || 'Error placing bet')
      setIsSubmitting(false)
    }
  }

  const betDisabled = isSubmitting || !betAmount || error

  const initialProb = getProbability(contract.pool)
  const resultProb = getProbabilityAfterBet(
    contract.pool,
    betChoice,
    betAmount ?? 0
  )
  const shares = calculateShares(contract.pool, betAmount ?? 0, betChoice)

  const estimatedWinnings = Math.floor(shares)
  const estimatedReturn = betAmount
    ? (estimatedWinnings - betAmount) / betAmount
    : 0
  const estimatedReturnPercent = (estimatedReturn * 100).toFixed() + '%'

  const remainingBalance = (user?.balance || 0) - (betAmount || 0)

  return (
    <Col
      className={clsx('bg-gray-100 shadow-md px-8 py-6 rounded-md', className)}
    >
      <Title className="!mt-0 whitespace-nowrap" text={`Buy ${betChoice}`} />

      <div className="mt-2 mb-1 text-sm text-gray-400">Outcome</div>
      <YesNoSelector
        className="my-2"
        selected={betChoice}
        onSelect={(choice) => onBetChoice(choice)}
      />

      <div className="mt-3 mb-1 text-sm text-gray-400">
        Amount{' '}
        {user && (
          <span className="float-right">
            {formatMoney(
              remainingBalance > 0 ? Math.floor(remainingBalance) : 0
            )}{' '}
            left
          </span>
        )}
      </div>
      <Col className="my-2">
        <label className="input-group">
          <span className="text-sm bg-gray-200">M$</span>
          <input
            className={clsx(
              'input input-bordered w-full',
              error && 'input-error'
            )}
            type="text"
            placeholder="0"
            maxLength={9}
            value={betAmount ?? ''}
            onChange={(e) => onBetChange(e.target.value)}
          />
        </label>
        {error && (
          <div className="font-medium tracking-wide text-red-500 text-xs mt-3">
            {error}
          </div>
        )}
        {user && <AddFundsButton className="self-end mt-3" />}
      </Col>

      <div className="mt-2 mb-1 text-sm text-gray-400">Implied probability</div>
      <Row>
        <div>{formatPercent(initialProb)}</div>
        <div className="mx-2">→</div>
        <div>{formatPercent(resultProb)}</div>
      </Row>

      <div className="mt-2 mb-1 text-sm text-gray-400">
        Estimated max payout
      </div>
      <div>
        {formatMoney(estimatedWinnings)} &nbsp;{' '}
        {estimatedWinnings ? <span>(+{estimatedReturnPercent})</span> : null}
      </div>

      <AdvancedPanel>
        <div className="mt-2 mb-1 text-sm text-gray-400">
          <OutcomeLabel outcome={betChoice} /> shares
        </div>
        <div>
          {formatWithCommas(shares)} of{' '}
          {formatWithCommas(shares + contract.totalShares[betChoice])}
        </div>

        <div className="mt-2 mb-1 text-sm text-gray-400">
          Current payout if <OutcomeLabel outcome={betChoice} />
        </div>
        <div>
          {formatMoney(
            betAmount
              ? calculatePayoutAfterCorrectBet(contract, {
                  outcome: betChoice,
                  amount: betAmount,
                  shares,
                } as Bet)
              : 0
          )}
        </div>
      </AdvancedPanel>

      <Spacer h={6} />

      {user ? (
        <button
          className={clsx(
            'btn',
            betDisabled
              ? 'btn-disabled'
              : betChoice === 'YES'
              ? 'btn-primary'
              : 'bg-red-400 hover:bg-red-500 border-none',
            isSubmitting ? 'loading' : ''
          )}
          onClick={betDisabled ? undefined : submitBet}
        >
          {isSubmitting ? 'Submitting...' : 'Submit trade'}
        </button>
      ) : (
        <button
          className="btn mt-4 border-none normal-case text-lg font-medium px-10 bg-gradient-to-r from-teal-500 to-green-500 hover:from-teal-600 hover:to-green-600"
          onClick={firebaseLogin}
        >
          Sign in to trade!
        </button>
      )}

      {wasSubmitted && <div className="mt-4">Trade submitted!</div>}
    </Col>
  )
}