Automation Wallet and Features

// Publish component automatically uses automation wallet
const { autoWallet, isAutomationEnabled } = useAutomation()
const [accountIdToUse, setAccountIdToUse] = useState<string>(accountId)
const [signerToUse, setSignerToUse] = useState(signer)

useEffect(() => {
  if (isAutomationEnabled && autoWallet?.address) {
    setAccountIdToUse(autoWallet.address)
    setSignerToUse(autoWallet)
  } else {
    setAccountIdToUse(accountId)
    setSignerToUse(signer)
  }
}, [isAutomationEnabled, autoWallet, accountId, signer])

// All publish transactions use automation wallet when enabled
const { erc721Address, datatokenAddress } = 
  await createTokensAndPricing(
    values,
    accountIdToUse,  // automation wallet address
    config,
    nftFactory
  )

// Download component checks automation status
const signerToUse = isAutomationEnabled ? autoWallet : signer
const accountToUse = isAutomationEnabled 
  ? autoWallet.address 
  : accountId

// Order asset
await order(
  signerToUse,
  asset,
  orderPriceAndFees,
  accountToUse,
  hasDatatoken
)

// Download file
await downloadFile(
  signerToUse,
  asset,
  accountToUse,
  validOrderTx
)

// Compute component uses automation wallet
const { autoWallet, isAutomationEnabled } = useAutomation()

const accountId = isAutomationEnabled 
  ? autoWallet?.address 
  : address

// Fetch compute jobs including automation wallet's jobs
const computeJobs = await getComputeJobs(
  [asset?.chainId],
  address,
  asset,
  cancelToken
)

if (autoWallet) {
  const autoComputeJobs = await getComputeJobs(
    [asset?.chainId],
    autoWallet?.address,
    asset,
    cancelToken
  )
  // Merge jobs from both wallets
  computeJobs.computeJobs.push(...autoComputeJobs.computeJobs)
}

const { balance, autoWalletAddress } = useAutomation()

// Display balances
console.log('Native:', balance.native.balance, balance.native.symbol)
console.log('Tokens:', balance.approved)

// Visual feedback based on balance status
const getFundingStatus = () => {
  if (nativeBalance > minRequired && hasAllTokens) {
    return 'green' // Fully funded
  } else if (nativeBalance > 0) {
    return 'yellow' // Partially funded
  } else {
    return 'red' // Insufficient funds
  }
}

// View on Etherscan/Polygonscan
const explorerUrl = `${networkConfig.explorerUri}/address/${autoWalletAddress}`

// Monitor in console (development)
LoggerInstance.log('[AutomationProvider] Using automation wallet', {
  address: autoWallet.address,
  balance: balance.native.balance,
  network: chain.name
})

// Strong encryption used
const wallet = await ethers.Wallet.fromEncryptedJson(
  automationWalletJSON,
  password
)

// Key derivation parameters (secure defaults)
// - 262144 iterations
// - PBKDF2 + AES-128-CTR

// Decrypted wallet exists only in memory
// Re-decryption required after:
// - Page refresh
// - Browser restart
// - Cache clear

const { isAutomationEnabled, setIsAutomationEnabled } = useAutomation()

// Disable automation mode
setIsAutomationEnabled(false)

// Re-enable when needed
setIsAutomationEnabled(true)

const { deleteCurrentAutomationWallet } = useAutomation()

// Triggers confirmation modal
deleteCurrentAutomationWallet()

function removeAutomationWalletAndCleanup() {
  setIsLoading(true)
  setIsAutomationEnabled(false)
  setAutoWallet(undefined)
  setAutoWalletAddress(undefined)
  setAutomationWalletJSON(undefined)
  setBalance(undefined)
  jsonWalletConnector.disconnect()
  toast.info('The automation wallet was removed from your machine.')
  setIsLoading(false)
}

<DeleteAutomationModal
  hasDeleteRequest={hasDeleteRequest}
  setHasDeleteRequest={setHasDeleteRequest}
  disabled={isLoading}
  onDeleteConfirm={() => removeAutomationWalletAndCleanup()}
/>

// External re-encryption
const { Wallet } = require('ethers')

const oldJson = 'OLD_ENCRYPTED_JSON'
const oldPassword = 'OLD_PASSWORD'
const newPassword = 'NEW_STRONG_PASSWORD'

Wallet.fromEncryptedJson(oldJson, oldPassword).then(wallet => {
  wallet.encrypt(newPassword).then(newJson => {
    console.log('New encrypted JSON:')
    console.log(newJson)
  })
})

if (!ethers.utils.isAddress(
  ethers.utils.getJsonWalletAddress(encryptedJson)
)) {
  // Invalid address in JSON
}

const [decryptPercentage, setDecryptPercentage] = useState<number>()

await ethers.Wallet.fromEncryptedJson(
  automationWalletJSON,
  password,
  (percent) => setDecryptPercentage(percent * 100)
)

useEffect(() => {
  if (isAutomationEnabled && chain?.id) {
    LoggerInstance.log(
      `[AutomationProvider] Network changed to: ${chain.name}`
    )
    setAutoWallet(jsonWalletConnector.getWallet())
  }
}, [chain])

// Check local storage availability
if (typeof localStorage !== 'undefined') {
  // Storage available
} else {
  // Enable local storage in browser settings
}

// EthersWalletConnector.ts
import { Wallet, ethers } from 'ethers'
import { Connector } from 'wagmi'

export const JSON_WALLET_CONNECTOR_ID = 'jsonWalletConnector'

export class EthersWalletConnector extends Connector {
  readonly id = JSON_WALLET_CONNECTOR_ID
  private wallet: Wallet

  async loadWallet(privateKey: string) {
    this.wallet = new ethers.Wallet(privateKey)
  }

  getWallet(): Wallet {
    return this.wallet
  }

  async connect() {
    return {
      account: this.wallet.address,
      chain: { id: await this.wallet.getChainId() },
      provider: this.wallet.provider
    }
  }
}

import { useAutomation } from '@context/Automation/AutomationProvider'

function MyCustomComponent() {
  const {
    autoWallet,
    isAutomationEnabled,
    balance,
    importAutomationWallet,
    decryptAutomationWallet
  } = useAutomation()

  async function setupAutomation() {
    // 1. Import wallet
    const imported = await importAutomationWallet(encryptedJson)
    
    // 2. Decrypt wallet
    const decrypted = await decryptAutomationWallet(password)
    
    // 3. Enable automation
    setIsAutomationEnabled(true)
    
    // 4. Use in operations
    if (isAutomationEnabled && autoWallet) {
      // Automated operations
    }
  }
}