github.com/vmware/govmomi@v0.37.2/govc/datastore/mkdir.go (about)

     1  /*
     2  Copyright (c) 2014-2015 VMware, Inc. All Rights Reserved.
     3  
     4  Licensed under the Apache License, Version 2.0 (the "License");
     5  you may not use this file except in compliance with the License.
     6  You may obtain a copy of the License at
     7  
     8      http://www.apache.org/licenses/LICENSE-2.0
     9  
    10  Unless required by applicable law or agreed to in writing, software
    11  distributed under the License is distributed on an "AS IS" BASIS,
    12  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    13  See the License for the specific language governing permissions and
    14  limitations under the License.
    15  */
    16  
    17  package datastore
    18  
    19  import (
    20  	"context"
    21  	"errors"
    22  	"flag"
    23  	"fmt"
    24  
    25  	"github.com/vmware/govmomi/govc/cli"
    26  	"github.com/vmware/govmomi/govc/flags"
    27  	"github.com/vmware/govmomi/object"
    28  	"github.com/vmware/govmomi/vim25/soap"
    29  	"github.com/vmware/govmomi/vim25/types"
    30  )
    31  
    32  type mkdir struct {
    33  	*flags.DatastoreFlag
    34  
    35  	createParents bool
    36  	isNamespace   bool
    37  }
    38  
    39  func init() {
    40  	cli.Register("datastore.mkdir", &mkdir{})
    41  }
    42  
    43  func (cmd *mkdir) Register(ctx context.Context, f *flag.FlagSet) {
    44  	cmd.DatastoreFlag, ctx = flags.NewDatastoreFlag(ctx)
    45  	cmd.DatastoreFlag.Register(ctx, f)
    46  
    47  	f.BoolVar(&cmd.createParents, "p", false, "Create intermediate directories as needed")
    48  	f.BoolVar(&cmd.isNamespace, "namespace", false, "Return uuid of namespace created on vsan datastore")
    49  }
    50  
    51  func (cmd *mkdir) Process(ctx context.Context) error {
    52  	if err := cmd.DatastoreFlag.Process(ctx); err != nil {
    53  		return err
    54  	}
    55  	return nil
    56  }
    57  
    58  func (cmd *mkdir) Usage() string {
    59  	return "DIRECTORY"
    60  }
    61  
    62  func (cmd *mkdir) Run(ctx context.Context, f *flag.FlagSet) error {
    63  	args := f.Args()
    64  	if len(args) == 0 {
    65  		return errors.New("missing operand")
    66  	}
    67  
    68  	c, err := cmd.Client()
    69  	if err != nil {
    70  		return err
    71  	}
    72  
    73  	if cmd.isNamespace {
    74  		var uuid string
    75  		var ds *object.Datastore
    76  
    77  		if ds, err = cmd.Datastore(); err != nil {
    78  			return err
    79  		}
    80  
    81  		path := args[0]
    82  
    83  		nm := object.NewDatastoreNamespaceManager(c)
    84  		if uuid, err = nm.CreateDirectory(ctx, ds, path, ""); err != nil {
    85  			return err
    86  		}
    87  
    88  		fmt.Println(uuid)
    89  	} else {
    90  		var dc *object.Datacenter
    91  		var path string
    92  
    93  		dc, err = cmd.Datacenter()
    94  		if err != nil {
    95  			return err
    96  		}
    97  
    98  		path, err = cmd.DatastorePath(args[0])
    99  		if err != nil {
   100  			return err
   101  		}
   102  
   103  		m := object.NewFileManager(c)
   104  		err = m.MakeDirectory(ctx, path, dc, cmd.createParents)
   105  
   106  		// ignore EEXIST if -p flag is given
   107  		if err != nil && cmd.createParents {
   108  			if soap.IsSoapFault(err) {
   109  				soapFault := soap.ToSoapFault(err)
   110  				if _, ok := soapFault.VimFault().(types.FileAlreadyExists); ok {
   111  					return nil
   112  				}
   113  			}
   114  		}
   115  	}
   116  
   117  	return err
   118  }