feat(indexd): add reset lost sector functionality for hosts

Author: telestrial

.changeset/funky-lamps-greet.md

@@ -0,0 +1,5 @@
+---
+'indexd': minor
+---
+
+Added button to reset lost sectors on hosts. Closes https://github.com/SiaFoundation/indexd/issues/732

apps/indexd/components/Data/Hosts/SidePanelHost.tsx

@@ -22,6 +22,7 @@ import { useHostsParams } from './useHostsParams'
 import { SidePanelHeadingCopyable } from '../SidePanelHeadingCopyable'
 import { SidePanelSkeleton } from '../SidePanelSkeleton'
 import { BlockReasons } from './BlockReasons'
+import { BulkHostResetLostSectors } from './bulkActions/BulkHostResetLostSectors'
 
 export function SidePanelHost() {
   const { panelId, setPanelId } = useHostsParams()
@@ -76,11 +77,14 @@ export function SidePanelHost() {
           }
           actions={
             host ? (
-              host.blocked ? (
-                <BulkHostBlocklistRemove hosts={[host]} />
-              ) : (
-                <BulkHostBlocklistAdd hosts={[host]} />
-              )
+              <>
+                {host.blocked ? (
+                  <BulkHostBlocklistRemove hosts={[host]} />
+                ) : (
+                  <BulkHostBlocklistAdd hosts={[host]} />
+                )}
+                <BulkHostResetLostSectors hosts={[host]} />
+              </>
             ) : null
           }
         >

apps/indexd/components/Data/Hosts/SidePanelHostList.tsx

@@ -3,6 +3,7 @@ import { HostDataTableState } from './types'
 import { useMapHosts } from '../useMapHosts'
 import { BulkHostBlocklistAdd } from './bulkActions/BulkHostBlocklistAdd'
 import { BulkHostBlocklistRemove } from './bulkActions/BulkHostBlocklistRemove'
+import { BulkHostResetLostSectors } from './bulkActions/BulkHostResetLostSectors'
 import { SidePanel } from '../SidePanel'
 import { useMemo } from 'react'
 import { useHostsParams } from './useHostsParams'
@@ -29,6 +30,7 @@ export function SidePanelHostList({ table }: { table: HostDataTableState }) {
           <>
             <BulkHostBlocklistAdd hosts={hosts} />
             <BulkHostBlocklistRemove hosts={hosts} />
+            <BulkHostResetLostSectors hosts={hosts} />
           </>
         ) : null
       }

apps/indexd/components/Data/Hosts/bulkActions/BulkHostResetLostSectors.tsx

@@ -0,0 +1,82 @@
+import { Button, handleBatchOperation } from '@siafoundation/design-system'
+import { useAdminHostResetLostSectors } from '@siafoundation/indexd-react'
+import { useCallback, useMemo } from 'react'
+import { pluralize } from '@siafoundation/units'
+import { HostData } from '../types'
+import { useMutate } from '@siafoundation/react-core'
+import { adminHostRoute, adminHostsRoute } from '@siafoundation/indexd-types'
+import { Row } from '@tanstack/react-table'
+import { ResetAlt16 } from '@siafoundation/react-icons'
+import { useDialog } from '../../../../contexts/dialog'
+
+export function BulkHostResetLostSectors({
+  hosts,
+}: {
+  hosts: HostData[] | Row<HostData>[]
+}) {
+  const resetLostSectors = useAdminHostResetLostSectors()
+  const mutate = useMutate()
+  const { openConfirmDialog } = useDialog()
+
+  const hostsWithLostSectors = useMemo(
+    () =>
+      hosts
+        .map((host) =>
+          'publicKey' in host ? host : (host.original as HostData)
+        )
+        .filter((host) => host.lostSectors > 0),
+    [hosts]
+  )
+
+  const operation = useCallback(async () => {
+    await handleBatchOperation(
+      hostsWithLostSectors.map((host) =>
+        resetLostSectors.post({
+          params: {
+            hostkey: host.publicKey,
+          },
+        })
+      ),
+      {
+        toastError: ({ successCount, errorCount, totalCount }) => ({
+          title: `Reset lost sectors for ${pluralize(successCount, 'host')}`,
+          body: `Error resetting lost sectors for ${errorCount}/${totalCount} total hosts.`,
+        }),
+        toastSuccess: ({ totalCount }) => ({
+          title: `Reset lost sectors for ${pluralize(totalCount, 'host')}`,
+        }),
+      }
+    )
+    await mutate((key) => key.startsWith(adminHostsRoute))
+    await Promise.all(
+      hostsWithLostSectors.map((host) =>
+        mutate((key) =>
+          key.startsWith(
+            adminHostRoute.replace(':hostkey', host.publicKey)
+          )
+        )
+      )
+    )
+  }, [hostsWithLostSectors, resetLostSectors, mutate])
+
+  if (hostsWithLostSectors.length === 0) {
+    return null
+  }
+
+  return (
+    <Button
+      onClick={() =>
+        openConfirmDialog({
+          title: `Reset lost sectors for ${pluralize(hostsWithLostSectors.length, 'host')}?`,
+          action: 'Reset',
+          variant: 'red',
+          body: null,
+          onConfirm: operation,
+        })
+      }
+    >
+      <ResetAlt16 />
+      Reset lost sectors{hosts.length > 1 ? ` (${hostsWithLostSectors.length})` : ''}
+    </Button>
+  )
+}