mango-ui-v3/components/Layout.tsx

import SideNav from './SideNav'
import { breakpoints } from '../components/TradePageGrid'
import { useViewport } from 'hooks/useViewport'
import BottomBar from './mobile/BottomBar'
import { ConnectWalletButton } from './ConnectWalletButton'
import GlobalNotification from './GlobalNotification'
import { abbreviateAddress } from 'utils'
import { useCallback, useEffect, useState } from 'react'
import AccountsModal from './AccountsModal'
import { useRouter } from 'next/router'
import FavoritesShortcutBar from './FavoritesShortcutBar'
import {
  ArrowRightIcon,
  ChevronRightIcon,
  CogIcon,
  ExclamationCircleIcon,
  UsersIcon,
} from '@heroicons/react/solid'
import Button, { IconButton } from './Button'
import SettingsModal from './SettingsModal'
import { useTranslation } from 'next-i18next'
import { useWallet } from '@solana/wallet-adapter-react'
import DepositModal from './DepositModal'
import WithdrawModal from './WithdrawModal'
import Tooltip from './Tooltip'
import useMangoStore from 'stores/useMangoStore'

const Layout = ({ children }) => {
  const [isCollapsed, setIsCollapsed] = useState(false)
  const { width } = useViewport()
  const isMobile = width ? width < breakpoints.sm : false
  const router = useRouter()
  const { pathname } = router

  useEffect(() => {
    const collapsed = width ? width <= breakpoints.xl : false
    setIsCollapsed(collapsed)
  }, [])

  const handleToggleSidebar = () => {
    setIsCollapsed(!isCollapsed)
    setTimeout(() => {
      window.dispatchEvent(new Event('resize'))
    }, 100)
  }

  return (
    <div className={`flex-grow bg-th-bkg-1 text-th-fgd-1 transition-all`}>
      <div className="flex">
        {isMobile ? (
          <div className="fixed bottom-0 left-0 z-20 w-full md:hidden">
            <BottomBar />
          </div>
        ) : (
          <div className={isCollapsed ? 'mr-[64px]' : 'mr-[220px]'}>
            <div className={`fixed z-20 h-screen`}>
              <button
                className="absolute -right-4 top-1/2 z-20 h-10 w-4 -translate-y-1/2 transform rounded-none rounded-r bg-th-bkg-4 focus:outline-none"
                onClick={handleToggleSidebar}
              >
                <ChevronRightIcon
                  className={`default-transition h-full w-full ${
                    !isCollapsed ? 'rotate-180' : 'rotate-360'
                  }`}
                />
              </button>
              <div
                className={`h-full ${!isCollapsed ? 'overflow-y-auto' : ''}`}
              >
                <SideNav collapsed={isCollapsed} />
              </div>
            </div>
          </div>
        )}
        <div className="w-full overflow-hidden">
          <GlobalNotification />
          <TopBar />
          {pathname === '/' ? <FavoritesShortcutBar /> : null}
          <div className={pathname === '/' ? 'px-3' : 'px-6 pb-16 md:pb-6'}>
            {children}
          </div>
        </div>
      </div>
    </div>
  )
}

const TopBar = () => {
  const { t } = useTranslation(['common', 'delegate'])
  const { connected, publicKey } = useWallet()
  const mangoAccount = useMangoStore((s) => s.selectedMangoAccount.current)
  const initialLoad = useMangoStore((s) => s.selectedMangoAccount.initialLoad)
  const loading = useMangoStore((s) => s.selectedMangoAccount.loading)
  const router = useRouter()
  const [showAccountsModal, setShowAccountsModal] = useState(false)
  const [showSettingsModal, setShowSettingsModal] = useState(false)
  const [showDepositModal, setShowDepositModal] = useState(false)
  const [showWithdrawModal, setShowWithdrawModal] = useState(false)
  const { pubkey } = router.query
  const { width } = useViewport()
  const isMobile = width ? width < breakpoints.sm : false

  const handleCloseAccounts = useCallback(() => {
    setShowAccountsModal(false)
  }, [])

  const canWithdraw =
    mangoAccount?.owner && publicKey
      ? mangoAccount?.owner?.equals(publicKey)
      : false

  return (
    <>
      <div className="flex h-14 items-center justify-between border-b border-th-bkg-3 bg-th-bkg-1 px-6">
        {mangoAccount && mangoAccount.beingLiquidated ? (
          <div className="flex items-center justify-center">
            <ExclamationCircleIcon className="mr-1.5 h-5 w-5 flex-shrink-0 text-th-red" />
            <span className="text-th-red">{t('being-liquidated')}</span>
          </div>
        ) : (
          <div className="flex items-center text-th-fgd-3">
            <span className="mb-0 mr-2 text-base">
              {pubkey
                ? '🔎'
                : connected
                ? initialLoad
                  ? ''
                  : mangoAccount
                  ? '🟢'
                  : '👋'
                : !isMobile
                ? '🔗'
                : ''}
            </span>
            {connected || pubkey ? (
              !initialLoad && !loading ? (
                mangoAccount ? (
                  <div
                    className="default-transition flex items-center font-bold text-th-fgd-1 hover:text-th-fgd-3"
                    role="button"
                    onClick={() => setShowAccountsModal(true)}
                  >
                    {`${
                      mangoAccount.name
                        ? mangoAccount.name
                        : abbreviateAddress(mangoAccount.publicKey)
                    }`}
                    {publicKey && !mangoAccount.owner.equals(publicKey) ? (
                      <Tooltip content={t('delegate:delegated-account')}>
                        <UsersIcon className="ml-2 h-5 w-5 text-th-fgd-3" />
                      </Tooltip>
                    ) : (
                      ''
                    )}
                  </div>
                ) : (
                  <span className="flex items-center text-th-fgd-3">
                    {t('create-account-helper')}
                    <ArrowRightIcon className="sideways-bounce ml-2 h-5 w-5 text-th-fgd-1" />
                  </span>
                )
              ) : (
                <div className="h-4 w-32 animate-pulse rounded bg-th-bkg-3" />
              )
            ) : !isMobile ? (
              <span className="flex items-center text-th-fgd-3">
                {t('connect-helper')}
                <ArrowRightIcon className="sideways-bounce ml-2 h-5 w-5 text-th-fgd-1" />
              </span>
            ) : null}
          </div>
        )}
        <div className="flex items-center space-x-4">
          {!isMobile && connected && !initialLoad ? (
            <div className="flex space-x-2">
              {mangoAccount ? (
                <Button
                  className="flex h-8 w-[86px] items-center justify-center pl-3 pr-3 text-xs"
                  onClick={() => setShowDepositModal(true)}
                >
                  {t('deposit')}
                </Button>
              ) : (
                <Button
                  className="flex h-8 w-32 items-center justify-center pl-3 pr-3 text-xs"
                  onClick={() => setShowAccountsModal(true)}
                >
                  {t('create-account')}
                </Button>
              )}
              {canWithdraw ? (
                <Button
                  className="flex h-8 w-[86px] items-center justify-center pl-3 pr-3 text-xs"
                  onClick={() => setShowWithdrawModal(true)}
                  primary={false}
                >
                  {t('withdraw')}
                </Button>
              ) : null}
            </div>
          ) : null}
          <IconButton
            className="h-8 w-8"
            onClick={() => setShowSettingsModal(true)}
          >
            <CogIcon className="h-5 w-5" />
          </IconButton>
          <ConnectWalletButton />
        </div>
      </div>
      {showAccountsModal ? (
        <AccountsModal
          onClose={handleCloseAccounts}
          isOpen={showAccountsModal}
        />
      ) : null}
      {showSettingsModal ? (
        <SettingsModal
          onClose={() => setShowSettingsModal(false)}
          isOpen={showSettingsModal}
        />
      ) : null}
      {showDepositModal ? (
        <DepositModal
          isOpen={showDepositModal}
          onClose={() => setShowDepositModal(false)}
        />
      ) : null}
      {showWithdrawModal ? (
        <WithdrawModal
          isOpen={showWithdrawModal}
          onClose={() => setShowWithdrawModal(false)}
        />
      ) : null}
    </>
  )
}

export default Layout