claude-code-best/src/context/mailbox.tsx

import React, { createContext, useContext, useMemo } from 'react'
import { Mailbox } from '../utils/mailbox.js'

const MailboxContext = createContext<Mailbox | undefined>(undefined)

type Props = {
  children: React.ReactNode
}

export function MailboxProvider({ children }: Props): React.ReactNode {
  const mailbox = useMemo(() => new Mailbox(), [])
  return (
    <MailboxContext.Provider value={mailbox}>
      {children}
    </MailboxContext.Provider>
  )
}

export function useMailbox(): Mailbox {
  const mailbox = useContext(MailboxContext)
  if (!mailbox) {
    throw new Error('useMailbox must be used within a MailboxProvider')
  }
  return mailbox
}