//===----------------------------------------------------------------------===//
// Copyright © 2025-2026 Apple Inc. and the container project authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//   https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//===----------------------------------------------------------------------===//

import ArgumentParser
import ContainerAPIClient
import Foundation

extension Application {
    public struct SystemDF: AsyncLoggableCommand {
        public static let configuration = CommandConfiguration(
            commandName: "df",
            abstract: "Show disk usage for images, containers, and volumes"
        )

        @Option(name: .long, help: "Format of the output")
        var format: ListFormat = .table

        @OptionGroup
        public var logOptions: Flags.Logging

        public init() {}

        public func run() async throws {
            let stats = try await ClientDiskUsage.get()
            try Output.render(payload: stats, format: format, jsonOptions: .pretty) {
                diskUsageTable(stats: stats)
            }
        }

        private func diskUsageTable(stats: DiskUsageStats) -> String {
            var rows: [[String]] = []

            // Header row
            rows.append(["TYPE", "TOTAL", "ACTIVE", "SIZE", "RECLAIMABLE"])

            // Images row
            rows.append([
                "Images",
                "\(stats.images.total)",
                "\(stats.images.active)",
                formatSize(stats.images.sizeInBytes),
                formatReclaimable(stats.images.reclaimable, total: stats.images.sizeInBytes),
            ])

            // Containers row
            rows.append([
                "Containers",
                "\(stats.containers.total)",
                "\(stats.containers.active)",
                formatSize(stats.containers.sizeInBytes),
                formatReclaimable(stats.containers.reclaimable, total: stats.containers.sizeInBytes),
            ])

            // Volumes row
            rows.append([
                "Local Volumes",
                "\(stats.volumes.total)",
                "\(stats.volumes.active)",
                formatSize(stats.volumes.sizeInBytes),
                formatReclaimable(stats.volumes.reclaimable, total: stats.volumes.sizeInBytes),
            ])

            let tableFormatter = TableOutput(rows: rows)
            return tableFormatter.format()
        }

        private func formatSize(_ bytes: UInt64) -> String {
            if bytes == 0 {
                return "0 B"
            }
            let formatter = ByteCountFormatter()
            formatter.countStyle = .file
            return formatter.string(fromByteCount: Int64(bytes))
        }

        private func formatReclaimable(_ reclaimable: UInt64, total: UInt64) -> String {
            let sizeStr = formatSize(reclaimable)

            if total == 0 {
                return "\(sizeStr) (0%)"
            }

            // Cap at 100% in case reclaimable > total (shouldn't happen but be defensive)
            let percentage = min(100, Int(round(Double(reclaimable) / Double(total) * 100.0)))
            return "\(sizeStr) (\(percentage)%)"
        }
    }
}
