vmReboot.go

package cmdVM

import (
    u "nexus-client/utils"
    g "nexus-client/globals"
)

type Reboot struct {
    Name string
}

func (cmd *Reboot)GetName() string {
    return cmd.Name
}

func (cmd *Reboot)GetDesc() []string {
    return []string{
        "Gracefully reboots one or more VMs.",
        "Requires VM_REBOOT VM access capability or VM_REBOOT_ANY user capability."}
}

func (cmd *Reboot)PrintUsage() {
    printRegexUsage(cmd)
    printRegexUsageDetails()
}

func (cmd *Reboot)Run(args []string) int {
    client := g.GetInstance().Client
    host := g.GetInstance().Host

    argc := len(args)
    if argc < 1 {
        cmd.PrintUsage()
        return 1
    }

    vms, err := getFilteredVMs("/vms/reboot", args)
    if err != nil {
        u.PrintlnErr(err.Error())
        return 1
    }

    if len(vms) == 0 {
        u.PrintlnErr("Error: VM(s) not found.")
        return 1
    }

    statusCode := 0

    for _, vm := range(vms) {
        uuid := vm.ID.String()
        resp, err := client.R().Put(host+"/vms/"+uuid+"/reboot")
        if err != nil {
            u.PrintlnErr("Reboot failed for VM \""+vm.Name+"\": "+err.Error())
            statusCode = 1
        } else {
            if resp.IsSuccess() {
                u.Println("Sent reboot message to VM \""+vm.Name+"\"")
            } else {
                u.PrintlnErr("Reboot failed for VM \""+vm.Name+"\": "+resp.Status()+": "+resp.String())
                statusCode = 1
            }
        }

    }

    return statusCode
}