public_git
/
tart
mirror of https://github.com/cirruslabs/tart.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

--wait option for IP command (#29)

This commit is contained in:
Fedor Korotkov 2022-04-19 12:56:59 -04:00 committed by GitHub
parent 4ca7787958
commit f07ffed6c8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
Sources/tart/Commands/IP.swift
View File

@ -1,5 +1,6 @@
import ArgumentParser
import Foundation
import Network
import SystemConfiguration
struct IP: AsyncParsableCommand {
@ -8,13 +9,15 @@ struct IP: AsyncParsableCommand {
@Argument(help: "VM name")
var name: String
@Option(help: "Number of seconds to wait for a potential VM booting")
var wait: UInt16 = 0
func run() async throws {
do {
let vmDir = try VMStorage().read(name)
let vmConfig = try VMConfig.init(fromURL: vmDir.configURL)
let vmMacAddress = MACAddress(fromString: vmConfig.macAddress.string)!
guard let ip = try ARPCache.ResolveMACAddress(macAddress: vmMacAddress) else {
guard let ip = try await resolveIP(vmConfig, secondsToWait: wait) else {
print("no IP address found, is your VM running?")
Foundation.exit(1)
@ -29,4 +32,19 @@ struct IP: AsyncParsableCommand {
Foundation.exit(1)
}
}
private func resolveIP(_ config: VMConfig, secondsToWait: UInt16) async throws -> IPv4Address? {
let waitUntil = Calendar.current.date(byAdding: .second, value: Int(secondsToWait), to: Date.now)!
let vmMacAddress = MACAddress(fromString: config.macAddress.string)!
repeat {
if let ip = try ARPCache.ResolveMACAddress(macAddress: vmMacAddress) {
return ip
}
try await Task.sleep(nanoseconds: 1_000_000)
} while Date.now < waitUntil
return nil
}
}