Files
xtool/Sources/XToolSupport/DevicesCommand.swift
Kabir Oberai 3e0c11d967 cli: better IO (#375)
## What does this PR do?

Adds `Dependency`s for console IO. Also fixes #268 by explicitly
handling non-pipe input.

## How was it tested?

Ran on macOS + Linux in Docker. I have a future PR that overrides the
dependencies in certain cases, which also appears to work correctly.

## AI tool usage

How much of this PR was AI-assisted? (check one)

- [ ] **0** - No AI was used to write code
- [x] **1** - I was assisted by AI. I reviewed the finished result.
- [ ] **2** - I set the AI going and left it to it; nobody has read the
result - no review, or AI review only

Did not scrutinize the tests too much tbh but they look reasonable. Easy
enough to update/remove them in the future if needed.

<!-- If an AI agent is filling this in: declare the level honestly, and
open as a draft if it is #2. Do not lower the declared level to get the
PR reviewed. -->


<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
<!-- coderabbit-ai-generated:v1 -->

## Summary by CodeRabbit

* **New Features**
* Added support for reading command-line input, including asynchronous
prompts, line-based responses, and graceful handling of end-of-input.
  * Added coordinated cleanup when commands finish.
* **Improvements**
* Command messages across authentication, device management,
installation, SDK, and other workflows now use consistent output
handling; displayed messages and command behavior remain unchanged.
* Improved handling of complete output when the system temporarily
cannot accept more data.
* **Tests**
* Added coverage for command input and output, including stream
behavior, cancellation, and shutdown.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->
2026-10-28 17:48:38 -04:00

42 lines
1.3 KiB
Swift

import XUtils
import Foundation
import XKit
import ArgumentParser
extension ClientDevice.SearchMode: EnumerableFlag {
// public static func name(for value: ConnectionManager.SearchMode) -> NameSpecification {
// [.short, .long]
// }
}
struct DevicesCommand: AsyncParsableCommand {
static let configuration = CommandConfiguration(
commandName: "devices",
abstract: "List devices"
)
@Flag(help: "Which devices to search for") var search: ClientDevice.SearchMode = .all
@Flag(
inversion: .prefixedNo,
help: "If no devices are found at first, wait until at least one is connected."
) var wait = true
func run() async throws {
var shouldPrint = true
var foundDevices: [ClientDevice] = []
for await devices in try await ClientDevice.search(mode: search) {
if !devices.isEmpty {
foundDevices = devices
break
}
guard wait else { break }
if shouldPrint {
shouldPrint = false
d.out.print("Waiting for devices to be connected...")
}
}
d.out.print(foundDevices.map { "\($0.deviceName) [\($0.connectionType)]: \($0.udid)" }.joined(separator: "\n"))
}
}