wehub-resource-sync
47 lines
1.6 KiB
Swift
47 lines
1.6 KiB
Swift
import Foundation
|
|
|
|
public struct AmpCLIProbe: Sendable {
|
|
private static let commandTimeout: TimeInterval = 15
|
|
private let arguments: [String]
|
|
|
|
public init() {
|
|
self.arguments = ["usage"]
|
|
}
|
|
|
|
init(arguments: [String]) {
|
|
self.arguments = arguments
|
|
}
|
|
|
|
public func fetch(
|
|
environment: [String: String] = ProcessInfo.processInfo.environment,
|
|
now: Date = Date()) async throws -> AmpUsageSnapshot
|
|
{
|
|
let loginPATH = LoginShellPathCache.shared.current
|
|
guard let executable = BinaryLocator.resolveAmpBinary(env: environment, loginPATH: loginPATH) else {
|
|
throw SubprocessRunnerError.binaryNotFound("amp")
|
|
}
|
|
|
|
var commandEnvironment = environment
|
|
commandEnvironment["NO_COLOR"] = "1"
|
|
commandEnvironment["PATH"] = PathBuilder.effectivePATH(
|
|
purposes: [.tty, .nodeTooling],
|
|
env: environment,
|
|
loginPATH: loginPATH)
|
|
|
|
let result = try await SubprocessRunner.run(
|
|
binary: executable,
|
|
arguments: self.arguments,
|
|
environment: commandEnvironment,
|
|
timeout: Self.commandTimeout,
|
|
standardInput: FileHandle.nullDevice,
|
|
label: "amp-usage")
|
|
let output = result.stdout.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty
|
|
? result.stderr
|
|
: result.stdout
|
|
guard !output.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty else {
|
|
throw AmpUsageError.parseFailed("The Amp CLI returned no usage data.")
|
|
}
|
|
return try AmpUsageParser.parse(displayText: output, now: now)
|
|
}
|
|
}
|