//===----------------------------------------------------------------------===//
// 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 ContainerPersistence
import ContainerizationError
import ContainerizationExtras
import ContainerizationOCI
import Foundation

extension Application {
    public struct PropertySet: AsyncLoggableCommand {
        public static let configuration = CommandConfiguration(
            commandName: "set",
            abstract: "Set a property value"
        )

        @OptionGroup
        public var logOptions: Flags.Logging

        @Argument(help: "The property ID")
        var id: String

        @Argument(help: "The property value")
        var value: String

        public init() {}

        public func run() async throws {
            guard let key = DefaultsStore.Keys(rawValue: id) else {
                throw ContainerizationError(.invalidArgument, message: "invalid property ID: \(id)")
            }

            switch key {
            case .buildRosetta:
                guard let boolValue = Parser.parseBool(string: value) else {
                    throw ContainerizationError(.invalidArgument, message: "invalid boolean value: \(value)")
                }
                DefaultsStore.setBool(value: boolValue, key: key)
            case .defaultBuildCPUs, .defaultContainerCPUs:
                guard let cpuCount = Int(value), cpuCount > 0 else {
                    throw ContainerizationError(.invalidArgument, message: "invalid CPU count: \(value)")
                }
                DefaultsStore.set(value: value, key: key)
            case .defaultBuildMemory, .defaultContainerMemory:
                guard let memoryMiB = try? Parser.memoryStringAsMiB(value), memoryMiB > 0 else {
                    throw ContainerizationError(.invalidArgument, message: "invalid memory value: \(value)")
                }
                DefaultsStore.set(value: value, key: key)
            case .defaultDNSDomain, .defaultRegistryDomain:
                guard Parser.isValidDomainName(value) else {
                    throw ContainerizationError(.invalidArgument, message: "invalid domain name: \(value)")
                }
                DefaultsStore.set(value: value, key: key)
            case .defaultBuilderImage, .defaultInitImage:
                guard (try? Reference.parse(value)) != nil else {
                    throw ContainerizationError(.invalidArgument, message: "invalid image reference: \(value)")
                }
                DefaultsStore.set(value: value, key: key)
            case .defaultKernelBinaryPath:
                DefaultsStore.set(value: value, key: key)
            case .defaultKernelURL:
                guard URL(string: value) != nil else {
                    throw ContainerizationError(.invalidArgument, message: "invalid URL: \(value)")
                }
                DefaultsStore.set(value: value, key: key)
                return
            case .defaultSubnet:
                guard (try? CIDRv4(value)) != nil else {
                    throw ContainerizationError(.invalidArgument, message: "invalid CIDRv4 address: \(value)")
                }
                DefaultsStore.set(value: value, key: key)
            case .defaultIPv6Subnet:
                guard (try? CIDRv6(value)) != nil else {
                    throw ContainerizationError(.invalidArgument, message: "invalid CIDRv6 address: \(value)")
                }
                DefaultsStore.set(value: value, key: key)
            }
        }
    }
}
