From 26c9555d2592096840b18fb6aeb3e97eabedaf4a Mon Sep 17 00:00:00 2001
From: cong_wang <congwang@blocksec.com>
Date: Wed, 5 Jun 2024 14:24:23 +0800
Subject: [PATCH] fix: Fix the address resolution error and FundFlow button
 style issue on the Solscan website

---
 CHANGELOG.md                                  |  4 +++
 package.json                                  |  2 +-
 src/common/utils/validator.ts                 | 12 +++++--
 .../FundFlowButton/index.module.less          |  1 -
 .../components/FundFlowButton/index.tsx       | 14 +++++++--
 .../solscan/feat-scripts/fund-flow.tsx        | 31 ++++++++++++++-----
 src/content/solscan/page-scripts/account.tsx  |  4 +--
 src/content/solscan/page-scripts/token.tsx    |  4 +--
 .../components/ExportTableDataBtn/index.tsx   |  4 +--
 .../feat-scripts/compliance-score.tsx         |  4 +--
 .../feat-scripts/decompile-in-ethervm.tsx     |  4 +--
 .../tronscan/feat-scripts/fund-flow.tsx       |  4 +--
 .../feat-scripts/main-address-label.tsx       |  4 +--
 13 files changed, 62 insertions(+), 30 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index fd9a74e..fbff69e 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,3 +1,7 @@
+### v5.0.5
+
+- [fix] Fix the bug in the Fund Flow feature on the Solscan website that causes the current Solana address to be retrieved incorrectly
+
 ### v5.0.4
 
 - [feat] Enhance the address label functionality on the Solscan token page
diff --git a/package.json b/package.json
index b0d19d7..b0a3b62 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "metasuites",
-  "version": "5.0.4",
+  "version": "5.0.5",
   "repository": {
     "type": "git",
     "url": "https://github.com/blocksecteam/metasuites.git"
diff --git a/src/common/utils/validator.ts b/src/common/utils/validator.ts
index a4b57f1..ec8fab2 100644
--- a/src/common/utils/validator.ts
+++ b/src/common/utils/validator.ts
@@ -37,8 +37,14 @@ export const isMethod = (str: string, isExact = true): boolean => {
 export const pickAddress = (str: string): string | undefined => {
   return (
     str.match(PATTERN_EVM_ADDRESS_LOOSE)?.[0] ||
-    str.match(PATTERN_BTC_ADDRESS_LOOSE)?.[0] ||
-    str.match(PATTERN_TRX_ADDRESS_LOOSE)?.[0] ||
-    str.match(PATTERN_SOLANA_ADDRESS_LOOSE)?.[0]
+    str.match(PATTERN_BTC_ADDRESS_LOOSE)?.[0]
   )
 }
+
+export const pickTrxAddress = (str: string): string | undefined => {
+  return str.match(PATTERN_TRX_ADDRESS_LOOSE)?.[0]
+}
+
+export const pickSolanaAddress = (str: string): string | undefined => {
+  return str.match(PATTERN_SOLANA_ADDRESS_LOOSE)?.[0]
+}
diff --git a/src/content/solscan/components/FundFlowButton/index.module.less b/src/content/solscan/components/FundFlowButton/index.module.less
index 2d95217..7e4752d 100644
--- a/src/content/solscan/components/FundFlowButton/index.module.less
+++ b/src/content/solscan/components/FundFlowButton/index.module.less
@@ -2,6 +2,5 @@
 
 .button {
   display: flex !important;
-  margin-left: 16px;
   .items-center;
 }
diff --git a/src/content/solscan/components/FundFlowButton/index.tsx b/src/content/solscan/components/FundFlowButton/index.tsx
index be42a83..8a25b20 100644
--- a/src/content/solscan/components/FundFlowButton/index.tsx
+++ b/src/content/solscan/components/FundFlowButton/index.tsx
@@ -1,17 +1,24 @@
 import { type FC, useState } from 'react'
 import { Button, ConfigProvider } from 'antd'
 import { StyleProvider } from '@ant-design/cssinjs'
+import cls from 'classnames'
 
 import { ModalFundFlow, IconFundflow } from '@common/components'
+import type { BaseComponent } from '@common/types'
 
 import styles from './index.module.less'
 
-interface Props {
+interface Props extends BaseComponent {
   chain: string
   mainAddress: string
 }
 
-const FundFlowButton: FC<Props> = ({ chain, mainAddress }) => {
+const FundFlowButton: FC<Props> = ({
+  chain,
+  mainAddress,
+  className,
+  style
+}) => {
   const [visible, setVisible] = useState(false)
 
   return (
@@ -23,7 +30,8 @@ const FundFlowButton: FC<Props> = ({ chain, mainAddress }) => {
         <Button
           type="primary"
           icon={<IconFundflow size={20} />}
-          className={styles.button}
+          style={style}
+          className={cls(styles.button, className)}
           onClick={() => setVisible(true)}
         >
           Fund Flow
diff --git a/src/content/solscan/feat-scripts/fund-flow.tsx b/src/content/solscan/feat-scripts/fund-flow.tsx
index c32dd77..63e26f3 100644
--- a/src/content/solscan/feat-scripts/fund-flow.tsx
+++ b/src/content/solscan/feat-scripts/fund-flow.tsx
@@ -1,22 +1,37 @@
 import { createRoot } from 'react-dom/client'
 import $ from 'jquery'
+import type { CSSProperties } from 'react'
 
-import { pickAddress } from '@common/utils'
+import { pickSolanaAddress } from '@common/utils'
+import { type SOLSCAN_PAGE_NAMES, SOLSCAN_PAGES } from '@common/constants'
 
 import { FundFlowButton } from '../components'
 
-const renderFundFlowButton = async () => {
+const renderFundFlowButton = async (
+  pageName: (typeof SOLSCAN_PAGE_NAMES)[number]
+) => {
   $('#__metadock-modal-fund-flow__, #__metadock-fund-flow-btn__').remove()
-  const mainAddressEl = $(
-    '#__next > div:nth-of-type(1) > div:nth-of-type(3) > div:first-child > div:first-child > div:last-child > div:first-child > div:first-child > div'
-  )
-  const mainAddress = pickAddress(window.location.pathname)
+  const isAccountPage = pageName === SOLSCAN_PAGES.ACCOUNT.name
+  const selector = isAccountPage
+    ? '#__next > div:nth-of-type(1) > div:nth-of-type(3) > div:first-child > div:first-child > div:last-child > div:first-child > div:first-child > div'
+    : '#__next > div:nth-of-type(1) > div:nth-of-type(3) > div:first-child > div:first-child > div:last-child > div:first-child'
+  const mainAddressEl = $(selector)
+  const mainAddress = pickSolanaAddress(window.location.pathname)
   if (mainAddress) {
     const btnRootEl = $('<div id="__metadock-fund-flow-btn__"></div>')
-    btnRootEl.css({ display: 'inline-block', verticalAlign: 'middle' })
+    btnRootEl.css({
+      display: isAccountPage ? 'inline-block' : 'block',
+      verticalAlign: 'middle'
+    })
     mainAddressEl.append(btnRootEl)
+    const style: CSSProperties = {}
+    if (isAccountPage) {
+      style.marginLeft = 16
+    } else {
+      style.marginTop = 16
+    }
     createRoot(btnRootEl[0]).render(
-      <FundFlowButton chain="solana" mainAddress={mainAddress} />
+      <FundFlowButton chain="solana" mainAddress={mainAddress} style={style} />
     )
   }
 }
diff --git a/src/content/solscan/page-scripts/account.tsx b/src/content/solscan/page-scripts/account.tsx
index 1d0aa9e..5c9a7d3 100644
--- a/src/content/solscan/page-scripts/account.tsx
+++ b/src/content/solscan/page-scripts/account.tsx
@@ -1,7 +1,7 @@
 import browser from 'webextension-polyfill'
 
 import { store } from '@src/store'
-import { GET_SOLSCAN_ACCOUNT_TAB_DATA } from '@common/constants'
+import { GET_SOLSCAN_ACCOUNT_TAB_DATA, SOLSCAN_PAGES } from '@common/constants'
 
 import {
   renderFundFlowButton,
@@ -14,7 +14,7 @@ const initAccountPageScript = async () => {
   const { fundFlow, enhancedLabels } = await store.get('options')
 
   lazyLoad(() => {
-    if (fundFlow) renderFundFlowButton()
+    if (fundFlow) renderFundFlowButton(SOLSCAN_PAGES.ACCOUNT.name)
     if (enhancedLabels) {
       renderMainAddressLabel()
       renderEnhancedLabels()
diff --git a/src/content/solscan/page-scripts/token.tsx b/src/content/solscan/page-scripts/token.tsx
index 0c3af9e..f25e784 100644
--- a/src/content/solscan/page-scripts/token.tsx
+++ b/src/content/solscan/page-scripts/token.tsx
@@ -1,7 +1,7 @@
 import browser from 'webextension-polyfill'
 
 import { store } from '@src/store'
-import { GET_SOLSCAN_ACCOUNT_TAB_DATA } from '@common/constants'
+import { GET_SOLSCAN_ACCOUNT_TAB_DATA, SOLSCAN_PAGES } from '@common/constants'
 
 import { renderFundFlowButton, renderEnhancedLabels } from '../feat-scripts'
 import { lazyLoad } from '../helper'
@@ -10,7 +10,7 @@ const initTokenPageScript = async () => {
   const { fundFlow, enhancedLabels } = await store.get('options')
 
   lazyLoad(() => {
-    if (fundFlow) renderFundFlowButton()
+    if (fundFlow) renderFundFlowButton(SOLSCAN_PAGES.TOKEN.name)
     if (enhancedLabels) {
       renderEnhancedLabels()
     }
diff --git a/src/content/tronscan/components/ExportTableDataBtn/index.tsx b/src/content/tronscan/components/ExportTableDataBtn/index.tsx
index d231d9d..2662827 100644
--- a/src/content/tronscan/components/ExportTableDataBtn/index.tsx
+++ b/src/content/tronscan/components/ExportTableDataBtn/index.tsx
@@ -3,7 +3,7 @@ import { ConfigProvider, Dropdown, type MenuProps, Button, Space } from 'antd'
 import $ from 'jquery'
 import { DownOutlined } from '@ant-design/icons'
 
-import { pickAddress, downloadJson, downloadCsv } from '@common/utils'
+import { pickTrxAddress, downloadJson, downloadCsv } from '@common/utils'
 import { IconDownload } from '@common/components'
 
 import { getTextWithoutRemarkLabel } from '../../helper'
@@ -68,7 +68,7 @@ const ExportTableDataBtn: FC<Props> = ({ chain, table }) => {
 
         if (addressLinkEl.length && href) {
           const addr =
-            pickAddress(href) || pickAddress(window.location.hash) || text
+            pickTrxAddress(href) || pickTrxAddress(window.location.hash) || text
           const label = addressLinkEl.attr('data-label')
           row.push(`${addr}${label ? `(${label})` : ''}`)
         } else {
diff --git a/src/content/tronscan/feat-scripts/compliance-score.tsx b/src/content/tronscan/feat-scripts/compliance-score.tsx
index e61425a..f638732 100644
--- a/src/content/tronscan/feat-scripts/compliance-score.tsx
+++ b/src/content/tronscan/feat-scripts/compliance-score.tsx
@@ -4,13 +4,13 @@ import $ from 'jquery'
 import { chromeEvent } from '@common/event'
 import type { RiskScore } from '@common/api/types'
 import { GET_ADDRESS_RISK_SCORE } from '@common/constants'
-import { pickAddress } from '@common/utils'
+import { pickTrxAddress } from '@common/utils'
 
 import { ComplianceScoreLabel } from '../components'
 
 /** address compliance score*/
 const genComplianceScoresBtn = async (container: JQuery<HTMLElement>) => {
-  const mainAddress = pickAddress(window.location.hash)
+  const mainAddress = pickTrxAddress(window.location.hash)
 
   if (!mainAddress) return
 
diff --git a/src/content/tronscan/feat-scripts/decompile-in-ethervm.tsx b/src/content/tronscan/feat-scripts/decompile-in-ethervm.tsx
index 0801093..4e0bd46 100644
--- a/src/content/tronscan/feat-scripts/decompile-in-ethervm.tsx
+++ b/src/content/tronscan/feat-scripts/decompile-in-ethervm.tsx
@@ -2,7 +2,7 @@ import { createRoot } from 'react-dom/client'
 import $ from 'jquery'
 import browser from 'webextension-polyfill'
 
-import { pickAddress } from '@common/utils'
+import { pickTrxAddress } from '@common/utils'
 import { TRONSCAN_TABS_CHANGED } from '@common/constants'
 
 import { DecompileInEthervmBtn } from '../components'
@@ -11,7 +11,7 @@ import { lazyLoad } from '../helper'
 const genDecompileInEthervmBtn = () => {
   const main = async () => {
     if ($('#decompile-btn').length) return
-    const mainAddress = pickAddress(window.location.hash)
+    const mainAddress = pickTrxAddress(window.location.hash)
 
     if (!mainAddress) return
 
diff --git a/src/content/tronscan/feat-scripts/fund-flow.tsx b/src/content/tronscan/feat-scripts/fund-flow.tsx
index a7b8926..685872b 100644
--- a/src/content/tronscan/feat-scripts/fund-flow.tsx
+++ b/src/content/tronscan/feat-scripts/fund-flow.tsx
@@ -1,13 +1,13 @@
 import { createRoot } from 'react-dom/client'
 import $ from 'jquery'
 
-import { pickAddress } from '@common/utils'
+import { pickTrxAddress } from '@common/utils'
 
 import { FundFlowBtn } from '../components'
 
 /** fund flow */
 const genFundFlowBtn = async (container: JQuery<HTMLElement>) => {
-  const mainAddress = pickAddress(window.location.hash)
+  const mainAddress = pickTrxAddress(window.location.hash)
 
   if (!mainAddress) return
 
diff --git a/src/content/tronscan/feat-scripts/main-address-label.tsx b/src/content/tronscan/feat-scripts/main-address-label.tsx
index fd736c8..d4264a2 100644
--- a/src/content/tronscan/feat-scripts/main-address-label.tsx
+++ b/src/content/tronscan/feat-scripts/main-address-label.tsx
@@ -5,12 +5,12 @@ import $ from 'jquery'
 import { chromeEvent } from '@common/event'
 import type { AddressLabel } from '@common/api/types'
 import { GET_IMPL_LABELS } from '@common/constants'
-import { pickAddress } from '@common/utils'
+import { pickTrxAddress } from '@common/utils'
 
 import { MainAddressLabel } from '../components'
 
 const genMainAddressLabel = async () => {
-  const mainAddress = pickAddress(window.location.hash)
+  const mainAddress = pickTrxAddress(window.location.hash)
 
   if (!mainAddress) return