github.com/vmware/govmomi@v0.51.0/cli/fields/rm.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 fields
     6  
     7  import (
     8  	"context"
     9  	"flag"
    10  
    11  	"github.com/vmware/govmomi/cli"
    12  	"github.com/vmware/govmomi/cli/flags"
    13  	"github.com/vmware/govmomi/object"
    14  )
    15  
    16  type rm struct {
    17  	*flags.ClientFlag
    18  }
    19  
    20  func init() {
    21  	cli.Register("fields.rm", &rm{})
    22  }
    23  
    24  func (cmd *rm) Register(ctx context.Context, f *flag.FlagSet) {
    25  	cmd.ClientFlag, ctx = flags.NewClientFlag(ctx)
    26  	cmd.ClientFlag.Register(ctx, f)
    27  }
    28  
    29  func (cmd *rm) Process(ctx context.Context) error {
    30  	if err := cmd.ClientFlag.Process(ctx); err != nil {
    31  		return err
    32  	}
    33  	return nil
    34  }
    35  
    36  func (cmd *rm) Usage() string {
    37  	return "KEY..."
    38  }
    39  
    40  func (cmd *rm) Run(ctx context.Context, f *flag.FlagSet) error {
    41  	c, err := cmd.Client()
    42  	if err != nil {
    43  		return err
    44  	}
    45  
    46  	m, err := object.GetCustomFieldsManager(c)
    47  	if err != nil {
    48  		return err
    49  	}
    50  
    51  	for _, name := range f.Args() {
    52  		key, err := m.FindKey(ctx, name)
    53  		if err != nil {
    54  			return err
    55  		}
    56  
    57  		if err := m.Remove(ctx, key); err != nil {
    58  			return err
    59  		}
    60  	}
    61  
    62  	return nil
    63  }