From f4d3afaeae8b0a71df069634760084e59c8b9400 Mon Sep 17 00:00:00 2001
From: reallybeard <89934888+reallybeard@users.noreply.github.com>
Date: Mon, 24 Jun 2024 16:05:06 -0500
Subject: [PATCH] feat: add copy button for accounts (#7197)

* add copy button for accounts

* move the logic into the hook

* Update useCopyToClipboard.tsx
---
 src/components/InlineCopyButton.tsx           | 49 +++++++++++++++++++
 .../components/ImportAccounts.tsx             | 13 +++--
 src/hooks/useCopyToClipboard.tsx              | 36 ++++++++++++++
 .../Accounts/components/AccountNumberRow.tsx  | 18 +++++--
 4 files changed, 108 insertions(+), 8 deletions(-)
 create mode 100644 src/components/InlineCopyButton.tsx
 create mode 100644 src/hooks/useCopyToClipboard.tsx

diff --git a/src/components/InlineCopyButton.tsx b/src/components/InlineCopyButton.tsx
new file mode 100644
index 00000000000..d3938e1cafc
--- /dev/null
+++ b/src/components/InlineCopyButton.tsx
@@ -0,0 +1,49 @@
+import { CheckIcon, CopyIcon } from '@chakra-ui/icons'
+import { Flex, IconButton } from '@chakra-ui/react'
+import type { PropsWithChildren } from 'react'
+import React, { useCallback } from 'react'
+import { useTranslate } from 'react-polyglot'
+import { useCopyToClipboard } from 'hooks/useCopyToClipboard'
+
+import { TooltipWithTouch } from './TooltipWithTouch'
+
+type InlineCopyButtonProps = {
+  value: string
+  isDisabled?: boolean
+} & PropsWithChildren
+
+const copyIcon = <CopyIcon />
+const checkIcon = <CheckIcon />
+
+export const InlineCopyButton: React.FC<InlineCopyButtonProps> = ({
+  value,
+  children,
+  isDisabled,
+}) => {
+  const translate = useTranslate()
+  const { isCopied, copyToClipboard } = useCopyToClipboard({ timeout: 2000 })
+
+  const handleCopyClick = useCallback(() => {
+    copyToClipboard(value)
+  }, [copyToClipboard, value])
+
+  // Hide the copy button if it is disabled
+  if (isDisabled) return <>{children}</>
+
+  return (
+    <Flex gap={2} alignItems='center'>
+      {children}
+      <TooltipWithTouch label={translate(isCopied ? 'common.copied' : 'common.copy')}>
+        <IconButton
+          icon={isCopied ? checkIcon : copyIcon}
+          colorScheme={isCopied ? 'green' : 'gray'}
+          size='sm'
+          variant='ghost'
+          fontSize='xl'
+          aria-label='Copy value'
+          onClick={handleCopyClick}
+        />
+      </TooltipWithTouch>
+    </Flex>
+  )
+}
diff --git a/src/components/ManageAccountsDrawer/components/ImportAccounts.tsx b/src/components/ManageAccountsDrawer/components/ImportAccounts.tsx
index a74f368ea71..10e063e03d3 100644
--- a/src/components/ManageAccountsDrawer/components/ImportAccounts.tsx
+++ b/src/components/ManageAccountsDrawer/components/ImportAccounts.tsx
@@ -20,6 +20,7 @@ import { useCallback, useEffect, useMemo, useState } from 'react'
 import { useTranslate } from 'react-polyglot'
 import { accountManagement } from 'react-queries/queries/accountManagement'
 import { Amount } from 'components/Amount/Amount'
+import { InlineCopyButton } from 'components/InlineCopyButton'
 import { RawText } from 'components/Text'
 import { WalletActions } from 'context/WalletProvider/actions'
 import {
@@ -78,11 +79,13 @@ const TableRowAccount = forwardRef<TableRowAccountProps, 'div'>(({ asset, accoun
   return (
     <>
       <Td fontWeight='bold'>
-        <Tooltip label={pubkey} isDisabled={isUtxoAccount}>
-          <div ref={ref}>
-            <RawText>{accountLabel}</RawText>
-          </div>
-        </Tooltip>
+        <InlineCopyButton value={pubkey} isDisabled={isUtxoAccount}>
+          <Tooltip label={pubkey} isDisabled={isUtxoAccount}>
+            <div ref={ref}>
+              <RawText>{accountLabel}</RawText>
+            </div>
+          </Tooltip>
+        </InlineCopyButton>
       </Td>
       <Td textAlign='right'>
         {isLoading ? (
diff --git a/src/hooks/useCopyToClipboard.tsx b/src/hooks/useCopyToClipboard.tsx
new file mode 100644
index 00000000000..53806afcb15
--- /dev/null
+++ b/src/hooks/useCopyToClipboard.tsx
@@ -0,0 +1,36 @@
+import { useState } from 'react'
+
+export interface useCopyToClipboardProps {
+  timeout?: number
+}
+
+export function useCopyToClipboard({ timeout = 2000 }: useCopyToClipboardProps) {
+  const [isCopied, setIsCopied] = useState<boolean>(false)
+  const [isCopying, setIsCopying] = useState<boolean>(false)
+
+  const copyToClipboard = (value: string) => {
+    if (typeof window === 'undefined' || !navigator.clipboard?.writeText) {
+      return
+    }
+
+    if (!value) {
+      return
+    }
+
+    // Prevent race condition if the user clicks copy multiple times
+    if (isCopying) return
+    setIsCopying(true)
+
+    void navigator.clipboard.writeText(value).then(() => {
+      setIsCopied(true)
+
+      setTimeout(() => {
+        // Reset the state after the timeout
+        setIsCopying(false)
+        setIsCopied(false)
+      }, timeout)
+    })
+  }
+
+  return { isCopied, copyToClipboard }
+}
diff --git a/src/pages/Accounts/components/AccountNumberRow.tsx b/src/pages/Accounts/components/AccountNumberRow.tsx
index fad7fbe7d04..12bc56af5c6 100644
--- a/src/pages/Accounts/components/AccountNumberRow.tsx
+++ b/src/pages/Accounts/components/AccountNumberRow.tsx
@@ -1,4 +1,4 @@
-import { ArrowDownIcon, ArrowUpIcon } from '@chakra-ui/icons'
+import { ArrowDownIcon, ArrowUpIcon, CheckIcon, CopyIcon } from '@chakra-ui/icons'
 import type { ButtonProps } from '@chakra-ui/react'
 import {
   Avatar,
@@ -15,8 +15,8 @@ import {
   Stack,
   useDisclosure,
 } from '@chakra-ui/react'
-import type { AccountId, ChainId } from '@shapeshiftoss/caip'
-import { useMemo } from 'react'
+import { type AccountId, type ChainId, fromAccountId } from '@shapeshiftoss/caip'
+import { useCallback, useMemo } from 'react'
 import { MdOutlineMoreVert } from 'react-icons/md'
 import { RiWindow2Line } from 'react-icons/ri'
 import { useTranslate } from 'react-polyglot'
@@ -24,6 +24,7 @@ import { useSelector } from 'react-redux'
 import { Amount } from 'components/Amount/Amount'
 import { NestedList } from 'components/NestedList'
 import { RawText } from 'components/Text'
+import { useCopyToClipboard } from 'hooks/useCopyToClipboard'
 import { getAccountTitle } from 'lib/utils/accounts'
 import { isUtxoAccountId, isUtxoChainId } from 'lib/utils/utxo'
 import {
@@ -51,6 +52,8 @@ const mdOutlineMoreVertIcon = <MdOutlineMoreVert />
 const riWindow2LineIcon = <RiWindow2Line />
 const arrowDownIcon = <ArrowDownIcon />
 const arrowUpIcon = <ArrowUpIcon />
+const copyIcon = <CopyIcon />
+const checkIcon = <CheckIcon />
 
 const UtxoAccountEntries: React.FC<UtxoAccountEntriesProps> = ({ accountIds, chainId }) => {
   const feeAsset = useAppSelector(s => selectFeeAssetByChainId(s, chainId))
@@ -108,6 +111,7 @@ export const AccountNumberRow: React.FC<AccountNumberRowProps> = ({
   ...buttonProps
 }) => {
   const { isOpen, onToggle } = useDisclosure()
+  const { isCopied, copyToClipboard } = useCopyToClipboard({ timeout: 2000 })
   const translate = useTranslate()
   const assets = useSelector(selectAssets)
   const accountId = useMemo(() => accountIds[0], [accountIds]) // all accountIds belong to the same chain
@@ -148,6 +152,11 @@ export const AccountNumberRow: React.FC<AccountNumberRowProps> = ({
 
   const fontFamily = useMemo(() => (!isUtxoChainId(chainId) ? 'monospace' : ''), [chainId])
 
+  const handleCopyClick = useCallback(() => {
+    const account = fromAccountId(accountId).account
+    copyToClipboard(account)
+  }, [accountId, copyToClipboard])
+
   return (
     <ListItem>
       <Flex p={0} flexDir='row' display='flex' gap={2} alignItems='center'>
@@ -198,6 +207,9 @@ export const AccountNumberRow: React.FC<AccountNumberRowProps> = ({
                 <MenuItem onClick={onToggle} icon={isOpen ? arrowUpIcon : arrowDownIcon}>
                   {translate(isOpen ? 'accounts.hideAssets' : 'accounts.showAssets')}
                 </MenuItem>
+                <MenuItem onClick={handleCopyClick} icon={isCopied ? checkIcon : copyIcon}>
+                  {translate(isCopied ? 'common.copied' : 'common.copy')}
+                </MenuItem>
               </MenuGroup>
             </MenuList>
           </Menu>