xtool-mirror/Sources/XUtils/Subprocess+Utils.swift

#if canImport(Subprocess)
import Subprocess
import Foundation

extension CollectedResult {
    @discardableResult
    public func checkSuccess() throws(SubprocessFailure) -> Self {
        try terminationStatus.checkSuccess()
        return self
    }
}

extension ExecutionResult {
    @discardableResult
    public func checkSuccess() throws(SubprocessFailure) -> Self {
        try terminationStatus.checkSuccess()
        return self
    }
}

extension TerminationStatus {
    fileprivate func checkSuccess() throws(SubprocessFailure) {
        if let failure { throw failure }
    }

    private var failure: SubprocessFailure? {
        if isSuccess { return nil }
        switch self {
        case .exited(let code): return .exited(code)
        case .unhandledException(let code): return .unhandledException(code)
        }
    }
}

public enum SubprocessFailure: Error {
    case exited(TerminationStatus.Code)
    case unhandledException(TerminationStatus.Code)
}

extension PlatformOptions {
    public static var withGracefulShutDown: Self {
        .init().withGracefulShutDown
    }

    public var withGracefulShutDown: Self {
        var copy = self
        copy.teardownSequence = [
            .gracefulShutDown(
                allowedDurationToNextStep: .milliseconds(500)
            )
        ]
        return copy
    }
}

extension Environment {
    public static func currentMap() -> [Environment.Key: String] {
        Dictionary(
            uniqueKeysWithValues: ProcessInfo.processInfo.environment.map {
                (Environment.Key(rawValue: $0)!, $1)
            }
        )
    }
}
#endif