1 files changed, 66 insertions, 0 deletions

diff --git a/frontend/lib/storage.ts b/frontend/lib/storage.ts
new file mode 100644
index 0000000..57b2429
--- /dev/null
+++ b/frontend/lib/storage.ts
@@ -0,0 +1,66 @@
+import { exec } from 'child_process'
+import { promisify } from 'util'
+
+const execAsync = promisify(exec)
+
+// Cache for 10 seconds
+let cacheTimestamp = 0
+let cachedData: StorageData | null = null
+
+export interface StorageData {
+  activeDriveUsage: number
+  totalDiskCapacity: number
+  totalDiskUsed: number
+  availableDisk: number
+}
+
+export async function fetchStorageData(): Promise<StorageData> {
+  const now = Date.now()
+  
+  // Return cached data if less than 10 seconds old
+  if (cachedData && now - cacheTimestamp < 10000) {
+    return cachedData
+  }
+
+  try {
+    // Get active drive usage in bytes
+    const { stdout: driveSize } = await execAsync('fctdrive drive-size')
+    const activeDriveUsage = parseInt(driveSize.trim()) || 0
+
+    // Get disk usage in bytes
+    const fctdrivePath = process.env.FCTDRIVE_PATH || '/home/diogo464/dev/fctdrive'
+    const { stdout: dfOutput } = await execAsync(`df -B1 "${fctdrivePath}"`)
+    
+    // Parse df output - second line contains the data
+    const lines = dfOutput.trim().split('\n')
+    const dataLine = lines[1] // Skip header line
+    const columns = dataLine.split(/\s+/)
+    
+    const totalDiskCapacity = parseInt(columns[1]) || 0
+    const totalDiskUsed = parseInt(columns[2]) || 0
+    const availableDisk = parseInt(columns[3]) || 0
+
+    const data: StorageData = {
+      activeDriveUsage,
+      totalDiskCapacity,
+      totalDiskUsed,
+      availableDisk
+    }
+
+    // Update cache
+    cachedData = data
+    cacheTimestamp = now
+
+    return data
+  } catch (error) {
+    console.error('Failed to fetch storage data:', error)
+    
+    // Return zeros on error as requested
+    return {
+      activeDriveUsage: 0,
+      totalDiskCapacity: 0,
+      totalDiskUsed: 0,
+      availableDisk: 0
+    }
+  }
+}
\ No newline at end of file