Merge pull request #2 from rajiteh/main

feat: ability to wake up from various domain states
This commit is contained in:
Scott Ellis 2021-12-28 11:21:20 -08:00 committed by GitHub
commit a372d706b8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -111,20 +111,32 @@ func WakeVirtualMachine(mac string) bool {
for _, list := range xmlquery.Find(querydoc, "//domain/devices/interface/mac/@address") {
// Use the strings.EqualFold function to do a case-insensitive comparison of MACs
if strings.EqualFold(list.InnerText(), mac) {
active, err := l.DomainIsActive(d)
stateInt, _, err := l.DomainGetState(d, 0)
if err != nil {
log.Fatalf("failed to check domain active state: %v", err)
log.Fatalf("failed to check domain state: %v", err)
}
// Check that the system isn't already active
// Silly that "active" is an int, not a bool
if active != 0 {
fmt.Printf("System %s is already awake\n", d.Name)
} else {
state := libvirt.DomainState(stateInt)
// fmt.Printf("Domain state is %v\n", state)
switch state {
case libvirt.DomainShutoff, libvirt.DomainCrashed:
fmt.Printf("Waking system: %s at MAC %s\n", d.Name, mac)
if err := l.DomainCreate(d); err != nil {
log.Fatalf("Failed to start domain: %v", err)
}
case libvirt.DomainPmsuspended:
fmt.Printf("PM Wakeup system: %s at MAC %s\n", d.Name, mac)
if err := l.DomainPmWakeup(d, 0); err != nil {
log.Fatalf("Failed to pm wakeup domain: %v", err)
}
case libvirt.DomainPaused:
fmt.Printf("Resume system: %s at MAC %s\n", d.Name, mac)
if err := l.DomainResume(d); err != nil {
log.Fatalf("Failed to resume domain: %v", err)
}
default:
fmt.Printf("System %s is already running or in a state that cannot be woken from. State: %d\n", d.Name, state)
}
}
}