feat: Update for new multisig script

Author: devchenyan

packages/neuron-ui/src/components/MultisigAddress/hooks.ts

@@ -1,5 +1,13 @@
 import React, { useCallback, useState, useEffect, useMemo } from 'react'
-import { isSuccessResponse, getMultisigAddress, DefaultLockInfo, addressToScript, scriptToAddress } from 'utils'
+import {
+  isSuccessResponse,
+  getMultisigAddress,
+  DefaultLockInfo,
+  addressToScript,
+  scriptToAddress,
+  LegacyMultiSigLockInfo,
+  MultiSigLockInfo,
+} from 'utils'
 import { MultisigOutputUpdate } from 'services/subjects'
 import {
   MultisigConfig,
@@ -11,11 +19,13 @@ import {
   exportMultisigConfig,
   deleteMultisigConfig,
   getMultisigBalances,
+  getMultisigDAOBalances,
   loadMultisigTxJson,
   OfflineSignJSON,
   getMultisigSyncProgress,
 } from 'services/remote'
 import { computeScriptHash } from '@ckb-lumos/lumos/utils'
+import { remaindRegenerateMultisigAddress } from 'services/localCache'
 
 export const useSearch = (clearSelected: () => void, onFilterConfig: (searchKey: string) => void) => {
   const [keywords, setKeywords] = useState('')
@@ -55,13 +65,45 @@ export const useSearch = (clearSelected: () => void, onFilterConfig: (searchKey:
 export const useConfigManage = ({ walletId, isMainnet }: { walletId: string; isMainnet: boolean }) => {
   const [entities, setEntities] = useState<MultisigEntity[]>([])
   const saveConfig = useCallback(
-    ({ m, n, r, addresses }: { m: number; n: number; r: number; addresses: string[] }) => {
+    ({
+      m,
+      n,
+      r,
+      addresses,
+      lockCodeHash = MultiSigLockInfo.CodeHash,
+    }: {
+      m: number
+      n: number
+      r: number
+      addresses: string[]
+      lockCodeHash?: string
+    }) => {
       return saveMultisigConfig({
         m,
         n,
         r,
         blake160s: addresses.map(v => addressToScript(v).args),
         walletId,
+        lockCodeHash,
+      }).then(res => {
+        if (isSuccessResponse(res)) {
+          setEntities(v => (res.result ? [res.result, ...v] : v))
+        } else {
+          throw new Error(typeof res.message === 'string' ? res.message : res.message.content)
+        }
+      })
+    },
+    [walletId, setEntities]
+  )
+  const regenerateConfig = useCallback(
+    (config: MultisigConfig) => {
+      return saveMultisigConfig({
+        m: config.m,
+        n: config.n,
+        r: config.r,
+        blake160s: config.blake160s,
+        walletId,
+        lockCodeHash: MultiSigLockInfo.CodeHash,
       }).then(res => {
         if (isSuccessResponse(res)) {
           setEntities(v => (res.result ? [res.result, ...v] : v))
@@ -132,7 +174,8 @@ export const useConfigManage = ({ walletId, isMainnet }: { walletId: string; isM
             { isMainnet }
           )
         ),
-        fullPayload: getMultisigAddress(entity.blake160s, entity.r, entity.m, entity.n, isMainnet),
+        fullPayload: getMultisigAddress(entity.blake160s, entity.r, entity.m, entity.n, isMainnet, entity.lockCodeHash),
+        isLegacy: entity.lockCodeHash === LegacyMultiSigLockInfo.CodeHash,
       })),
     [entities, isMainnet]
   )
@@ -145,6 +188,7 @@ export const useConfigManage = ({ walletId, isMainnet }: { walletId: string; isM
   )
   return {
     saveConfig,
+    regenerateConfig,
     allConfigs,
     onUpdateConfigAlias,
     onUpdateConfig,
@@ -267,6 +311,47 @@ const useDeleteAction = (deleteConfigById: (id: number) => void) => {
   }
 }
 
+const useRegenerateAction = (regenerateConfig: (config: MultisigConfig) => Promise<void>) => {
+  const [isDialogOpen, setIsDialogOpen] = useState(false)
+  const [config, setConfig] = useState<MultisigConfig>()
+  const [isNoRemind, setIsNoRemind] = useState(false)
+  const [regenerateErrorMessage, setRegenerateErrorMessage] = useState<string | undefined>()
+
+  const handleCheckbox = useCallback(
+    (e: React.ChangeEvent<HTMLInputElement>) => {
+      const { checked } = e.target
+      setIsNoRemind(checked)
+      remaindRegenerateMultisigAddress.save(checked)
+    },
+    [setIsNoRemind]
+  )
+
+  const regenerate = useCallback(
+    async (multisigConfig: MultisigConfig) => {
+      try {
+        await regenerateConfig(multisigConfig)
+      } catch (error) {
+        setIsDialogOpen(true)
+        setRegenerateErrorMessage(error instanceof Error ? error.message : '')
+      }
+    },
+    [setIsDialogOpen, config, isNoRemind]
+  )
+  const closeDialog = useCallback(() => {
+    setIsDialogOpen(false)
+  }, [setIsDialogOpen])
+  return {
+    action: regenerate,
+    closeDialog,
+    config,
+    setConfig,
+    isDialogOpen,
+    isNoRemind,
+    handleCheckbox,
+    regenerateErrorMessage,
+  }
+}
+
 const useApproveAction = () => {
   const [isDialogOpen, setIsDialogOpen] = useState(false)
   const [multisigConfig, setMultisigConfig] = useState<MultisigConfig | undefined>()
@@ -342,14 +427,21 @@ const useDaoWithdrawAction = () => {
   }
 }
 
-export const useActions = ({ deleteConfigById }: { deleteConfigById: (id: number) => void }) => {
+export const useActions = ({
+  deleteConfigById,
+  regenerateConfig,
+}: {
+  deleteConfigById: (id: number) => void
+  regenerateConfig: (config: MultisigConfig) => Promise<void>
+}) => {
   return {
     deleteAction: useDeleteAction(deleteConfigById),
     infoAction: useInfoAction(),
     sendAction: useSendAction(),
     approveAction: useApproveAction(),
     daoDepositAction: useDaoDepositAction(),
     daoWithdrawAction: useDaoWithdrawAction(),
+    regenerateAction: useRegenerateAction(regenerateConfig),
   }
 }
 
@@ -365,14 +457,20 @@ export const useSubscription = ({
   isLightClient: boolean
 }) => {
   const [multisigBanlances, setMultisigBanlances] = useState<Record<string, string>>({})
+  const [multisigDaoBalances, setMultisigDaoBalances] = useState<Record<string, string>>({})
   const [multisigSyncProgress, setMultisigSyncProgress] = useState<Record<string, number>>({})
   const getAndSaveMultisigBalances = useCallback(() => {
     getMultisigBalances({ isMainnet, multisigAddresses: configs.map(v => v.fullPayload) }).then(res => {
       if (isSuccessResponse(res) && res.result) {
         setMultisigBanlances(res.result)
       }
     })
-  }, [setMultisigBanlances, isMainnet, configs])
+    getMultisigDAOBalances({ isMainnet, multisigAddresses: configs.map(v => v.fullPayload) }).then(res => {
+      if (isSuccessResponse(res) && res.result) {
+        setMultisigDaoBalances(res.result)
+      }
+    })
+  }, [setMultisigBanlances, setMultisigDaoBalances, isMainnet, configs])
   const hashToPayload = useMemo(
     () =>
       configs.reduce<Record<string, string>>(
@@ -415,7 +513,7 @@ export const useSubscription = ({
       clearInterval(interval)
     }
   }, [isLightClient, getAndSaveMultisigSyncProgress])
-  return { multisigBanlances, multisigSyncProgress }
+  return { multisigBanlances, multisigDaoBalances, multisigSyncProgress }
 }
 
 export const useCancelWithLightClient = () => {