Mac-utlity-app

//
//  DiskInfo.swift
//  MUA
//
//  Created by Mitchel Volkering on 21/12/2025.
//

import Foundation

/// Information about a disk/volume
struct DiskInfo: Identifiable {
    let id = UUID()
    let name: String
    let mountPoint: URL
    let totalSpace: Int64
    let freeSpace: Int64
    let isRemovable: Bool
    let isInternal: Bool

    var usedSpace: Int64 {
        totalSpace - freeSpace
    }

    var usedPercentage: Double {
        guard totalSpace > 0 else { return 0 }
        return (Double(usedSpace) / Double(totalSpace)) * 100
    }

    var formattedTotalSpace: String {
        ByteCountFormatter.string(fromByteCount: totalSpace, countStyle: .file)
    }

    var formattedFreeSpace: String {
        ByteCountFormatter.string(fromByteCount: freeSpace, countStyle: .file)
    }

    var formattedUsedSpace: String {
        ByteCountFormatter.string(fromByteCount: usedSpace, countStyle: .file)
    }

    var icon: String {
        if isRemovable {
            return "externaldrive.fill"
        }
        return isInternal ? "internaldrive.fill" : "externaldrive.fill"
    }

    /// Get disk info for all mounted volumes
    static func getAllDisks() -> [DiskInfo] {
        var disks: [DiskInfo] = []

        let fileManager = FileManager.default
        let keys: [URLResourceKey] = [
            .volumeNameKey,
            .volumeTotalCapacityKey,
            .volumeAvailableCapacityKey,
            .volumeIsRemovableKey,
            .volumeIsInternalKey
        ]

        guard let volumes = fileManager.mountedVolumeURLs(includingResourceValuesForKeys: keys, options: [.skipHiddenVolumes]) else {
            return disks
        }

        for volume in volumes {
            do {
                let resourceValues = try volume.resourceValues(forKeys: Set(keys))
                let name = resourceValues.volumeName ?? volume.lastPathComponent
                let total = Int64(resourceValues.volumeTotalCapacity ?? 0)
                let free = Int64(resourceValues.volumeAvailableCapacity ?? 0)
                let isRemovable = resourceValues.volumeIsRemovable ?? false
                let isInternal = resourceValues.volumeIsInternal ?? true

                let disk = DiskInfo(
                    name: name,
                    mountPoint: volume,
                    totalSpace: total,
                    freeSpace: free,
                    isRemovable: isRemovable,
                    isInternal: isInternal
                )
                disks.append(disk)
            } catch {
                continue
            }
        }

        return disks
    }

    /// Get the main system disk
    static func getSystemDisk() -> DiskInfo? {
        getAllDisks().first { $0.mountPoint.path == "/" }
    }
}