github.com/vmware/govmomi@v0.51.0/cli/cluster/module/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 module
     6  
     7  import (
     8  	"bufio"
     9  	"context"
    10  	"flag"
    11  	"os"
    12  	"strings"
    13  
    14  	"github.com/vmware/govmomi/cli"
    15  	"github.com/vmware/govmomi/cli/flags"
    16  	"github.com/vmware/govmomi/vapi/cluster"
    17  )
    18  
    19  type rm struct {
    20  	*flags.ClientFlag
    21  	ignoreNotFound bool
    22  }
    23  
    24  func init() {
    25  	cli.Register("cluster.module.rm", &rm{})
    26  }
    27  
    28  func (cmd *rm) Register(ctx context.Context, f *flag.FlagSet) {
    29  	cmd.ClientFlag, ctx = flags.NewClientFlag(ctx)
    30  	cmd.ClientFlag.Register(ctx, f)
    31  
    32  	f.BoolVar(&cmd.ignoreNotFound, "ignore-not-found", false, "Treat \"404 Not Found\" as a successful delete.")
    33  }
    34  
    35  func (cmd *rm) Usage() string {
    36  	return "ID"
    37  }
    38  
    39  func (cmd *rm) Description() string {
    40  	return `Delete cluster module ID.
    41  
    42  If ID is "-", read a list from stdin.
    43  
    44  Examples:
    45    govc cluster.module.rm module_id
    46    govc cluster.module.rm - < input-file.txt`
    47  }
    48  
    49  func (cmd *rm) Run(ctx context.Context, f *flag.FlagSet) error {
    50  	if f.NArg() != 1 {
    51  		return flag.ErrHelp
    52  	}
    53  
    54  	moduleID := f.Arg(0)
    55  
    56  	c, err := cmd.RestClient()
    57  	if err != nil {
    58  		return err
    59  	}
    60  	m := cluster.NewManager(c)
    61  
    62  	if moduleID == "-" {
    63  		scanner := bufio.NewScanner(os.Stdin)
    64  		for scanner.Scan() {
    65  			moduleID := scanner.Text()
    66  			if moduleID == "" {
    67  				continue
    68  			}
    69  			if err := cmd.deleteModule(ctx, m, moduleID); err != nil {
    70  				return err
    71  			}
    72  		}
    73  		return nil
    74  	}
    75  
    76  	return cmd.deleteModule(ctx, m, moduleID)
    77  }
    78  
    79  func (cmd *rm) deleteModule(ctx context.Context, m *cluster.Manager, moduleID string) error {
    80  	if err := m.DeleteModule(ctx, moduleID); err != nil {
    81  		if cmd.ignoreNotFound && strings.HasSuffix(err.Error(), "404 Not Found") {
    82  			return nil
    83  		}
    84  		return err
    85  	}
    86  	return nil
    87  }