github.com/vmware/govmomi@v0.51.0/cli/importx/spec.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 importx
     6  
     7  import (
     8  	"context"
     9  	"flag"
    10  	"fmt"
    11  	"io"
    12  	"path"
    13  
    14  	"github.com/vmware/govmomi/cli"
    15  	"github.com/vmware/govmomi/cli/flags"
    16  	"github.com/vmware/govmomi/ovf/importer"
    17  )
    18  
    19  type spec struct {
    20  	*flags.ClientFlag
    21  	*flags.OutputFlag
    22  
    23  	Archive importer.Archive
    24  
    25  	hidden bool
    26  }
    27  
    28  func init() {
    29  	cli.Register("import.spec", &spec{})
    30  }
    31  
    32  func (cmd *spec) Register(ctx context.Context, f *flag.FlagSet) {
    33  	cmd.ClientFlag, ctx = flags.NewClientFlag(ctx)
    34  	cmd.ClientFlag.Register(ctx, f)
    35  
    36  	cmd.OutputFlag, ctx = flags.NewOutputFlag(ctx)
    37  	cmd.OutputFlag.Register(ctx, f)
    38  
    39  	f.BoolVar(&cmd.hidden, "hidden", false, "Enable hidden properties")
    40  }
    41  
    42  func (cmd *spec) Process(ctx context.Context) error {
    43  	if err := cmd.ClientFlag.Process(ctx); err != nil {
    44  		return err
    45  	}
    46  	return cmd.OutputFlag.Process(ctx)
    47  }
    48  
    49  func (cmd *spec) Usage() string {
    50  	return "PATH_TO_OVF_OR_OVA"
    51  }
    52  
    53  func (cmd *spec) Run(ctx context.Context, f *flag.FlagSet) error {
    54  	fpath := ""
    55  	args := f.Args()
    56  	if len(args) == 1 {
    57  		fpath = f.Arg(0)
    58  	}
    59  
    60  	if len(fpath) > 0 {
    61  		switch path.Ext(fpath) {
    62  		case ".ovf":
    63  			cmd.Archive = &importer.FileArchive{Path: fpath}
    64  		case "", ".ova":
    65  			cmd.Archive = &importer.TapeArchive{Path: fpath}
    66  			fpath = "*.ovf"
    67  		default:
    68  			return fmt.Errorf("invalid file extension %s", path.Ext(fpath))
    69  		}
    70  
    71  		if importer.IsRemotePath(f.Arg(0)) {
    72  			client, err := cmd.Client()
    73  			if err != nil {
    74  				return err
    75  			}
    76  			switch archive := cmd.Archive.(type) {
    77  			case *importer.FileArchive:
    78  				archive.Client = client
    79  			case *importer.TapeArchive:
    80  				archive.Client = client
    81  			}
    82  		}
    83  	}
    84  
    85  	env, err := importer.Spec(fpath, cmd.Archive, cmd.hidden, cmd.Verbose())
    86  	if err != nil {
    87  		return err
    88  	}
    89  
    90  	if !cmd.All() {
    91  		cmd.JSON = true
    92  	}
    93  	return cmd.WriteResult(&specResult{env})
    94  }
    95  
    96  type specResult struct {
    97  	*importer.Options
    98  }
    99  
   100  func (*specResult) Write(w io.Writer) error {
   101  	return nil
   102  }