mac-drive/PDCore/PDCoreIOS/Extensions/FileManager+Extension.swift

// Copyright (c) 2026 Proton AG
//
// This file is part of Proton Drive.
//
// Proton Drive is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Proton Drive is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Proton Drive. If not, see https://www.gnu.org/licenses/.

import Foundation
import PDCore

extension FileManager {
    public var availabeCapacityInBytes: Int64? {
        let fileURL = URL(fileURLWithPath: "/")
        do {
            let values = try fileURL.resourceValues(forKeys: [.volumeAvailableCapacityForImportantUsageKey])
            if let capacity = values.volumeAvailableCapacityForImportantUsage {
                return capacity
            } else {
                Log.warning("Unable to get available storage", domain: .storage)
                return nil
            }
        } catch {
            Log.error("Unable to get available storage", error: error, domain: .storage, sendToSentryIfPossible: false)
            return nil
        }
    }
    
    /// Returns a human-readable formatted string of the available device storage capacity
    public var availableCapacity: String {
        if let capacity = availabeCapacityInBytes {
            return ByteCountFormatter.storageSizeString(forByteCount: capacity)
        }
        return "Unknown available capacity"
    }
}