feat: upgrade module query

.changeset/good-pets-cover.md

@@ -0,0 +1,6 @@
+---
+"@abstract-money/core": patch
+"@abstract-money/react": patch
+---
+
+Added query and action to upgrade a module.

packages/core/src/actions/account/wallet/upgrade-module.ts

@@ -0,0 +1,36 @@
+import { SigningCosmWasmClient } from '@cosmjs/cosmwasm-stargate'
+import { ManagerClient, VersionControlTypes } from '../../../codegen/abstract'
+import { WithCosmWasmSignOptions } from '../../../types/parameters'
+import { getManagerClientFromApi } from './get-manager-client-from-api'
+
+export type UpgradeModuleParameters = WithCosmWasmSignOptions<
+  {
+    accountId: VersionControlTypes.AccountId
+    signingCosmWasmClient: SigningCosmWasmClient
+    apiUrl: string
+    sender: string
+  } & Omit<
+    Parameters<typeof ManagerClient.prototype.upgrade>[0],
+    'accountId'
+  > & { subAccountId?: number }
+>
+
+export async function upgradeModule({
+  accountId,
+  subAccountId,
+  signingCosmWasmClient,
+  apiUrl,
+  sender,
+  fee,
+  memo,
+  funds,
+  ...rest
+}: UpgradeModuleParameters) {
+  const managerClient = await getManagerClientFromApi({
+    accountId,
+    signingCosmWasmClient,
+    sender,
+    apiUrl,
+  })
+  return managerClient.upgrade(rest, fee, memo, funds)
+}

packages/core/src/actions/index.ts

@@ -30,6 +30,7 @@ export * from './account/wallet/execute'
 export * from './account/wallet/get-manager-client-from-api'
 export * from './account/wallet/get-proxy-client-from-api'
 export * from './account/wallet/remove-namespace'
+export * from './account/wallet/upgrade-module'
 export * from './account/wallet/withdraw'
 
 export * from './public/get-abstract-module-address-from-version-control'

packages/core/src/clients/decorators/account-wallet.ts

@@ -6,6 +6,7 @@ import { execute } from '../../actions/account/wallet/execute'
 import { getManagerClientFromApi } from '../../actions/account/wallet/get-manager-client-from-api'
 import { getProxyClientFromApi } from '../../actions/account/wallet/get-proxy-client-from-api'
 import { revokeNamespace } from '../../actions/account/wallet/remove-namespace'
+import { upgradeModule } from '../../actions/account/wallet/upgrade-module'
 import { withdraw } from '../../actions/account/wallet/withdraw'
 import { VersionControlTypes } from '../../codegen/abstract/index'
 import { ExtractAndOmitParameters } from '../../types/parameters'
@@ -49,6 +50,11 @@ export type AccountWalletActions = {
       typeof revokeNamespace
     >,
   ): ReturnType<typeof revokeNamespace>
+  upgradeModule(
+    parameters: ExtractAndOmitDecoratedParametersFromParameters<
+      typeof upgradeModule
+    >,
+  ): ReturnType<typeof upgradeModule>
   withdraw(
     parameters: ExtractAndOmitDecoratedParametersFromParameters<
       typeof withdraw
@@ -95,6 +101,14 @@ export function accountWalletActions(
         apiUrl,
         sender,
       }),
+    upgradeModule: (parameters) =>
+      upgradeModule({
+        ...parameters,
+        accountId,
+        signingCosmWasmClient,
+        apiUrl,
+        sender,
+      }),
     withdraw: (parameters) =>
       withdraw({
         ...parameters,

packages/react/src/hooks/account/wallet/index.ts

@@ -1,4 +1,5 @@
 export * from './use-deposit'
 export * from './use-create-sub-account'
 export * from './use-execute'
+export * from './use-upgrade-module'
 export * from './use-withdraw'

packages/react/src/hooks/account/wallet/use-upgrade-module.ts

@@ -0,0 +1,37 @@
+import { AccountWalletClient } from '@abstract-money/core/clients'
+import { AccountId } from '@abstract-money/core/utils'
+import { useMutation } from '@tanstack/react-query'
+import { useConfig } from '../../../contexts'
+import { ExtractArgsFromParameters } from '../../../types/args'
+import { UseMutationParameters } from '../../../types/queries'
+
+type UpgradeModuleMutation = ExtractArgsFromParameters<
+  Parameters<AccountWalletClient['upgradeModule']>[0]
+>
+
+export type UseUpgradeModuleParameters = {
+  accountId: AccountId | undefined
+  chainName: string | undefined
+  mutation?: UseMutationParameters<
+    Awaited<ReturnType<AccountWalletClient['upgradeModule']>>,
+    unknown,
+    UpgradeModuleMutation
+  >
+}
+
+export function useUpgradeModule({
+  accountId,
+  chainName,
+  mutation,
+}: UseUpgradeModuleParameters) {
+  const config = useConfig()
+  const walletClient = config.useAccountWalletClient({
+    chainName,
+    accountId,
+  })
+
+  return useMutation(({ args, ...cosmWasmSignOptions }) => {
+    if (!walletClient) throw new Error('walletClient is not defined')
+    return walletClient.upgradeModule({ ...cosmWasmSignOptions, ...args })
+  }, mutation)
+}