core: frontend: disk: Add speed test

Signed-off-by: Patrick José Pereira <patrickelectric@gmail.com>

Author: patrickelectric

core/frontend/src/store/disk.ts

@@ -3,7 +3,7 @@ import {
 } from 'vuex-module-decorators'
 
 import store from '@/store'
-import { DiskUsageQuery, DiskUsageResponse } from '@/types/disk'
+import { DiskSpeedResult, DiskUsageQuery, DiskUsageResponse } from '@/types/disk'
 import back_axios, { isBackendOffline } from '@/utils/api'
 
 @Module({ dynamic: true, store, name: 'disk' })
@@ -18,6 +18,12 @@ class DiskStore extends VuexModule {
 
   error: string | null = null
 
+  speedResult: DiskSpeedResult | null = null
+
+  speedTesting = false
+
+  speedError: string | null = null
+
   @Mutation
   setUsage(value: DiskUsageResponse | null): void {
     this.usage = value
@@ -38,6 +44,21 @@ class DiskStore extends VuexModule {
     this.error = message
   }
 
+  @Mutation
+  setSpeedResult(value: DiskSpeedResult | null): void {
+    this.speedResult = value
+  }
+
+  @Mutation
+  setSpeedTesting(value: boolean): void {
+    this.speedTesting = value
+  }
+
+  @Mutation
+  setSpeedError(message: string | null): void {
+    this.speedError = message
+  }
+
   @Action
   async fetchUsage(query: DiskUsageQuery): Promise<void> {
     this.setLoading(true)
@@ -109,6 +130,34 @@ class DiskStore extends VuexModule {
     this.setDeleting(false)
     return { succeeded, failed }
   }
+
+  @Action
+  async runSpeedTest(sizeBytes: number): Promise<void> {
+    this.setSpeedTesting(true)
+    this.setSpeedError(null)
+    this.setSpeedResult(null)
+
+    await back_axios({
+      method: 'get',
+      url: `${this.API_URL}/speed`,
+      params: { size_bytes: sizeBytes },
+      timeout: 600000,
+    })
+      .then((response) => {
+        this.setSpeedResult(response.data as DiskSpeedResult)
+      })
+      .catch((error) => {
+        if (isBackendOffline(error)) {
+          return
+        }
+        const axiosError = error as { response?: { data?: { detail?: string } }; message?: string }
+        const message = axiosError.response?.data?.detail || axiosError.message || 'Unknown error'
+        this.setSpeedError(`Speed test failed: ${message}`)
+      })
+      .finally(() => {
+        this.setSpeedTesting(false)
+      })
+  }
 }
 
 const disk_store = getModule(DiskStore)

core/frontend/src/types/disk.ts

@@ -20,3 +20,12 @@ export interface DiskUsageQuery {
   include_files?: boolean
   min_size_bytes?: number
 }
+
+export interface DiskSpeedResult {
+  write_speed_mbps: number | null
+  read_speed_mbps: number | null
+  bytes_tested: number
+  seed: string
+  success: boolean
+  error: string | null
+}