github.com/vmware/govmomi@v0.51.0/cli/device/remove.go (about)

     1  // © Broadcom. All Rights Reserved.
     2  // The term “Broadcom” refers to Broadcom Inc. and/or its subsidiaries.
     3  // SPDX-License-Identifier: Apache-2.0
     4  
     5  package device
     6  
     7  import (
     8  	"context"
     9  	"flag"
    10  	"fmt"
    11  
    12  	"github.com/vmware/govmomi/cli"
    13  	"github.com/vmware/govmomi/cli/flags"
    14  )
    15  
    16  type remove struct {
    17  	*flags.VirtualMachineFlag
    18  	keepFiles bool
    19  }
    20  
    21  func init() {
    22  	cli.Register("device.remove", &remove{})
    23  }
    24  
    25  func (cmd *remove) Register(ctx context.Context, f *flag.FlagSet) {
    26  	cmd.VirtualMachineFlag, ctx = flags.NewVirtualMachineFlag(ctx)
    27  	cmd.VirtualMachineFlag.Register(ctx, f)
    28  	f.BoolVar(&cmd.keepFiles, "keep", false, "Keep files in datastore")
    29  }
    30  
    31  func (cmd *remove) Process(ctx context.Context) error {
    32  	if err := cmd.VirtualMachineFlag.Process(ctx); err != nil {
    33  		return err
    34  	}
    35  	return nil
    36  }
    37  
    38  func (cmd *remove) Usage() string {
    39  	return "DEVICE..."
    40  }
    41  
    42  func (cmd *remove) Description() string {
    43  	return `Remove DEVICE from VM.
    44  
    45  Examples:
    46    govc device.remove -vm $name cdrom-3000
    47    govc device.remove -vm $name disk-1000
    48    govc device.remove -vm $name -keep disk-*`
    49  }
    50  
    51  func (cmd *remove) Run(ctx context.Context, f *flag.FlagSet) error {
    52  	vm, err := cmd.VirtualMachine()
    53  	if err != nil {
    54  		return err
    55  	}
    56  
    57  	if vm == nil {
    58  		return flag.ErrHelp
    59  	}
    60  
    61  	devices, err := vm.Device(ctx)
    62  	if err != nil {
    63  		return err
    64  	}
    65  
    66  	for _, name := range f.Args() {
    67  		device := match(name, devices)
    68  		if len(device) == 0 {
    69  			return fmt.Errorf("device '%s' not found", name)
    70  		}
    71  
    72  		if err = vm.RemoveDevice(ctx, cmd.keepFiles, device...); err != nil {
    73  			return err
    74  		}
    75  	}
    76  
    77  	return nil
    78  }