github.com/vmware/govmomi@v0.51.0/cli/extension/unregister.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 extension
     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 unregister struct {
    17  	*flags.ClientFlag
    18  }
    19  
    20  func init() {
    21  	cli.Register("extension.unregister", &unregister{})
    22  }
    23  
    24  func (cmd *unregister) 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 *unregister) 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 *unregister) Run(ctx context.Context, f *flag.FlagSet) error {
    37  	c, err := cmd.Client()
    38  	if err != nil {
    39  		return err
    40  	}
    41  
    42  	m, err := object.GetExtensionManager(c)
    43  	if err != nil {
    44  		return err
    45  	}
    46  
    47  	for _, key := range f.Args() {
    48  		if err = m.Unregister(ctx, key); err != nil {
    49  			return err
    50  		}
    51  	}
    52  
    53  	return nil
    54  }